blob: 7d91af22da08ad0c24161b2aeba1a9a93ed79aa3 [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,
20 a subclass of socket.error
21- socket.herror: exception raised for gethostby* errors,
22 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
30 (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])
37 --> List of (family, socktype, proto, canonname, sockaddr)
38- 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:
57 TIPC_ADDR_NAMESEQ, TIPC_ADDR_NAME, and TIPC_ADDR_ID;
58 and scope can be one of:
59 TIPC_ZONE_SCOPE, TIPC_CLUSTER_SCOPE, and TIPC_NODE_SCOPE.
60 The meaning of v1, v2 and v3 depends on the value of addr_type:
61 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
73
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
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +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{
354 HANDLE newhandle;
355
356 if (!DuplicateHandle(GetCurrentProcess(), (HANDLE)handle,
357 GetCurrentProcess(), &newhandle,
358 0, FALSE, DUPLICATE_SAME_ACCESS))
359 {
360 WSASetLastError(GetLastError());
361 return INVALID_SOCKET;
362 }
363 return (SOCKET)newhandle;
364}
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000365#define SOCKETCLOSE closesocket
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000366#else
367/* On Unix we can use dup to duplicate the file descriptor of a socket*/
368#define dup_socket(fd) dup(fd)
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000369#endif
370
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000371#ifdef MS_WIN32
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000372#define EAFNOSUPPORT WSAEAFNOSUPPORT
373#define snprintf _snprintf
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000374#endif
Fred Drakea04eaad2000-06-30 02:46:07 +0000375
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000376#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000377#define SOCKETCLOSE soclose
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000378#define NO_DUP /* Sockets are Not Actual File Handles under OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000379#endif
380
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000381#ifndef SOCKETCLOSE
382#define SOCKETCLOSE close
383#endif
384
Hye-Shik Chang81268602004-02-02 06:05:24 +0000385#if defined(HAVE_BLUETOOTH_H) || defined(HAVE_BLUETOOTH_BLUETOOTH_H)
386#define USE_BLUETOOTH 1
387#if defined(__FreeBSD__)
388#define BTPROTO_L2CAP BLUETOOTH_PROTO_L2CAP
389#define BTPROTO_RFCOMM BLUETOOTH_PROTO_RFCOMM
Thomas Wouterscf297e42007-02-23 15:07:44 +0000390#define BTPROTO_HCI BLUETOOTH_PROTO_HCI
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000391#define SOL_HCI SOL_HCI_RAW
392#define HCI_FILTER SO_HCI_RAW_FILTER
Hye-Shik Chang81268602004-02-02 06:05:24 +0000393#define sockaddr_l2 sockaddr_l2cap
394#define sockaddr_rc sockaddr_rfcomm
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000395#define hci_dev hci_node
Hye-Shik Chang81268602004-02-02 06:05:24 +0000396#define _BT_L2_MEMB(sa, memb) ((sa)->l2cap_##memb)
397#define _BT_RC_MEMB(sa, memb) ((sa)->rfcomm_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000398#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000399#elif defined(__NetBSD__)
400#define sockaddr_l2 sockaddr_bt
401#define sockaddr_rc sockaddr_bt
Thomas Wouterscf297e42007-02-23 15:07:44 +0000402#define sockaddr_hci sockaddr_bt
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000403#define sockaddr_sco sockaddr_bt
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000404#define _BT_L2_MEMB(sa, memb) ((sa)->bt_##memb)
405#define _BT_RC_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000406#define _BT_HCI_MEMB(sa, memb) ((sa)->bt_##memb)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000407#define _BT_SCO_MEMB(sa, memb) ((sa)->bt_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000408#else
Hye-Shik Chang81268602004-02-02 06:05:24 +0000409#define _BT_L2_MEMB(sa, memb) ((sa)->l2_##memb)
410#define _BT_RC_MEMB(sa, memb) ((sa)->rc_##memb)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000411#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000412#define _BT_SCO_MEMB(sa, memb) ((sa)->sco_##memb)
413#endif
414#endif
415
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000416#ifdef __VMS
417/* TCP/IP Services for VMS uses a maximum send/recv buffer length */
418#define SEGMENT_SIZE (32 * 1024 -1)
419#endif
420
Thomas Wouters89f507f2006-12-13 04:49:30 +0000421#define SAS2SA(x) ((struct sockaddr *)(x))
422
Martin v. Löwise9416172003-05-03 10:12:45 +0000423/*
424 * Constants for getnameinfo()
425 */
426#if !defined(NI_MAXHOST)
427#define NI_MAXHOST 1025
428#endif
429#if !defined(NI_MAXSERV)
430#define NI_MAXSERV 32
431#endif
432
Guido van Rossum7d0a8262007-05-21 23:13:11 +0000433#ifndef INVALID_SOCKET /* MS defines this */
434#define INVALID_SOCKET (-1)
435#endif
436
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000437/* XXX There's a problem here: *static* functions are not supposed to have
438 a Py prefix (or use CapitalizedWords). Later... */
439
Guido van Rossum30a685f1991-06-27 15:51:29 +0000440/* Global variable holding the exception type for errors detected
441 by this module (but not argument type or memory errors, etc.). */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000442static PyObject *socket_error;
443static PyObject *socket_herror;
444static PyObject *socket_gaierror;
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000445static PyObject *socket_timeout;
Guido van Rossum30a685f1991-06-27 15:51:29 +0000446
Tim Peters643a7fc2002-02-17 04:13:21 +0000447/* A forward reference to the socket type object.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000448 The sock_type variable contains pointers to various functions,
449 some of which call new_sockobject(), which uses sock_type, so
Tim Peters643a7fc2002-02-17 04:13:21 +0000450 there has to be a circular reference. */
Jeremy Hylton938ace62002-07-17 16:30:39 +0000451static PyTypeObject sock_type;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000452
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000453#if defined(HAVE_POLL_H)
454#include <poll.h>
455#elif defined(HAVE_SYS_POLL_H)
456#include <sys/poll.h>
457#endif
458
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000459#ifdef Py_SOCKET_FD_CAN_BE_GE_FD_SETSIZE
460/* Platform can select file descriptors beyond FD_SETSIZE */
461#define IS_SELECTABLE(s) 1
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000462#elif defined(HAVE_POLL)
463/* Instead of select(), we'll use poll() since poll() works on any fd. */
464#define IS_SELECTABLE(s) 1
465/* Can we call select() with this socket without a buffer overrun? */
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000466#else
467/* POSIX says selecting file descriptors beyond FD_SETSIZE
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000468 has undefined behaviour. If there's no timeout left, we don't have to
469 call select, so it's a safe, little white lie. */
470#define IS_SELECTABLE(s) ((s)->sock_fd < FD_SETSIZE || s->sock_timeout <= 0.0)
Martin v. Löwisf84d1b92006-02-11 09:27:05 +0000471#endif
Neal Norwitz082b2df2006-02-07 07:04:46 +0000472
473static PyObject*
474select_error(void)
475{
476 PyErr_SetString(socket_error, "unable to select on socket");
477 return NULL;
478}
479
Guido van Rossum30a685f1991-06-27 15:51:29 +0000480/* Convenience function to raise an error according to errno
481 and return a NULL pointer from a function. */
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000482
Guido van Rossum73624e91994-10-10 17:59:00 +0000483static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000484set_error(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000485{
Guido van Rossum8d665e61996-06-26 18:22:49 +0000486#ifdef MS_WINDOWS
Mark Hammond46a733d2000-07-24 01:45:11 +0000487 int err_no = WSAGetLastError();
Guido van Rossum3d392eb2007-11-16 00:35:22 +0000488 /* PyErr_SetExcFromWindowsErr() invokes FormatMessage() which
489 recognizes the error codes used by both GetLastError() and
490 WSAGetLastError */
491 if (err_no)
492 return PyErr_SetExcFromWindowsErr(socket_error, err_no);
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000493#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000494
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000495#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000496 if (sock_errno() != NO_ERROR) {
497 APIRET rc;
498 ULONG msglen;
499 char outbuf[100];
500 int myerrorcode = sock_errno();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000501
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000502 /* Retrieve socket-related error message from MPTN.MSG file */
503 rc = DosGetMessage(NULL, 0, outbuf, sizeof(outbuf),
504 myerrorcode - SOCBASEERR + 26,
505 "mptn.msg",
506 &msglen);
507 if (rc == NO_ERROR) {
508 PyObject *v;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000509
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000510 /* OS/2 doesn't guarantee a terminator */
511 outbuf[msglen] = '\0';
512 if (strlen(outbuf) > 0) {
513 /* If non-empty msg, trim CRLF */
514 char *lastc = &outbuf[ strlen(outbuf)-1 ];
Neal Norwitz30b5c5d2005-12-19 06:05:18 +0000515 while (lastc > outbuf &&
516 isspace(Py_CHARMASK(*lastc))) {
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000517 /* Trim trailing whitespace (CRLF) */
518 *lastc-- = '\0';
519 }
520 }
521 v = Py_BuildValue("(is)", myerrorcode, outbuf);
522 if (v != NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000523 PyErr_SetObject(socket_error, v);
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000524 Py_DECREF(v);
525 }
526 return NULL;
527 }
528 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000529#endif
530
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000531 return PyErr_SetFromErrno(socket_error);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000532}
533
Guido van Rossum30a685f1991-06-27 15:51:29 +0000534
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000535static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000536set_herror(int h_error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000537{
538 PyObject *v;
539
540#ifdef HAVE_HSTRERROR
541 v = Py_BuildValue("(is)", h_error, (char *)hstrerror(h_error));
542#else
543 v = Py_BuildValue("(is)", h_error, "host not found");
544#endif
545 if (v != NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000546 PyErr_SetObject(socket_herror, v);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000547 Py_DECREF(v);
548 }
549
550 return NULL;
551}
552
553
554static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000555set_gaierror(int error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000556{
557 PyObject *v;
558
Martin v. Löwis272cb402002-03-01 08:31:07 +0000559#ifdef EAI_SYSTEM
560 /* EAI_SYSTEM is not available on Windows XP. */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000561 if (error == EAI_SYSTEM)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000562 return set_error();
Martin v. Löwis272cb402002-03-01 08:31:07 +0000563#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000564
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000565#ifdef HAVE_GAI_STRERROR
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000566 v = Py_BuildValue("(is)", error, gai_strerror(error));
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000567#else
568 v = Py_BuildValue("(is)", error, "getaddrinfo failed");
569#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000570 if (v != NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000571 PyErr_SetObject(socket_gaierror, v);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000572 Py_DECREF(v);
573 }
574
575 return NULL;
576}
577
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000578#ifdef __VMS
579/* Function to send in segments */
580static int
581sendsegmented(int sock_fd, char *buf, int len, int flags)
582{
583 int n = 0;
584 int remaining = len;
585
586 while (remaining > 0) {
587 unsigned int segment;
588
589 segment = (remaining >= SEGMENT_SIZE ? SEGMENT_SIZE : remaining);
590 n = send(sock_fd, buf, segment, flags);
591 if (n < 0) {
592 return n;
593 }
594 remaining -= segment;
595 buf += segment;
596 } /* end while */
597
598 return len;
599}
600#endif
601
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000602/* Function to perform the setting of socket blocking mode
603 internally. block = (1 | 0). */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000604static int
605internal_setblocking(PySocketSockObject *s, int block)
606{
Guido van Rossum67f7a382002-06-06 21:08:16 +0000607#ifndef MS_WINDOWS
608 int delay_flag;
609#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +0000610
611 Py_BEGIN_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +0000612#ifndef MS_WINDOWS
613#if defined(PYOS_OS2) && !defined(PYCC_GCC)
614 block = !block;
615 ioctl(s->sock_fd, FIONBIO, (caddr_t)&block, sizeof(block));
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000616#elif defined(__VMS)
617 block = !block;
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000618 ioctl(s->sock_fd, FIONBIO, (unsigned int *)&block);
619#else /* !PYOS_OS2 && !__VMS */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000620 delay_flag = fcntl(s->sock_fd, F_GETFL, 0);
621 if (block)
Neal Norwitz39d22e52002-11-02 19:55:21 +0000622 delay_flag &= (~O_NONBLOCK);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000623 else
Neal Norwitz39d22e52002-11-02 19:55:21 +0000624 delay_flag |= O_NONBLOCK;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000625 fcntl(s->sock_fd, F_SETFL, delay_flag);
626#endif /* !PYOS_OS2 */
627#else /* MS_WINDOWS */
628 block = !block;
629 ioctlsocket(s->sock_fd, FIONBIO, (u_long*)&block);
630#endif /* MS_WINDOWS */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000631 Py_END_ALLOW_THREADS
632
633 /* Since these don't return anything */
634 return 1;
635}
636
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000637/* Do a select()/poll() on the socket, if necessary (sock_timeout > 0).
Guido van Rossum11ba0942002-06-13 15:07:44 +0000638 The argument writing indicates the direction.
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000639 This does not raise an exception; we'll let our caller do that
640 after they've reacquired the interpreter lock.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000641 Returns 1 on timeout, -1 on error, 0 otherwise. */
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000642static int
Guido van Rossumb9e916a2002-06-07 01:42:47 +0000643internal_select(PySocketSockObject *s, int writing)
Guido van Rossum67f7a382002-06-06 21:08:16 +0000644{
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000645 int n;
Guido van Rossum11ba0942002-06-13 15:07:44 +0000646
Guido van Rossumad654902002-07-19 12:44:59 +0000647 /* Nothing to do unless we're in timeout mode (not non-blocking) */
Guido van Rossum11ba0942002-06-13 15:07:44 +0000648 if (s->sock_timeout <= 0.0)
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000649 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000650
Guido van Rossumad654902002-07-19 12:44:59 +0000651 /* Guard against closed socket */
652 if (s->sock_fd < 0)
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000653 return 0;
Guido van Rossumad654902002-07-19 12:44:59 +0000654
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000655 /* Prefer poll, if available, since you can poll() any fd
656 * which can't be done with select(). */
657#ifdef HAVE_POLL
658 {
659 struct pollfd pollfd;
660 int timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000661
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000662 pollfd.fd = s->sock_fd;
663 pollfd.events = writing ? POLLOUT : POLLIN;
664
665 /* s->sock_timeout is in seconds, timeout in ms */
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000666 timeout = (int)(s->sock_timeout * 1000 + 0.5);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000667 n = poll(&pollfd, 1, timeout);
668 }
669#else
670 {
671 /* Construct the arguments to select */
672 fd_set fds;
673 struct timeval tv;
674 tv.tv_sec = (int)s->sock_timeout;
675 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
676 FD_ZERO(&fds);
677 FD_SET(s->sock_fd, &fds);
678
679 /* See if the socket is ready */
680 if (writing)
681 n = select(s->sock_fd+1, NULL, &fds, NULL, &tv);
682 else
683 n = select(s->sock_fd+1, &fds, NULL, NULL, &tv);
684 }
685#endif
Guido van Rossum39eb8fa2007-11-16 01:24:05 +0000686
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000687 if (n < 0)
688 return -1;
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000689 if (n == 0)
690 return 1;
691 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000692}
693
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000694/* Initialize a new socket object. */
695
Tim Petersa12b4cf2002-07-18 22:38:44 +0000696static double defaulttimeout = -1.0; /* Default timeout for new sockets */
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000697
Martin v. Löwis1a214512008-06-11 05:26:20 +0000698static void
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000699init_sockobject(PySocketSockObject *s,
700 SOCKET_T fd, int family, int type, int proto)
701{
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000702 s->sock_fd = fd;
703 s->sock_family = family;
704 s->sock_type = type;
705 s->sock_proto = proto;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000706 s->sock_timeout = defaulttimeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000707
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000708 s->errorhandler = &set_error;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000709
710 if (defaulttimeout >= 0.0)
711 internal_setblocking(s, 0);
712
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000713}
714
715
Guido van Rossum30a685f1991-06-27 15:51:29 +0000716/* Create a new socket object.
717 This just creates the object and initializes it.
718 If the creation fails, return NULL and set an exception (implicit
719 in NEWOBJ()). */
720
Guido van Rossum73624e91994-10-10 17:59:00 +0000721static PySocketSockObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000722new_sockobject(SOCKET_T fd, int family, int type, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000723{
Guido van Rossum73624e91994-10-10 17:59:00 +0000724 PySocketSockObject *s;
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000725 s = (PySocketSockObject *)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000726 PyType_GenericNew(&sock_type, NULL, NULL);
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000727 if (s != NULL)
728 init_sockobject(s, fd, family, type, proto);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000729 return s;
730}
731
Guido van Rossum30a685f1991-06-27 15:51:29 +0000732
Guido van Rossum48a680c2001-03-02 06:34:14 +0000733/* Lock to allow python interpreter to continue, but only allow one
Just van Rossum1040d2c2003-05-09 07:53:18 +0000734 thread to be in gethostbyname or getaddrinfo */
735#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
736PyThread_type_lock netdb_lock;
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000737#endif
738
739
Guido van Rossum30a685f1991-06-27 15:51:29 +0000740/* Convert a string specifying a host name or one of a few symbolic
741 names to a numeric IP address. This usually calls gethostbyname()
742 to do the work; the names "" and "<broadcast>" are special.
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000743 Return the length (IPv4 should be 4 bytes), or negative if
Guido van Rossum30a685f1991-06-27 15:51:29 +0000744 an error occurred; then an exception is raised. */
745
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000746static int
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000747setipaddr(char *name, struct sockaddr *addr_ret, size_t addr_ret_size, int af)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000748{
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000749 struct addrinfo hints, *res;
750 int error;
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000751 int d1, d2, d3, d4;
752 char ch;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000753
Guido van Rossuma376cc51996-12-05 23:43:35 +0000754 memset((void *) addr_ret, '\0', sizeof(*addr_ret));
Guido van Rossum30a685f1991-06-27 15:51:29 +0000755 if (name[0] == '\0') {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000756 int siz;
757 memset(&hints, 0, sizeof(hints));
758 hints.ai_family = af;
759 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
760 hints.ai_flags = AI_PASSIVE;
Just van Rossum1040d2c2003-05-09 07:53:18 +0000761 Py_BEGIN_ALLOW_THREADS
762 ACQUIRE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000763 error = getaddrinfo(NULL, "0", &hints, &res);
Just van Rossum1040d2c2003-05-09 07:53:18 +0000764 Py_END_ALLOW_THREADS
765 /* We assume that those thread-unsafe getaddrinfo() versions
766 *are* safe regarding their return value, ie. that a
767 subsequent call to getaddrinfo() does not destroy the
768 outcome of the first call. */
769 RELEASE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000770 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000771 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000772 return -1;
773 }
774 switch (res->ai_family) {
775 case AF_INET:
776 siz = 4;
777 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000778#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000779 case AF_INET6:
780 siz = 16;
781 break;
782#endif
783 default:
784 freeaddrinfo(res);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000785 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000786 "unsupported address family");
787 return -1;
788 }
789 if (res->ai_next) {
Martin v. Löwisf0b11d22001-11-07 08:31:03 +0000790 freeaddrinfo(res);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000791 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000792 "wildcard resolved to multiple address");
793 return -1;
794 }
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000795 if (res->ai_addrlen < addr_ret_size)
796 addr_ret_size = res->ai_addrlen;
797 memcpy(addr_ret, res->ai_addr, addr_ret_size);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000798 freeaddrinfo(res);
799 return siz;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000800 }
Guido van Rossum30a685f1991-06-27 15:51:29 +0000801 if (name[0] == '<' && strcmp(name, "<broadcast>") == 0) {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000802 struct sockaddr_in *sin;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +0000803 if (af != AF_INET && af != AF_UNSPEC) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000804 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000805 "address family mismatched");
806 return -1;
807 }
808 sin = (struct sockaddr_in *)addr_ret;
809 memset((void *) sin, '\0', sizeof(*sin));
810 sin->sin_family = AF_INET;
811#ifdef HAVE_SOCKADDR_SA_LEN
812 sin->sin_len = sizeof(*sin);
813#endif
814 sin->sin_addr.s_addr = INADDR_BROADCAST;
815 return sizeof(sin->sin_addr);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000816 }
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000817 if (sscanf(name, "%d.%d.%d.%d%c", &d1, &d2, &d3, &d4, &ch) == 4 &&
818 0 <= d1 && d1 <= 255 && 0 <= d2 && d2 <= 255 &&
819 0 <= d3 && d3 <= 255 && 0 <= d4 && d4 <= 255) {
820 struct sockaddr_in *sin;
821 sin = (struct sockaddr_in *)addr_ret;
822 sin->sin_addr.s_addr = htonl(
823 ((long) d1 << 24) | ((long) d2 << 16) |
824 ((long) d3 << 8) | ((long) d4 << 0));
825 sin->sin_family = AF_INET;
826#ifdef HAVE_SOCKADDR_SA_LEN
827 sin->sin_len = sizeof(*sin);
828#endif
829 return 4;
830 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000831 memset(&hints, 0, sizeof(hints));
832 hints.ai_family = af;
Just van Rossum1040d2c2003-05-09 07:53:18 +0000833 Py_BEGIN_ALLOW_THREADS
834 ACQUIRE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000835 error = getaddrinfo(name, NULL, &hints, &res);
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000836#if defined(__digital__) && defined(__unix__)
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000837 if (error == EAI_NONAME && af == AF_UNSPEC) {
838 /* On Tru64 V5.1, numeric-to-addr conversion fails
839 if no address family is given. Assume IPv4 for now.*/
840 hints.ai_family = AF_INET;
841 error = getaddrinfo(name, NULL, &hints, &res);
842 }
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000843#endif
Just van Rossum1040d2c2003-05-09 07:53:18 +0000844 Py_END_ALLOW_THREADS
845 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000846 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000847 set_gaierror(error);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000848 return -1;
849 }
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000850 if (res->ai_addrlen < addr_ret_size)
851 addr_ret_size = res->ai_addrlen;
852 memcpy((char *) addr_ret, res->ai_addr, addr_ret_size);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000853 freeaddrinfo(res);
854 switch (addr_ret->sa_family) {
855 case AF_INET:
856 return 4;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000857#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000858 case AF_INET6:
859 return 16;
Guido van Rossum955becc1999-03-22 20:14:53 +0000860#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000861 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000862 PyErr_SetString(socket_error, "unknown address family");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000863 return -1;
864 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000865}
866
Guido van Rossum30a685f1991-06-27 15:51:29 +0000867
Guido van Rossum30a685f1991-06-27 15:51:29 +0000868/* Create a string object representing an IP address.
869 This is always a string of the form 'dd.dd.dd.dd' (with variable
870 size numbers). */
871
Guido van Rossum73624e91994-10-10 17:59:00 +0000872static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000873makeipaddr(struct sockaddr *addr, int addrlen)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000874{
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000875 char buf[NI_MAXHOST];
876 int error;
877
878 error = getnameinfo(addr, addrlen, buf, sizeof(buf), NULL, 0,
879 NI_NUMERICHOST);
880 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000881 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000882 return NULL;
883 }
Guido van Rossum32c4ac02007-08-15 03:56:40 +0000884 return PyUnicode_FromString(buf);
Guido van Rossum30a685f1991-06-27 15:51:29 +0000885}
886
887
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000888#ifdef USE_BLUETOOTH
889/* Convert a string representation of a Bluetooth address into a numeric
890 address. Returns the length (6), or raises an exception and returns -1 if
891 an error occurred. */
892
893static int
894setbdaddr(char *name, bdaddr_t *bdaddr)
895{
896 unsigned int b0, b1, b2, b3, b4, b5;
897 char ch;
898 int n;
899
900 n = sscanf(name, "%X:%X:%X:%X:%X:%X%c",
901 &b5, &b4, &b3, &b2, &b1, &b0, &ch);
902 if (n == 6 && (b0 | b1 | b2 | b3 | b4 | b5) < 256) {
903 bdaddr->b[0] = b0;
904 bdaddr->b[1] = b1;
905 bdaddr->b[2] = b2;
906 bdaddr->b[3] = b3;
907 bdaddr->b[4] = b4;
908 bdaddr->b[5] = b5;
909 return 6;
910 } else {
911 PyErr_SetString(socket_error, "bad bluetooth address");
912 return -1;
913 }
914}
915
916/* Create a string representation of the Bluetooth address. This is always a
917 string of the form 'XX:XX:XX:XX:XX:XX' where XX is a two digit hexadecimal
918 value (zero padded if necessary). */
919
920static PyObject *
921makebdaddr(bdaddr_t *bdaddr)
922{
923 char buf[(6 * 2) + 5 + 1];
924
925 sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
926 bdaddr->b[5], bdaddr->b[4], bdaddr->b[3],
927 bdaddr->b[2], bdaddr->b[1], bdaddr->b[0]);
Guido van Rossum32c4ac02007-08-15 03:56:40 +0000928 return PyUnicode_FromString(buf);
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000929}
930#endif
931
932
Guido van Rossum30a685f1991-06-27 15:51:29 +0000933/* Create an object representing the given socket address,
934 suitable for passing it back to bind(), connect() etc.
935 The family field of the sockaddr structure is inspected
936 to determine what kind of address it really is. */
937
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000938/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +0000939static PyObject *
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000940makesockaddr(int sockfd, struct sockaddr *addr, int addrlen, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000941{
Guido van Rossum25bec8c1992-08-05 19:00:45 +0000942 if (addrlen == 0) {
943 /* No address -- may be recvfrom() from known socket */
Guido van Rossum73624e91994-10-10 17:59:00 +0000944 Py_INCREF(Py_None);
945 return Py_None;
Guido van Rossum25bec8c1992-08-05 19:00:45 +0000946 }
947
Guido van Rossum30a685f1991-06-27 15:51:29 +0000948 switch (addr->sa_family) {
949
950 case AF_INET:
951 {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000952 struct sockaddr_in *a;
953 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
Barry Warsaw752300b1997-01-03 17:18:10 +0000954 PyObject *ret = NULL;
955 if (addrobj) {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000956 a = (struct sockaddr_in *)addr;
Barry Warsaw752300b1997-01-03 17:18:10 +0000957 ret = Py_BuildValue("Oi", addrobj, ntohs(a->sin_port));
958 Py_DECREF(addrobj);
959 }
Guido van Rossum6f5afc91993-02-05 09:46:15 +0000960 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000961 }
Guido van Rossum30a685f1991-06-27 15:51:29 +0000962
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +0000963#if defined(AF_UNIX)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000964 case AF_UNIX:
965 {
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000966 struct sockaddr_un *a = (struct sockaddr_un *) addr;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000967#ifdef linux
968 if (a->sun_path[0] == 0) { /* Linux abstract namespace */
Christian Heimesbbffeb62008-01-24 09:42:52 +0000969 addrlen -= offsetof(struct sockaddr_un, sun_path);
Christian Heimes72b710a2008-05-26 13:28:38 +0000970 return PyBytes_FromStringAndSize(a->sun_path, addrlen);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000971 }
972 else
973#endif /* linux */
974 {
975 /* regular NULL-terminated string */
Guido van Rossum32c4ac02007-08-15 03:56:40 +0000976 return PyUnicode_FromString(a->sun_path);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000977 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000978 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000979#endif /* AF_UNIX */
980
Martin v. Löwis11017b12006-01-14 18:12:57 +0000981#if defined(AF_NETLINK)
982 case AF_NETLINK:
983 {
984 struct sockaddr_nl *a = (struct sockaddr_nl *) addr;
Martin v. Löwisd0560052006-01-15 07:49:20 +0000985 return Py_BuildValue("II", a->nl_pid, a->nl_groups);
Martin v. Löwis11017b12006-01-14 18:12:57 +0000986 }
987#endif /* AF_NETLINK */
988
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000989#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000990 case AF_INET6:
991 {
992 struct sockaddr_in6 *a;
993 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
994 PyObject *ret = NULL;
995 if (addrobj) {
996 a = (struct sockaddr_in6 *)addr;
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000997 ret = Py_BuildValue("Oiii",
998 addrobj,
999 ntohs(a->sin6_port),
1000 a->sin6_flowinfo,
1001 a->sin6_scope_id);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001002 Py_DECREF(addrobj);
1003 }
1004 return ret;
1005 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001006#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00001007
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001008#ifdef USE_BLUETOOTH
1009 case AF_BLUETOOTH:
1010 switch (proto) {
1011
1012 case BTPROTO_L2CAP:
1013 {
1014 struct sockaddr_l2 *a = (struct sockaddr_l2 *) addr;
1015 PyObject *addrobj = makebdaddr(&_BT_L2_MEMB(a, bdaddr));
1016 PyObject *ret = NULL;
1017 if (addrobj) {
1018 ret = Py_BuildValue("Oi",
1019 addrobj,
1020 _BT_L2_MEMB(a, psm));
1021 Py_DECREF(addrobj);
1022 }
1023 return ret;
1024 }
1025
1026 case BTPROTO_RFCOMM:
1027 {
1028 struct sockaddr_rc *a = (struct sockaddr_rc *) addr;
1029 PyObject *addrobj = makebdaddr(&_BT_RC_MEMB(a, bdaddr));
1030 PyObject *ret = NULL;
1031 if (addrobj) {
1032 ret = Py_BuildValue("Oi",
1033 addrobj,
1034 _BT_RC_MEMB(a, channel));
1035 Py_DECREF(addrobj);
1036 }
1037 return ret;
1038 }
1039
Thomas Wouterscf297e42007-02-23 15:07:44 +00001040 case BTPROTO_HCI:
1041 {
1042 struct sockaddr_hci *a = (struct sockaddr_hci *) addr;
1043 PyObject *ret = NULL;
1044 ret = Py_BuildValue("i", _BT_HCI_MEMB(a, dev));
1045 return ret;
1046 }
1047
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001048#if !defined(__FreeBSD__)
1049 case BTPROTO_SCO:
1050 {
1051 struct sockaddr_sco *a = (struct sockaddr_sco *) addr;
1052 return makebdaddr(&_BT_SCO_MEMB(a, bdaddr));
1053 }
1054#endif
1055
1056 }
1057#endif
1058
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001059#ifdef HAVE_NETPACKET_PACKET_H
Jeremy Hylton22308652001-02-02 03:23:09 +00001060 case AF_PACKET:
1061 {
1062 struct sockaddr_ll *a = (struct sockaddr_ll *)addr;
1063 char *ifname = "";
1064 struct ifreq ifr;
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001065 /* need to look up interface name give index */
1066 if (a->sll_ifindex) {
Jeremy Hylton22308652001-02-02 03:23:09 +00001067 ifr.ifr_ifindex = a->sll_ifindex;
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001068 if (ioctl(sockfd, SIOCGIFNAME, &ifr) == 0)
Jeremy Hylton22308652001-02-02 03:23:09 +00001069 ifname = ifr.ifr_name;
Jeremy Hylton22308652001-02-02 03:23:09 +00001070 }
Guido van Rossum32c4ac02007-08-15 03:56:40 +00001071 return Py_BuildValue("shbhy#",
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001072 ifname,
1073 ntohs(a->sll_protocol),
1074 a->sll_pkttype,
1075 a->sll_hatype,
1076 a->sll_addr,
1077 a->sll_halen);
Jeremy Hylton22308652001-02-02 03:23:09 +00001078 }
1079#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001080
Christian Heimes043d6f62008-01-07 17:19:16 +00001081#ifdef HAVE_LINUX_TIPC_H
1082 case AF_TIPC:
1083 {
1084 struct sockaddr_tipc *a = (struct sockaddr_tipc *) addr;
1085 if (a->addrtype == TIPC_ADDR_NAMESEQ) {
1086 return Py_BuildValue("IIIII",
1087 a->addrtype,
1088 a->addr.nameseq.type,
1089 a->addr.nameseq.lower,
1090 a->addr.nameseq.upper,
1091 a->scope);
1092 } else if (a->addrtype == TIPC_ADDR_NAME) {
1093 return Py_BuildValue("IIIII",
1094 a->addrtype,
1095 a->addr.name.name.type,
1096 a->addr.name.name.instance,
1097 a->addr.name.name.instance,
1098 a->scope);
1099 } else if (a->addrtype == TIPC_ADDR_ID) {
1100 return Py_BuildValue("IIIII",
1101 a->addrtype,
1102 a->addr.id.node,
1103 a->addr.id.ref,
1104 0,
1105 a->scope);
1106 } else {
1107 PyErr_SetString(PyExc_TypeError,
1108 "Invalid address type");
1109 return NULL;
1110 }
1111 }
1112#endif
1113
Guido van Rossum30a685f1991-06-27 15:51:29 +00001114 /* More cases here... */
1115
1116 default:
Guido van Rossumaa948df1997-05-07 17:41:48 +00001117 /* If we don't know the address family, don't raise an
Guido van Rossum9ce8e382007-08-09 21:06:37 +00001118 exception -- return it as an (int, bytes) tuple. */
1119 return Py_BuildValue("iy#",
Guido van Rossumaa948df1997-05-07 17:41:48 +00001120 addr->sa_family,
1121 addr->sa_data,
1122 sizeof(addr->sa_data));
Guido van Rossum25bec8c1992-08-05 19:00:45 +00001123
Guido van Rossum30a685f1991-06-27 15:51:29 +00001124 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001125}
1126
Guido van Rossum30a685f1991-06-27 15:51:29 +00001127
1128/* Parse a socket address argument according to the socket object's
1129 address family. Return 1 if the address was in the proper format,
1130 0 of not. The address is returned through addr_ret, its length
1131 through len_ret. */
1132
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001133static int
Guido van Rossum48a680c2001-03-02 06:34:14 +00001134getsockaddrarg(PySocketSockObject *s, PyObject *args,
Thomas Wouters89f507f2006-12-13 04:49:30 +00001135 struct sockaddr *addr_ret, int *len_ret)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001136{
Guido van Rossum30a685f1991-06-27 15:51:29 +00001137 switch (s->sock_family) {
1138
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001139#if defined(AF_UNIX)
Guido van Rossum30a685f1991-06-27 15:51:29 +00001140 case AF_UNIX:
1141 {
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001142 struct sockaddr_un* addr;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001143 char *path;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001144 int len;
Guido van Rossume22905a2007-08-27 23:09:25 +00001145 if (!PyArg_Parse(args, "s#", &path, &len))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001146 return 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001147
1148 addr = (struct sockaddr_un*)addr_ret;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001149#ifdef linux
1150 if (len > 0 && path[0] == 0) {
1151 /* Linux abstract namespace extension */
1152 if (len > sizeof addr->sun_path) {
1153 PyErr_SetString(socket_error,
1154 "AF_UNIX path too long");
1155 return 0;
1156 }
1157 }
1158 else
1159#endif /* linux */
1160 {
1161 /* regular NULL-terminated string */
1162 if (len >= sizeof addr->sun_path) {
1163 PyErr_SetString(socket_error,
1164 "AF_UNIX path too long");
1165 return 0;
1166 }
1167 addr->sun_path[len] = 0;
Guido van Rossum30a685f1991-06-27 15:51:29 +00001168 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001169 addr->sun_family = s->sock_family;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001170 memcpy(addr->sun_path, path, len);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001171#if defined(PYOS_OS2)
1172 *len_ret = sizeof(*addr);
1173#else
Christian Heimesbbffeb62008-01-24 09:42:52 +00001174 *len_ret = len + offsetof(struct sockaddr_un, sun_path);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001175#endif
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001176 return 1;
1177 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001178#endif /* AF_UNIX */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001179
Martin v. Löwis11017b12006-01-14 18:12:57 +00001180#if defined(AF_NETLINK)
1181 case AF_NETLINK:
1182 {
1183 struct sockaddr_nl* addr;
1184 int pid, groups;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001185 addr = (struct sockaddr_nl *)addr_ret;
Martin v. Löwis11017b12006-01-14 18:12:57 +00001186 if (!PyTuple_Check(args)) {
1187 PyErr_Format(
1188 PyExc_TypeError,
1189 "getsockaddrarg: "
1190 "AF_NETLINK address must be tuple, not %.500s",
Christian Heimes90aa7642007-12-19 02:45:37 +00001191 Py_TYPE(args)->tp_name);
Martin v. Löwis11017b12006-01-14 18:12:57 +00001192 return 0;
1193 }
1194 if (!PyArg_ParseTuple(args, "II:getsockaddrarg", &pid, &groups))
1195 return 0;
1196 addr->nl_family = AF_NETLINK;
1197 addr->nl_pid = pid;
1198 addr->nl_groups = groups;
Martin v. Löwis11017b12006-01-14 18:12:57 +00001199 *len_ret = sizeof(*addr);
1200 return 1;
1201 }
1202#endif
1203
Guido van Rossum30a685f1991-06-27 15:51:29 +00001204 case AF_INET:
1205 {
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001206 struct sockaddr_in* addr;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001207 char *host;
Martin v. Löwis5db099a2003-08-07 11:55:15 +00001208 int port, result;
Guido van Rossume4dad902000-12-01 13:13:11 +00001209 if (!PyTuple_Check(args)) {
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001210 PyErr_Format(
1211 PyExc_TypeError,
1212 "getsockaddrarg: "
1213 "AF_INET address must be tuple, not %.500s",
Christian Heimes90aa7642007-12-19 02:45:37 +00001214 Py_TYPE(args)->tp_name);
Guido van Rossume4dad902000-12-01 13:13:11 +00001215 return 0;
1216 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001217 if (!PyArg_ParseTuple(args, "eti:getsockaddrarg",
Martin v. Löwis2548c732003-04-18 10:39:54 +00001218 "idna", &host, &port))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001219 return 0;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001220 addr=(struct sockaddr_in*)addr_ret;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001221 result = setipaddr(host, (struct sockaddr *)addr,
Martin v. Löwis5db099a2003-08-07 11:55:15 +00001222 sizeof(*addr), AF_INET);
1223 PyMem_Free(host);
1224 if (result < 0)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001225 return 0;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001226 addr->sin_family = AF_INET;
Guido van Rossum644a12b1997-04-09 19:24:53 +00001227 addr->sin_port = htons((short)port);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001228 *len_ret = sizeof *addr;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001229 return 1;
1230 }
1231
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001232#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001233 case AF_INET6:
1234 {
1235 struct sockaddr_in6* addr;
1236 char *host;
Martin v. Löwis5db099a2003-08-07 11:55:15 +00001237 int port, flowinfo, scope_id, result;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001238 flowinfo = scope_id = 0;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001239 if (!PyTuple_Check(args)) {
1240 PyErr_Format(
1241 PyExc_TypeError,
1242 "getsockaddrarg: "
1243 "AF_INET6 address must be tuple, not %.500s",
Christian Heimes90aa7642007-12-19 02:45:37 +00001244 Py_TYPE(args)->tp_name);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001245 return 0;
1246 }
1247 if (!PyArg_ParseTuple(args, "eti|ii",
Martin v. Löwis2548c732003-04-18 10:39:54 +00001248 "idna", &host, &port, &flowinfo,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001249 &scope_id)) {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001250 return 0;
1251 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00001252 addr = (struct sockaddr_in6*)addr_ret;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001253 result = setipaddr(host, (struct sockaddr *)addr,
Martin v. Löwis5db099a2003-08-07 11:55:15 +00001254 sizeof(*addr), AF_INET6);
1255 PyMem_Free(host);
1256 if (result < 0)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001257 return 0;
1258 addr->sin6_family = s->sock_family;
1259 addr->sin6_port = htons((short)port);
1260 addr->sin6_flowinfo = flowinfo;
1261 addr->sin6_scope_id = scope_id;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001262 *len_ret = sizeof *addr;
1263 return 1;
1264 }
1265#endif
1266
Hye-Shik Chang81268602004-02-02 06:05:24 +00001267#ifdef USE_BLUETOOTH
Martin v. Löwis12af0482004-01-31 12:34:17 +00001268 case AF_BLUETOOTH:
1269 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001270 switch (s->sock_proto) {
1271 case BTPROTO_L2CAP:
Martin v. Löwis12af0482004-01-31 12:34:17 +00001272 {
Thomas Wouters89f507f2006-12-13 04:49:30 +00001273 struct sockaddr_l2 *addr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001274 char *straddr;
Martin v. Löwis12af0482004-01-31 12:34:17 +00001275
Thomas Wouters89f507f2006-12-13 04:49:30 +00001276 addr = (struct sockaddr_l2 *)addr_ret;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001277 _BT_L2_MEMB(addr, family) = AF_BLUETOOTH;
1278 if (!PyArg_ParseTuple(args, "si", &straddr,
1279 &_BT_L2_MEMB(addr, psm))) {
1280 PyErr_SetString(socket_error, "getsockaddrarg: "
1281 "wrong format");
Martin v. Löwis12af0482004-01-31 12:34:17 +00001282 return 0;
1283 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001284 if (setbdaddr(straddr, &_BT_L2_MEMB(addr, bdaddr)) < 0)
1285 return 0;
1286
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001287 *len_ret = sizeof *addr;
1288 return 1;
1289 }
1290 case BTPROTO_RFCOMM:
1291 {
Thomas Wouters89f507f2006-12-13 04:49:30 +00001292 struct sockaddr_rc *addr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001293 char *straddr;
1294
Thomas Wouters89f507f2006-12-13 04:49:30 +00001295 addr = (struct sockaddr_rc *)addr_ret;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001296 _BT_RC_MEMB(addr, family) = AF_BLUETOOTH;
1297 if (!PyArg_ParseTuple(args, "si", &straddr,
1298 &_BT_RC_MEMB(addr, channel))) {
1299 PyErr_SetString(socket_error, "getsockaddrarg: "
1300 "wrong format");
1301 return 0;
1302 }
1303 if (setbdaddr(straddr, &_BT_RC_MEMB(addr, bdaddr)) < 0)
1304 return 0;
1305
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001306 *len_ret = sizeof *addr;
1307 return 1;
1308 }
Thomas Wouterscf297e42007-02-23 15:07:44 +00001309 case BTPROTO_HCI:
1310 {
1311 struct sockaddr_hci *addr = (struct sockaddr_hci *)addr_ret;
1312 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
1313 if (!PyArg_ParseTuple(args, "i", &_BT_HCI_MEMB(addr, dev))) {
1314 PyErr_SetString(socket_error, "getsockaddrarg: "
1315 "wrong format");
1316 return 0;
1317 }
1318 *len_ret = sizeof *addr;
1319 return 1;
1320 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001321#if !defined(__FreeBSD__)
1322 case BTPROTO_SCO:
1323 {
Thomas Wouters89f507f2006-12-13 04:49:30 +00001324 struct sockaddr_sco *addr;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001325 char *straddr;
1326
Thomas Wouters89f507f2006-12-13 04:49:30 +00001327 addr = (struct sockaddr_sco *)addr_ret;
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001328 _BT_SCO_MEMB(addr, family) = AF_BLUETOOTH;
Christian Heimes72b710a2008-05-26 13:28:38 +00001329 if (!PyBytes_Check(args)) {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001330 PyErr_SetString(socket_error, "getsockaddrarg: "
1331 "wrong format");
1332 return 0;
1333 }
Christian Heimes72b710a2008-05-26 13:28:38 +00001334 straddr = PyBytes_AS_STRING(args);
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001335 if (setbdaddr(straddr, &_BT_SCO_MEMB(addr, bdaddr)) < 0)
1336 return 0;
1337
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001338 *len_ret = sizeof *addr;
1339 return 1;
1340 }
1341#endif
1342 default:
1343 PyErr_SetString(socket_error, "getsockaddrarg: unknown Bluetooth protocol");
1344 return 0;
Martin v. Löwis12af0482004-01-31 12:34:17 +00001345 }
1346 }
1347#endif
1348
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001349#ifdef HAVE_NETPACKET_PACKET_H
Jeremy Hylton22308652001-02-02 03:23:09 +00001350 case AF_PACKET:
1351 {
1352 struct sockaddr_ll* addr;
1353 struct ifreq ifr;
1354 char *interfaceName;
1355 int protoNumber;
1356 int hatype = 0;
1357 int pkttype = 0;
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +00001358 char *haddr = NULL;
1359 unsigned int halen = 0;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001360
Thomas Wouters477c8d52006-05-27 19:21:47 +00001361 if (!PyTuple_Check(args)) {
1362 PyErr_Format(
1363 PyExc_TypeError,
1364 "getsockaddrarg: "
1365 "AF_PACKET address must be tuple, not %.500s",
Christian Heimes90aa7642007-12-19 02:45:37 +00001366 Py_TYPE(args)->tp_name);
Thomas Wouters477c8d52006-05-27 19:21:47 +00001367 return 0;
1368 }
Guido van Rossum8a392d72007-11-21 22:09:45 +00001369 if (!PyArg_ParseTuple(args, "si|iiy#", &interfaceName,
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +00001370 &protoNumber, &pkttype, &hatype,
1371 &haddr, &halen))
Jeremy Hylton22308652001-02-02 03:23:09 +00001372 return 0;
1373 strncpy(ifr.ifr_name, interfaceName, sizeof(ifr.ifr_name));
1374 ifr.ifr_name[(sizeof(ifr.ifr_name))-1] = '\0';
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001375 if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001376 s->errorhandler();
Jeremy Hylton22308652001-02-02 03:23:09 +00001377 return 0;
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001378 }
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +00001379 if (halen > 8) {
1380 PyErr_SetString(PyExc_ValueError,
1381 "Hardware address must be 8 bytes or less");
1382 return 0;
1383 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00001384 addr = (struct sockaddr_ll*)addr_ret;
1385 addr->sll_family = AF_PACKET;
1386 addr->sll_protocol = htons((short)protoNumber);
1387 addr->sll_ifindex = ifr.ifr_ifindex;
1388 addr->sll_pkttype = pkttype;
1389 addr->sll_hatype = hatype;
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +00001390 if (halen != 0) {
1391 memcpy(&addr->sll_addr, haddr, halen);
1392 }
1393 addr->sll_halen = halen;
Jeremy Hylton22308652001-02-02 03:23:09 +00001394 *len_ret = sizeof *addr;
1395 return 1;
1396 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001397#endif
1398
Christian Heimes043d6f62008-01-07 17:19:16 +00001399#ifdef HAVE_LINUX_TIPC_H
1400 case AF_TIPC:
1401 {
1402 unsigned int atype, v1, v2, v3;
1403 unsigned int scope = TIPC_CLUSTER_SCOPE;
1404 struct sockaddr_tipc *addr;
1405
1406 if (!PyTuple_Check(args)) {
1407 PyErr_Format(
1408 PyExc_TypeError,
1409 "getsockaddrarg: "
1410 "AF_TIPC address must be tuple, not %.500s",
1411 Py_TYPE(args)->tp_name);
1412 return 0;
1413 }
1414
1415 if (!PyArg_ParseTuple(args,
1416 "IIII|I;Invalid TIPC address format",
1417 &atype, &v1, &v2, &v3, &scope))
1418 return 0;
1419
1420 addr = (struct sockaddr_tipc *) addr_ret;
1421 memset(addr, 0, sizeof(struct sockaddr_tipc));
1422
1423 addr->family = AF_TIPC;
1424 addr->scope = scope;
1425 addr->addrtype = atype;
1426
1427 if (atype == TIPC_ADDR_NAMESEQ) {
1428 addr->addr.nameseq.type = v1;
1429 addr->addr.nameseq.lower = v2;
1430 addr->addr.nameseq.upper = v3;
1431 } else if (atype == TIPC_ADDR_NAME) {
1432 addr->addr.name.name.type = v1;
1433 addr->addr.name.name.instance = v2;
1434 } else if (atype == TIPC_ADDR_ID) {
1435 addr->addr.id.node = v1;
1436 addr->addr.id.ref = v2;
1437 } else {
1438 /* Shouldn't happen */
1439 PyErr_SetString(PyExc_TypeError, "Invalid address type");
1440 return 0;
1441 }
1442
1443 *len_ret = sizeof(*addr);
1444
1445 return 1;
1446 }
1447#endif
1448
Guido van Rossum30a685f1991-06-27 15:51:29 +00001449 /* More cases here... */
1450
1451 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001452 PyErr_SetString(socket_error, "getsockaddrarg: bad family");
Guido van Rossum30a685f1991-06-27 15:51:29 +00001453 return 0;
1454
1455 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001456}
1457
Guido van Rossum30a685f1991-06-27 15:51:29 +00001458
Guido van Rossum48a680c2001-03-02 06:34:14 +00001459/* Get the address length according to the socket object's address family.
Guido van Rossum710e1df1992-06-12 10:39:36 +00001460 Return 1 if the family is known, 0 otherwise. The length is returned
1461 through len_ret. */
1462
1463static int
Peter Schneider-Kamp618e25d2000-07-11 23:00:12 +00001464getsockaddrlen(PySocketSockObject *s, socklen_t *len_ret)
Guido van Rossum710e1df1992-06-12 10:39:36 +00001465{
1466 switch (s->sock_family) {
1467
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001468#if defined(AF_UNIX)
Guido van Rossum710e1df1992-06-12 10:39:36 +00001469 case AF_UNIX:
1470 {
1471 *len_ret = sizeof (struct sockaddr_un);
1472 return 1;
1473 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001474#endif /* AF_UNIX */
Martin v. Löwis11017b12006-01-14 18:12:57 +00001475#if defined(AF_NETLINK)
1476 case AF_NETLINK:
1477 {
1478 *len_ret = sizeof (struct sockaddr_nl);
1479 return 1;
1480 }
1481#endif
Guido van Rossum710e1df1992-06-12 10:39:36 +00001482
1483 case AF_INET:
1484 {
1485 *len_ret = sizeof (struct sockaddr_in);
1486 return 1;
1487 }
1488
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001489#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001490 case AF_INET6:
1491 {
1492 *len_ret = sizeof (struct sockaddr_in6);
1493 return 1;
1494 }
1495#endif
1496
Hye-Shik Chang81268602004-02-02 06:05:24 +00001497#ifdef USE_BLUETOOTH
Martin v. Löwis12af0482004-01-31 12:34:17 +00001498 case AF_BLUETOOTH:
1499 {
1500 switch(s->sock_proto)
1501 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001502
1503 case BTPROTO_L2CAP:
1504 *len_ret = sizeof (struct sockaddr_l2);
1505 return 1;
1506 case BTPROTO_RFCOMM:
1507 *len_ret = sizeof (struct sockaddr_rc);
1508 return 1;
Thomas Wouterscf297e42007-02-23 15:07:44 +00001509 case BTPROTO_HCI:
1510 *len_ret = sizeof (struct sockaddr_hci);
1511 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001512#if !defined(__FreeBSD__)
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001513 case BTPROTO_SCO:
1514 *len_ret = sizeof (struct sockaddr_sco);
1515 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001516#endif
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001517 default:
1518 PyErr_SetString(socket_error, "getsockaddrlen: "
1519 "unknown BT protocol");
1520 return 0;
1521
Martin v. Löwis12af0482004-01-31 12:34:17 +00001522 }
1523 }
1524#endif
1525
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001526#ifdef HAVE_NETPACKET_PACKET_H
Jeremy Hylton22308652001-02-02 03:23:09 +00001527 case AF_PACKET:
1528 {
1529 *len_ret = sizeof (struct sockaddr_ll);
1530 return 1;
1531 }
1532#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001533
Christian Heimes043d6f62008-01-07 17:19:16 +00001534#ifdef HAVE_LINUX_TIPC_H
1535 case AF_TIPC:
1536 {
1537 *len_ret = sizeof (struct sockaddr_tipc);
1538 return 1;
1539 }
1540#endif
1541
Guido van Rossum710e1df1992-06-12 10:39:36 +00001542 /* More cases here... */
1543
1544 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001545 PyErr_SetString(socket_error, "getsockaddrlen: bad family");
Guido van Rossum710e1df1992-06-12 10:39:36 +00001546 return 0;
1547
1548 }
1549}
1550
1551
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001552/* s._accept() -> (fd, address) */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001553
Guido van Rossum73624e91994-10-10 17:59:00 +00001554static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001555sock_accept(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001556{
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001557 sock_addr_t addrbuf;
Guido van Rossum7d0a8262007-05-21 23:13:11 +00001558 SOCKET_T newfd = INVALID_SOCKET;
Guido van Rossumff3ab422000-04-24 15:16:03 +00001559 socklen_t addrlen;
Barry Warsaw752300b1997-01-03 17:18:10 +00001560 PyObject *sock = NULL;
1561 PyObject *addr = NULL;
1562 PyObject *res = NULL;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001563 int timeout;
Barry Warsaw752300b1997-01-03 17:18:10 +00001564
Guido van Rossum710e1df1992-06-12 10:39:36 +00001565 if (!getsockaddrlen(s, &addrlen))
1566 return NULL;
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001567 memset(&addrbuf, 0, addrlen);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001568
Neal Norwitz082b2df2006-02-07 07:04:46 +00001569 if (!IS_SELECTABLE(s))
1570 return select_error();
1571
Guido van Rossum73624e91994-10-10 17:59:00 +00001572 Py_BEGIN_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001573 timeout = internal_select(s, 0);
1574 if (!timeout)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001575 newfd = accept(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001576 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001577
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001578 if (timeout == 1) {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001579 PyErr_SetString(socket_timeout, "timed out");
1580 return NULL;
1581 }
1582
Fred Drakea04eaad2000-06-30 02:46:07 +00001583 if (newfd == INVALID_SOCKET)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001584 return s->errorhandler();
Barry Warsaw752300b1997-01-03 17:18:10 +00001585
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001586 sock = PyLong_FromSocket_t(newfd);
Barry Warsaw752300b1997-01-03 17:18:10 +00001587 if (sock == NULL) {
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +00001588 SOCKETCLOSE(newfd);
Barry Warsaw752300b1997-01-03 17:18:10 +00001589 goto finally;
1590 }
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001591
Thomas Wouters89f507f2006-12-13 04:49:30 +00001592 addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001593 addrlen, s->sock_proto);
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001594 if (addr == NULL)
Barry Warsaw752300b1997-01-03 17:18:10 +00001595 goto finally;
1596
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001597 res = PyTuple_Pack(2, sock, addr);
Barry Warsaw752300b1997-01-03 17:18:10 +00001598
Guido van Rossum67f7a382002-06-06 21:08:16 +00001599finally:
Guido van Rossum73624e91994-10-10 17:59:00 +00001600 Py_XDECREF(sock);
1601 Py_XDECREF(addr);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001602 return res;
1603}
1604
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001605PyDoc_STRVAR(accept_doc,
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001606"_accept() -> (integer, address info)\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00001607\n\
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00001608Wait for an incoming connection. Return a new socket file descriptor\n\
1609representing the connection, and the address of the client.\n\
1610For IP sockets, the address info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001611
Guido van Rossum11ba0942002-06-13 15:07:44 +00001612/* s.setblocking(flag) method. Argument:
1613 False -- non-blocking mode; same as settimeout(0)
1614 True -- blocking mode; same as settimeout(None)
1615*/
Guido van Rossume4485b01994-09-07 14:32:49 +00001616
Guido van Rossum73624e91994-10-10 17:59:00 +00001617static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001618sock_setblocking(PySocketSockObject *s, PyObject *arg)
Guido van Rossume4485b01994-09-07 14:32:49 +00001619{
1620 int block;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001621
Christian Heimes217cfd12007-12-02 14:31:20 +00001622 block = PyLong_AsLong(arg);
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001623 if (block == -1 && PyErr_Occurred())
Guido van Rossume4485b01994-09-07 14:32:49 +00001624 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001625
Guido van Rossum11ba0942002-06-13 15:07:44 +00001626 s->sock_timeout = block ? -1.0 : 0.0;
Guido van Rossume8008f02002-06-07 03:36:20 +00001627 internal_setblocking(s, block);
Guido van Rossume4485b01994-09-07 14:32:49 +00001628
Guido van Rossum73624e91994-10-10 17:59:00 +00001629 Py_INCREF(Py_None);
1630 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +00001631}
Guido van Rossume4485b01994-09-07 14:32:49 +00001632
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001633PyDoc_STRVAR(setblocking_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001634"setblocking(flag)\n\
1635\n\
1636Set the socket to blocking (flag is true) or non-blocking (false).\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001637setblocking(True) is equivalent to settimeout(None);\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001638setblocking(False) is equivalent to settimeout(0.0).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001639
Guido van Rossum11ba0942002-06-13 15:07:44 +00001640/* s.settimeout(timeout) method. Argument:
1641 None -- no timeout, blocking mode; same as setblocking(True)
1642 0.0 -- non-blocking mode; same as setblocking(False)
1643 > 0 -- timeout mode; operations time out after timeout seconds
1644 < 0 -- illegal; raises an exception
1645*/
Guido van Rossum67f7a382002-06-06 21:08:16 +00001646static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001647sock_settimeout(PySocketSockObject *s, PyObject *arg)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001648{
Guido van Rossum11ba0942002-06-13 15:07:44 +00001649 double timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001650
1651 if (arg == Py_None)
Guido van Rossum11ba0942002-06-13 15:07:44 +00001652 timeout = -1.0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001653 else {
Guido van Rossum11ba0942002-06-13 15:07:44 +00001654 timeout = PyFloat_AsDouble(arg);
1655 if (timeout < 0.0) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00001656 if (!PyErr_Occurred())
1657 PyErr_SetString(PyExc_ValueError,
Guido van Rossum11ba0942002-06-13 15:07:44 +00001658 "Timeout value out of range");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001659 return NULL;
1660 }
1661 }
1662
Guido van Rossum11ba0942002-06-13 15:07:44 +00001663 s->sock_timeout = timeout;
1664 internal_setblocking(s, timeout < 0.0);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001665
1666 Py_INCREF(Py_None);
1667 return Py_None;
1668}
1669
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001670PyDoc_STRVAR(settimeout_doc,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001671"settimeout(timeout)\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001672\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001673Set a timeout on socket operations. 'timeout' can be a float,\n\
1674giving in seconds, or None. Setting a timeout of None disables\n\
1675the timeout feature and is equivalent to setblocking(1).\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001676Setting a timeout of zero is the same as setblocking(0).");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001677
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001678/* s.gettimeout() method.
1679 Returns the timeout associated with a socket. */
Guido van Rossum67f7a382002-06-06 21:08:16 +00001680static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001681sock_gettimeout(PySocketSockObject *s)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001682{
1683 if (s->sock_timeout < 0.0) {
1684 Py_INCREF(Py_None);
1685 return Py_None;
1686 }
1687 else
1688 return PyFloat_FromDouble(s->sock_timeout);
1689}
1690
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001691PyDoc_STRVAR(gettimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001692"gettimeout() -> timeout\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001693\n\
1694Returns the timeout in floating seconds associated with socket \n\
1695operations. A timeout of None indicates that timeouts on socket \n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001696operations are disabled.");
Guido van Rossume4485b01994-09-07 14:32:49 +00001697
Guido van Rossumaee08791992-09-08 09:05:33 +00001698/* s.setsockopt() method.
1699 With an integer third argument, sets an integer option.
1700 With a string third argument, sets an option from a buffer;
1701 use optional built-in module 'struct' to encode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001702
Guido van Rossum73624e91994-10-10 17:59:00 +00001703static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001704sock_setsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001705{
1706 int level;
1707 int optname;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001708 int res;
Guido van Rossumaee08791992-09-08 09:05:33 +00001709 char *buf;
1710 int buflen;
1711 int flag;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001712
Guido van Rossum27fc3c02000-03-24 20:56:56 +00001713 if (PyArg_ParseTuple(args, "iii:setsockopt",
1714 &level, &optname, &flag)) {
Guido van Rossumaee08791992-09-08 09:05:33 +00001715 buf = (char *) &flag;
1716 buflen = sizeof flag;
1717 }
1718 else {
Guido van Rossum73624e91994-10-10 17:59:00 +00001719 PyErr_Clear();
Guido van Rossum8a392d72007-11-21 22:09:45 +00001720 if (!PyArg_ParseTuple(args, "iiy#:setsockopt",
Guido van Rossum27fc3c02000-03-24 20:56:56 +00001721 &level, &optname, &buf, &buflen))
Guido van Rossumaee08791992-09-08 09:05:33 +00001722 return NULL;
1723 }
Thomas Wouters334fb892000-07-25 12:56:38 +00001724 res = setsockopt(s->sock_fd, level, optname, (void *)buf, buflen);
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001725 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001726 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001727 Py_INCREF(Py_None);
1728 return Py_None;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001729}
1730
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001731PyDoc_STRVAR(setsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001732"setsockopt(level, option, value)\n\
1733\n\
1734Set a socket option. See the Unix manual for level and option.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001735The value argument can either be an integer or a string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001736
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001737
Guido van Rossumaee08791992-09-08 09:05:33 +00001738/* s.getsockopt() method.
1739 With two arguments, retrieves an integer option.
1740 With a third integer argument, retrieves a string buffer of that size;
1741 use optional built-in module 'struct' to decode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001742
Guido van Rossum73624e91994-10-10 17:59:00 +00001743static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001744sock_getsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001745{
1746 int level;
1747 int optname;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001748 int res;
Guido van Rossum73624e91994-10-10 17:59:00 +00001749 PyObject *buf;
Guido van Rossumff3ab422000-04-24 15:16:03 +00001750 socklen_t buflen = 0;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001751
Guido van Rossum27fc3c02000-03-24 20:56:56 +00001752 if (!PyArg_ParseTuple(args, "ii|i:getsockopt",
1753 &level, &optname, &buflen))
Guido van Rossumbe32c891996-06-20 16:25:29 +00001754 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001755
Guido van Rossumbe32c891996-06-20 16:25:29 +00001756 if (buflen == 0) {
Guido van Rossumaee08791992-09-08 09:05:33 +00001757 int flag = 0;
Guido van Rossumff3ab422000-04-24 15:16:03 +00001758 socklen_t flagsize = sizeof flag;
Guido van Rossumb376a4a1993-11-23 17:53:17 +00001759 res = getsockopt(s->sock_fd, level, optname,
Thomas Wouters334fb892000-07-25 12:56:38 +00001760 (void *)&flag, &flagsize);
Guido van Rossumaee08791992-09-08 09:05:33 +00001761 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001762 return s->errorhandler();
Christian Heimes217cfd12007-12-02 14:31:20 +00001763 return PyLong_FromLong(flag);
Guido van Rossumaee08791992-09-08 09:05:33 +00001764 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001765#ifdef __VMS
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001766 /* socklen_t is unsigned so no negative test is needed,
1767 test buflen == 0 is previously done */
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001768 if (buflen > 1024) {
1769#else
Guido van Rossumaee08791992-09-08 09:05:33 +00001770 if (buflen <= 0 || buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001771#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001772 PyErr_SetString(socket_error,
Barry Warsaw752300b1997-01-03 17:18:10 +00001773 "getsockopt buflen out of range");
Guido van Rossumaee08791992-09-08 09:05:33 +00001774 return NULL;
1775 }
Christian Heimes72b710a2008-05-26 13:28:38 +00001776 buf = PyBytes_FromStringAndSize((char *)NULL, buflen);
Guido van Rossumaee08791992-09-08 09:05:33 +00001777 if (buf == NULL)
1778 return NULL;
Guido van Rossumb376a4a1993-11-23 17:53:17 +00001779 res = getsockopt(s->sock_fd, level, optname,
Christian Heimes72b710a2008-05-26 13:28:38 +00001780 (void *)PyBytes_AS_STRING(buf), &buflen);
Guido van Rossumaee08791992-09-08 09:05:33 +00001781 if (res < 0) {
Guido van Rossum73624e91994-10-10 17:59:00 +00001782 Py_DECREF(buf);
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001783 return s->errorhandler();
Guido van Rossumaee08791992-09-08 09:05:33 +00001784 }
Christian Heimes72b710a2008-05-26 13:28:38 +00001785 _PyBytes_Resize(&buf, buflen);
Guido van Rossumaee08791992-09-08 09:05:33 +00001786 return buf;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001787}
1788
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001789PyDoc_STRVAR(getsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001790"getsockopt(level, option[, buffersize]) -> value\n\
1791\n\
1792Get a socket option. See the Unix manual for level and option.\n\
1793If a nonzero buffersize argument is given, the return value is a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001794string of that length; otherwise it is an integer.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001795
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001796
Fred Drake728819a2000-07-01 03:40:12 +00001797/* s.bind(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001798
Guido van Rossum73624e91994-10-10 17:59:00 +00001799static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001800sock_bind(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001801{
Thomas Wouters89f507f2006-12-13 04:49:30 +00001802 sock_addr_t addrbuf;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001803 int addrlen;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001804 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001805
Thomas Wouters89f507f2006-12-13 04:49:30 +00001806 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001807 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00001808 Py_BEGIN_ALLOW_THREADS
Thomas Wouters89f507f2006-12-13 04:49:30 +00001809 res = bind(s->sock_fd, SAS2SA(&addrbuf), addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001810 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001811 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001812 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001813 Py_INCREF(Py_None);
1814 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001815}
1816
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001817PyDoc_STRVAR(bind_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001818"bind(address)\n\
1819\n\
1820Bind the socket to a local address. For IP sockets, the address is a\n\
Jeremy Hylton22308652001-02-02 03:23:09 +00001821pair (host, port); the host must refer to the local host. For raw packet\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001822sockets the address is a tuple (ifname, proto [,pkttype [,hatype]])");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001823
Guido van Rossum30a685f1991-06-27 15:51:29 +00001824
1825/* s.close() method.
1826 Set the file descriptor to -1 so operations tried subsequently
1827 will surely fail. */
1828
Guido van Rossum73624e91994-10-10 17:59:00 +00001829static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001830sock_close(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001831{
Guido van Rossum20d3fc02000-12-18 22:23:44 +00001832 SOCKET_T fd;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001833
Guido van Rossum20d3fc02000-12-18 22:23:44 +00001834 if ((fd = s->sock_fd) != -1) {
1835 s->sock_fd = -1;
Guido van Rossum08481461996-10-12 14:07:22 +00001836 Py_BEGIN_ALLOW_THREADS
Guido van Rossum20d3fc02000-12-18 22:23:44 +00001837 (void) SOCKETCLOSE(fd);
Guido van Rossum08481461996-10-12 14:07:22 +00001838 Py_END_ALLOW_THREADS
1839 }
Guido van Rossum73624e91994-10-10 17:59:00 +00001840 Py_INCREF(Py_None);
1841 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001842}
1843
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001844PyDoc_STRVAR(close_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001845"close()\n\
1846\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001847Close the socket. It cannot be used after this call.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001848
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001849static int
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001850internal_connect(PySocketSockObject *s, struct sockaddr *addr, int addrlen,
1851 int *timeoutp)
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001852{
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001853 int res, timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001854
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001855 timeout = 0;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001856 res = connect(s->sock_fd, addr, addrlen);
1857
1858#ifdef MS_WINDOWS
1859
1860 if (s->sock_timeout > 0.0) {
Neal Norwitz082b2df2006-02-07 07:04:46 +00001861 if (res < 0 && WSAGetLastError() == WSAEWOULDBLOCK &&
1862 IS_SELECTABLE(s)) {
Guido van Rossumb76bdf82003-02-19 17:50:16 +00001863 /* This is a mess. Best solution: trust select */
1864 fd_set fds;
Mark Hammonda57ec932004-08-03 05:06:26 +00001865 fd_set fds_exc;
Guido van Rossumb76bdf82003-02-19 17:50:16 +00001866 struct timeval tv;
1867 tv.tv_sec = (int)s->sock_timeout;
1868 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
1869 FD_ZERO(&fds);
1870 FD_SET(s->sock_fd, &fds);
Mark Hammonda57ec932004-08-03 05:06:26 +00001871 FD_ZERO(&fds_exc);
1872 FD_SET(s->sock_fd, &fds_exc);
1873 res = select(s->sock_fd+1, NULL, &fds, &fds_exc, &tv);
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001874 if (res == 0) {
Guido van Rossumb76bdf82003-02-19 17:50:16 +00001875 res = WSAEWOULDBLOCK;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001876 timeout = 1;
Mark Hammonda57ec932004-08-03 05:06:26 +00001877 } else if (res > 0) {
1878 if (FD_ISSET(s->sock_fd, &fds))
Sean Reifscheider54cf12b2007-09-17 17:55:36 +00001879 /* The socket is in the writable set - this
Mark Hammonda57ec932004-08-03 05:06:26 +00001880 means connected */
1881 res = 0;
1882 else {
1883 /* As per MS docs, we need to call getsockopt()
1884 to get the underlying error */
1885 int res_size = sizeof res;
1886 /* It must be in the exception set */
1887 assert(FD_ISSET(s->sock_fd, &fds_exc));
Thomas Wouters477c8d52006-05-27 19:21:47 +00001888 if (0 == getsockopt(s->sock_fd, SOL_SOCKET, SO_ERROR,
Mark Hammonda57ec932004-08-03 05:06:26 +00001889 (char *)&res, &res_size))
1890 /* getsockopt also clears WSAGetLastError,
1891 so reset it back. */
1892 WSASetLastError(res);
1893 else
1894 res = WSAGetLastError();
1895 }
1896 }
Guido van Rossumb76bdf82003-02-19 17:50:16 +00001897 /* else if (res < 0) an error occurred */
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001898 }
1899 }
1900
1901 if (res < 0)
1902 res = WSAGetLastError();
1903
1904#else
1905
1906 if (s->sock_timeout > 0.0) {
Christian Heimesfdab48e2008-01-20 09:06:41 +00001907 if (res < 0 && errno == EINPROGRESS && IS_SELECTABLE(s)) {
1908 timeout = internal_select(s, 1);
1909 if (timeout == 0) {
1910 /* Bug #1019808: in case of an EINPROGRESS,
1911 use getsockopt(SO_ERROR) to get the real
1912 error. */
1913 socklen_t res_size = sizeof res;
1914 (void)getsockopt(s->sock_fd, SOL_SOCKET,
1915 SO_ERROR, &res, &res_size);
1916 if (res == EISCONN)
1917 res = 0;
1918 errno = res;
1919 }
1920 else if (timeout == -1) {
1921 res = errno; /* had error */
1922 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001923 else
1924 res = EWOULDBLOCK; /* timed out */
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001925 }
1926 }
1927
1928 if (res < 0)
1929 res = errno;
1930
1931#endif
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001932 *timeoutp = timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001933
1934 return res;
1935}
Guido van Rossum30a685f1991-06-27 15:51:29 +00001936
Fred Drake728819a2000-07-01 03:40:12 +00001937/* s.connect(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001938
Guido van Rossum73624e91994-10-10 17:59:00 +00001939static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001940sock_connect(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001941{
Thomas Wouters89f507f2006-12-13 04:49:30 +00001942 sock_addr_t addrbuf;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001943 int addrlen;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001944 int res;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001945 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001946
Thomas Wouters89f507f2006-12-13 04:49:30 +00001947 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001948 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001949
Guido van Rossum73624e91994-10-10 17:59:00 +00001950 Py_BEGIN_ALLOW_THREADS
Thomas Wouters89f507f2006-12-13 04:49:30 +00001951 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
Guido van Rossum11ba0942002-06-13 15:07:44 +00001952 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001953
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001954 if (timeout == 1) {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001955 PyErr_SetString(socket_timeout, "timed out");
1956 return NULL;
1957 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001958 if (res != 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001959 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001960 Py_INCREF(Py_None);
1961 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001962}
1963
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001964PyDoc_STRVAR(connect_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001965"connect(address)\n\
1966\n\
1967Connect the socket to a remote address. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001968is a pair (host, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001969
Guido van Rossum30a685f1991-06-27 15:51:29 +00001970
Fred Drake728819a2000-07-01 03:40:12 +00001971/* s.connect_ex(sockaddr) method */
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001972
1973static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001974sock_connect_ex(PySocketSockObject *s, PyObject *addro)
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001975{
Thomas Wouters89f507f2006-12-13 04:49:30 +00001976 sock_addr_t addrbuf;
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001977 int addrlen;
1978 int res;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001979 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001980
Thomas Wouters89f507f2006-12-13 04:49:30 +00001981 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001982 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001983
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001984 Py_BEGIN_ALLOW_THREADS
Thomas Wouters89f507f2006-12-13 04:49:30 +00001985 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001986 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001987
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001988 /* Signals are not errors (though they may raise exceptions). Adapted
1989 from PyErr_SetFromErrnoWithFilenameObject(). */
1990#ifdef EINTR
1991 if (res == EINTR && PyErr_CheckSignals())
1992 return NULL;
1993#endif
1994
Christian Heimes217cfd12007-12-02 14:31:20 +00001995 return PyLong_FromLong((long) res);
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001996}
1997
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001998PyDoc_STRVAR(connect_ex_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001999"connect_ex(address) -> errno\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002000\n\
2001This is like connect(address), but returns an error code (the errno value)\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002002instead of raising an exception when an error occurs.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002003
Guido van Rossumfc4255d1997-11-19 18:57:13 +00002004
Guido van Rossumed233a51992-06-23 09:07:03 +00002005/* s.fileno() method */
2006
Guido van Rossum73624e91994-10-10 17:59:00 +00002007static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002008sock_fileno(PySocketSockObject *s)
Guido van Rossumed233a51992-06-23 09:07:03 +00002009{
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00002010 return PyLong_FromSocket_t(s->sock_fd);
Guido van Rossumed233a51992-06-23 09:07:03 +00002011}
2012
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002013PyDoc_STRVAR(fileno_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002014"fileno() -> integer\n\
2015\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002016Return the integer file descriptor of the socket.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002017
Guido van Rossumed233a51992-06-23 09:07:03 +00002018
Guido van Rossumc89705d1992-11-26 08:54:07 +00002019/* s.getsockname() method */
2020
Guido van Rossum73624e91994-10-10 17:59:00 +00002021static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002022sock_getsockname(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002023{
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00002024 sock_addr_t addrbuf;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002025 int res;
2026 socklen_t addrlen;
2027
Guido van Rossumc89705d1992-11-26 08:54:07 +00002028 if (!getsockaddrlen(s, &addrlen))
2029 return NULL;
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00002030 memset(&addrbuf, 0, addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00002031 Py_BEGIN_ALLOW_THREADS
Thomas Wouters89f507f2006-12-13 04:49:30 +00002032 res = getsockname(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00002033 Py_END_ALLOW_THREADS
Guido van Rossumc89705d1992-11-26 08:54:07 +00002034 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002035 return s->errorhandler();
Thomas Wouters89f507f2006-12-13 04:49:30 +00002036 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00002037 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002038}
2039
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002040PyDoc_STRVAR(getsockname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002041"getsockname() -> address info\n\
2042\n\
2043Return the address of the local endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002044info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002045
Guido van Rossumc89705d1992-11-26 08:54:07 +00002046
Guido van Rossumb6775db1994-08-01 11:34:53 +00002047#ifdef HAVE_GETPEERNAME /* Cray APP doesn't have this :-( */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002048/* s.getpeername() method */
2049
Guido van Rossum73624e91994-10-10 17:59:00 +00002050static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002051sock_getpeername(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00002052{
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00002053 sock_addr_t addrbuf;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002054 int res;
2055 socklen_t addrlen;
2056
Guido van Rossumc89705d1992-11-26 08:54:07 +00002057 if (!getsockaddrlen(s, &addrlen))
2058 return NULL;
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00002059 memset(&addrbuf, 0, addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00002060 Py_BEGIN_ALLOW_THREADS
Thomas Wouters89f507f2006-12-13 04:49:30 +00002061 res = getpeername(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00002062 Py_END_ALLOW_THREADS
Guido van Rossumc89705d1992-11-26 08:54:07 +00002063 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002064 return s->errorhandler();
Thomas Wouters89f507f2006-12-13 04:49:30 +00002065 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00002066 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00002067}
Guido van Rossum82a5c661998-07-07 20:45:43 +00002068
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002069PyDoc_STRVAR(getpeername_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002070"getpeername() -> address info\n\
2071\n\
2072Return the address of the remote endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002073info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002074
Guido van Rossumb6775db1994-08-01 11:34:53 +00002075#endif /* HAVE_GETPEERNAME */
Guido van Rossumc89705d1992-11-26 08:54:07 +00002076
2077
Guido van Rossum30a685f1991-06-27 15:51:29 +00002078/* s.listen(n) method */
2079
Guido van Rossum73624e91994-10-10 17:59:00 +00002080static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002081sock_listen(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002082{
2083 int backlog;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002084 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002085
Christian Heimes217cfd12007-12-02 14:31:20 +00002086 backlog = PyLong_AsLong(arg);
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002087 if (backlog == -1 && PyErr_Occurred())
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002088 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00002089 Py_BEGIN_ALLOW_THREADS
Guido van Rossumb6775db1994-08-01 11:34:53 +00002090 if (backlog < 1)
2091 backlog = 1;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002092 res = listen(s->sock_fd, backlog);
Guido van Rossum73624e91994-10-10 17:59:00 +00002093 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002094 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002095 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00002096 Py_INCREF(Py_None);
2097 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002098}
2099
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002100PyDoc_STRVAR(listen_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002101"listen(backlog)\n\
2102\n\
2103Enable a server to accept connections. The backlog argument must be at\n\
2104least 1; it specifies the number of unaccepted connection that the system\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002105will allow before refusing new connections.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002106
2107
Thomas Wouters477c8d52006-05-27 19:21:47 +00002108/*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002109 * This is the guts of the recv() and recv_into() methods, which reads into a
Thomas Wouters902d6eb2007-01-09 23:18:33 +00002110 * char buffer. If you have any inc/dec ref to do to the objects that contain
Thomas Wouters477c8d52006-05-27 19:21:47 +00002111 * the buffer, do it in the caller. This function returns the number of bytes
2112 * succesfully read. If there was an error, it returns -1. Note that it is
2113 * also possible that we return a number of bytes smaller than the request
2114 * bytes.
2115 */
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002116static ssize_t
Thomas Wouters477c8d52006-05-27 19:21:47 +00002117sock_recv_guts(PySocketSockObject *s, char* cbuf, int len, int flags)
2118{
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002119 ssize_t outlen = -1;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002120 int timeout;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002121#ifdef __VMS
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002122 int remaining;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002123 char *read_buf;
2124#endif
2125
2126 if (!IS_SELECTABLE(s)) {
2127 select_error();
2128 return -1;
2129 }
Guido van Rossumc2de7c02007-08-03 22:27:51 +00002130 if (len == 0) {
2131 /* If 0 bytes were requested, do nothing. */
2132 return 0;
2133 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002134
2135#ifndef __VMS
2136 Py_BEGIN_ALLOW_THREADS
2137 timeout = internal_select(s, 0);
2138 if (!timeout)
2139 outlen = recv(s->sock_fd, cbuf, len, flags);
2140 Py_END_ALLOW_THREADS
2141
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002142 if (timeout == 1) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002143 PyErr_SetString(socket_timeout, "timed out");
2144 return -1;
2145 }
2146 if (outlen < 0) {
2147 /* Note: the call to errorhandler() ALWAYS indirectly returned
2148 NULL, so ignore its return value */
2149 s->errorhandler();
2150 return -1;
2151 }
2152#else
2153 read_buf = cbuf;
2154 remaining = len;
2155 while (remaining != 0) {
2156 unsigned int segment;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002157 int nread = -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002158
2159 segment = remaining /SEGMENT_SIZE;
2160 if (segment != 0) {
2161 segment = SEGMENT_SIZE;
2162 }
2163 else {
2164 segment = remaining;
2165 }
2166
2167 Py_BEGIN_ALLOW_THREADS
2168 timeout = internal_select(s, 0);
2169 if (!timeout)
2170 nread = recv(s->sock_fd, read_buf, segment, flags);
2171 Py_END_ALLOW_THREADS
2172
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002173 if (timeout == 1) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002174 PyErr_SetString(socket_timeout, "timed out");
2175 return -1;
2176 }
2177 if (nread < 0) {
2178 s->errorhandler();
2179 return -1;
2180 }
2181 if (nread != remaining) {
2182 read_buf += nread;
2183 break;
2184 }
2185
2186 remaining -= segment;
2187 read_buf += segment;
2188 }
2189 outlen = read_buf - cbuf;
2190#endif /* !__VMS */
2191
2192 return outlen;
2193}
2194
Guido van Rossum48a680c2001-03-02 06:34:14 +00002195
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002196/* s.recv(nbytes [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002197
Guido van Rossum73624e91994-10-10 17:59:00 +00002198static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002199sock_recv(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002200{
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002201 int recvlen, flags = 0;
2202 ssize_t outlen;
Guido van Rossum73624e91994-10-10 17:59:00 +00002203 PyObject *buf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002204
Thomas Wouters477c8d52006-05-27 19:21:47 +00002205 if (!PyArg_ParseTuple(args, "i|i:recv", &recvlen, &flags))
Guido van Rossumbe32c891996-06-20 16:25:29 +00002206 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002207
Thomas Wouters477c8d52006-05-27 19:21:47 +00002208 if (recvlen < 0) {
Martin v. Löwisfba64e12001-11-19 10:41:26 +00002209 PyErr_SetString(PyExc_ValueError,
Guido van Rossum11ba0942002-06-13 15:07:44 +00002210 "negative buffersize in recv");
Martin v. Löwisfba64e12001-11-19 10:41:26 +00002211 return NULL;
2212 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002213
Thomas Wouters477c8d52006-05-27 19:21:47 +00002214 /* Allocate a new string. */
Christian Heimes72b710a2008-05-26 13:28:38 +00002215 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002216 if (buf == NULL)
2217 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002218
Thomas Wouters477c8d52006-05-27 19:21:47 +00002219 /* Call the guts */
Christian Heimes72b710a2008-05-26 13:28:38 +00002220 outlen = sock_recv_guts(s, PyBytes_AS_STRING(buf), recvlen, flags);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002221 if (outlen < 0) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002222 /* An error occurred, release the string and return an
Thomas Wouters477c8d52006-05-27 19:21:47 +00002223 error. */
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002224 Py_DECREF(buf);
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002225 return NULL;
2226 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002227 if (outlen != recvlen) {
2228 /* We did not read as many bytes as we anticipated, resize the
Kurt B. Kaisere4050372007-08-22 21:38:31 +00002229 string if possible and be successful. */
Christian Heimes72b710a2008-05-26 13:28:38 +00002230 _PyBytes_Resize(&buf, outlen);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002231 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002232
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002233 return buf;
2234}
2235
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002236PyDoc_STRVAR(recv_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002237"recv(buffersize[, flags]) -> data\n\
2238\n\
2239Receive up to buffersize bytes from the socket. For the optional flags\n\
2240argument, see the Unix manual. When no data is available, block until\n\
2241at least one byte is available or until the remote end is closed. When\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002242the remote end is closed and all data is read, return the empty string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002243
Guido van Rossum30a685f1991-06-27 15:51:29 +00002244
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002245/* s.recv_into(buffer, [nbytes [,flags]]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002246
Thomas Wouters477c8d52006-05-27 19:21:47 +00002247static PyObject*
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002248sock_recv_into(PySocketSockObject *s, PyObject *args, PyObject *kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002249{
2250 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
2251
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002252 int recvlen = 0, flags = 0;
2253 ssize_t readlen;
Martin v. Löwis423be952008-08-13 15:53:07 +00002254 Py_buffer pbuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002255 char *buf;
2256 int buflen;
2257
2258 /* Get the buffer's memory */
Martin v. Löwis423be952008-08-13 15:53:07 +00002259 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ii:recv_into", kwlist,
2260 &pbuf, &recvlen, &flags))
Thomas Wouters477c8d52006-05-27 19:21:47 +00002261 return NULL;
Martin v. Löwis423be952008-08-13 15:53:07 +00002262 buf = pbuf.buf;
2263 buflen = pbuf.len;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002264
2265 if (recvlen < 0) {
Martin v. Löwis423be952008-08-13 15:53:07 +00002266 PyBuffer_Release(&pbuf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002267 PyErr_SetString(PyExc_ValueError,
Guido van Rossumd8faa362007-04-27 19:54:29 +00002268 "negative buffersize in recv_into");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002269 return NULL;
2270 }
2271 if (recvlen == 0) {
2272 /* If nbytes was not specified, use the buffer's length */
2273 recvlen = buflen;
2274 }
2275
2276 /* Check if the buffer is large enough */
2277 if (buflen < recvlen) {
Martin v. Löwis423be952008-08-13 15:53:07 +00002278 PyBuffer_Release(&pbuf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002279 PyErr_SetString(PyExc_ValueError,
2280 "buffer too small for requested bytes");
2281 return NULL;
2282 }
2283
2284 /* Call the guts */
2285 readlen = sock_recv_guts(s, buf, recvlen, flags);
2286 if (readlen < 0) {
2287 /* Return an error. */
Martin v. Löwis423be952008-08-13 15:53:07 +00002288 PyBuffer_Release(&pbuf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002289 return NULL;
2290 }
2291
Martin v. Löwis423be952008-08-13 15:53:07 +00002292 PyBuffer_Release(&pbuf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002293 /* Return the number of bytes read. Note that we do not do anything
2294 special here in the case that readlen < recvlen. */
Christian Heimes217cfd12007-12-02 14:31:20 +00002295 return PyLong_FromSsize_t(readlen);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002296}
2297
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002298PyDoc_STRVAR(recv_into_doc,
2299"recv_into(buffer, [nbytes[, flags]]) -> nbytes_read\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002300\n\
2301A version of recv() that stores its data into a buffer rather than creating \n\
2302a new string. Receive up to buffersize bytes from the socket. If buffersize \n\
2303is not specified (or 0), receive up to the size available in the given buffer.\n\
2304\n\
2305See recv() for documentation about the flags.");
2306
2307
2308/*
Christian Heimes99170a52007-12-19 02:07:34 +00002309 * This is the guts of the recvfrom() and recvfrom_into() methods, which reads
2310 * into a char buffer. If you have any inc/def ref to do to the objects that
2311 * contain the buffer, do it in the caller. This function returns the number
2312 * of bytes succesfully read. If there was an error, it returns -1. Note
2313 * that it is also possible that we return a number of bytes smaller than the
2314 * request bytes.
Thomas Wouters477c8d52006-05-27 19:21:47 +00002315 *
2316 * 'addr' is a return value for the address object. Note that you must decref
2317 * it yourself.
2318 */
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002319static ssize_t
Thomas Wouters477c8d52006-05-27 19:21:47 +00002320sock_recvfrom_guts(PySocketSockObject *s, char* cbuf, int len, int flags,
2321 PyObject** addr)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002322{
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00002323 sock_addr_t addrbuf;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002324 int timeout;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002325 ssize_t n = -1;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002326 socklen_t addrlen;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002327
Thomas Wouters477c8d52006-05-27 19:21:47 +00002328 *addr = NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002329
Guido van Rossum18c9a4f1993-05-25 12:16:29 +00002330 if (!getsockaddrlen(s, &addrlen))
Thomas Wouters477c8d52006-05-27 19:21:47 +00002331 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002332
Thomas Wouters477c8d52006-05-27 19:21:47 +00002333 if (!IS_SELECTABLE(s)) {
2334 select_error();
2335 return -1;
2336 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002337
Guido van Rossum73624e91994-10-10 17:59:00 +00002338 Py_BEGIN_ALLOW_THREADS
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00002339 memset(&addrbuf, 0, addrlen);
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002340 timeout = internal_select(s, 0);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002341 if (!timeout) {
Guido van Rossum8d665e61996-06-26 18:22:49 +00002342#ifndef MS_WINDOWS
Andrew MacIntyreba43e872002-03-03 03:03:52 +00002343#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002344 n = recvfrom(s->sock_fd, cbuf, len, flags,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002345 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002346#else
Thomas Wouters477c8d52006-05-27 19:21:47 +00002347 n = recvfrom(s->sock_fd, cbuf, len, flags,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002348 (void *) &addrbuf, &addrlen);
Guido van Rossum32c575d1997-12-02 20:37:32 +00002349#endif
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002350#else
Thomas Wouters477c8d52006-05-27 19:21:47 +00002351 n = recvfrom(s->sock_fd, cbuf, len, flags,
Thomas Wouters89f507f2006-12-13 04:49:30 +00002352 SAS2SA(&addrbuf), &addrlen);
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002353#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002354 }
Guido van Rossum73624e91994-10-10 17:59:00 +00002355 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002356
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002357 if (timeout == 1) {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002358 PyErr_SetString(socket_timeout, "timed out");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002359 return -1;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002360 }
Guido van Rossum7c53b771995-09-13 18:39:47 +00002361 if (n < 0) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00002362 s->errorhandler();
2363 return -1;
Guido van Rossum7c53b771995-09-13 18:39:47 +00002364 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002365
Thomas Wouters89f507f2006-12-13 04:49:30 +00002366 if (!(*addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
Thomas Wouters477c8d52006-05-27 19:21:47 +00002367 addrlen, s->sock_proto)))
2368 return -1;
2369
2370 return n;
2371}
2372
2373/* s.recvfrom(nbytes [,flags]) method */
2374
2375static PyObject *
2376sock_recvfrom(PySocketSockObject *s, PyObject *args)
2377{
2378 PyObject *buf = NULL;
2379 PyObject *addr = NULL;
2380 PyObject *ret = NULL;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002381 int recvlen, flags = 0;
2382 ssize_t outlen;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002383
2384 if (!PyArg_ParseTuple(args, "i|i:recvfrom", &recvlen, &flags))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002385 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00002386
Guido van Rossumd8faa362007-04-27 19:54:29 +00002387 if (recvlen < 0) {
2388 PyErr_SetString(PyExc_ValueError,
2389 "negative buffersize in recvfrom");
2390 return NULL;
2391 }
2392
Christian Heimes72b710a2008-05-26 13:28:38 +00002393 buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002394 if (buf == NULL)
2395 return NULL;
2396
Christian Heimes72b710a2008-05-26 13:28:38 +00002397 outlen = sock_recvfrom_guts(s, PyBytes_AS_STRING(buf),
Thomas Wouters477c8d52006-05-27 19:21:47 +00002398 recvlen, flags, &addr);
2399 if (outlen < 0) {
Barry Warsaw752300b1997-01-03 17:18:10 +00002400 goto finally;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002401 }
2402
2403 if (outlen != recvlen) {
2404 /* We did not read as many bytes as we anticipated, resize the
2405 string if possible and be succesful. */
Christian Heimes72b710a2008-05-26 13:28:38 +00002406 if (_PyBytes_Resize(&buf, outlen) < 0)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002407 /* Oopsy, not so succesful after all. */
2408 goto finally;
2409 }
Barry Warsaw752300b1997-01-03 17:18:10 +00002410
Raymond Hettinger8ae46892003-10-12 19:09:37 +00002411 ret = PyTuple_Pack(2, buf, addr);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002412
2413finally:
Guido van Rossum73624e91994-10-10 17:59:00 +00002414 Py_XDECREF(buf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002415 Py_XDECREF(addr);
Guido van Rossum6f5afc91993-02-05 09:46:15 +00002416 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002417}
2418
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002419PyDoc_STRVAR(recvfrom_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002420"recvfrom(buffersize[, flags]) -> (data, address info)\n\
2421\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002422Like recv(buffersize, flags) but also return the sender's address info.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002423
Thomas Wouters477c8d52006-05-27 19:21:47 +00002424
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002425/* s.recvfrom_into(buffer[, nbytes [,flags]]) method */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002426
2427static PyObject *
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002428sock_recvfrom_into(PySocketSockObject *s, PyObject *args, PyObject* kwds)
Thomas Wouters477c8d52006-05-27 19:21:47 +00002429{
2430 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
2431
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002432 int recvlen = 0, flags = 0;
2433 ssize_t readlen;
Martin v. Löwis423be952008-08-13 15:53:07 +00002434 Py_buffer pbuf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002435 char *buf;
2436 int buflen;
2437
2438 PyObject *addr = NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002439
Martin v. Löwis423be952008-08-13 15:53:07 +00002440 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ii:recvfrom_into",
2441 kwlist, &pbuf,
Guido van Rossumd8faa362007-04-27 19:54:29 +00002442 &recvlen, &flags))
Thomas Wouters477c8d52006-05-27 19:21:47 +00002443 return NULL;
Martin v. Löwis423be952008-08-13 15:53:07 +00002444 buf = pbuf.buf;
2445 buflen = pbuf.len;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002446 assert(buf != 0 && buflen > 0);
2447
2448 if (recvlen < 0) {
Martin v. Löwis423be952008-08-13 15:53:07 +00002449 PyBuffer_Release(&pbuf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002450 PyErr_SetString(PyExc_ValueError,
Guido van Rossumd8faa362007-04-27 19:54:29 +00002451 "negative buffersize in recvfrom_into");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002452 return NULL;
2453 }
2454 if (recvlen == 0) {
2455 /* If nbytes was not specified, use the buffer's length */
2456 recvlen = buflen;
2457 }
2458
2459 readlen = sock_recvfrom_guts(s, buf, recvlen, flags, &addr);
2460 if (readlen < 0) {
Martin v. Löwis423be952008-08-13 15:53:07 +00002461 PyBuffer_Release(&pbuf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002462 /* Return an error */
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002463 Py_XDECREF(addr);
2464 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002465 }
2466
Martin v. Löwis423be952008-08-13 15:53:07 +00002467 PyBuffer_Release(&pbuf);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002468 /* Return the number of bytes read and the address. Note that we do
2469 not do anything special here in the case that readlen < recvlen. */
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002470 return Py_BuildValue("lN", readlen, addr);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002471}
2472
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002473PyDoc_STRVAR(recvfrom_into_doc,
2474"recvfrom_into(buffer[, nbytes[, flags]]) -> (nbytes, address info)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00002475\n\
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002476Like recv_into(buffer[, nbytes[, flags]]) but also return the sender's address info.");
Thomas Wouters477c8d52006-05-27 19:21:47 +00002477
2478
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002479/* s.send(data [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002480
Guido van Rossum73624e91994-10-10 17:59:00 +00002481static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002482sock_send(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002483{
Guido van Rossumff4949e1992-08-05 19:58:53 +00002484 char *buf;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002485 int len, n = -1, flags = 0, timeout;
Martin v. Löwis423be952008-08-13 15:53:07 +00002486 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002487
Martin v. Löwis423be952008-08-13 15:53:07 +00002488 if (!PyArg_ParseTuple(args, "y*|i:send", &pbuf, &flags))
Guido van Rossumbe32c891996-06-20 16:25:29 +00002489 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002490
Martin v. Löwis423be952008-08-13 15:53:07 +00002491 if (!IS_SELECTABLE(s)) {
2492 PyBuffer_Release(&pbuf);
Neal Norwitz082b2df2006-02-07 07:04:46 +00002493 return select_error();
Martin v. Löwis423be952008-08-13 15:53:07 +00002494 }
2495 buf = pbuf.buf;
2496 len = pbuf.len;
Neal Norwitz082b2df2006-02-07 07:04:46 +00002497
Guido van Rossum73624e91994-10-10 17:59:00 +00002498 Py_BEGIN_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002499 timeout = internal_select(s, 1);
2500 if (!timeout)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002501#ifdef __VMS
2502 n = sendsegmented(s->sock_fd, buf, len, flags);
2503#else
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002504 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002505#endif
Guido van Rossum73624e91994-10-10 17:59:00 +00002506 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002507
Martin v. Löwis423be952008-08-13 15:53:07 +00002508 PyBuffer_Release(&pbuf);
2509
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002510 if (timeout == 1) {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002511 PyErr_SetString(socket_timeout, "timed out");
2512 return NULL;
2513 }
Guido van Rossum30a685f1991-06-27 15:51:29 +00002514 if (n < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002515 return s->errorhandler();
Christian Heimes217cfd12007-12-02 14:31:20 +00002516 return PyLong_FromLong((long)n);
Guido van Rossum30a685f1991-06-27 15:51:29 +00002517}
2518
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002519PyDoc_STRVAR(send_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002520"send(data[, flags]) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002521\n\
2522Send a data string to the socket. For the optional flags\n\
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002523argument, see the Unix manual. Return the number of bytes\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002524sent; this may be less than len(data) if the network is busy.");
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002525
2526
2527/* s.sendall(data [,flags]) method */
2528
2529static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002530sock_sendall(PySocketSockObject *s, PyObject *args)
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002531{
2532 char *buf;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002533 int len, n = -1, flags = 0, timeout;
Martin v. Löwis423be952008-08-13 15:53:07 +00002534 Py_buffer pbuf;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002535
Martin v. Löwis423be952008-08-13 15:53:07 +00002536 if (!PyArg_ParseTuple(args, "y*|i:sendall", &pbuf, &flags))
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002537 return NULL;
Martin v. Löwis423be952008-08-13 15:53:07 +00002538 buf = pbuf.buf;
2539 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002540
Martin v. Löwis423be952008-08-13 15:53:07 +00002541 if (!IS_SELECTABLE(s)) {
2542 PyBuffer_Release(&pbuf);
Neal Norwitz082b2df2006-02-07 07:04:46 +00002543 return select_error();
Martin v. Löwis423be952008-08-13 15:53:07 +00002544 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002545
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002546 Py_BEGIN_ALLOW_THREADS
2547 do {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002548 timeout = internal_select(s, 1);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002549 n = -1;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002550 if (timeout)
2551 break;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002552#ifdef __VMS
2553 n = sendsegmented(s->sock_fd, buf, len, flags);
2554#else
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002555 n = send(s->sock_fd, buf, len, flags);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002556#endif
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002557 if (n < 0)
2558 break;
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002559 buf += n;
2560 len -= n;
2561 } while (len > 0);
2562 Py_END_ALLOW_THREADS
Martin v. Löwis423be952008-08-13 15:53:07 +00002563 PyBuffer_Release(&pbuf);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002564
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002565 if (timeout == 1) {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002566 PyErr_SetString(socket_timeout, "timed out");
2567 return NULL;
2568 }
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002569 if (n < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002570 return s->errorhandler();
Guido van Rossum67f7a382002-06-06 21:08:16 +00002571
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002572 Py_INCREF(Py_None);
2573 return Py_None;
2574}
2575
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002576PyDoc_STRVAR(sendall_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002577"sendall(data[, flags])\n\
2578\n\
2579Send a data string to the socket. For the optional flags\n\
2580argument, see the Unix manual. This calls send() repeatedly\n\
2581until all data is sent. If an error occurs, it's impossible\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002582to tell how much data has been sent.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002583
Guido van Rossum30a685f1991-06-27 15:51:29 +00002584
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002585/* s.sendto(data, [flags,] sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002586
Guido van Rossum73624e91994-10-10 17:59:00 +00002587static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002588sock_sendto(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002589{
Martin v. Löwis423be952008-08-13 15:53:07 +00002590 Py_buffer pbuf;
Guido van Rossum73624e91994-10-10 17:59:00 +00002591 PyObject *addro;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002592 char *buf;
Martin v. Löwis423be952008-08-13 15:53:07 +00002593 Py_ssize_t len;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002594 sock_addr_t addrbuf;
Martin v. Löwis423be952008-08-13 15:53:07 +00002595 int addrlen, n = -1, flags, timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002596
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002597 flags = 0;
Martin v. Löwis423be952008-08-13 15:53:07 +00002598 if (!PyArg_ParseTuple(args, "y*O:sendto", &pbuf, &addro)) {
Guido van Rossum73624e91994-10-10 17:59:00 +00002599 PyErr_Clear();
Martin v. Löwis423be952008-08-13 15:53:07 +00002600 if (!PyArg_ParseTuple(args, "y*iO:sendto",
2601 &pbuf, &flags, &addro))
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002602 return NULL;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002603 }
Martin v. Löwis423be952008-08-13 15:53:07 +00002604 buf = pbuf.buf;
2605 len = pbuf.len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002606
Martin v. Löwis423be952008-08-13 15:53:07 +00002607 if (!IS_SELECTABLE(s)) {
2608 PyBuffer_Release(&pbuf);
Neal Norwitz082b2df2006-02-07 07:04:46 +00002609 return select_error();
Martin v. Löwis423be952008-08-13 15:53:07 +00002610 }
Neal Norwitz082b2df2006-02-07 07:04:46 +00002611
Martin v. Löwis423be952008-08-13 15:53:07 +00002612 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen)) {
2613 PyBuffer_Release(&pbuf);
Thomas Wouters89f507f2006-12-13 04:49:30 +00002614 return NULL;
Martin v. Löwis423be952008-08-13 15:53:07 +00002615 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00002616
Guido van Rossum73624e91994-10-10 17:59:00 +00002617 Py_BEGIN_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002618 timeout = internal_select(s, 1);
2619 if (!timeout)
Thomas Wouters89f507f2006-12-13 04:49:30 +00002620 n = sendto(s->sock_fd, buf, len, flags, SAS2SA(&addrbuf), addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00002621 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002622
Martin v. Löwis423be952008-08-13 15:53:07 +00002623 PyBuffer_Release(&pbuf);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002624 if (timeout == 1) {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002625 PyErr_SetString(socket_timeout, "timed out");
2626 return NULL;
2627 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002628 if (n < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002629 return s->errorhandler();
Christian Heimes217cfd12007-12-02 14:31:20 +00002630 return PyLong_FromLong((long)n);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002631}
2632
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002633PyDoc_STRVAR(sendto_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002634"sendto(data[, flags], address) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002635\n\
2636Like send(data, flags) but allows specifying the destination address.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002637For IP sockets, the address is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002638
Guido van Rossum30a685f1991-06-27 15:51:29 +00002639
2640/* s.shutdown(how) method */
2641
Guido van Rossum73624e91994-10-10 17:59:00 +00002642static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002643sock_shutdown(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002644{
2645 int how;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002646 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002647
Christian Heimes217cfd12007-12-02 14:31:20 +00002648 how = PyLong_AsLong(arg);
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002649 if (how == -1 && PyErr_Occurred())
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002650 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00002651 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002652 res = shutdown(s->sock_fd, how);
Guido van Rossum73624e91994-10-10 17:59:00 +00002653 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002654 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002655 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00002656 Py_INCREF(Py_None);
2657 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002658}
2659
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002660PyDoc_STRVAR(shutdown_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002661"shutdown(flag)\n\
2662\n\
Martin v. Löwis94681fc2003-11-27 19:40:22 +00002663Shut down the reading side of the socket (flag == SHUT_RD), the writing side\n\
2664of the socket (flag == SHUT_WR), or both ends (flag == SHUT_RDWR).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002665
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002666#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Christian Heimesfaf2f632008-01-06 16:59:19 +00002667static PyObject*
2668sock_ioctl(PySocketSockObject *s, PyObject *arg)
2669{
2670 unsigned long cmd = SIO_RCVALL;
2671 unsigned int option = RCVALL_ON;
2672 DWORD recv;
2673
2674 if (!PyArg_ParseTuple(arg, "kI:ioctl", &cmd, &option))
2675 return NULL;
2676
2677 if (WSAIoctl(s->sock_fd, cmd, &option, sizeof(option),
2678 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2679 return set_error();
2680 }
2681 return PyLong_FromUnsignedLong(recv);
2682}
2683PyDoc_STRVAR(sock_ioctl_doc,
2684"ioctl(cmd, option) -> long\n\
2685\n\
2686Control the socket with WSAIoctl syscall. Currently only socket.SIO_RCVALL\n\
2687is supported as control. Options must be one of the socket.RCVALL_*\n\
2688constants.");
2689
2690#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00002691
2692/* List of methods for socket objects */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002693
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002694static PyMethodDef sock_methods[] = {
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00002695 {"_accept", (PyCFunction)sock_accept, METH_NOARGS,
Thomas Wouters477c8d52006-05-27 19:21:47 +00002696 accept_doc},
2697 {"bind", (PyCFunction)sock_bind, METH_O,
2698 bind_doc},
2699 {"close", (PyCFunction)sock_close, METH_NOARGS,
2700 close_doc},
2701 {"connect", (PyCFunction)sock_connect, METH_O,
2702 connect_doc},
2703 {"connect_ex", (PyCFunction)sock_connect_ex, METH_O,
2704 connect_ex_doc},
Thomas Wouters477c8d52006-05-27 19:21:47 +00002705 {"fileno", (PyCFunction)sock_fileno, METH_NOARGS,
2706 fileno_doc},
Guido van Rossumb6775db1994-08-01 11:34:53 +00002707#ifdef HAVE_GETPEERNAME
Thomas Wouters477c8d52006-05-27 19:21:47 +00002708 {"getpeername", (PyCFunction)sock_getpeername,
2709 METH_NOARGS, getpeername_doc},
Guido van Rossum9575a441993-04-07 14:06:14 +00002710#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00002711 {"getsockname", (PyCFunction)sock_getsockname,
2712 METH_NOARGS, getsockname_doc},
2713 {"getsockopt", (PyCFunction)sock_getsockopt, METH_VARARGS,
2714 getsockopt_doc},
Amaury Forgeot d'Arc3d17a5c2008-06-13 01:09:34 +00002715#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
Christian Heimesfaf2f632008-01-06 16:59:19 +00002716 {"ioctl", (PyCFunction)sock_ioctl, METH_VARARGS,
2717 sock_ioctl_doc},
2718#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00002719 {"listen", (PyCFunction)sock_listen, METH_O,
2720 listen_doc},
Thomas Wouters477c8d52006-05-27 19:21:47 +00002721 {"recv", (PyCFunction)sock_recv, METH_VARARGS,
2722 recv_doc},
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002723 {"recv_into", (PyCFunction)sock_recv_into, METH_VARARGS | METH_KEYWORDS,
2724 recv_into_doc},
Thomas Wouters477c8d52006-05-27 19:21:47 +00002725 {"recvfrom", (PyCFunction)sock_recvfrom, METH_VARARGS,
2726 recvfrom_doc},
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00002727 {"recvfrom_into", (PyCFunction)sock_recvfrom_into, METH_VARARGS | METH_KEYWORDS,
2728 recvfrom_into_doc},
Thomas Wouters477c8d52006-05-27 19:21:47 +00002729 {"send", (PyCFunction)sock_send, METH_VARARGS,
2730 send_doc},
2731 {"sendall", (PyCFunction)sock_sendall, METH_VARARGS,
2732 sendall_doc},
2733 {"sendto", (PyCFunction)sock_sendto, METH_VARARGS,
2734 sendto_doc},
2735 {"setblocking", (PyCFunction)sock_setblocking, METH_O,
2736 setblocking_doc},
2737 {"settimeout", (PyCFunction)sock_settimeout, METH_O,
2738 settimeout_doc},
2739 {"gettimeout", (PyCFunction)sock_gettimeout, METH_NOARGS,
2740 gettimeout_doc},
2741 {"setsockopt", (PyCFunction)sock_setsockopt, METH_VARARGS,
2742 setsockopt_doc},
2743 {"shutdown", (PyCFunction)sock_shutdown, METH_O,
2744 shutdown_doc},
Guido van Rossumb6775db1994-08-01 11:34:53 +00002745 {NULL, NULL} /* sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002746};
2747
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002748/* SockObject members */
2749static PyMemberDef sock_memberlist[] = {
2750 {"family", T_INT, offsetof(PySocketSockObject, sock_family), READONLY, "the socket family"},
2751 {"type", T_INT, offsetof(PySocketSockObject, sock_type), READONLY, "the socket type"},
2752 {"proto", T_INT, offsetof(PySocketSockObject, sock_proto), READONLY, "the socket protocol"},
2753 {"timeout", T_DOUBLE, offsetof(PySocketSockObject, sock_timeout), READONLY, "the socket timeout"},
2754 {0},
2755};
Guido van Rossum30a685f1991-06-27 15:51:29 +00002756
Guido van Rossum73624e91994-10-10 17:59:00 +00002757/* Deallocate a socket object in response to the last Py_DECREF().
Guido van Rossum30a685f1991-06-27 15:51:29 +00002758 First close the file description. */
2759
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002760static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002761sock_dealloc(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002762{
Guido van Rossumfa972c92000-04-10 12:45:45 +00002763 if (s->sock_fd != -1)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +00002764 (void) SOCKETCLOSE(s->sock_fd);
Christian Heimes90aa7642007-12-19 02:45:37 +00002765 Py_TYPE(s)->tp_free((PyObject *)s);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002766}
2767
Guido van Rossum30a685f1991-06-27 15:51:29 +00002768
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002769static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002770sock_repr(PySocketSockObject *s)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002771{
Fred Drakea04eaad2000-06-30 02:46:07 +00002772#if SIZEOF_SOCKET_T > SIZEOF_LONG
2773 if (s->sock_fd > LONG_MAX) {
2774 /* this can occur on Win64, and actually there is a special
2775 ugly printf formatter for decimal pointer length integer
2776 printing, only bother if necessary*/
2777 PyErr_SetString(PyExc_OverflowError,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002778 "no printf formatter to display "
2779 "the socket descriptor in decimal");
Fred Drakea04eaad2000-06-30 02:46:07 +00002780 return NULL;
2781 }
2782#endif
Walter Dörwalda7eb93e2007-06-05 13:41:53 +00002783 return PyUnicode_FromFormat(
Guido van Rossum7d0a8262007-05-21 23:13:11 +00002784 "<socket object, fd=%ld, family=%d, type=%d, proto=%d>",
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002785 (long)s->sock_fd, s->sock_family,
2786 s->sock_type,
2787 s->sock_proto);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002788}
2789
2790
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002791/* Create a new, uninitialized socket object. */
2792
2793static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002794sock_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002795{
2796 PyObject *new;
2797
2798 new = type->tp_alloc(type, 0);
Guido van Rossume1c478f2002-06-06 20:08:25 +00002799 if (new != NULL) {
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002800 ((PySocketSockObject *)new)->sock_fd = -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002801 ((PySocketSockObject *)new)->sock_timeout = -1.0;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002802 ((PySocketSockObject *)new)->errorhandler = &set_error;
Guido van Rossume1c478f2002-06-06 20:08:25 +00002803 }
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002804 return new;
2805}
2806
2807
2808/* Initialize a new socket object. */
2809
2810/*ARGSUSED*/
2811static int
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00002812sock_initobj(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002813{
2814 PySocketSockObject *s = (PySocketSockObject *)self;
Guido van Rossum7d0a8262007-05-21 23:13:11 +00002815 PyObject *fdobj = NULL;
2816 SOCKET_T fd = INVALID_SOCKET;
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002817 int family = AF_INET, type = SOCK_STREAM, proto = 0;
Guido van Rossum7d0a8262007-05-21 23:13:11 +00002818 static char *keywords[] = {"family", "type", "proto", "fileno", 0};
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002819
2820 if (!PyArg_ParseTupleAndKeywords(args, kwds,
Guido van Rossum7d0a8262007-05-21 23:13:11 +00002821 "|iiiO:socket", keywords,
2822 &family, &type, &proto, &fdobj))
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002823 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002824
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00002825 if (fdobj != NULL && fdobj != Py_None) {
2826 fd = PyLong_AsSocket_t(fdobj);
Guido van Rossum7d0a8262007-05-21 23:13:11 +00002827 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
2828 return -1;
2829 if (fd == INVALID_SOCKET) {
2830 PyErr_SetString(PyExc_ValueError,
2831 "can't use invalid socket value");
2832 return -1;
2833 }
2834 }
2835 else {
2836 Py_BEGIN_ALLOW_THREADS
2837 fd = socket(family, type, proto);
2838 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002839
Guido van Rossum7d0a8262007-05-21 23:13:11 +00002840 if (fd == INVALID_SOCKET) {
2841 set_error();
2842 return -1;
2843 }
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002844 }
2845 init_sockobject(s, fd, family, type, proto);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002846
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002847 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002848
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002849}
2850
2851
Guido van Rossumb6775db1994-08-01 11:34:53 +00002852/* Type object for socket objects. */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002853
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002854static PyTypeObject sock_type = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00002855 PyVarObject_HEAD_INIT(0, 0) /* Must fill in type value later */
Guido van Rossumca6dfa52001-10-28 12:31:33 +00002856 "_socket.socket", /* tp_name */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002857 sizeof(PySocketSockObject), /* tp_basicsize */
2858 0, /* tp_itemsize */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002859 (destructor)sock_dealloc, /* tp_dealloc */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002860 0, /* tp_print */
2861 0, /* tp_getattr */
2862 0, /* tp_setattr */
2863 0, /* tp_compare */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002864 (reprfunc)sock_repr, /* tp_repr */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002865 0, /* tp_as_number */
2866 0, /* tp_as_sequence */
2867 0, /* tp_as_mapping */
2868 0, /* tp_hash */
2869 0, /* tp_call */
2870 0, /* tp_str */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00002871 PyObject_GenericGetAttr, /* tp_getattro */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002872 0, /* tp_setattro */
2873 0, /* tp_as_buffer */
2874 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002875 sock_doc, /* tp_doc */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002876 0, /* tp_traverse */
2877 0, /* tp_clear */
2878 0, /* tp_richcompare */
2879 0, /* tp_weaklistoffset */
2880 0, /* tp_iter */
2881 0, /* tp_iternext */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002882 sock_methods, /* tp_methods */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002883 sock_memberlist, /* tp_members */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002884 0, /* tp_getset */
2885 0, /* tp_base */
2886 0, /* tp_dict */
2887 0, /* tp_descr_get */
2888 0, /* tp_descr_set */
2889 0, /* tp_dictoffset */
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00002890 sock_initobj, /* tp_init */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00002891 PyType_GenericAlloc, /* tp_alloc */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002892 sock_new, /* tp_new */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00002893 PyObject_Del, /* tp_free */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002894};
2895
Guido van Rossum30a685f1991-06-27 15:51:29 +00002896
Guido van Rossum81194471991-07-27 21:42:02 +00002897/* Python interface to gethostname(). */
2898
2899/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002900static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00002901socket_gethostname(PyObject *self, PyObject *unused)
Guido van Rossum81194471991-07-27 21:42:02 +00002902{
2903 char buf[1024];
Guido van Rossumff4949e1992-08-05 19:58:53 +00002904 int res;
Guido van Rossum73624e91994-10-10 17:59:00 +00002905 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002906 res = gethostname(buf, (int) sizeof buf - 1);
Guido van Rossum73624e91994-10-10 17:59:00 +00002907 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002908 if (res < 0)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002909 return set_error();
Guido van Rossum81194471991-07-27 21:42:02 +00002910 buf[sizeof buf - 1] = '\0';
Guido van Rossum32c4ac02007-08-15 03:56:40 +00002911 return PyUnicode_FromString(buf);
Guido van Rossum81194471991-07-27 21:42:02 +00002912}
Guido van Rossumff4949e1992-08-05 19:58:53 +00002913
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002914PyDoc_STRVAR(gethostname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002915"gethostname() -> string\n\
2916\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002917Return the current host name.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002918
Guido van Rossumff4949e1992-08-05 19:58:53 +00002919
Guido van Rossum30a685f1991-06-27 15:51:29 +00002920/* Python interface to gethostbyname(name). */
2921
2922/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002923static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002924socket_gethostbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002925{
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002926 char *name;
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00002927 sock_addr_t addrbuf;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002928
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002929 if (!PyArg_ParseTuple(args, "s:gethostbyname", &name))
Guido van Rossum30a685f1991-06-27 15:51:29 +00002930 return NULL;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002931 if (setipaddr(name, SAS2SA(&addrbuf), sizeof(addrbuf), AF_INET) < 0)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002932 return NULL;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002933 return makeipaddr(SAS2SA(&addrbuf), sizeof(struct sockaddr_in));
Guido van Rossum30a685f1991-06-27 15:51:29 +00002934}
2935
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002936PyDoc_STRVAR(gethostbyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002937"gethostbyname(host) -> address\n\
2938\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002939Return the IP address (a string of the form '255.255.255.255') for a host.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002940
2941
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002942/* Convenience function common to gethostbyname_ex and gethostbyaddr */
2943
2944static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002945gethost_common(struct hostent *h, struct sockaddr *addr, int alen, int af)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002946{
2947 char **pch;
2948 PyObject *rtn_tuple = (PyObject *)NULL;
2949 PyObject *name_list = (PyObject *)NULL;
2950 PyObject *addr_list = (PyObject *)NULL;
2951 PyObject *tmp;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002952
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002953 if (h == NULL) {
Guido van Rossum48a680c2001-03-02 06:34:14 +00002954 /* Let's get real error message to return */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002955 set_herror(h_errno);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002956 return NULL;
2957 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002958
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002959 if (h->h_addrtype != af) {
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002960 /* Let's get real error message to return */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002961 PyErr_SetString(socket_error,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002962 (char *)strerror(EAFNOSUPPORT));
Christian Heimesada8c3b2008-03-18 18:26:33 +00002963
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002964 return NULL;
2965 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002966
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002967 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00002968
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002969 case AF_INET:
2970 if (alen < sizeof(struct sockaddr_in))
2971 return NULL;
2972 break;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002973
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00002974#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002975 case AF_INET6:
2976 if (alen < sizeof(struct sockaddr_in6))
2977 return NULL;
2978 break;
2979#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00002980
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002981 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002982
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002983 if ((name_list = PyList_New(0)) == NULL)
2984 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002985
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002986 if ((addr_list = PyList_New(0)) == NULL)
2987 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002988
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00002989 /* SF #1511317: h_aliases can be NULL */
2990 if (h->h_aliases) {
2991 for (pch = h->h_aliases; *pch != NULL; pch++) {
2992 int status;
Guido van Rossum32c4ac02007-08-15 03:56:40 +00002993 tmp = PyUnicode_FromString(*pch);
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00002994 if (tmp == NULL)
2995 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002996
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00002997 status = PyList_Append(name_list, tmp);
2998 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002999
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003000 if (status)
3001 goto err;
3002 }
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003003 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003004
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003005 for (pch = h->h_addr_list; *pch != NULL; pch++) {
3006 int status;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003007
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003008 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00003009
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003010 case AF_INET:
3011 {
3012 struct sockaddr_in sin;
3013 memset(&sin, 0, sizeof(sin));
3014 sin.sin_family = af;
3015#ifdef HAVE_SOCKADDR_SA_LEN
3016 sin.sin_len = sizeof(sin);
3017#endif
3018 memcpy(&sin.sin_addr, *pch, sizeof(sin.sin_addr));
3019 tmp = makeipaddr((struct sockaddr *)&sin, sizeof(sin));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003020
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003021 if (pch == h->h_addr_list && alen >= sizeof(sin))
3022 memcpy((char *) addr, &sin, sizeof(sin));
3023 break;
3024 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003025
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003026#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003027 case AF_INET6:
3028 {
3029 struct sockaddr_in6 sin6;
3030 memset(&sin6, 0, sizeof(sin6));
3031 sin6.sin6_family = af;
3032#ifdef HAVE_SOCKADDR_SA_LEN
3033 sin6.sin6_len = sizeof(sin6);
3034#endif
3035 memcpy(&sin6.sin6_addr, *pch, sizeof(sin6.sin6_addr));
3036 tmp = makeipaddr((struct sockaddr *)&sin6,
3037 sizeof(sin6));
Guido van Rossum67f7a382002-06-06 21:08:16 +00003038
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003039 if (pch == h->h_addr_list && alen >= sizeof(sin6))
3040 memcpy((char *) addr, &sin6, sizeof(sin6));
3041 break;
3042 }
3043#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00003044
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003045 default: /* can't happen */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003046 PyErr_SetString(socket_error,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003047 "unsupported address family");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003048 return NULL;
3049 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003050
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003051 if (tmp == NULL)
3052 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003053
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003054 status = PyList_Append(addr_list, tmp);
3055 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003056
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003057 if (status)
3058 goto err;
3059 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00003060
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003061 rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list);
Guido van Rossum67f7a382002-06-06 21:08:16 +00003062
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003063 err:
3064 Py_XDECREF(name_list);
3065 Py_XDECREF(addr_list);
3066 return rtn_tuple;
3067}
3068
3069
3070/* Python interface to gethostbyname_ex(name). */
3071
3072/*ARGSUSED*/
3073static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003074socket_gethostbyname_ex(PyObject *self, PyObject *args)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003075{
3076 char *name;
3077 struct hostent *h;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00003078#ifdef ENABLE_IPV6
3079 struct sockaddr_storage addr;
3080#else
3081 struct sockaddr_in addr;
3082#endif
Martin v. Löwis9db2f572001-07-23 01:30:10 +00003083 struct sockaddr *sa;
Guido van Rossum955becc1999-03-22 20:14:53 +00003084 PyObject *ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003085#ifdef HAVE_GETHOSTBYNAME_R
3086 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003087#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
3088 struct hostent_data data;
3089#else
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003090 char buf[16384];
3091 int buf_len = (sizeof buf) - 1;
3092 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003093#endif
3094#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003095 int result;
3096#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003097#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003098
Guido van Rossum27fc3c02000-03-24 20:56:56 +00003099 if (!PyArg_ParseTuple(args, "s:gethostbyname_ex", &name))
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003100 return NULL;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00003101 if (setipaddr(name, (struct sockaddr *)&addr, sizeof(addr), AF_INET) < 0)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003102 return NULL;
3103 Py_BEGIN_ALLOW_THREADS
3104#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003105#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003106 result = gethostbyname_r(name, &hp_allocated, buf, buf_len,
3107 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003108#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003109 h = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003110#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Guido van Rossume7de2061999-03-24 17:24:33 +00003111 memset((void *) &data, '\0', sizeof(data));
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003112 result = gethostbyname_r(name, &hp_allocated, &data);
3113 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003114#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003115#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003116#ifdef USE_GETHOSTBYNAME_LOCK
Just van Rossum1040d2c2003-05-09 07:53:18 +00003117 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003118#endif
3119 h = gethostbyname(name);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003120#endif /* HAVE_GETHOSTBYNAME_R */
3121 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00003122 /* Some C libraries would require addr.__ss_family instead of
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003123 addr.ss_family.
3124 Therefore, we cast the sockaddr_storage into sockaddr to
3125 access sa_family. */
Martin v. Löwis9db2f572001-07-23 01:30:10 +00003126 sa = (struct sockaddr*)&addr;
Guido van Rossum67f7a382002-06-06 21:08:16 +00003127 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr),
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003128 sa->sa_family);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003129#ifdef USE_GETHOSTBYNAME_LOCK
Just van Rossum1040d2c2003-05-09 07:53:18 +00003130 PyThread_release_lock(netdb_lock);
Guido van Rossum955becc1999-03-22 20:14:53 +00003131#endif
3132 return ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003133}
3134
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003135PyDoc_STRVAR(ghbn_ex_doc,
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003136"gethostbyname_ex(host) -> (name, aliaslist, addresslist)\n\
3137\n\
3138Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003139for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum7d896ab1998-08-04 22:16:43 +00003140
3141
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003142/* Python interface to gethostbyaddr(IP). */
3143
3144/*ARGSUSED*/
3145static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003146socket_gethostbyaddr(PyObject *self, PyObject *args)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003147{
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003148#ifdef ENABLE_IPV6
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003149 struct sockaddr_storage addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003150#else
Guido van Rossum48a680c2001-03-02 06:34:14 +00003151 struct sockaddr_in addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003152#endif
3153 struct sockaddr *sa = (struct sockaddr *)&addr;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003154 char *ip_num;
3155 struct hostent *h;
Guido van Rossum3baaa131999-03-22 21:44:51 +00003156 PyObject *ret;
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003157#ifdef HAVE_GETHOSTBYNAME_R
3158 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003159#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
3160 struct hostent_data data;
3161#else
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003162 char buf[16384];
3163 int buf_len = (sizeof buf) - 1;
3164 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003165#endif
3166#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003167 int result;
3168#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003169#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003170 char *ap;
3171 int al;
3172 int af;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003173
Guido van Rossum27fc3c02000-03-24 20:56:56 +00003174 if (!PyArg_ParseTuple(args, "s:gethostbyaddr", &ip_num))
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003175 return NULL;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00003176 af = AF_UNSPEC;
Martin v. Löwisddc6f472002-07-28 16:10:31 +00003177 if (setipaddr(ip_num, sa, sizeof(addr), af) < 0)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003178 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003179 af = sa->sa_family;
3180 ap = NULL;
3181 al = 0;
3182 switch (af) {
3183 case AF_INET:
3184 ap = (char *)&((struct sockaddr_in *)sa)->sin_addr;
3185 al = sizeof(((struct sockaddr_in *)sa)->sin_addr);
3186 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003187#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003188 case AF_INET6:
3189 ap = (char *)&((struct sockaddr_in6 *)sa)->sin6_addr;
3190 al = sizeof(((struct sockaddr_in6 *)sa)->sin6_addr);
3191 break;
3192#endif
3193 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003194 PyErr_SetString(socket_error, "unsupported address family");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003195 return NULL;
3196 }
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003197 Py_BEGIN_ALLOW_THREADS
3198#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003199#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003200 result = gethostbyaddr_r(ap, al, af,
3201 &hp_allocated, buf, buf_len,
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003202 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003203#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003204 h = gethostbyaddr_r(ap, al, af,
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003205 &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003206#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Guido van Rossume7de2061999-03-24 17:24:33 +00003207 memset((void *) &data, '\0', sizeof(data));
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003208 result = gethostbyaddr_r(ap, al, af, &hp_allocated, &data);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00003209 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00003210#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003211#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003212#ifdef USE_GETHOSTBYNAME_LOCK
Just van Rossum1040d2c2003-05-09 07:53:18 +00003213 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003214#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003215 h = gethostbyaddr(ap, al, af);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003216#endif /* HAVE_GETHOSTBYNAME_R */
3217 Py_END_ALLOW_THREADS
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003218 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr), af);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003219#ifdef USE_GETHOSTBYNAME_LOCK
Just van Rossum1040d2c2003-05-09 07:53:18 +00003220 PyThread_release_lock(netdb_lock);
Guido van Rossum3baaa131999-03-22 21:44:51 +00003221#endif
3222 return ret;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00003223}
3224
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003225PyDoc_STRVAR(gethostbyaddr_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003226"gethostbyaddr(host) -> (name, aliaslist, addresslist)\n\
3227\n\
3228Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003229for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003230
Guido van Rossum30a685f1991-06-27 15:51:29 +00003231
3232/* Python interface to getservbyname(name).
3233 This only returns the port number, since the other info is already
3234 known or not useful (like the list of aliases). */
3235
3236/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00003237static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003238socket_getservbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00003239{
Barry Warsaw11b91a02004-06-28 00:50:43 +00003240 char *name, *proto=NULL;
Guido van Rossum30a685f1991-06-27 15:51:29 +00003241 struct servent *sp;
Barry Warsaw11b91a02004-06-28 00:50:43 +00003242 if (!PyArg_ParseTuple(args, "s|s:getservbyname", &name, &proto))
Guido van Rossum30a685f1991-06-27 15:51:29 +00003243 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00003244 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00003245 sp = getservbyname(name, proto);
Guido van Rossum73624e91994-10-10 17:59:00 +00003246 Py_END_ALLOW_THREADS
Guido van Rossum30a685f1991-06-27 15:51:29 +00003247 if (sp == NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003248 PyErr_SetString(socket_error, "service/proto not found");
Guido van Rossum30a685f1991-06-27 15:51:29 +00003249 return NULL;
3250 }
Christian Heimes217cfd12007-12-02 14:31:20 +00003251 return PyLong_FromLong((long) ntohs(sp->s_port));
Guido van Rossum30a685f1991-06-27 15:51:29 +00003252}
3253
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003254PyDoc_STRVAR(getservbyname_doc,
Barry Warsaw11b91a02004-06-28 00:50:43 +00003255"getservbyname(servicename[, protocolname]) -> integer\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00003256\n\
3257Return a port number from a service name and protocol name.\n\
Barry Warsaw11b91a02004-06-28 00:50:43 +00003258The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3259otherwise any protocol will match.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003260
Guido van Rossum30a685f1991-06-27 15:51:29 +00003261
Barry Warsaw11b91a02004-06-28 00:50:43 +00003262/* Python interface to getservbyport(port).
3263 This only returns the service name, since the other info is already
3264 known or not useful (like the list of aliases). */
3265
3266/*ARGSUSED*/
3267static PyObject *
3268socket_getservbyport(PyObject *self, PyObject *args)
3269{
Raymond Hettinger87de0ca2004-09-28 02:19:40 +00003270 unsigned short port;
Barry Warsaw11b91a02004-06-28 00:50:43 +00003271 char *proto=NULL;
3272 struct servent *sp;
Raymond Hettinger87de0ca2004-09-28 02:19:40 +00003273 if (!PyArg_ParseTuple(args, "H|s:getservbyport", &port, &proto))
Barry Warsaw11b91a02004-06-28 00:50:43 +00003274 return NULL;
3275 Py_BEGIN_ALLOW_THREADS
3276 sp = getservbyport(htons(port), proto);
3277 Py_END_ALLOW_THREADS
3278 if (sp == NULL) {
3279 PyErr_SetString(socket_error, "port/proto not found");
3280 return NULL;
3281 }
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003282 return PyUnicode_FromString(sp->s_name);
Barry Warsaw11b91a02004-06-28 00:50:43 +00003283}
3284
3285PyDoc_STRVAR(getservbyport_doc,
3286"getservbyport(port[, protocolname]) -> string\n\
3287\n\
3288Return the service name from a port number and protocol name.\n\
3289The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3290otherwise any protocol will match.");
3291
Guido van Rossum3901d851996-12-19 16:35:04 +00003292/* Python interface to getprotobyname(name).
3293 This only returns the protocol number, since the other info is
3294 already known or not useful (like the list of aliases). */
3295
3296/*ARGSUSED*/
3297static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003298socket_getprotobyname(PyObject *self, PyObject *args)
Guido van Rossum3901d851996-12-19 16:35:04 +00003299{
3300 char *name;
3301 struct protoent *sp;
Guido van Rossum27fc3c02000-03-24 20:56:56 +00003302 if (!PyArg_ParseTuple(args, "s:getprotobyname", &name))
Guido van Rossum3901d851996-12-19 16:35:04 +00003303 return NULL;
3304 Py_BEGIN_ALLOW_THREADS
3305 sp = getprotobyname(name);
3306 Py_END_ALLOW_THREADS
3307 if (sp == NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003308 PyErr_SetString(socket_error, "protocol not found");
Guido van Rossum3901d851996-12-19 16:35:04 +00003309 return NULL;
3310 }
Christian Heimes217cfd12007-12-02 14:31:20 +00003311 return PyLong_FromLong((long) sp->p_proto);
Guido van Rossum3901d851996-12-19 16:35:04 +00003312}
3313
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003314PyDoc_STRVAR(getprotobyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003315"getprotobyname(name) -> integer\n\
3316\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003317Return the protocol number for the named protocol. (Rarely used.)");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003318
Guido van Rossum3901d851996-12-19 16:35:04 +00003319
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003320#ifndef NO_DUP
3321/* dup() function for socket fds */
3322
3323static PyObject *
3324socket_dup(PyObject *self, PyObject *fdobj)
3325{
3326 SOCKET_T fd, newfd;
3327 PyObject *newfdobj;
3328
3329
3330 fd = PyLong_AsSocket_t(fdobj);
3331 if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
3332 return NULL;
3333
3334 newfd = dup_socket(fd);
3335 if (newfd == INVALID_SOCKET)
3336 return set_error();
3337
3338 newfdobj = PyLong_FromSocket_t(newfd);
3339 if (newfdobj == NULL)
3340 SOCKETCLOSE(newfd);
3341 return newfdobj;
3342}
3343
3344PyDoc_STRVAR(dup_doc,
3345"dup(integer) -> integer\n\
3346\n\
3347Duplicate an integer socket file descriptor. This is like os.dup(), but for\n\
3348sockets; on some platforms os.dup() won't work for socket file descriptors.");
3349#endif
3350
3351
Dave Cole331708b2004-08-09 04:51:41 +00003352#ifdef HAVE_SOCKETPAIR
3353/* Create a pair of sockets using the socketpair() function.
Dave Cole07fda7e2004-08-23 05:16:23 +00003354 Arguments as for socket() except the default family is AF_UNIX if
Dave Colee8bbfe42004-08-26 00:51:16 +00003355 defined on the platform; otherwise, the default is AF_INET. */
Dave Cole331708b2004-08-09 04:51:41 +00003356
3357/*ARGSUSED*/
3358static PyObject *
3359socket_socketpair(PyObject *self, PyObject *args)
3360{
3361 PySocketSockObject *s0 = NULL, *s1 = NULL;
3362 SOCKET_T sv[2];
3363 int family, type = SOCK_STREAM, proto = 0;
3364 PyObject *res = NULL;
3365
3366#if defined(AF_UNIX)
3367 family = AF_UNIX;
3368#else
3369 family = AF_INET;
3370#endif
3371 if (!PyArg_ParseTuple(args, "|iii:socketpair",
3372 &family, &type, &proto))
3373 return NULL;
3374 /* Create a pair of socket fds */
3375 if (socketpair(family, type, proto, sv) < 0)
3376 return set_error();
Dave Cole331708b2004-08-09 04:51:41 +00003377 s0 = new_sockobject(sv[0], family, type, proto);
3378 if (s0 == NULL)
3379 goto finally;
3380 s1 = new_sockobject(sv[1], family, type, proto);
3381 if (s1 == NULL)
3382 goto finally;
3383 res = PyTuple_Pack(2, s0, s1);
3384
3385finally:
3386 if (res == NULL) {
3387 if (s0 == NULL)
3388 SOCKETCLOSE(sv[0]);
3389 if (s1 == NULL)
3390 SOCKETCLOSE(sv[1]);
3391 }
3392 Py_XDECREF(s0);
3393 Py_XDECREF(s1);
3394 return res;
3395}
3396
3397PyDoc_STRVAR(socketpair_doc,
3398"socketpair([family[, type[, proto]]]) -> (socket object, socket object)\n\
3399\n\
3400Create a pair of socket objects from the sockets returned by the platform\n\
3401socketpair() function.\n\
Dave Cole07fda7e2004-08-23 05:16:23 +00003402The arguments are the same as for socket() except the default family is\n\
Dave Colee8bbfe42004-08-26 00:51:16 +00003403AF_UNIX if defined on the platform; otherwise, the default is AF_INET.");
Dave Cole331708b2004-08-09 04:51:41 +00003404
3405#endif /* HAVE_SOCKETPAIR */
3406
3407
Guido van Rossum006bf911996-06-12 04:04:55 +00003408static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003409socket_ntohs(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003410{
3411 int x1, x2;
3412
Guido van Rossum27fc3c02000-03-24 20:56:56 +00003413 if (!PyArg_ParseTuple(args, "i:ntohs", &x1)) {
Guido van Rossum006bf911996-06-12 04:04:55 +00003414 return NULL;
3415 }
Guido van Rossum018919a2007-01-15 00:07:32 +00003416 if (x1 < 0) {
3417 PyErr_SetString(PyExc_OverflowError,
3418 "can't convert negative number to unsigned long");
3419 return NULL;
3420 }
3421 x2 = (unsigned int)ntohs((unsigned short)x1);
Christian Heimes217cfd12007-12-02 14:31:20 +00003422 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003423}
3424
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003425PyDoc_STRVAR(ntohs_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003426"ntohs(integer) -> integer\n\
3427\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003428Convert a 16-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003429
3430
Guido van Rossum006bf911996-06-12 04:04:55 +00003431static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003432socket_ntohl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003433{
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003434 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003435
Guido van Rossumddefaf32007-01-14 03:31:43 +00003436 if (PyLong_Check(arg)) {
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003437 x = PyLong_AsUnsignedLong(arg);
Jeremy Hylton825e47b2002-07-25 16:37:51 +00003438 if (x == (unsigned long) -1 && PyErr_Occurred())
3439 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003440#if SIZEOF_LONG > 4
3441 {
3442 unsigned long y;
3443 /* only want the trailing 32 bits */
3444 y = x & 0xFFFFFFFFUL;
3445 if (y ^ x)
3446 return PyErr_Format(PyExc_OverflowError,
3447 "long int larger than 32 bits");
3448 x = y;
Tim Peters58141872002-08-06 22:25:02 +00003449 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003450#endif
3451 }
3452 else
Tim Peters58141872002-08-06 22:25:02 +00003453 return PyErr_Format(PyExc_TypeError,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003454 "expected int/long, %s found",
Christian Heimes90aa7642007-12-19 02:45:37 +00003455 Py_TYPE(arg)->tp_name);
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003456 if (x == (unsigned long) -1 && PyErr_Occurred())
3457 return NULL;
Guido van Rossum018919a2007-01-15 00:07:32 +00003458 return PyLong_FromUnsignedLong(ntohl(x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003459}
3460
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003461PyDoc_STRVAR(ntohl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003462"ntohl(integer) -> integer\n\
3463\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003464Convert a 32-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003465
3466
Guido van Rossum006bf911996-06-12 04:04:55 +00003467static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003468socket_htons(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003469{
Hye-Shik Changa7b673f2003-12-17 09:40:23 +00003470 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003471
Guido van Rossum27fc3c02000-03-24 20:56:56 +00003472 if (!PyArg_ParseTuple(args, "i:htons", &x1)) {
Guido van Rossum006bf911996-06-12 04:04:55 +00003473 return NULL;
3474 }
Guido van Rossum018919a2007-01-15 00:07:32 +00003475 if (x1 < 0) {
3476 PyErr_SetString(PyExc_OverflowError,
3477 "can't convert negative number to unsigned long");
3478 return NULL;
3479 }
3480 x2 = (unsigned int)htons((unsigned short)x1);
Christian Heimes217cfd12007-12-02 14:31:20 +00003481 return PyLong_FromLong(x2);
Guido van Rossum006bf911996-06-12 04:04:55 +00003482}
3483
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003484PyDoc_STRVAR(htons_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003485"htons(integer) -> integer\n\
3486\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003487Convert a 16-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003488
3489
Guido van Rossum006bf911996-06-12 04:04:55 +00003490static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003491socket_htonl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003492{
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003493 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003494
Guido van Rossumddefaf32007-01-14 03:31:43 +00003495 if (PyLong_Check(arg)) {
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003496 x = PyLong_AsUnsignedLong(arg);
Jeremy Hylton825e47b2002-07-25 16:37:51 +00003497 if (x == (unsigned long) -1 && PyErr_Occurred())
3498 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003499#if SIZEOF_LONG > 4
3500 {
3501 unsigned long y;
3502 /* only want the trailing 32 bits */
3503 y = x & 0xFFFFFFFFUL;
3504 if (y ^ x)
3505 return PyErr_Format(PyExc_OverflowError,
3506 "long int larger than 32 bits");
3507 x = y;
Tim Peters58141872002-08-06 22:25:02 +00003508 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003509#endif
3510 }
3511 else
Tim Peters58141872002-08-06 22:25:02 +00003512 return PyErr_Format(PyExc_TypeError,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003513 "expected int/long, %s found",
Christian Heimes90aa7642007-12-19 02:45:37 +00003514 Py_TYPE(arg)->tp_name);
Guido van Rossum018919a2007-01-15 00:07:32 +00003515 return PyLong_FromUnsignedLong(htonl((unsigned long)x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003516}
3517
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003518PyDoc_STRVAR(htonl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003519"htonl(integer) -> integer\n\
3520\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003521Convert a 32-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003522
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003523/* socket.inet_aton() and socket.inet_ntoa() functions. */
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003524
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003525PyDoc_STRVAR(inet_aton_doc,
Guido van Rossum7d0a8262007-05-21 23:13:11 +00003526"inet_aton(string) -> bytes giving packed 32-bit IP representation\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003527\n\
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003528Convert 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 +00003529binary format used in low-level network functions.");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003530
3531static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003532socket_inet_aton(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003533{
Guido van Rossuma2e48551999-09-09 15:42:59 +00003534#ifndef INADDR_NONE
3535#define INADDR_NONE (-1)
3536#endif
Neal Norwitz88f115b2003-02-13 02:15:42 +00003537#ifdef HAVE_INET_ATON
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003538 struct in_addr buf;
Thomas Wouters477c8d52006-05-27 19:21:47 +00003539#endif
3540
3541#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
Tim Peters1df9fdd2003-02-13 03:13:40 +00003542 /* Have to use inet_addr() instead */
3543 unsigned long packed_addr;
3544#endif
3545 char *ip_addr;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003546
Tim Peters1df9fdd2003-02-13 03:13:40 +00003547 if (!PyArg_ParseTuple(args, "s:inet_aton", &ip_addr))
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003548 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003549
Tim Peters1df9fdd2003-02-13 03:13:40 +00003550
3551#ifdef HAVE_INET_ATON
Thomas Wouters477c8d52006-05-27 19:21:47 +00003552
3553#ifdef USE_INET_ATON_WEAKLINK
3554 if (inet_aton != NULL) {
3555#endif
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003556 if (inet_aton(ip_addr, &buf))
Christian Heimes72b710a2008-05-26 13:28:38 +00003557 return PyBytes_FromStringAndSize((char *)(&buf),
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003558 sizeof(buf));
3559
3560 PyErr_SetString(socket_error,
3561 "illegal IP address string passed to inet_aton");
3562 return NULL;
3563
Thomas Wouters477c8d52006-05-27 19:21:47 +00003564#ifdef USE_INET_ATON_WEAKLINK
3565 } else {
3566#endif
3567
3568#endif
3569
3570#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
3571
Georg Brandld2e3ba72005-08-26 08:34:00 +00003572 /* special-case this address as inet_addr might return INADDR_NONE
3573 * for this */
3574 if (strcmp(ip_addr, "255.255.255.255") == 0) {
3575 packed_addr = 0xFFFFFFFF;
3576 } else {
Thomas Wouters477c8d52006-05-27 19:21:47 +00003577
Georg Brandld2e3ba72005-08-26 08:34:00 +00003578 packed_addr = inet_addr(ip_addr);
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003579
Georg Brandld2e3ba72005-08-26 08:34:00 +00003580 if (packed_addr == INADDR_NONE) { /* invalid address */
3581 PyErr_SetString(socket_error,
3582 "illegal IP address string passed to inet_aton");
3583 return NULL;
3584 }
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003585 }
Christian Heimes72b710a2008-05-26 13:28:38 +00003586 return PyBytes_FromStringAndSize((char *) &packed_addr,
Guido van Rossum8a392d72007-11-21 22:09:45 +00003587 sizeof(packed_addr));
Thomas Wouters477c8d52006-05-27 19:21:47 +00003588
3589#ifdef USE_INET_ATON_WEAKLINK
3590 }
3591#endif
3592
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003593#endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003594}
3595
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003596PyDoc_STRVAR(inet_ntoa_doc,
Fred Drakee0661342000-03-07 14:05:16 +00003597"inet_ntoa(packed_ip) -> ip_address_string\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003598\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003599Convert an IP address from 32-bit packed binary format to string format");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003600
3601static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003602socket_inet_ntoa(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003603{
3604 char *packed_str;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003605 int addr_len;
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003606 struct in_addr packed_addr;
3607
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003608 if (!PyArg_ParseTuple(args, "y#:inet_ntoa", &packed_str, &addr_len)) {
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003609 return NULL;
3610 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00003611
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003612 if (addr_len != sizeof(packed_addr)) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003613 PyErr_SetString(socket_error,
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003614 "packed IP wrong length for inet_ntoa");
3615 return NULL;
3616 }
3617
3618 memcpy(&packed_addr, packed_str, addr_len);
3619
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003620 return PyUnicode_FromString(inet_ntoa(packed_addr));
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003621}
Guido van Rossum82a5c661998-07-07 20:45:43 +00003622
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003623#ifdef HAVE_INET_PTON
3624
3625PyDoc_STRVAR(inet_pton_doc,
3626"inet_pton(af, ip) -> packed IP address string\n\
3627\n\
3628Convert an IP address from string format to a packed string suitable\n\
3629for use with low-level network functions.");
3630
3631static PyObject *
3632socket_inet_pton(PyObject *self, PyObject *args)
3633{
3634 int af;
3635 char* ip;
3636 int retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003637#ifdef ENABLE_IPV6
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003638 char packed[MAX(sizeof(struct in_addr), sizeof(struct in6_addr))];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003639#else
3640 char packed[sizeof(struct in_addr)];
3641#endif
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003642 if (!PyArg_ParseTuple(args, "is:inet_pton", &af, &ip)) {
3643 return NULL;
3644 }
3645
Martin v. Löwis04697e82004-06-02 12:35:29 +00003646#if !defined(ENABLE_IPV6) && defined(AF_INET6)
Martin v. Löwis10649092003-08-05 06:25:06 +00003647 if(af == AF_INET6) {
3648 PyErr_SetString(socket_error,
3649 "can't use AF_INET6, IPv6 is disabled");
3650 return NULL;
3651 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00003652#endif
Martin v. Löwis10649092003-08-05 06:25:06 +00003653
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003654 retval = inet_pton(af, ip, packed);
3655 if (retval < 0) {
3656 PyErr_SetFromErrno(socket_error);
3657 return NULL;
3658 } else if (retval == 0) {
3659 PyErr_SetString(socket_error,
3660 "illegal IP address string passed to inet_pton");
3661 return NULL;
3662 } else if (af == AF_INET) {
Christian Heimes72b710a2008-05-26 13:28:38 +00003663 return PyBytes_FromStringAndSize(packed,
Guido van Rossum8a392d72007-11-21 22:09:45 +00003664 sizeof(struct in_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003665#ifdef ENABLE_IPV6
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003666 } else if (af == AF_INET6) {
Christian Heimes72b710a2008-05-26 13:28:38 +00003667 return PyBytes_FromStringAndSize(packed,
Guido van Rossum8a392d72007-11-21 22:09:45 +00003668 sizeof(struct in6_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003669#endif
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003670 } else {
3671 PyErr_SetString(socket_error, "unknown address family");
3672 return NULL;
3673 }
3674}
Thomas Wouters477c8d52006-05-27 19:21:47 +00003675
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003676PyDoc_STRVAR(inet_ntop_doc,
3677"inet_ntop(af, packed_ip) -> string formatted IP address\n\
3678\n\
3679Convert a packed IP address of the given family to string format.");
3680
3681static PyObject *
3682socket_inet_ntop(PyObject *self, PyObject *args)
3683{
3684 int af;
3685 char* packed;
3686 int len;
3687 const char* retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003688#ifdef ENABLE_IPV6
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003689 char ip[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN) + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003690#else
3691 char ip[INET_ADDRSTRLEN + 1];
3692#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +00003693
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003694 /* Guarantee NUL-termination for PyUnicode_FromString() below */
Jeremy Hylton80961f32004-11-07 14:24:25 +00003695 memset((void *) &ip[0], '\0', sizeof(ip));
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003696
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003697 if (!PyArg_ParseTuple(args, "iy#:inet_ntop", &af, &packed, &len)) {
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003698 return NULL;
3699 }
3700
3701 if (af == AF_INET) {
3702 if (len != sizeof(struct in_addr)) {
3703 PyErr_SetString(PyExc_ValueError,
3704 "invalid length of packed IP address string");
3705 return NULL;
3706 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003707#ifdef ENABLE_IPV6
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003708 } else if (af == AF_INET6) {
3709 if (len != sizeof(struct in6_addr)) {
3710 PyErr_SetString(PyExc_ValueError,
3711 "invalid length of packed IP address string");
3712 return NULL;
3713 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003714#endif
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003715 } else {
3716 PyErr_Format(PyExc_ValueError,
3717 "unknown address family %d", af);
3718 return NULL;
3719 }
3720
3721 retval = inet_ntop(af, packed, ip, sizeof(ip));
3722 if (!retval) {
3723 PyErr_SetFromErrno(socket_error);
3724 return NULL;
3725 } else {
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003726 return PyUnicode_FromString(retval);
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003727 }
3728
3729 /* NOTREACHED */
3730 PyErr_SetString(PyExc_RuntimeError, "invalid handling of inet_ntop");
3731 return NULL;
3732}
3733
3734#endif /* HAVE_INET_PTON */
3735
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003736/* Python interface to getaddrinfo(host, port). */
3737
3738/*ARGSUSED*/
3739static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003740socket_getaddrinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003741{
Martin v. Löwisf0b11d22001-11-07 08:31:03 +00003742 struct addrinfo hints, *res;
3743 struct addrinfo *res0 = NULL;
Martin v. Löwis2548c732003-04-18 10:39:54 +00003744 PyObject *hobj = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003745 PyObject *pobj = (PyObject *)NULL;
Martin v. Löwisf65b1a12001-08-12 09:28:40 +00003746 char pbuf[30];
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003747 char *hptr, *pptr;
3748 int family, socktype, protocol, flags;
3749 int error;
3750 PyObject *all = (PyObject *)NULL;
Martin v. Löwis2548c732003-04-18 10:39:54 +00003751 PyObject *idna = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003752
3753 family = socktype = protocol = flags = 0;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00003754 family = AF_UNSPEC;
Martin v. Löwis2548c732003-04-18 10:39:54 +00003755 if (!PyArg_ParseTuple(args, "OO|iiii:getaddrinfo",
3756 &hobj, &pobj, &family, &socktype,
3757 &protocol, &flags)) {
3758 return NULL;
3759 }
3760 if (hobj == Py_None) {
3761 hptr = NULL;
3762 } else if (PyUnicode_Check(hobj)) {
3763 idna = PyObject_CallMethod(hobj, "encode", "s", "idna");
3764 if (!idna)
3765 return NULL;
Christian Heimes72b710a2008-05-26 13:28:38 +00003766 assert(PyBytes_Check(idna));
3767 hptr = PyBytes_AS_STRING(idna);
3768 } else if (PyBytes_Check(hobj)) {
3769 hptr = PyBytes_AsString(hobj);
Martin v. Löwis2548c732003-04-18 10:39:54 +00003770 } else {
Thomas Wouters477c8d52006-05-27 19:21:47 +00003771 PyErr_SetString(PyExc_TypeError,
Martin v. Löwis2548c732003-04-18 10:39:54 +00003772 "getaddrinfo() argument 1 must be string or None");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003773 return NULL;
3774 }
Martin v. Löwisd1a1d1e2007-12-04 22:10:37 +00003775 if (PyLong_CheckExact(pobj)) {
3776 long value = PyLong_AsLong(pobj);
3777 if (value == -1 && PyErr_Occurred())
3778 goto err;
3779 PyOS_snprintf(pbuf, sizeof(pbuf), "%ld", value);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003780 pptr = pbuf;
Guido van Rossum32c4ac02007-08-15 03:56:40 +00003781 } else if (PyUnicode_Check(pobj)) {
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00003782 pptr = _PyUnicode_AsString(pobj);
Christian Heimes72b710a2008-05-26 13:28:38 +00003783 } else if (PyBytes_Check(pobj)) {
3784 pptr = PyBytes_AsString(pobj);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003785 } else if (pobj == Py_None) {
3786 pptr = (char *)NULL;
3787 } else {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003788 PyErr_SetString(socket_error, "Int or String expected");
Martin v. Löwis5db099a2003-08-07 11:55:15 +00003789 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003790 }
3791 memset(&hints, 0, sizeof(hints));
3792 hints.ai_family = family;
3793 hints.ai_socktype = socktype;
3794 hints.ai_protocol = protocol;
3795 hints.ai_flags = flags;
Just van Rossum1040d2c2003-05-09 07:53:18 +00003796 Py_BEGIN_ALLOW_THREADS
3797 ACQUIRE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003798 error = getaddrinfo(hptr, pptr, &hints, &res0);
Just van Rossum1040d2c2003-05-09 07:53:18 +00003799 Py_END_ALLOW_THREADS
3800 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003801 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003802 set_gaierror(error);
Martin v. Löwis5db099a2003-08-07 11:55:15 +00003803 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003804 }
3805
3806 if ((all = PyList_New(0)) == NULL)
3807 goto err;
3808 for (res = res0; res; res = res->ai_next) {
Guido van Rossum9ce8e382007-08-09 21:06:37 +00003809 PyObject *single;
Guido van Rossum716aac02001-10-12 18:59:27 +00003810 PyObject *addr =
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00003811 makesockaddr(-1, res->ai_addr, res->ai_addrlen, protocol);
Guido van Rossum716aac02001-10-12 18:59:27 +00003812 if (addr == NULL)
3813 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003814 single = Py_BuildValue("iiisO", res->ai_family,
3815 res->ai_socktype, res->ai_protocol,
3816 res->ai_canonname ? res->ai_canonname : "",
Guido van Rossum716aac02001-10-12 18:59:27 +00003817 addr);
3818 Py_DECREF(addr);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003819 if (single == NULL)
3820 goto err;
3821
3822 if (PyList_Append(all, single))
3823 goto err;
3824 Py_XDECREF(single);
3825 }
Martin v. Löwis2548c732003-04-18 10:39:54 +00003826 Py_XDECREF(idna);
Neal Norwitz90128ba2002-08-09 03:37:42 +00003827 if (res0)
3828 freeaddrinfo(res0);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003829 return all;
3830 err:
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003831 Py_XDECREF(all);
Martin v. Löwis2548c732003-04-18 10:39:54 +00003832 Py_XDECREF(idna);
Martin v. Löwisf0b11d22001-11-07 08:31:03 +00003833 if (res0)
3834 freeaddrinfo(res0);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003835 return (PyObject *)NULL;
3836}
3837
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003838PyDoc_STRVAR(getaddrinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003839"getaddrinfo(host, port [, family, socktype, proto, flags])\n\
3840 -> list of (family, socktype, proto, canonname, sockaddr)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003841\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003842Resolve host and port into addrinfo struct.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003843
3844/* Python interface to getnameinfo(sa, flags). */
3845
3846/*ARGSUSED*/
3847static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003848socket_getnameinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003849{
3850 PyObject *sa = (PyObject *)NULL;
3851 int flags;
3852 char *hostp;
Martin v. Löwis06b1d212001-11-02 23:34:52 +00003853 int port, flowinfo, scope_id;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003854 char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
3855 struct addrinfo hints, *res = NULL;
3856 int error;
3857 PyObject *ret = (PyObject *)NULL;
3858
3859 flags = flowinfo = scope_id = 0;
Martin v. Löwis06b1d212001-11-02 23:34:52 +00003860 if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags))
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003861 return NULL;
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003862 if (!PyArg_ParseTuple(sa, "si|ii",
3863 &hostp, &port, &flowinfo, &scope_id))
Martin v. Löwis06b1d212001-11-02 23:34:52 +00003864 return NULL;
Marc-André Lemburg03d1b182001-07-31 18:05:33 +00003865 PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003866 memset(&hints, 0, sizeof(hints));
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00003867 hints.ai_family = AF_UNSPEC;
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003868 hints.ai_socktype = SOCK_DGRAM; /* make numeric port happy */
Just van Rossum1040d2c2003-05-09 07:53:18 +00003869 Py_BEGIN_ALLOW_THREADS
3870 ACQUIRE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003871 error = getaddrinfo(hostp, pbuf, &hints, &res);
Just van Rossum1040d2c2003-05-09 07:53:18 +00003872 Py_END_ALLOW_THREADS
3873 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003874 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003875 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003876 goto fail;
3877 }
3878 if (res->ai_next) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003879 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003880 "sockaddr resolved to multiple addresses");
3881 goto fail;
3882 }
3883 switch (res->ai_family) {
3884 case AF_INET:
3885 {
3886 char *t1;
3887 int t2;
3888 if (PyArg_ParseTuple(sa, "si", &t1, &t2) == 0) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003889 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003890 "IPv4 sockaddr must be 2 tuple");
3891 goto fail;
3892 }
3893 break;
3894 }
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003895#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003896 case AF_INET6:
3897 {
3898 struct sockaddr_in6 *sin6;
3899 sin6 = (struct sockaddr_in6 *)res->ai_addr;
3900 sin6->sin6_flowinfo = flowinfo;
3901 sin6->sin6_scope_id = scope_id;
3902 break;
3903 }
3904#endif
3905 }
3906 error = getnameinfo(res->ai_addr, res->ai_addrlen,
3907 hbuf, sizeof(hbuf), pbuf, sizeof(pbuf), flags);
3908 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003909 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003910 goto fail;
3911 }
3912 ret = Py_BuildValue("ss", hbuf, pbuf);
3913
3914fail:
3915 if (res)
3916 freeaddrinfo(res);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003917 return ret;
3918}
3919
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003920PyDoc_STRVAR(getnameinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003921"getnameinfo(sockaddr, flags) --> (host, port)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003922\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003923Get host and port for a sockaddr.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003924
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00003925
3926/* Python API to getting and setting the default timeout value. */
3927
3928static PyObject *
3929socket_getdefaulttimeout(PyObject *self)
3930{
3931 if (defaulttimeout < 0.0) {
3932 Py_INCREF(Py_None);
3933 return Py_None;
3934 }
3935 else
3936 return PyFloat_FromDouble(defaulttimeout);
3937}
3938
3939PyDoc_STRVAR(getdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003940"getdefaulttimeout() -> timeout\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00003941\n\
3942Returns the default timeout in floating seconds for new socket objects.\n\
3943A value of None indicates that new socket objects have no timeout.\n\
3944When the socket module is first imported, the default is None.");
3945
3946static PyObject *
3947socket_setdefaulttimeout(PyObject *self, PyObject *arg)
3948{
3949 double timeout;
3950
3951 if (arg == Py_None)
3952 timeout = -1.0;
3953 else {
3954 timeout = PyFloat_AsDouble(arg);
3955 if (timeout < 0.0) {
3956 if (!PyErr_Occurred())
3957 PyErr_SetString(PyExc_ValueError,
3958 "Timeout value out of range");
3959 return NULL;
3960 }
3961 }
3962
3963 defaulttimeout = timeout;
3964
3965 Py_INCREF(Py_None);
3966 return Py_None;
3967}
3968
3969PyDoc_STRVAR(setdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003970"setdefaulttimeout(timeout)\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00003971\n\
3972Set the default timeout in floating seconds for new socket objects.\n\
3973A value of None indicates that new socket objects have no timeout.\n\
3974When the socket module is first imported, the default is None.");
3975
3976
Guido van Rossum30a685f1991-06-27 15:51:29 +00003977/* List of functions exported by this module. */
3978
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003979static PyMethodDef socket_methods[] = {
3980 {"gethostbyname", socket_gethostbyname,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003981 METH_VARARGS, gethostbyname_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003982 {"gethostbyname_ex", socket_gethostbyname_ex,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003983 METH_VARARGS, ghbn_ex_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003984 {"gethostbyaddr", socket_gethostbyaddr,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003985 METH_VARARGS, gethostbyaddr_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003986 {"gethostname", socket_gethostname,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00003987 METH_NOARGS, gethostname_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003988 {"getservbyname", socket_getservbyname,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003989 METH_VARARGS, getservbyname_doc},
Barry Warsaw11b91a02004-06-28 00:50:43 +00003990 {"getservbyport", socket_getservbyport,
3991 METH_VARARGS, getservbyport_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003992 {"getprotobyname", socket_getprotobyname,
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00003993 METH_VARARGS, getprotobyname_doc},
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00003994#ifndef NO_DUP
3995 {"dup", socket_dup,
3996 METH_O, dup_doc},
3997#endif
Dave Cole331708b2004-08-09 04:51:41 +00003998#ifdef HAVE_SOCKETPAIR
3999 {"socketpair", socket_socketpair,
4000 METH_VARARGS, socketpair_doc},
4001#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004002 {"ntohs", socket_ntohs,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00004003 METH_VARARGS, ntohs_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004004 {"ntohl", socket_ntohl,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00004005 METH_O, ntohl_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004006 {"htons", socket_htons,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00004007 METH_VARARGS, htons_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004008 {"htonl", socket_htonl,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00004009 METH_O, htonl_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004010 {"inet_aton", socket_inet_aton,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00004011 METH_VARARGS, inet_aton_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004012 {"inet_ntoa", socket_inet_ntoa,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00004013 METH_VARARGS, inet_ntoa_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004014#ifdef HAVE_INET_PTON
4015 {"inet_pton", socket_inet_pton,
4016 METH_VARARGS, inet_pton_doc},
4017 {"inet_ntop", socket_inet_ntop,
4018 METH_VARARGS, inet_ntop_doc},
4019#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004020 {"getaddrinfo", socket_getaddrinfo,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004021 METH_VARARGS, getaddrinfo_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004022 {"getnameinfo", socket_getnameinfo,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004023 METH_VARARGS, getnameinfo_doc},
Guido van Rossum1693ba82002-07-18 21:11:26 +00004024 {"getdefaulttimeout", (PyCFunction)socket_getdefaulttimeout,
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00004025 METH_NOARGS, getdefaulttimeout_doc},
4026 {"setdefaulttimeout", socket_setdefaulttimeout,
4027 METH_O, setdefaulttimeout_doc},
Guido van Rossum30a685f1991-06-27 15:51:29 +00004028 {NULL, NULL} /* Sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004029};
4030
Guido van Rossum30a685f1991-06-27 15:51:29 +00004031
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004032#ifdef MS_WINDOWS
4033#define OS_INIT_DEFINED
4034
4035/* Additional initialization and cleanup for Windows */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004036
4037static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004038os_cleanup(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004039{
4040 WSACleanup();
Guido van Rossumbe32c891996-06-20 16:25:29 +00004041}
4042
4043static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004044os_init(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00004045{
4046 WSADATA WSAData;
4047 int ret;
Guido van Rossumbe32c891996-06-20 16:25:29 +00004048 ret = WSAStartup(0x0101, &WSAData);
4049 switch (ret) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004050 case 0: /* No error */
Mark Hammond14350ab2003-07-09 04:57:46 +00004051 Py_AtExit(os_cleanup);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004052 return 1; /* Success */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004053 case WSASYSNOTREADY:
4054 PyErr_SetString(PyExc_ImportError,
4055 "WSAStartup failed: network not ready");
4056 break;
4057 case WSAVERNOTSUPPORTED:
4058 case WSAEINVAL:
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004059 PyErr_SetString(
4060 PyExc_ImportError,
4061 "WSAStartup failed: requested version not supported");
Guido van Rossumbe32c891996-06-20 16:25:29 +00004062 break;
4063 default:
Walter Dörwald0bbd8ab2007-06-05 13:49:43 +00004064 PyErr_Format(PyExc_ImportError, "WSAStartup failed: error code %d", ret);
Guido van Rossumbe32c891996-06-20 16:25:29 +00004065 break;
4066 }
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004067 return 0; /* Failure */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004068}
4069
Guido van Rossum8d665e61996-06-26 18:22:49 +00004070#endif /* MS_WINDOWS */
Guido van Rossumbe32c891996-06-20 16:25:29 +00004071
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004072
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004073#ifdef PYOS_OS2
4074#define OS_INIT_DEFINED
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004075
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004076/* Additional initialization for OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004077
4078static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004079os_init(void)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004080{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004081#ifndef PYCC_GCC
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004082 int rc = sock_init();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004083
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004084 if (rc == 0) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004085 return 1; /* Success */
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004086 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004087
Walter Dörwald0bbd8ab2007-06-05 13:49:43 +00004088 PyErr_Format(PyExc_ImportError, "OS/2 TCP/IP Error# %d", sock_errno());
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004089
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004090 return 0; /* Failure */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004091#else
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004092 /* No need to initialise sockets with GCC/EMX */
4093 return 1; /* Success */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00004094#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00004095}
4096
4097#endif /* PYOS_OS2 */
4098
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004099
4100#ifndef OS_INIT_DEFINED
4101static int
4102os_init(void)
4103{
4104 return 1; /* Success */
4105}
4106#endif
4107
4108
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004109/* C API table - always add new things to the end for binary
4110 compatibility. */
4111static
4112PySocketModule_APIObject PySocketModuleAPI =
4113{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004114 &sock_type,
Brett Cannon06c34792004-03-23 23:16:54 +00004115 NULL
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004116};
4117
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004118
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004119/* Initialize the _socket module.
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004120
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004121 This module is actually called "_socket", and there's a wrapper
Guido van Rossum7d0a8262007-05-21 23:13:11 +00004122 "socket.py" which implements some additional functionality.
4123 The import of "_socket" may fail with an ImportError exception if
4124 os-specific initialization fails. On Windows, this does WINSOCK
4125 initialization. When WINSOCK is initialized succesfully, a call to
4126 WSACleanup() is scheduled to be made at exit time.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004127*/
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004128
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00004129PyDoc_STRVAR(socket_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00004130"Implementation module for socket operations.\n\
4131\n\
4132See the socket module for documentation.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00004133
Martin v. Löwis1a214512008-06-11 05:26:20 +00004134static struct PyModuleDef socketmodule = {
4135 PyModuleDef_HEAD_INIT,
4136 PySocket_MODULE_NAME,
4137 socket_doc,
4138 -1,
4139 socket_methods,
4140 NULL,
4141 NULL,
4142 NULL,
4143 NULL
4144};
4145
Mark Hammondfe51c6d2002-08-02 02:27:13 +00004146PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00004147PyInit__socket(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004148{
Christian Heimes25bb7832008-01-11 16:17:00 +00004149 PyObject *m, *has_ipv6;
Fred Drake4baedc12002-04-01 14:53:37 +00004150
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004151 if (!os_init())
Martin v. Löwis1a214512008-06-11 05:26:20 +00004152 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004153
Christian Heimes90aa7642007-12-19 02:45:37 +00004154 Py_TYPE(&sock_type) = &PyType_Type;
Martin v. Löwis1a214512008-06-11 05:26:20 +00004155 m = PyModule_Create(&socketmodule);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00004156 if (m == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004157 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004158
Gregory P. Smithc111d9f2007-09-09 23:55:55 +00004159 socket_error = PyErr_NewException("socket.error",
4160 PyExc_IOError, NULL);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004161 if (socket_error == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004162 return NULL;
Brett Cannon06c34792004-03-23 23:16:54 +00004163 PySocketModuleAPI.error = socket_error;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004164 Py_INCREF(socket_error);
4165 PyModule_AddObject(m, "error", socket_error);
4166 socket_herror = PyErr_NewException("socket.herror",
4167 socket_error, NULL);
4168 if (socket_herror == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004169 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004170 Py_INCREF(socket_herror);
4171 PyModule_AddObject(m, "herror", socket_herror);
4172 socket_gaierror = PyErr_NewException("socket.gaierror", socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004173 NULL);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004174 if (socket_gaierror == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004175 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004176 Py_INCREF(socket_gaierror);
4177 PyModule_AddObject(m, "gaierror", socket_gaierror);
Raymond Hettingeref7343c2003-06-29 03:08:05 +00004178 socket_timeout = PyErr_NewException("socket.timeout",
4179 socket_error, NULL);
4180 if (socket_timeout == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004181 return NULL;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00004182 Py_INCREF(socket_timeout);
4183 PyModule_AddObject(m, "timeout", socket_timeout);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004184 Py_INCREF((PyObject *)&sock_type);
Fred Drake4baedc12002-04-01 14:53:37 +00004185 if (PyModule_AddObject(m, "SocketType",
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004186 (PyObject *)&sock_type) != 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004187 return NULL;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004188 Py_INCREF((PyObject *)&sock_type);
Fred Drake4baedc12002-04-01 14:53:37 +00004189 if (PyModule_AddObject(m, "socket",
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004190 (PyObject *)&sock_type) != 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004191 return NULL;
Guido van Rossum09be4091999-08-09 14:40:40 +00004192
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004193#ifdef ENABLE_IPV6
4194 has_ipv6 = Py_True;
4195#else
4196 has_ipv6 = Py_False;
4197#endif
4198 Py_INCREF(has_ipv6);
4199 PyModule_AddObject(m, "has_ipv6", has_ipv6);
4200
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004201 /* Export C API */
Fred Drake4baedc12002-04-01 14:53:37 +00004202 if (PyModule_AddObject(m, PySocket_CAPI_NAME,
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004203 PyCObject_FromVoidPtr((void *)&PySocketModuleAPI, NULL)
4204 ) != 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00004205 return NULL;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00004206
Guido van Rossum09be4091999-08-09 14:40:40 +00004207 /* Address families (we only support AF_INET and AF_UNIX) */
4208#ifdef AF_UNSPEC
Fred Drake4baedc12002-04-01 14:53:37 +00004209 PyModule_AddIntConstant(m, "AF_UNSPEC", AF_UNSPEC);
Guido van Rossum09be4091999-08-09 14:40:40 +00004210#endif
Fred Drake4baedc12002-04-01 14:53:37 +00004211 PyModule_AddIntConstant(m, "AF_INET", AF_INET);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004212#ifdef AF_INET6
Fred Drake4baedc12002-04-01 14:53:37 +00004213 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004214#endif /* AF_INET6 */
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00004215#if defined(AF_UNIX)
Fred Drake4baedc12002-04-01 14:53:37 +00004216 PyModule_AddIntConstant(m, "AF_UNIX", AF_UNIX);
Guido van Rossumb6775db1994-08-01 11:34:53 +00004217#endif /* AF_UNIX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004218#ifdef AF_AX25
Fred Drake4baedc12002-04-01 14:53:37 +00004219 /* Amateur Radio AX.25 */
4220 PyModule_AddIntConstant(m, "AF_AX25", AF_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004221#endif
4222#ifdef AF_IPX
Fred Drake4baedc12002-04-01 14:53:37 +00004223 PyModule_AddIntConstant(m, "AF_IPX", AF_IPX); /* Novell IPX */
Guido van Rossum09be4091999-08-09 14:40:40 +00004224#endif
4225#ifdef AF_APPLETALK
Fred Drake4baedc12002-04-01 14:53:37 +00004226 /* Appletalk DDP */
4227 PyModule_AddIntConstant(m, "AF_APPLETALK", AF_APPLETALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004228#endif
4229#ifdef AF_NETROM
Fred Drake4baedc12002-04-01 14:53:37 +00004230 /* Amateur radio NetROM */
4231 PyModule_AddIntConstant(m, "AF_NETROM", AF_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004232#endif
4233#ifdef AF_BRIDGE
Fred Drake4baedc12002-04-01 14:53:37 +00004234 /* Multiprotocol bridge */
4235 PyModule_AddIntConstant(m, "AF_BRIDGE", AF_BRIDGE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004236#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004237#ifdef AF_ATMPVC
4238 /* ATM PVCs */
4239 PyModule_AddIntConstant(m, "AF_ATMPVC", AF_ATMPVC);
4240#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004241#ifdef AF_AAL5
Fred Drake4baedc12002-04-01 14:53:37 +00004242 /* Reserved for Werner's ATM */
4243 PyModule_AddIntConstant(m, "AF_AAL5", AF_AAL5);
Guido van Rossum09be4091999-08-09 14:40:40 +00004244#endif
4245#ifdef AF_X25
Fred Drake4baedc12002-04-01 14:53:37 +00004246 /* Reserved for X.25 project */
4247 PyModule_AddIntConstant(m, "AF_X25", AF_X25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004248#endif
4249#ifdef AF_INET6
Fred Drake4baedc12002-04-01 14:53:37 +00004250 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6); /* IP version 6 */
Guido van Rossum09be4091999-08-09 14:40:40 +00004251#endif
4252#ifdef AF_ROSE
Fred Drake4baedc12002-04-01 14:53:37 +00004253 /* Amateur Radio X.25 PLP */
4254 PyModule_AddIntConstant(m, "AF_ROSE", AF_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004255#endif
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004256#ifdef AF_DECnet
4257 /* Reserved for DECnet project */
4258 PyModule_AddIntConstant(m, "AF_DECnet", AF_DECnet);
4259#endif
4260#ifdef AF_NETBEUI
4261 /* Reserved for 802.2LLC project */
4262 PyModule_AddIntConstant(m, "AF_NETBEUI", AF_NETBEUI);
4263#endif
4264#ifdef AF_SECURITY
4265 /* Security callback pseudo AF */
4266 PyModule_AddIntConstant(m, "AF_SECURITY", AF_SECURITY);
4267#endif
4268#ifdef AF_KEY
4269 /* PF_KEY key management API */
4270 PyModule_AddIntConstant(m, "AF_KEY", AF_KEY);
4271#endif
4272#ifdef AF_NETLINK
4273 /* */
4274 PyModule_AddIntConstant(m, "AF_NETLINK", AF_NETLINK);
Martin v. Löwis11017b12006-01-14 18:12:57 +00004275 PyModule_AddIntConstant(m, "NETLINK_ROUTE", NETLINK_ROUTE);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004276#ifdef NETLINK_SKIP
Martin v. Löwis11017b12006-01-14 18:12:57 +00004277 PyModule_AddIntConstant(m, "NETLINK_SKIP", NETLINK_SKIP);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004278#endif
4279#ifdef NETLINK_W1
4280 PyModule_AddIntConstant(m, "NETLINK_W1", NETLINK_W1);
4281#endif
Martin v. Löwis11017b12006-01-14 18:12:57 +00004282 PyModule_AddIntConstant(m, "NETLINK_USERSOCK", NETLINK_USERSOCK);
4283 PyModule_AddIntConstant(m, "NETLINK_FIREWALL", NETLINK_FIREWALL);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004284#ifdef NETLINK_TCPDIAG
Martin v. Löwis11017b12006-01-14 18:12:57 +00004285 PyModule_AddIntConstant(m, "NETLINK_TCPDIAG", NETLINK_TCPDIAG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004286#endif
4287#ifdef NETLINK_NFLOG
Martin v. Löwis11017b12006-01-14 18:12:57 +00004288 PyModule_AddIntConstant(m, "NETLINK_NFLOG", NETLINK_NFLOG);
Guido van Rossum668a94a2006-02-21 01:07:27 +00004289#endif
Neal Norwitz65851662006-01-16 04:31:40 +00004290#ifdef NETLINK_XFRM
Martin v. Löwis11017b12006-01-14 18:12:57 +00004291 PyModule_AddIntConstant(m, "NETLINK_XFRM", NETLINK_XFRM);
Neal Norwitz65851662006-01-16 04:31:40 +00004292#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004293#ifdef NETLINK_ARPD
Martin v. Löwis11017b12006-01-14 18:12:57 +00004294 PyModule_AddIntConstant(m, "NETLINK_ARPD", NETLINK_ARPD);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004295#endif
4296#ifdef NETLINK_ROUTE6
Martin v. Löwis11017b12006-01-14 18:12:57 +00004297 PyModule_AddIntConstant(m, "NETLINK_ROUTE6", NETLINK_ROUTE6);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004298#endif
Martin v. Löwis11017b12006-01-14 18:12:57 +00004299 PyModule_AddIntConstant(m, "NETLINK_IP6_FW", NETLINK_IP6_FW);
Thomas Wouterscf297e42007-02-23 15:07:44 +00004300#ifdef NETLINK_DNRTMSG
Martin v. Löwis11017b12006-01-14 18:12:57 +00004301 PyModule_AddIntConstant(m, "NETLINK_DNRTMSG", NETLINK_DNRTMSG);
Guido van Rossum39eb8fa2007-11-16 01:24:05 +00004302#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004303#ifdef NETLINK_TAPBASE
Martin v. Löwis11017b12006-01-14 18:12:57 +00004304 PyModule_AddIntConstant(m, "NETLINK_TAPBASE", NETLINK_TAPBASE);
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004305#endif
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004306#endif /* AF_NETLINK */
Martin v. Löwis81aec4b2004-07-19 17:01:20 +00004307#ifdef AF_ROUTE
4308 /* Alias to emulate 4.4BSD */
4309 PyModule_AddIntConstant(m, "AF_ROUTE", AF_ROUTE);
4310#endif
4311#ifdef AF_ASH
4312 /* Ash */
4313 PyModule_AddIntConstant(m, "AF_ASH", AF_ASH);
4314#endif
4315#ifdef AF_ECONET
4316 /* Acorn Econet */
4317 PyModule_AddIntConstant(m, "AF_ECONET", AF_ECONET);
4318#endif
4319#ifdef AF_ATMSVC
4320 /* ATM SVCs */
4321 PyModule_AddIntConstant(m, "AF_ATMSVC", AF_ATMSVC);
4322#endif
4323#ifdef AF_SNA
4324 /* Linux SNA Project (nutters!) */
4325 PyModule_AddIntConstant(m, "AF_SNA", AF_SNA);
4326#endif
4327#ifdef AF_IRDA
4328 /* IRDA sockets */
4329 PyModule_AddIntConstant(m, "AF_IRDA", AF_IRDA);
4330#endif
4331#ifdef AF_PPPOX
4332 /* PPPoX sockets */
4333 PyModule_AddIntConstant(m, "AF_PPPOX", AF_PPPOX);
4334#endif
4335#ifdef AF_WANPIPE
4336 /* Wanpipe API Sockets */
4337 PyModule_AddIntConstant(m, "AF_WANPIPE", AF_WANPIPE);
4338#endif
4339#ifdef AF_LLC
4340 /* Linux LLC */
4341 PyModule_AddIntConstant(m, "AF_LLC", AF_LLC);
4342#endif
Martin v. Löwis12af0482004-01-31 12:34:17 +00004343
Hye-Shik Chang81268602004-02-02 06:05:24 +00004344#ifdef USE_BLUETOOTH
Martin v. Löwis12af0482004-01-31 12:34:17 +00004345 PyModule_AddIntConstant(m, "AF_BLUETOOTH", AF_BLUETOOTH);
4346 PyModule_AddIntConstant(m, "BTPROTO_L2CAP", BTPROTO_L2CAP);
Thomas Wouterscf297e42007-02-23 15:07:44 +00004347 PyModule_AddIntConstant(m, "BTPROTO_HCI", BTPROTO_HCI);
4348 PyModule_AddIntConstant(m, "SOL_HCI", SOL_HCI);
Thomas Wouterscf297e42007-02-23 15:07:44 +00004349 PyModule_AddIntConstant(m, "HCI_FILTER", HCI_FILTER);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004350#if !defined(__FreeBSD__)
Guido van Rossumcd16bf62007-06-13 18:07:49 +00004351 PyModule_AddIntConstant(m, "HCI_TIME_STAMP", HCI_TIME_STAMP);
4352 PyModule_AddIntConstant(m, "HCI_DATA_DIR", HCI_DATA_DIR);
Martin v. Löwis12af0482004-01-31 12:34:17 +00004353 PyModule_AddIntConstant(m, "BTPROTO_SCO", BTPROTO_SCO);
Hye-Shik Chang81268602004-02-02 06:05:24 +00004354#endif
Martin v. Löwis12af0482004-01-31 12:34:17 +00004355 PyModule_AddIntConstant(m, "BTPROTO_RFCOMM", BTPROTO_RFCOMM);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00004356 PyModule_AddStringConstant(m, "BDADDR_ANY", "00:00:00:00:00:00");
4357 PyModule_AddStringConstant(m, "BDADDR_LOCAL", "00:00:00:FF:FF:FF");
Martin v. Löwis12af0482004-01-31 12:34:17 +00004358#endif
4359
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00004360#ifdef HAVE_NETPACKET_PACKET_H
Fred Drake4baedc12002-04-01 14:53:37 +00004361 PyModule_AddIntConstant(m, "AF_PACKET", AF_PACKET);
4362 PyModule_AddIntConstant(m, "PF_PACKET", PF_PACKET);
4363 PyModule_AddIntConstant(m, "PACKET_HOST", PACKET_HOST);
4364 PyModule_AddIntConstant(m, "PACKET_BROADCAST", PACKET_BROADCAST);
4365 PyModule_AddIntConstant(m, "PACKET_MULTICAST", PACKET_MULTICAST);
4366 PyModule_AddIntConstant(m, "PACKET_OTHERHOST", PACKET_OTHERHOST);
4367 PyModule_AddIntConstant(m, "PACKET_OUTGOING", PACKET_OUTGOING);
4368 PyModule_AddIntConstant(m, "PACKET_LOOPBACK", PACKET_LOOPBACK);
4369 PyModule_AddIntConstant(m, "PACKET_FASTROUTE", PACKET_FASTROUTE);
Guido van Rossum48a680c2001-03-02 06:34:14 +00004370#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004371
Christian Heimes043d6f62008-01-07 17:19:16 +00004372#ifdef HAVE_LINUX_TIPC_H
4373 PyModule_AddIntConstant(m, "AF_TIPC", AF_TIPC);
4374
4375 /* for addresses */
4376 PyModule_AddIntConstant(m, "TIPC_ADDR_NAMESEQ", TIPC_ADDR_NAMESEQ);
4377 PyModule_AddIntConstant(m, "TIPC_ADDR_NAME", TIPC_ADDR_NAME);
4378 PyModule_AddIntConstant(m, "TIPC_ADDR_ID", TIPC_ADDR_ID);
4379
4380 PyModule_AddIntConstant(m, "TIPC_ZONE_SCOPE", TIPC_ZONE_SCOPE);
4381 PyModule_AddIntConstant(m, "TIPC_CLUSTER_SCOPE", TIPC_CLUSTER_SCOPE);
4382 PyModule_AddIntConstant(m, "TIPC_NODE_SCOPE", TIPC_NODE_SCOPE);
4383
4384 /* for setsockopt() */
4385 PyModule_AddIntConstant(m, "SOL_TIPC", SOL_TIPC);
4386 PyModule_AddIntConstant(m, "TIPC_IMPORTANCE", TIPC_IMPORTANCE);
4387 PyModule_AddIntConstant(m, "TIPC_SRC_DROPPABLE", TIPC_SRC_DROPPABLE);
4388 PyModule_AddIntConstant(m, "TIPC_DEST_DROPPABLE",
4389 TIPC_DEST_DROPPABLE);
4390 PyModule_AddIntConstant(m, "TIPC_CONN_TIMEOUT", TIPC_CONN_TIMEOUT);
4391
4392 PyModule_AddIntConstant(m, "TIPC_LOW_IMPORTANCE",
4393 TIPC_LOW_IMPORTANCE);
4394 PyModule_AddIntConstant(m, "TIPC_MEDIUM_IMPORTANCE",
4395 TIPC_MEDIUM_IMPORTANCE);
4396 PyModule_AddIntConstant(m, "TIPC_HIGH_IMPORTANCE",
4397 TIPC_HIGH_IMPORTANCE);
4398 PyModule_AddIntConstant(m, "TIPC_CRITICAL_IMPORTANCE",
4399 TIPC_CRITICAL_IMPORTANCE);
4400
4401 /* for subscriptions */
4402 PyModule_AddIntConstant(m, "TIPC_SUB_PORTS", TIPC_SUB_PORTS);
4403 PyModule_AddIntConstant(m, "TIPC_SUB_SERVICE", TIPC_SUB_SERVICE);
Christian Heimes25bb7832008-01-11 16:17:00 +00004404#ifdef TIPC_SUB_CANCEL
4405 /* doesn't seem to be available everywhere */
Christian Heimes043d6f62008-01-07 17:19:16 +00004406 PyModule_AddIntConstant(m, "TIPC_SUB_CANCEL", TIPC_SUB_CANCEL);
Christian Heimes25bb7832008-01-11 16:17:00 +00004407#endif
Christian Heimes043d6f62008-01-07 17:19:16 +00004408 PyModule_AddIntConstant(m, "TIPC_WAIT_FOREVER", TIPC_WAIT_FOREVER);
4409 PyModule_AddIntConstant(m, "TIPC_PUBLISHED", TIPC_PUBLISHED);
4410 PyModule_AddIntConstant(m, "TIPC_WITHDRAWN", TIPC_WITHDRAWN);
4411 PyModule_AddIntConstant(m, "TIPC_SUBSCR_TIMEOUT", TIPC_SUBSCR_TIMEOUT);
4412 PyModule_AddIntConstant(m, "TIPC_CFG_SRV", TIPC_CFG_SRV);
4413 PyModule_AddIntConstant(m, "TIPC_TOP_SRV", TIPC_TOP_SRV);
4414#endif
4415
Guido van Rossum09be4091999-08-09 14:40:40 +00004416 /* Socket types */
Fred Drake4baedc12002-04-01 14:53:37 +00004417 PyModule_AddIntConstant(m, "SOCK_STREAM", SOCK_STREAM);
4418 PyModule_AddIntConstant(m, "SOCK_DGRAM", SOCK_DGRAM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004419/* We have incomplete socket support. */
Fred Drake4baedc12002-04-01 14:53:37 +00004420 PyModule_AddIntConstant(m, "SOCK_RAW", SOCK_RAW);
4421 PyModule_AddIntConstant(m, "SOCK_SEQPACKET", SOCK_SEQPACKET);
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00004422#if defined(SOCK_RDM)
Fred Drake4baedc12002-04-01 14:53:37 +00004423 PyModule_AddIntConstant(m, "SOCK_RDM", SOCK_RDM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00004424#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004425
4426#ifdef SO_DEBUG
Fred Drake4baedc12002-04-01 14:53:37 +00004427 PyModule_AddIntConstant(m, "SO_DEBUG", SO_DEBUG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004428#endif
4429#ifdef SO_ACCEPTCONN
Fred Drake4baedc12002-04-01 14:53:37 +00004430 PyModule_AddIntConstant(m, "SO_ACCEPTCONN", SO_ACCEPTCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004431#endif
4432#ifdef SO_REUSEADDR
Fred Drake4baedc12002-04-01 14:53:37 +00004433 PyModule_AddIntConstant(m, "SO_REUSEADDR", SO_REUSEADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004434#endif
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00004435#ifdef SO_EXCLUSIVEADDRUSE
4436 PyModule_AddIntConstant(m, "SO_EXCLUSIVEADDRUSE", SO_EXCLUSIVEADDRUSE);
4437#endif
4438
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004439#ifdef SO_KEEPALIVE
Fred Drake4baedc12002-04-01 14:53:37 +00004440 PyModule_AddIntConstant(m, "SO_KEEPALIVE", SO_KEEPALIVE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004441#endif
4442#ifdef SO_DONTROUTE
Fred Drake4baedc12002-04-01 14:53:37 +00004443 PyModule_AddIntConstant(m, "SO_DONTROUTE", SO_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004444#endif
4445#ifdef SO_BROADCAST
Fred Drake4baedc12002-04-01 14:53:37 +00004446 PyModule_AddIntConstant(m, "SO_BROADCAST", SO_BROADCAST);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004447#endif
4448#ifdef SO_USELOOPBACK
Fred Drake4baedc12002-04-01 14:53:37 +00004449 PyModule_AddIntConstant(m, "SO_USELOOPBACK", SO_USELOOPBACK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004450#endif
4451#ifdef SO_LINGER
Fred Drake4baedc12002-04-01 14:53:37 +00004452 PyModule_AddIntConstant(m, "SO_LINGER", SO_LINGER);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004453#endif
4454#ifdef SO_OOBINLINE
Fred Drake4baedc12002-04-01 14:53:37 +00004455 PyModule_AddIntConstant(m, "SO_OOBINLINE", SO_OOBINLINE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004456#endif
4457#ifdef SO_REUSEPORT
Fred Drake4baedc12002-04-01 14:53:37 +00004458 PyModule_AddIntConstant(m, "SO_REUSEPORT", SO_REUSEPORT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004459#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004460#ifdef SO_SNDBUF
Fred Drake4baedc12002-04-01 14:53:37 +00004461 PyModule_AddIntConstant(m, "SO_SNDBUF", SO_SNDBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004462#endif
4463#ifdef SO_RCVBUF
Fred Drake4baedc12002-04-01 14:53:37 +00004464 PyModule_AddIntConstant(m, "SO_RCVBUF", SO_RCVBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004465#endif
4466#ifdef SO_SNDLOWAT
Fred Drake4baedc12002-04-01 14:53:37 +00004467 PyModule_AddIntConstant(m, "SO_SNDLOWAT", SO_SNDLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004468#endif
4469#ifdef SO_RCVLOWAT
Fred Drake4baedc12002-04-01 14:53:37 +00004470 PyModule_AddIntConstant(m, "SO_RCVLOWAT", SO_RCVLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004471#endif
4472#ifdef SO_SNDTIMEO
Fred Drake4baedc12002-04-01 14:53:37 +00004473 PyModule_AddIntConstant(m, "SO_SNDTIMEO", SO_SNDTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004474#endif
4475#ifdef SO_RCVTIMEO
Fred Drake4baedc12002-04-01 14:53:37 +00004476 PyModule_AddIntConstant(m, "SO_RCVTIMEO", SO_RCVTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004477#endif
4478#ifdef SO_ERROR
Fred Drake4baedc12002-04-01 14:53:37 +00004479 PyModule_AddIntConstant(m, "SO_ERROR", SO_ERROR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004480#endif
4481#ifdef SO_TYPE
Fred Drake4baedc12002-04-01 14:53:37 +00004482 PyModule_AddIntConstant(m, "SO_TYPE", SO_TYPE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004483#endif
4484
4485 /* Maximum number of connections for "listen" */
4486#ifdef SOMAXCONN
Fred Drake4baedc12002-04-01 14:53:37 +00004487 PyModule_AddIntConstant(m, "SOMAXCONN", SOMAXCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004488#else
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004489 PyModule_AddIntConstant(m, "SOMAXCONN", 5); /* Common value */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004490#endif
4491
4492 /* Flags for send, recv */
4493#ifdef MSG_OOB
Fred Drake4baedc12002-04-01 14:53:37 +00004494 PyModule_AddIntConstant(m, "MSG_OOB", MSG_OOB);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004495#endif
4496#ifdef MSG_PEEK
Fred Drake4baedc12002-04-01 14:53:37 +00004497 PyModule_AddIntConstant(m, "MSG_PEEK", MSG_PEEK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004498#endif
4499#ifdef MSG_DONTROUTE
Fred Drake4baedc12002-04-01 14:53:37 +00004500 PyModule_AddIntConstant(m, "MSG_DONTROUTE", MSG_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004501#endif
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00004502#ifdef MSG_DONTWAIT
Fred Drake4baedc12002-04-01 14:53:37 +00004503 PyModule_AddIntConstant(m, "MSG_DONTWAIT", MSG_DONTWAIT);
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00004504#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004505#ifdef MSG_EOR
Fred Drake4baedc12002-04-01 14:53:37 +00004506 PyModule_AddIntConstant(m, "MSG_EOR", MSG_EOR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004507#endif
4508#ifdef MSG_TRUNC
Fred Drake4baedc12002-04-01 14:53:37 +00004509 PyModule_AddIntConstant(m, "MSG_TRUNC", MSG_TRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004510#endif
4511#ifdef MSG_CTRUNC
Fred Drake4baedc12002-04-01 14:53:37 +00004512 PyModule_AddIntConstant(m, "MSG_CTRUNC", MSG_CTRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004513#endif
4514#ifdef MSG_WAITALL
Fred Drake4baedc12002-04-01 14:53:37 +00004515 PyModule_AddIntConstant(m, "MSG_WAITALL", MSG_WAITALL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004516#endif
4517#ifdef MSG_BTAG
Fred Drake4baedc12002-04-01 14:53:37 +00004518 PyModule_AddIntConstant(m, "MSG_BTAG", MSG_BTAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004519#endif
4520#ifdef MSG_ETAG
Fred Drake4baedc12002-04-01 14:53:37 +00004521 PyModule_AddIntConstant(m, "MSG_ETAG", MSG_ETAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004522#endif
4523
4524 /* Protocol level and numbers, usable for [gs]etsockopt */
4525#ifdef SOL_SOCKET
Fred Drake4baedc12002-04-01 14:53:37 +00004526 PyModule_AddIntConstant(m, "SOL_SOCKET", SOL_SOCKET);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004527#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004528#ifdef SOL_IP
Fred Drake4baedc12002-04-01 14:53:37 +00004529 PyModule_AddIntConstant(m, "SOL_IP", SOL_IP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004530#else
Fred Drake4baedc12002-04-01 14:53:37 +00004531 PyModule_AddIntConstant(m, "SOL_IP", 0);
Guido van Rossum09be4091999-08-09 14:40:40 +00004532#endif
4533#ifdef SOL_IPX
Fred Drake4baedc12002-04-01 14:53:37 +00004534 PyModule_AddIntConstant(m, "SOL_IPX", SOL_IPX);
Guido van Rossum09be4091999-08-09 14:40:40 +00004535#endif
4536#ifdef SOL_AX25
Fred Drake4baedc12002-04-01 14:53:37 +00004537 PyModule_AddIntConstant(m, "SOL_AX25", SOL_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00004538#endif
4539#ifdef SOL_ATALK
Fred Drake4baedc12002-04-01 14:53:37 +00004540 PyModule_AddIntConstant(m, "SOL_ATALK", SOL_ATALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004541#endif
4542#ifdef SOL_NETROM
Fred Drake4baedc12002-04-01 14:53:37 +00004543 PyModule_AddIntConstant(m, "SOL_NETROM", SOL_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004544#endif
4545#ifdef SOL_ROSE
Fred Drake4baedc12002-04-01 14:53:37 +00004546 PyModule_AddIntConstant(m, "SOL_ROSE", SOL_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004547#endif
4548#ifdef SOL_TCP
Fred Drake4baedc12002-04-01 14:53:37 +00004549 PyModule_AddIntConstant(m, "SOL_TCP", SOL_TCP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004550#else
Fred Drake4baedc12002-04-01 14:53:37 +00004551 PyModule_AddIntConstant(m, "SOL_TCP", 6);
Guido van Rossum09be4091999-08-09 14:40:40 +00004552#endif
4553#ifdef SOL_UDP
Fred Drake4baedc12002-04-01 14:53:37 +00004554 PyModule_AddIntConstant(m, "SOL_UDP", SOL_UDP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004555#else
Fred Drake4baedc12002-04-01 14:53:37 +00004556 PyModule_AddIntConstant(m, "SOL_UDP", 17);
Guido van Rossum09be4091999-08-09 14:40:40 +00004557#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004558#ifdef IPPROTO_IP
Fred Drake4baedc12002-04-01 14:53:37 +00004559 PyModule_AddIntConstant(m, "IPPROTO_IP", IPPROTO_IP);
Guido van Rossum578de301998-05-28 20:18:18 +00004560#else
Fred Drake4baedc12002-04-01 14:53:37 +00004561 PyModule_AddIntConstant(m, "IPPROTO_IP", 0);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004562#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004563#ifdef IPPROTO_HOPOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004564 PyModule_AddIntConstant(m, "IPPROTO_HOPOPTS", IPPROTO_HOPOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004565#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004566#ifdef IPPROTO_ICMP
Fred Drake4baedc12002-04-01 14:53:37 +00004567 PyModule_AddIntConstant(m, "IPPROTO_ICMP", IPPROTO_ICMP);
Guido van Rossum578de301998-05-28 20:18:18 +00004568#else
Fred Drake4baedc12002-04-01 14:53:37 +00004569 PyModule_AddIntConstant(m, "IPPROTO_ICMP", 1);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004570#endif
4571#ifdef IPPROTO_IGMP
Fred Drake4baedc12002-04-01 14:53:37 +00004572 PyModule_AddIntConstant(m, "IPPROTO_IGMP", IPPROTO_IGMP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004573#endif
4574#ifdef IPPROTO_GGP
Fred Drake4baedc12002-04-01 14:53:37 +00004575 PyModule_AddIntConstant(m, "IPPROTO_GGP", IPPROTO_GGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004576#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004577#ifdef IPPROTO_IPV4
Fred Drake4baedc12002-04-01 14:53:37 +00004578 PyModule_AddIntConstant(m, "IPPROTO_IPV4", IPPROTO_IPV4);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004579#endif
Martin v. Löwisa0f17342003-10-03 13:56:20 +00004580#ifdef IPPROTO_IPV6
4581 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
4582#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004583#ifdef IPPROTO_IPIP
Fred Drake4baedc12002-04-01 14:53:37 +00004584 PyModule_AddIntConstant(m, "IPPROTO_IPIP", IPPROTO_IPIP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004585#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004586#ifdef IPPROTO_TCP
Fred Drake4baedc12002-04-01 14:53:37 +00004587 PyModule_AddIntConstant(m, "IPPROTO_TCP", IPPROTO_TCP);
Guido van Rossum578de301998-05-28 20:18:18 +00004588#else
Fred Drake4baedc12002-04-01 14:53:37 +00004589 PyModule_AddIntConstant(m, "IPPROTO_TCP", 6);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004590#endif
4591#ifdef IPPROTO_EGP
Fred Drake4baedc12002-04-01 14:53:37 +00004592 PyModule_AddIntConstant(m, "IPPROTO_EGP", IPPROTO_EGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004593#endif
4594#ifdef IPPROTO_PUP
Fred Drake4baedc12002-04-01 14:53:37 +00004595 PyModule_AddIntConstant(m, "IPPROTO_PUP", IPPROTO_PUP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004596#endif
4597#ifdef IPPROTO_UDP
Fred Drake4baedc12002-04-01 14:53:37 +00004598 PyModule_AddIntConstant(m, "IPPROTO_UDP", IPPROTO_UDP);
Guido van Rossum578de301998-05-28 20:18:18 +00004599#else
Fred Drake4baedc12002-04-01 14:53:37 +00004600 PyModule_AddIntConstant(m, "IPPROTO_UDP", 17);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004601#endif
4602#ifdef IPPROTO_IDP
Fred Drake4baedc12002-04-01 14:53:37 +00004603 PyModule_AddIntConstant(m, "IPPROTO_IDP", IPPROTO_IDP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004604#endif
4605#ifdef IPPROTO_HELLO
Fred Drake4baedc12002-04-01 14:53:37 +00004606 PyModule_AddIntConstant(m, "IPPROTO_HELLO", IPPROTO_HELLO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004607#endif
4608#ifdef IPPROTO_ND
Fred Drake4baedc12002-04-01 14:53:37 +00004609 PyModule_AddIntConstant(m, "IPPROTO_ND", IPPROTO_ND);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004610#endif
4611#ifdef IPPROTO_TP
Fred Drake4baedc12002-04-01 14:53:37 +00004612 PyModule_AddIntConstant(m, "IPPROTO_TP", IPPROTO_TP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004613#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004614#ifdef IPPROTO_IPV6
Fred Drake4baedc12002-04-01 14:53:37 +00004615 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004616#endif
4617#ifdef IPPROTO_ROUTING
Fred Drake4baedc12002-04-01 14:53:37 +00004618 PyModule_AddIntConstant(m, "IPPROTO_ROUTING", IPPROTO_ROUTING);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004619#endif
4620#ifdef IPPROTO_FRAGMENT
Fred Drake4baedc12002-04-01 14:53:37 +00004621 PyModule_AddIntConstant(m, "IPPROTO_FRAGMENT", IPPROTO_FRAGMENT);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004622#endif
4623#ifdef IPPROTO_RSVP
Fred Drake4baedc12002-04-01 14:53:37 +00004624 PyModule_AddIntConstant(m, "IPPROTO_RSVP", IPPROTO_RSVP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004625#endif
4626#ifdef IPPROTO_GRE
Fred Drake4baedc12002-04-01 14:53:37 +00004627 PyModule_AddIntConstant(m, "IPPROTO_GRE", IPPROTO_GRE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004628#endif
4629#ifdef IPPROTO_ESP
Fred Drake4baedc12002-04-01 14:53:37 +00004630 PyModule_AddIntConstant(m, "IPPROTO_ESP", IPPROTO_ESP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004631#endif
4632#ifdef IPPROTO_AH
Fred Drake4baedc12002-04-01 14:53:37 +00004633 PyModule_AddIntConstant(m, "IPPROTO_AH", IPPROTO_AH);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004634#endif
4635#ifdef IPPROTO_MOBILE
Fred Drake4baedc12002-04-01 14:53:37 +00004636 PyModule_AddIntConstant(m, "IPPROTO_MOBILE", IPPROTO_MOBILE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004637#endif
4638#ifdef IPPROTO_ICMPV6
Fred Drake4baedc12002-04-01 14:53:37 +00004639 PyModule_AddIntConstant(m, "IPPROTO_ICMPV6", IPPROTO_ICMPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004640#endif
4641#ifdef IPPROTO_NONE
Fred Drake4baedc12002-04-01 14:53:37 +00004642 PyModule_AddIntConstant(m, "IPPROTO_NONE", IPPROTO_NONE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004643#endif
4644#ifdef IPPROTO_DSTOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004645 PyModule_AddIntConstant(m, "IPPROTO_DSTOPTS", IPPROTO_DSTOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004646#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004647#ifdef IPPROTO_XTP
Fred Drake4baedc12002-04-01 14:53:37 +00004648 PyModule_AddIntConstant(m, "IPPROTO_XTP", IPPROTO_XTP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004649#endif
4650#ifdef IPPROTO_EON
Fred Drake4baedc12002-04-01 14:53:37 +00004651 PyModule_AddIntConstant(m, "IPPROTO_EON", IPPROTO_EON);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004652#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004653#ifdef IPPROTO_PIM
Fred Drake4baedc12002-04-01 14:53:37 +00004654 PyModule_AddIntConstant(m, "IPPROTO_PIM", IPPROTO_PIM);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004655#endif
4656#ifdef IPPROTO_IPCOMP
Fred Drake4baedc12002-04-01 14:53:37 +00004657 PyModule_AddIntConstant(m, "IPPROTO_IPCOMP", IPPROTO_IPCOMP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004658#endif
4659#ifdef IPPROTO_VRRP
Fred Drake4baedc12002-04-01 14:53:37 +00004660 PyModule_AddIntConstant(m, "IPPROTO_VRRP", IPPROTO_VRRP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004661#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004662#ifdef IPPROTO_BIP
Fred Drake4baedc12002-04-01 14:53:37 +00004663 PyModule_AddIntConstant(m, "IPPROTO_BIP", IPPROTO_BIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004664#endif
4665/**/
4666#ifdef IPPROTO_RAW
Fred Drake4baedc12002-04-01 14:53:37 +00004667 PyModule_AddIntConstant(m, "IPPROTO_RAW", IPPROTO_RAW);
Guido van Rossum578de301998-05-28 20:18:18 +00004668#else
Fred Drake4baedc12002-04-01 14:53:37 +00004669 PyModule_AddIntConstant(m, "IPPROTO_RAW", 255);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004670#endif
4671#ifdef IPPROTO_MAX
Fred Drake4baedc12002-04-01 14:53:37 +00004672 PyModule_AddIntConstant(m, "IPPROTO_MAX", IPPROTO_MAX);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004673#endif
4674
4675 /* Some port configuration */
4676#ifdef IPPORT_RESERVED
Fred Drake4baedc12002-04-01 14:53:37 +00004677 PyModule_AddIntConstant(m, "IPPORT_RESERVED", IPPORT_RESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004678#else
Fred Drake4baedc12002-04-01 14:53:37 +00004679 PyModule_AddIntConstant(m, "IPPORT_RESERVED", 1024);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004680#endif
4681#ifdef IPPORT_USERRESERVED
Fred Drake4baedc12002-04-01 14:53:37 +00004682 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", IPPORT_USERRESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004683#else
Fred Drake4baedc12002-04-01 14:53:37 +00004684 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", 5000);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004685#endif
4686
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004687 /* Some reserved IP v.4 addresses */
4688#ifdef INADDR_ANY
Fred Drake4baedc12002-04-01 14:53:37 +00004689 PyModule_AddIntConstant(m, "INADDR_ANY", INADDR_ANY);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004690#else
Fred Drake4baedc12002-04-01 14:53:37 +00004691 PyModule_AddIntConstant(m, "INADDR_ANY", 0x00000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004692#endif
4693#ifdef INADDR_BROADCAST
Fred Drake4baedc12002-04-01 14:53:37 +00004694 PyModule_AddIntConstant(m, "INADDR_BROADCAST", INADDR_BROADCAST);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004695#else
Fred Drake4baedc12002-04-01 14:53:37 +00004696 PyModule_AddIntConstant(m, "INADDR_BROADCAST", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004697#endif
4698#ifdef INADDR_LOOPBACK
Fred Drake4baedc12002-04-01 14:53:37 +00004699 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", INADDR_LOOPBACK);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004700#else
Fred Drake4baedc12002-04-01 14:53:37 +00004701 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", 0x7F000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004702#endif
4703#ifdef INADDR_UNSPEC_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004704 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004705#else
Fred Drake4baedc12002-04-01 14:53:37 +00004706 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", 0xe0000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004707#endif
4708#ifdef INADDR_ALLHOSTS_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004709 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP",
4710 INADDR_ALLHOSTS_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004711#else
Fred Drake4baedc12002-04-01 14:53:37 +00004712 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP", 0xe0000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004713#endif
4714#ifdef INADDR_MAX_LOCAL_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004715 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP",
4716 INADDR_MAX_LOCAL_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004717#else
Fred Drake4baedc12002-04-01 14:53:37 +00004718 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004719#endif
4720#ifdef INADDR_NONE
Fred Drake4baedc12002-04-01 14:53:37 +00004721 PyModule_AddIntConstant(m, "INADDR_NONE", INADDR_NONE);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004722#else
Fred Drake4baedc12002-04-01 14:53:37 +00004723 PyModule_AddIntConstant(m, "INADDR_NONE", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004724#endif
4725
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004726 /* IPv4 [gs]etsockopt options */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004727#ifdef IP_OPTIONS
Fred Drake4baedc12002-04-01 14:53:37 +00004728 PyModule_AddIntConstant(m, "IP_OPTIONS", IP_OPTIONS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004729#endif
4730#ifdef IP_HDRINCL
Fred Drake4baedc12002-04-01 14:53:37 +00004731 PyModule_AddIntConstant(m, "IP_HDRINCL", IP_HDRINCL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004732#endif
4733#ifdef IP_TOS
Fred Drake4baedc12002-04-01 14:53:37 +00004734 PyModule_AddIntConstant(m, "IP_TOS", IP_TOS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004735#endif
4736#ifdef IP_TTL
Fred Drake4baedc12002-04-01 14:53:37 +00004737 PyModule_AddIntConstant(m, "IP_TTL", IP_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004738#endif
4739#ifdef IP_RECVOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004740 PyModule_AddIntConstant(m, "IP_RECVOPTS", IP_RECVOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004741#endif
4742#ifdef IP_RECVRETOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004743 PyModule_AddIntConstant(m, "IP_RECVRETOPTS", IP_RECVRETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004744#endif
4745#ifdef IP_RECVDSTADDR
Fred Drake4baedc12002-04-01 14:53:37 +00004746 PyModule_AddIntConstant(m, "IP_RECVDSTADDR", IP_RECVDSTADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004747#endif
4748#ifdef IP_RETOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004749 PyModule_AddIntConstant(m, "IP_RETOPTS", IP_RETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004750#endif
4751#ifdef IP_MULTICAST_IF
Fred Drake4baedc12002-04-01 14:53:37 +00004752 PyModule_AddIntConstant(m, "IP_MULTICAST_IF", IP_MULTICAST_IF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004753#endif
4754#ifdef IP_MULTICAST_TTL
Fred Drake4baedc12002-04-01 14:53:37 +00004755 PyModule_AddIntConstant(m, "IP_MULTICAST_TTL", IP_MULTICAST_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004756#endif
4757#ifdef IP_MULTICAST_LOOP
Fred Drake4baedc12002-04-01 14:53:37 +00004758 PyModule_AddIntConstant(m, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004759#endif
4760#ifdef IP_ADD_MEMBERSHIP
Fred Drake4baedc12002-04-01 14:53:37 +00004761 PyModule_AddIntConstant(m, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004762#endif
4763#ifdef IP_DROP_MEMBERSHIP
Fred Drake4baedc12002-04-01 14:53:37 +00004764 PyModule_AddIntConstant(m, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004765#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004766#ifdef IP_DEFAULT_MULTICAST_TTL
Fred Drake4baedc12002-04-01 14:53:37 +00004767 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_TTL",
4768 IP_DEFAULT_MULTICAST_TTL);
Guido van Rossum09be4091999-08-09 14:40:40 +00004769#endif
4770#ifdef IP_DEFAULT_MULTICAST_LOOP
Fred Drake4baedc12002-04-01 14:53:37 +00004771 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_LOOP",
4772 IP_DEFAULT_MULTICAST_LOOP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004773#endif
4774#ifdef IP_MAX_MEMBERSHIPS
Fred Drake4baedc12002-04-01 14:53:37 +00004775 PyModule_AddIntConstant(m, "IP_MAX_MEMBERSHIPS", IP_MAX_MEMBERSHIPS);
Guido van Rossum09be4091999-08-09 14:40:40 +00004776#endif
4777
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004778 /* IPv6 [gs]etsockopt options, defined in RFC2553 */
4779#ifdef IPV6_JOIN_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004780 PyModule_AddIntConstant(m, "IPV6_JOIN_GROUP", IPV6_JOIN_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004781#endif
4782#ifdef IPV6_LEAVE_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004783 PyModule_AddIntConstant(m, "IPV6_LEAVE_GROUP", IPV6_LEAVE_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004784#endif
4785#ifdef IPV6_MULTICAST_HOPS
Fred Drake4baedc12002-04-01 14:53:37 +00004786 PyModule_AddIntConstant(m, "IPV6_MULTICAST_HOPS", IPV6_MULTICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004787#endif
4788#ifdef IPV6_MULTICAST_IF
Fred Drake4baedc12002-04-01 14:53:37 +00004789 PyModule_AddIntConstant(m, "IPV6_MULTICAST_IF", IPV6_MULTICAST_IF);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004790#endif
4791#ifdef IPV6_MULTICAST_LOOP
Fred Drake4baedc12002-04-01 14:53:37 +00004792 PyModule_AddIntConstant(m, "IPV6_MULTICAST_LOOP", IPV6_MULTICAST_LOOP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004793#endif
4794#ifdef IPV6_UNICAST_HOPS
Fred Drake4baedc12002-04-01 14:53:37 +00004795 PyModule_AddIntConstant(m, "IPV6_UNICAST_HOPS", IPV6_UNICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004796#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00004797 /* Additional IPV6 socket options, defined in RFC 3493 */
4798#ifdef IPV6_V6ONLY
4799 PyModule_AddIntConstant(m, "IPV6_V6ONLY", IPV6_V6ONLY);
4800#endif
4801 /* Advanced IPV6 socket options, from RFC 3542 */
4802#ifdef IPV6_CHECKSUM
4803 PyModule_AddIntConstant(m, "IPV6_CHECKSUM", IPV6_CHECKSUM);
4804#endif
4805#ifdef IPV6_DONTFRAG
4806 PyModule_AddIntConstant(m, "IPV6_DONTFRAG", IPV6_DONTFRAG);
4807#endif
4808#ifdef IPV6_DSTOPTS
4809 PyModule_AddIntConstant(m, "IPV6_DSTOPTS", IPV6_DSTOPTS);
4810#endif
4811#ifdef IPV6_HOPLIMIT
4812 PyModule_AddIntConstant(m, "IPV6_HOPLIMIT", IPV6_HOPLIMIT);
4813#endif
4814#ifdef IPV6_HOPOPTS
4815 PyModule_AddIntConstant(m, "IPV6_HOPOPTS", IPV6_HOPOPTS);
4816#endif
4817#ifdef IPV6_NEXTHOP
4818 PyModule_AddIntConstant(m, "IPV6_NEXTHOP", IPV6_NEXTHOP);
4819#endif
4820#ifdef IPV6_PATHMTU
4821 PyModule_AddIntConstant(m, "IPV6_PATHMTU", IPV6_PATHMTU);
4822#endif
4823#ifdef IPV6_PKTINFO
4824 PyModule_AddIntConstant(m, "IPV6_PKTINFO", IPV6_PKTINFO);
4825#endif
4826#ifdef IPV6_RECVDSTOPTS
4827 PyModule_AddIntConstant(m, "IPV6_RECVDSTOPTS", IPV6_RECVDSTOPTS);
4828#endif
4829#ifdef IPV6_RECVHOPLIMIT
4830 PyModule_AddIntConstant(m, "IPV6_RECVHOPLIMIT", IPV6_RECVHOPLIMIT);
4831#endif
4832#ifdef IPV6_RECVHOPOPTS
4833 PyModule_AddIntConstant(m, "IPV6_RECVHOPOPTS", IPV6_RECVHOPOPTS);
4834#endif
4835#ifdef IPV6_RECVPKTINFO
4836 PyModule_AddIntConstant(m, "IPV6_RECVPKTINFO", IPV6_RECVPKTINFO);
4837#endif
4838#ifdef IPV6_RECVRTHDR
4839 PyModule_AddIntConstant(m, "IPV6_RECVRTHDR", IPV6_RECVRTHDR);
4840#endif
4841#ifdef IPV6_RECVTCLASS
4842 PyModule_AddIntConstant(m, "IPV6_RECVTCLASS", IPV6_RECVTCLASS);
4843#endif
4844#ifdef IPV6_RTHDR
4845 PyModule_AddIntConstant(m, "IPV6_RTHDR", IPV6_RTHDR);
4846#endif
4847#ifdef IPV6_RTHDRDSTOPTS
4848 PyModule_AddIntConstant(m, "IPV6_RTHDRDSTOPTS", IPV6_RTHDRDSTOPTS);
4849#endif
4850#ifdef IPV6_RTHDR_TYPE_0
4851 PyModule_AddIntConstant(m, "IPV6_RTHDR_TYPE_0", IPV6_RTHDR_TYPE_0);
4852#endif
4853#ifdef IPV6_RECVPATHMTU
4854 PyModule_AddIntConstant(m, "IPV6_RECVPATHMTU", IPV6_RECVPATHMTU);
4855#endif
4856#ifdef IPV6_TCLASS
4857 PyModule_AddIntConstant(m, "IPV6_TCLASS", IPV6_TCLASS);
4858#endif
4859#ifdef IPV6_USE_MIN_MTU
4860 PyModule_AddIntConstant(m, "IPV6_USE_MIN_MTU", IPV6_USE_MIN_MTU);
4861#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004862
Guido van Rossum09be4091999-08-09 14:40:40 +00004863 /* TCP options */
4864#ifdef TCP_NODELAY
Fred Drake4baedc12002-04-01 14:53:37 +00004865 PyModule_AddIntConstant(m, "TCP_NODELAY", TCP_NODELAY);
Guido van Rossum09be4091999-08-09 14:40:40 +00004866#endif
4867#ifdef TCP_MAXSEG
Fred Drake4baedc12002-04-01 14:53:37 +00004868 PyModule_AddIntConstant(m, "TCP_MAXSEG", TCP_MAXSEG);
Guido van Rossum09be4091999-08-09 14:40:40 +00004869#endif
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004870#ifdef TCP_CORK
Fred Drake4baedc12002-04-01 14:53:37 +00004871 PyModule_AddIntConstant(m, "TCP_CORK", TCP_CORK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004872#endif
4873#ifdef TCP_KEEPIDLE
Fred Drake4baedc12002-04-01 14:53:37 +00004874 PyModule_AddIntConstant(m, "TCP_KEEPIDLE", TCP_KEEPIDLE);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004875#endif
4876#ifdef TCP_KEEPINTVL
Fred Drake4baedc12002-04-01 14:53:37 +00004877 PyModule_AddIntConstant(m, "TCP_KEEPINTVL", TCP_KEEPINTVL);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004878#endif
4879#ifdef TCP_KEEPCNT
Fred Drake4baedc12002-04-01 14:53:37 +00004880 PyModule_AddIntConstant(m, "TCP_KEEPCNT", TCP_KEEPCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004881#endif
4882#ifdef TCP_SYNCNT
Fred Drake4baedc12002-04-01 14:53:37 +00004883 PyModule_AddIntConstant(m, "TCP_SYNCNT", TCP_SYNCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004884#endif
4885#ifdef TCP_LINGER2
Fred Drake4baedc12002-04-01 14:53:37 +00004886 PyModule_AddIntConstant(m, "TCP_LINGER2", TCP_LINGER2);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004887#endif
4888#ifdef TCP_DEFER_ACCEPT
Fred Drake4baedc12002-04-01 14:53:37 +00004889 PyModule_AddIntConstant(m, "TCP_DEFER_ACCEPT", TCP_DEFER_ACCEPT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004890#endif
4891#ifdef TCP_WINDOW_CLAMP
Fred Drake4baedc12002-04-01 14:53:37 +00004892 PyModule_AddIntConstant(m, "TCP_WINDOW_CLAMP", TCP_WINDOW_CLAMP);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004893#endif
4894#ifdef TCP_INFO
Fred Drake4baedc12002-04-01 14:53:37 +00004895 PyModule_AddIntConstant(m, "TCP_INFO", TCP_INFO);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004896#endif
4897#ifdef TCP_QUICKACK
Fred Drake4baedc12002-04-01 14:53:37 +00004898 PyModule_AddIntConstant(m, "TCP_QUICKACK", TCP_QUICKACK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004899#endif
4900
Guido van Rossum09be4091999-08-09 14:40:40 +00004901
4902 /* IPX options */
4903#ifdef IPX_TYPE
Fred Drake4baedc12002-04-01 14:53:37 +00004904 PyModule_AddIntConstant(m, "IPX_TYPE", IPX_TYPE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004905#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00004906
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004907 /* get{addr,name}info parameters */
4908#ifdef EAI_ADDRFAMILY
Fred Drake4baedc12002-04-01 14:53:37 +00004909 PyModule_AddIntConstant(m, "EAI_ADDRFAMILY", EAI_ADDRFAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004910#endif
4911#ifdef EAI_AGAIN
Fred Drake4baedc12002-04-01 14:53:37 +00004912 PyModule_AddIntConstant(m, "EAI_AGAIN", EAI_AGAIN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004913#endif
4914#ifdef EAI_BADFLAGS
Fred Drake4baedc12002-04-01 14:53:37 +00004915 PyModule_AddIntConstant(m, "EAI_BADFLAGS", EAI_BADFLAGS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004916#endif
4917#ifdef EAI_FAIL
Fred Drake4baedc12002-04-01 14:53:37 +00004918 PyModule_AddIntConstant(m, "EAI_FAIL", EAI_FAIL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004919#endif
4920#ifdef EAI_FAMILY
Fred Drake4baedc12002-04-01 14:53:37 +00004921 PyModule_AddIntConstant(m, "EAI_FAMILY", EAI_FAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004922#endif
4923#ifdef EAI_MEMORY
Fred Drake4baedc12002-04-01 14:53:37 +00004924 PyModule_AddIntConstant(m, "EAI_MEMORY", EAI_MEMORY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004925#endif
4926#ifdef EAI_NODATA
Fred Drake4baedc12002-04-01 14:53:37 +00004927 PyModule_AddIntConstant(m, "EAI_NODATA", EAI_NODATA);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004928#endif
4929#ifdef EAI_NONAME
Fred Drake4baedc12002-04-01 14:53:37 +00004930 PyModule_AddIntConstant(m, "EAI_NONAME", EAI_NONAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004931#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00004932#ifdef EAI_OVERFLOW
4933 PyModule_AddIntConstant(m, "EAI_OVERFLOW", EAI_OVERFLOW);
4934#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004935#ifdef EAI_SERVICE
Fred Drake4baedc12002-04-01 14:53:37 +00004936 PyModule_AddIntConstant(m, "EAI_SERVICE", EAI_SERVICE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004937#endif
4938#ifdef EAI_SOCKTYPE
Fred Drake4baedc12002-04-01 14:53:37 +00004939 PyModule_AddIntConstant(m, "EAI_SOCKTYPE", EAI_SOCKTYPE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004940#endif
4941#ifdef EAI_SYSTEM
Fred Drake4baedc12002-04-01 14:53:37 +00004942 PyModule_AddIntConstant(m, "EAI_SYSTEM", EAI_SYSTEM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004943#endif
4944#ifdef EAI_BADHINTS
Fred Drake4baedc12002-04-01 14:53:37 +00004945 PyModule_AddIntConstant(m, "EAI_BADHINTS", EAI_BADHINTS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004946#endif
4947#ifdef EAI_PROTOCOL
Fred Drake4baedc12002-04-01 14:53:37 +00004948 PyModule_AddIntConstant(m, "EAI_PROTOCOL", EAI_PROTOCOL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004949#endif
4950#ifdef EAI_MAX
Fred Drake4baedc12002-04-01 14:53:37 +00004951 PyModule_AddIntConstant(m, "EAI_MAX", EAI_MAX);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004952#endif
4953#ifdef AI_PASSIVE
Fred Drake4baedc12002-04-01 14:53:37 +00004954 PyModule_AddIntConstant(m, "AI_PASSIVE", AI_PASSIVE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004955#endif
4956#ifdef AI_CANONNAME
Fred Drake4baedc12002-04-01 14:53:37 +00004957 PyModule_AddIntConstant(m, "AI_CANONNAME", AI_CANONNAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004958#endif
4959#ifdef AI_NUMERICHOST
Fred Drake4baedc12002-04-01 14:53:37 +00004960 PyModule_AddIntConstant(m, "AI_NUMERICHOST", AI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004961#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00004962#ifdef AI_NUMERICSERV
4963 PyModule_AddIntConstant(m, "AI_NUMERICSERV", AI_NUMERICSERV);
4964#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004965#ifdef AI_MASK
Fred Drake4baedc12002-04-01 14:53:37 +00004966 PyModule_AddIntConstant(m, "AI_MASK", AI_MASK);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004967#endif
4968#ifdef AI_ALL
Fred Drake4baedc12002-04-01 14:53:37 +00004969 PyModule_AddIntConstant(m, "AI_ALL", AI_ALL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004970#endif
4971#ifdef AI_V4MAPPED_CFG
Fred Drake4baedc12002-04-01 14:53:37 +00004972 PyModule_AddIntConstant(m, "AI_V4MAPPED_CFG", AI_V4MAPPED_CFG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004973#endif
4974#ifdef AI_ADDRCONFIG
Fred Drake4baedc12002-04-01 14:53:37 +00004975 PyModule_AddIntConstant(m, "AI_ADDRCONFIG", AI_ADDRCONFIG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004976#endif
4977#ifdef AI_V4MAPPED
Fred Drake4baedc12002-04-01 14:53:37 +00004978 PyModule_AddIntConstant(m, "AI_V4MAPPED", AI_V4MAPPED);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004979#endif
4980#ifdef AI_DEFAULT
Fred Drake4baedc12002-04-01 14:53:37 +00004981 PyModule_AddIntConstant(m, "AI_DEFAULT", AI_DEFAULT);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004982#endif
4983#ifdef NI_MAXHOST
Fred Drake4baedc12002-04-01 14:53:37 +00004984 PyModule_AddIntConstant(m, "NI_MAXHOST", NI_MAXHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004985#endif
4986#ifdef NI_MAXSERV
Fred Drake4baedc12002-04-01 14:53:37 +00004987 PyModule_AddIntConstant(m, "NI_MAXSERV", NI_MAXSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004988#endif
4989#ifdef NI_NOFQDN
Fred Drake4baedc12002-04-01 14:53:37 +00004990 PyModule_AddIntConstant(m, "NI_NOFQDN", NI_NOFQDN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004991#endif
4992#ifdef NI_NUMERICHOST
Fred Drake4baedc12002-04-01 14:53:37 +00004993 PyModule_AddIntConstant(m, "NI_NUMERICHOST", NI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004994#endif
4995#ifdef NI_NAMEREQD
Fred Drake4baedc12002-04-01 14:53:37 +00004996 PyModule_AddIntConstant(m, "NI_NAMEREQD", NI_NAMEREQD);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004997#endif
4998#ifdef NI_NUMERICSERV
Fred Drake4baedc12002-04-01 14:53:37 +00004999 PyModule_AddIntConstant(m, "NI_NUMERICSERV", NI_NUMERICSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005000#endif
5001#ifdef NI_DGRAM
Fred Drake4baedc12002-04-01 14:53:37 +00005002 PyModule_AddIntConstant(m, "NI_DGRAM", NI_DGRAM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00005003#endif
5004
Martin v. Löwis94681fc2003-11-27 19:40:22 +00005005 /* shutdown() parameters */
5006#ifdef SHUT_RD
5007 PyModule_AddIntConstant(m, "SHUT_RD", SHUT_RD);
5008#elif defined(SD_RECEIVE)
5009 PyModule_AddIntConstant(m, "SHUT_RD", SD_RECEIVE);
5010#else
5011 PyModule_AddIntConstant(m, "SHUT_RD", 0);
5012#endif
5013#ifdef SHUT_WR
5014 PyModule_AddIntConstant(m, "SHUT_WR", SHUT_WR);
5015#elif defined(SD_SEND)
5016 PyModule_AddIntConstant(m, "SHUT_WR", SD_SEND);
5017#else
5018 PyModule_AddIntConstant(m, "SHUT_WR", 1);
5019#endif
5020#ifdef SHUT_RDWR
5021 PyModule_AddIntConstant(m, "SHUT_RDWR", SHUT_RDWR);
5022#elif defined(SD_BOTH)
5023 PyModule_AddIntConstant(m, "SHUT_RDWR", SD_BOTH);
5024#else
5025 PyModule_AddIntConstant(m, "SHUT_RDWR", 2);
5026#endif
5027
Christian Heimesfaf2f632008-01-06 16:59:19 +00005028#ifdef SIO_RCVALL
Christian Heimesd32ed6f2008-01-14 18:49:24 +00005029 {
5030 PyObject *tmp;
5031 tmp = PyLong_FromUnsignedLong(SIO_RCVALL);
5032 if (tmp == NULL)
Martin v. Löwisb9e8dcf2008-06-11 05:59:46 +00005033 return NULL;
Christian Heimesd32ed6f2008-01-14 18:49:24 +00005034 PyModule_AddObject(m, "SIO_RCVALL", tmp);
5035 }
Christian Heimesfaf2f632008-01-06 16:59:19 +00005036 PyModule_AddIntConstant(m, "RCVALL_OFF", RCVALL_OFF);
5037 PyModule_AddIntConstant(m, "RCVALL_ON", RCVALL_ON);
5038 PyModule_AddIntConstant(m, "RCVALL_SOCKETLEVELONLY", RCVALL_SOCKETLEVELONLY);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005039#ifdef RCVALL_IPLEVEL
Christian Heimesfaf2f632008-01-06 16:59:19 +00005040 PyModule_AddIntConstant(m, "RCVALL_IPLEVEL", RCVALL_IPLEVEL);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005041#endif
5042#ifdef RCVALL_MAX
Christian Heimesfaf2f632008-01-06 16:59:19 +00005043 PyModule_AddIntConstant(m, "RCVALL_MAX", RCVALL_MAX);
Amaury Forgeot d'Arc762681b2008-06-12 23:03:41 +00005044#endif
Christian Heimesfaf2f632008-01-06 16:59:19 +00005045#endif /* _MSTCPIP_ */
5046
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005047 /* Initialize gethostbyname lock */
Just van Rossum1040d2c2003-05-09 07:53:18 +00005048#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
5049 netdb_lock = PyThread_allocate_lock();
Guido van Rossum4f199ea1998-04-09 20:56:35 +00005050#endif
Martin v. Löwis1a214512008-06-11 05:26:20 +00005051 return m;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00005052}
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005053
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005054
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005055#ifndef HAVE_INET_PTON
Christian Heimes96e7b3d2007-11-20 06:51:17 +00005056#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005057
5058/* Simplistic emulation code for inet_pton that only works for IPv4 */
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00005059/* These are not exposed because they do not set errno properly */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005060
Guido van Rossum3eede5a2002-06-07 02:08:35 +00005061int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005062inet_pton(int af, const char *src, void *dst)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005063{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005064 if (af == AF_INET) {
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005065 long packed_addr;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005066 packed_addr = inet_addr(src);
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005067 if (packed_addr == INADDR_NONE)
5068 return 0;
5069 memcpy(dst, &packed_addr, 4);
5070 return 1;
5071 }
5072 /* Should set errno to EAFNOSUPPORT */
5073 return -1;
5074}
5075
Martin v. Löwisc925b1532001-07-21 09:42:15 +00005076const char *
5077inet_ntop(int af, const void *src, char *dst, socklen_t size)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005078{
5079 if (af == AF_INET) {
5080 struct in_addr packed_addr;
5081 if (size < 16)
5082 /* Should set errno to ENOSPC. */
5083 return NULL;
5084 memcpy(&packed_addr, src, sizeof(packed_addr));
5085 return strncpy(dst, inet_ntoa(packed_addr), size);
5086 }
5087 /* Should set errno to EAFNOSUPPORT */
5088 return NULL;
5089}
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00005090
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00005091#endif
Christian Heimesb6150692007-11-15 23:37:07 +00005092#endif