blob: 463cd8e3181061f9db9bbc5e3151e36ebde9efbc [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
Jeremy Hylton22308652001-02-02 03:23:09 +000010 portable manner, though AF_PACKET is supported under Linux.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000011- No read/write operations (use sendall/recv or makefile instead).
12- Additional restrictions apply on some non-Unix platforms (compensated
13 for by socket.py).
Guido van Rossum6574b3e1991-06-25 21:36:08 +000014
Guido van Rossum27e177d1995-03-16 15:43:47 +000015Module interface:
Guido van Rossum6574b3e1991-06-25 21:36:08 +000016
Guido van Rossum27e177d1995-03-16 15:43:47 +000017- socket.error: exception raised for socket specific errors
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000018- socket.gaierror: exception raised for getaddrinfo/getnameinfo errors,
19 a subclass of socket.error
20- socket.herror: exception raised for gethostby* errors,
21 a subclass of socket.error
Guido van Rossum83a072d2002-09-03 19:10:18 +000022- socket.fromfd(fd, family, type[, proto]) --> new socket object (created
23 from an existing file descriptor)
Guido van Rossum30a685f1991-06-27 15:51:29 +000024- socket.gethostbyname(hostname) --> host IP address (string: 'dd.dd.dd.dd')
Guido van Rossum3bbc62e1995-01-02 19:30:30 +000025- socket.gethostbyaddr(IP address) --> (hostname, [alias, ...], [IP addr, ...])
Guido van Rossum27e177d1995-03-16 15:43:47 +000026- socket.gethostname() --> host name (string: 'spam' or 'spam.domain.com')
Guido van Rossum25405c71996-12-19 16:42:52 +000027- socket.getprotobyname(protocolname) --> protocol number
Barry Warsaw11b91a02004-06-28 00:50:43 +000028- socket.getservbyname(servicename[, protocolname]) --> port number
29- socket.getservbyport(portnumber[, protocolname]) --> service name
Guido van Rossum83a072d2002-09-03 19:10:18 +000030- socket.socket([family[, type [, proto]]]) --> new socket object
Guido van Rossum006bf911996-06-12 04:04:55 +000031- socket.ntohs(16 bit value) --> new int object
32- socket.ntohl(32 bit value) --> new int object
33- socket.htons(16 bit value) --> new int object
34- socket.htonl(32 bit value) --> new int object
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000035- socket.getaddrinfo(host, port [, family, socktype, proto, flags])
36 --> List of (family, socktype, proto, canonname, sockaddr)
37- socket.getnameinfo(sockaddr, flags) --> (host, port)
Guido van Rossum27e177d1995-03-16 15:43:47 +000038- socket.AF_INET, socket.SOCK_STREAM, etc.: constants from <socket.h>
Guido van Rossum47dfa4a2003-04-25 05:48:32 +000039- socket.has_ipv6: boolean value indicating if IPv6 is supported
Guido van Rossum5c9eb211999-08-20 18:21:51 +000040- socket.inet_aton(IP address) -> 32-bit packed IP representation
41- socket.inet_ntoa(packed IP) -> IP address string
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +000042- socket.getdefaulttimeout() -> None | float
43- socket.setdefaulttimeout(None | float)
Guido van Rossum6574b3e1991-06-25 21:36:08 +000044- an Internet socket address is a pair (hostname, port)
45 where hostname can be anything recognized by gethostbyname()
46 (including the dd.dd.dd.dd notation) and port is in host byte order
47- where a hostname is returned, the dd.dd.dd.dd notation is used
Guido van Rossum27e177d1995-03-16 15:43:47 +000048- a UNIX domain socket address is a string specifying the pathname
Jeremy Hylton22308652001-02-02 03:23:09 +000049- an AF_PACKET socket address is a tuple containing a string
50 specifying the ethernet interface and an integer specifying
51 the Ethernet protocol number to be received. For example:
Jeremy Hyltondbfb6622001-02-02 19:55:17 +000052 ("eth0",0x1234). Optional 3rd,4th,5th elements in the tuple
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +000053 specify packet-type and ha-type/addr.
Guido van Rossum6574b3e1991-06-25 21:36:08 +000054
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000055Local naming conventions:
Guido van Rossum6574b3e1991-06-25 21:36:08 +000056
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000057- names starting with sock_ are socket object methods
58- names starting with socket_ are module-level functions
59- names starting with PySocket are exported through socketmodule.h
Guido van Rossum30a685f1991-06-27 15:51:29 +000060
Guido van Rossum6574b3e1991-06-25 21:36:08 +000061*/
62
Michael W. Hudson76b8cc82004-05-26 17:06:31 +000063#if defined(__sgi) && _COMPILER_VERSION>700 && !_SGIAPI
64#define _BSD_TYPES
65#endif
66
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000067#include "Python.h"
68
Guido van Rossum47dfa4a2003-04-25 05:48:32 +000069#undef MAX
70#define MAX(x, y) ((x) < (y) ? (y) : (x))
71
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000072/* Socket object documentation */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000073PyDoc_STRVAR(sock_doc,
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000074"socket([family[, type[, proto]]]) -> socket object\n\
75\n\
76Open a socket of the given type. The family argument specifies the\n\
77address family; it defaults to AF_INET. The type argument specifies\n\
78whether this is a stream (SOCK_STREAM, this is the default)\n\
79or datagram (SOCK_DGRAM) socket. The protocol argument defaults to 0,\n\
80specifying the default protocol. Keyword arguments are accepted.\n\
81\n\
82A socket object represents one endpoint of a network connection.\n\
83\n\
84Methods of socket objects (keyword arguments not allowed):\n\
85\n\
86accept() -- accept a connection, returning new socket and client address\n\
87bind(addr) -- bind the socket to a local address\n\
88close() -- close the socket\n\
89connect(addr) -- connect the socket to a remote address\n\
90connect_ex(addr) -- connect, return an error code instead of an exception\n\
91dup() -- return a new socket object identical to the current one [*]\n\
92fileno() -- return underlying file descriptor\n\
93getpeername() -- return remote address [*]\n\
94getsockname() -- return local address\n\
95getsockopt(level, optname[, buflen]) -- get socket options\n\
96gettimeout() -- return timeout or None\n\
97listen(n) -- start listening for incoming connections\n\
98makefile([mode, [bufsize]]) -- return a file object for the socket [*]\n\
99recv(buflen[, flags]) -- receive data\n\
100recvfrom(buflen[, flags]) -- receive data and sender's address\n\
101sendall(data[, flags]) -- send all data\n\
102send(data[, flags]) -- send data, may not send all of it\n\
103sendto(data[, flags], addr) -- send data to a given address\n\
104setblocking(0 | 1) -- set or clear the blocking I/O flag\n\
105setsockopt(level, optname, value) -- set socket options\n\
106settimeout(None | float) -- set or clear the timeout\n\
107shutdown(how) -- shut down traffic in one or both directions\n\
108\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000109 [*] not available on all platforms!");
Guido van Rossum3baaa131999-03-22 21:44:51 +0000110
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000111/* XXX This is a terrible mess of platform-dependent preprocessor hacks.
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000112 I hope some day someone can clean this up please... */
113
Guido van Rossum9376b741999-09-15 22:01:40 +0000114/* Hacks for gethostbyname_r(). On some non-Linux platforms, the configure
115 script doesn't get this right, so we hardcode some platform checks below.
116 On the other hand, not all Linux versions agree, so there the settings
117 computed by the configure script are needed! */
118
119#ifndef linux
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000120# undef HAVE_GETHOSTBYNAME_R_3_ARG
121# undef HAVE_GETHOSTBYNAME_R_5_ARG
122# undef HAVE_GETHOSTBYNAME_R_6_ARG
Guido van Rossum9376b741999-09-15 22:01:40 +0000123#endif
Guido van Rossume7de2061999-03-24 17:24:33 +0000124
Guido van Rossum7a122991999-04-13 04:07:32 +0000125#ifndef WITH_THREAD
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000126# undef HAVE_GETHOSTBYNAME_R
Guido van Rossum7a122991999-04-13 04:07:32 +0000127#endif
128
Guido van Rossume7de2061999-03-24 17:24:33 +0000129#ifdef HAVE_GETHOSTBYNAME_R
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000130# if defined(_AIX) || defined(__osf__)
131# define HAVE_GETHOSTBYNAME_R_3_ARG
132# elif defined(__sun) || defined(__sgi)
133# define HAVE_GETHOSTBYNAME_R_5_ARG
134# elif defined(linux)
Guido van Rossum9376b741999-09-15 22:01:40 +0000135/* Rely on the configure script */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000136# else
137# undef HAVE_GETHOSTBYNAME_R
138# endif
Guido van Rossume7de2061999-03-24 17:24:33 +0000139#endif
140
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000141#if !defined(HAVE_GETHOSTBYNAME_R) && defined(WITH_THREAD) && \
142 !defined(MS_WINDOWS)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000143# define USE_GETHOSTBYNAME_LOCK
Guido van Rossum3baaa131999-03-22 21:44:51 +0000144#endif
145
Just van Rossum1040d2c2003-05-09 07:53:18 +0000146/* On systems on which getaddrinfo() is believed to not be thread-safe,
Just van Rossum09aecd72003-05-09 08:03:44 +0000147 (this includes the getaddrinfo emulation) protect access with a lock. */
Just van Rossum16e426b2003-05-09 08:12:00 +0000148#if defined(WITH_THREAD) && (defined(__APPLE__) || defined(__FreeBSD__) || \
149 defined(__OpenBSD__) || defined(__NetBSD__) || !defined(HAVE_GETADDRINFO))
Just van Rossum1040d2c2003-05-09 07:53:18 +0000150#define USE_GETADDRINFO_LOCK
151#endif
152
153#ifdef USE_GETADDRINFO_LOCK
154#define ACQUIRE_GETADDRINFO_LOCK PyThread_acquire_lock(netdb_lock, 1);
155#define RELEASE_GETADDRINFO_LOCK PyThread_release_lock(netdb_lock);
156#else
157#define ACQUIRE_GETADDRINFO_LOCK
158#define RELEASE_GETADDRINFO_LOCK
159#endif
160
161#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000162# include "pythread.h"
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000163#endif
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000164
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000165#if defined(PYCC_VACPP)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000166# include <types.h>
167# include <io.h>
168# include <sys/ioctl.h>
169# include <utils.h>
170# include <ctype.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000171#endif
172
Martin v. Löwis9e437302002-12-06 12:57:26 +0000173#if defined(__VMS)
174#if ! defined(_SOCKADDR_LEN)
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000175# ifdef getaddrinfo
176# undef getaddrinfo
177# endif
178# include "TCPIP_IOCTL_ROUTINE"
179#else
180# include <ioctl.h>
181#endif
Martin v. Löwis9e437302002-12-06 12:57:26 +0000182#endif
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000183
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000184#if defined(PYOS_OS2)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000185# define INCL_DOS
186# define INCL_DOSERRORS
187# define INCL_NOPMAPI
188# include <os2.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000189#endif
190
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000191#if defined(__sgi) && _COMPILER_VERSION>700 && !_SGIAPI
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000192/* make sure that the reentrant (gethostbyaddr_r etc)
193 functions are declared correctly if compiling with
194 MIPSPro 7.x in ANSI C mode (default) */
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000195
196/* XXX Using _SGIAPI is the wrong thing,
197 but I don't know what the right thing is. */
Michael W. Hudson76b8cc82004-05-26 17:06:31 +0000198#undef _SGIAPI /* to avoid warning */
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000199#define _SGIAPI 1
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000200
Michael W. Hudson76b8cc82004-05-26 17:06:31 +0000201#undef _XOPEN_SOURCE
202#include <sys/socket.h>
203#include <sys/types.h>
204#include <netinet/in.h>
205#ifdef _SS_ALIGNSIZE
206#define HAVE_GETADDRINFO 1
207#define HAVE_GETNAMEINFO 1
208#endif
209
Jeremy Hyltonfb509a32003-07-17 16:58:48 +0000210#define HAVE_INET_PTON
211#include <netdb.h>
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000212#endif
213
Anthony Baxterbab23cf2003-10-04 08:00:49 +0000214/* Irix 6.5 fails to define this variable at all. This is needed
215 for both GCC and SGI's compiler. I'd say that the SGI headers
216 are just busted. */
217#if defined(__sgi) && !defined(INET_ADDRSTRLEN)
218#define INET_ADDRSTRLEN 16
219#endif
220
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000221/* Generic includes */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000222#include <sys/types.h>
Guido van Rossum81194471991-07-27 21:42:02 +0000223#include <signal.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000224
Marc-André Lemburg976ade62002-02-16 18:47:07 +0000225/* Generic socket object definitions and includes */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000226#define PySocket_BUILDING_SOCKET
Marc-André Lemburgbb8b78b2002-02-16 18:44:52 +0000227#include "socketmodule.h"
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000228
229/* Addressing includes */
230
Guido van Rossum6f489d91996-06-28 20:15:15 +0000231#ifndef MS_WINDOWS
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000232
233/* Non-MS WINDOWS includes */
234# include <netdb.h>
Guido van Rossum5c9eb211999-08-20 18:21:51 +0000235
Guido van Rossum9376b741999-09-15 22:01:40 +0000236/* Headers needed for inet_ntoa() and inet_addr() */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000237# ifdef __BEOS__
238# include <net/netdb.h>
239# elif defined(PYOS_OS2) && defined(PYCC_VACPP)
240# include <netdb.h>
Tim Peters603c6832001-11-05 02:45:59 +0000241typedef size_t socklen_t;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000242# else
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000243# include <arpa/inet.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000244# endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +0000245
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000246# ifndef RISCOS
247# include <fcntl.h>
248# else
Martin v. Löwisa94568a2003-05-10 07:36:56 +0000249# include <sys/ioctl.h>
250# include <socklib.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000251# define NO_DUP
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000252int h_errno; /* not used */
Martin v. Löwisa94568a2003-05-10 07:36:56 +0000253# define INET_ADDRSTRLEN 16
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000254# endif
255
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000256#else
Guido van Rossum48a680c2001-03-02 06:34:14 +0000257
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000258/* MS_WINDOWS includes */
259# include <fcntl.h>
Guido van Rossum48a680c2001-03-02 06:34:14 +0000260
Jeremy Hylton22308652001-02-02 03:23:09 +0000261#endif
262
Skip Montanaro7befb992004-02-10 16:50:21 +0000263#include <stddef.h>
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000264
265#ifndef offsetof
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000266# define offsetof(type, member) ((size_t)(&((type *)0)->member))
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000267#endif
268
Neal Norwitz39d22e52002-11-02 19:55:21 +0000269#ifndef O_NONBLOCK
270# define O_NONBLOCK O_NDELAY
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000271#endif
272
Michael W. Hudson76b8cc82004-05-26 17:06:31 +0000273#if defined(__sgi) && _COMPILER_VERSION>700 \
274 && !defined(_SS_ALIGNSIZE) /* defined in sys/socket.h */
275 /* by some newer versions of IRIX */
276 /* (e.g. not by 6.5.10 but by 6.5.21) */
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000277#include "addrinfo.h"
Michael W. Hudson76b8cc82004-05-26 17:06:31 +0000278#endif
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000279
Andrew MacIntyre5a8b4592004-07-07 14:02:59 +0000280#if defined(PYOS_OS2) && defined(PYCC_GCC)
281#include "addrinfo.h"
282#endif
283
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000284#ifndef HAVE_INET_PTON
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000285int inet_pton(int af, const char *src, void *dst);
Martin v. Löwisc925b1532001-07-21 09:42:15 +0000286const char *inet_ntop(int af, const void *src, char *dst, socklen_t size);
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000287#endif
288
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000289#ifdef __APPLE__
290/* On OS X, getaddrinfo returns no error indication of lookup
291 failure, so we must use the emulation instead of the libinfo
292 implementation. Unfortunately, performing an autoconf test
293 for this bug would require DNS access for the machine performing
294 the configuration, which is not acceptable. Therefore, we
295 determine the bug just by checking for __APPLE__. If this bug
296 gets ever fixed, perhaps checking for sys/version.h would be
297 appropriate, which is 10/0 on the system with the bug. */
Jack Jansen84262fb2002-07-02 14:40:42 +0000298#ifndef HAVE_GETNAMEINFO
299/* This bug seems to be fixed in Jaguar. Ths easiest way I could
300 Find to check for Jaguar is that it has getnameinfo(), which
301 older releases don't have */
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000302#undef HAVE_GETADDRINFO
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000303#endif
Jack Jansen84262fb2002-07-02 14:40:42 +0000304#endif
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000305
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000306/* I know this is a bad practice, but it is the easiest... */
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000307#if !defined(HAVE_GETADDRINFO)
Martin v. Löwisfccac2e2003-05-01 05:20:46 +0000308/* avoid clashes with the C library definition of the symbol. */
309#define getaddrinfo fake_getaddrinfo
310#define gai_strerror fake_gai_strerror
311#define freeaddrinfo fake_freeaddrinfo
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000312#include "getaddrinfo.c"
313#endif
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000314#if !defined(HAVE_GETNAMEINFO)
Martin v. Löwisfccac2e2003-05-01 05:20:46 +0000315#define getnameinfo fake_getnameinfo
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000316#include "getnameinfo.c"
317#endif
318
Guido van Rossumbcc20741998-08-04 22:53:56 +0000319#if defined(MS_WINDOWS) || defined(__BEOS__)
320/* BeOS suffers from the same socket dichotomy as Win32... - [cjh] */
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000321/* seem to be a few differences in the API */
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000322#define SOCKETCLOSE closesocket
Guido van Rossumbe32c891996-06-20 16:25:29 +0000323#define NO_DUP /* Actually it exists on NT 3.5, but what the heck... */
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000324#endif
325
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000326#ifdef MS_WIN32
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000327#define EAFNOSUPPORT WSAEAFNOSUPPORT
328#define snprintf _snprintf
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000329#endif
Fred Drakea04eaad2000-06-30 02:46:07 +0000330
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000331#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000332#define SOCKETCLOSE soclose
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000333#define NO_DUP /* Sockets are Not Actual File Handles under OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000334#endif
335
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000336#ifndef SOCKETCLOSE
337#define SOCKETCLOSE close
338#endif
339
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000340#ifdef __VMS
341/* TCP/IP Services for VMS uses a maximum send/revc buffer length of 65535 */
342#define SEGMENT_SIZE 65535
343#endif
344
Hye-Shik Chang81268602004-02-02 06:05:24 +0000345#if defined(HAVE_BLUETOOTH_H) || defined(HAVE_BLUETOOTH_BLUETOOTH_H)
346#define USE_BLUETOOTH 1
347#if defined(__FreeBSD__)
348#define BTPROTO_L2CAP BLUETOOTH_PROTO_L2CAP
349#define BTPROTO_RFCOMM BLUETOOTH_PROTO_RFCOMM
350#define sockaddr_l2 sockaddr_l2cap
351#define sockaddr_rc sockaddr_rfcomm
352#define _BT_SOCKADDR_MEMB(s, proto) &((s)->sock_addr)
353#define _BT_L2_MEMB(sa, memb) ((sa)->l2cap_##memb)
354#define _BT_RC_MEMB(sa, memb) ((sa)->rfcomm_##memb)
355#else
Anthony Baxter5d7c0672004-02-16 05:35:28 +0000356#define _BT_SOCKADDR_MEMB(s, proto) (&((s)->sock_addr).bt_##proto)
Hye-Shik Chang81268602004-02-02 06:05:24 +0000357#define _BT_L2_MEMB(sa, memb) ((sa)->l2_##memb)
358#define _BT_RC_MEMB(sa, memb) ((sa)->rc_##memb)
359#define _BT_SCO_MEMB(sa, memb) ((sa)->sco_##memb)
360#endif
361#endif
362
Martin v. Löwise9416172003-05-03 10:12:45 +0000363/*
364 * Constants for getnameinfo()
365 */
366#if !defined(NI_MAXHOST)
367#define NI_MAXHOST 1025
368#endif
369#if !defined(NI_MAXSERV)
370#define NI_MAXSERV 32
371#endif
372
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000373/* XXX There's a problem here: *static* functions are not supposed to have
374 a Py prefix (or use CapitalizedWords). Later... */
375
Guido van Rossum30a685f1991-06-27 15:51:29 +0000376/* Global variable holding the exception type for errors detected
377 by this module (but not argument type or memory errors, etc.). */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000378static PyObject *socket_error;
379static PyObject *socket_herror;
380static PyObject *socket_gaierror;
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000381static PyObject *socket_timeout;
Guido van Rossum30a685f1991-06-27 15:51:29 +0000382
Guido van Rossum48a680c2001-03-02 06:34:14 +0000383#ifdef RISCOS
384/* Global variable which is !=0 if Python is running in a RISC OS taskwindow */
385static int taskwindow;
386#endif
387
Tim Peters643a7fc2002-02-17 04:13:21 +0000388/* A forward reference to the socket type object.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000389 The sock_type variable contains pointers to various functions,
390 some of which call new_sockobject(), which uses sock_type, so
Tim Peters643a7fc2002-02-17 04:13:21 +0000391 there has to be a circular reference. */
Jeremy Hylton938ace62002-07-17 16:30:39 +0000392static PyTypeObject sock_type;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000393
Guido van Rossum30a685f1991-06-27 15:51:29 +0000394/* Convenience function to raise an error according to errno
395 and return a NULL pointer from a function. */
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000396
Guido van Rossum73624e91994-10-10 17:59:00 +0000397static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000398set_error(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000399{
Guido van Rossum8d665e61996-06-26 18:22:49 +0000400#ifdef MS_WINDOWS
Mark Hammond46a733d2000-07-24 01:45:11 +0000401 int err_no = WSAGetLastError();
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000402 static struct {
403 int no;
404 const char *msg;
405 } *msgp, msgs[] = {
406 {WSAEINTR, "Interrupted system call"},
407 {WSAEBADF, "Bad file descriptor"},
408 {WSAEACCES, "Permission denied"},
409 {WSAEFAULT, "Bad address"},
410 {WSAEINVAL, "Invalid argument"},
411 {WSAEMFILE, "Too many open files"},
412 {WSAEWOULDBLOCK,
413 "The socket operation could not complete "
414 "without blocking"},
415 {WSAEINPROGRESS, "Operation now in progress"},
416 {WSAEALREADY, "Operation already in progress"},
417 {WSAENOTSOCK, "Socket operation on non-socket"},
418 {WSAEDESTADDRREQ, "Destination address required"},
419 {WSAEMSGSIZE, "Message too long"},
420 {WSAEPROTOTYPE, "Protocol wrong type for socket"},
421 {WSAENOPROTOOPT, "Protocol not available"},
422 {WSAEPROTONOSUPPORT, "Protocol not supported"},
423 {WSAESOCKTNOSUPPORT, "Socket type not supported"},
424 {WSAEOPNOTSUPP, "Operation not supported"},
425 {WSAEPFNOSUPPORT, "Protocol family not supported"},
426 {WSAEAFNOSUPPORT, "Address family not supported"},
427 {WSAEADDRINUSE, "Address already in use"},
428 {WSAEADDRNOTAVAIL, "Can't assign requested address"},
429 {WSAENETDOWN, "Network is down"},
430 {WSAENETUNREACH, "Network is unreachable"},
431 {WSAENETRESET, "Network dropped connection on reset"},
432 {WSAECONNABORTED, "Software caused connection abort"},
433 {WSAECONNRESET, "Connection reset by peer"},
434 {WSAENOBUFS, "No buffer space available"},
435 {WSAEISCONN, "Socket is already connected"},
436 {WSAENOTCONN, "Socket is not connected"},
437 {WSAESHUTDOWN, "Can't send after socket shutdown"},
438 {WSAETOOMANYREFS, "Too many references: can't splice"},
439 {WSAETIMEDOUT, "Operation timed out"},
440 {WSAECONNREFUSED, "Connection refused"},
441 {WSAELOOP, "Too many levels of symbolic links"},
442 {WSAENAMETOOLONG, "File name too long"},
443 {WSAEHOSTDOWN, "Host is down"},
444 {WSAEHOSTUNREACH, "No route to host"},
445 {WSAENOTEMPTY, "Directory not empty"},
446 {WSAEPROCLIM, "Too many processes"},
447 {WSAEUSERS, "Too many users"},
448 {WSAEDQUOT, "Disc quota exceeded"},
449 {WSAESTALE, "Stale NFS file handle"},
450 {WSAEREMOTE, "Too many levels of remote in path"},
451 {WSASYSNOTREADY, "Network subsystem is unvailable"},
452 {WSAVERNOTSUPPORTED, "WinSock version is not supported"},
453 {WSANOTINITIALISED,
454 "Successful WSAStartup() not yet performed"},
455 {WSAEDISCON, "Graceful shutdown in progress"},
456 /* Resolver errors */
457 {WSAHOST_NOT_FOUND, "No such host is known"},
458 {WSATRY_AGAIN, "Host not found, or server failed"},
459 {WSANO_RECOVERY, "Unexpected server error encountered"},
460 {WSANO_DATA, "Valid name without requested data"},
461 {WSANO_ADDRESS, "No address, look for MX record"},
462 {0, NULL}
463 };
Mark Hammond46a733d2000-07-24 01:45:11 +0000464 if (err_no) {
Guido van Rossum73624e91994-10-10 17:59:00 +0000465 PyObject *v;
Mark Hammond46a733d2000-07-24 01:45:11 +0000466 const char *msg = "winsock error";
Guido van Rossum48a680c2001-03-02 06:34:14 +0000467
Mark Hammond46a733d2000-07-24 01:45:11 +0000468 for (msgp = msgs; msgp->msg; msgp++) {
469 if (err_no == msgp->no) {
470 msg = msgp->msg;
471 break;
472 }
473 }
474
475 v = Py_BuildValue("(is)", err_no, msg);
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000476 if (v != NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000477 PyErr_SetObject(socket_error, v);
Guido van Rossum73624e91994-10-10 17:59:00 +0000478 Py_DECREF(v);
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000479 }
480 return NULL;
481 }
482 else
483#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000484
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000485#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000486 if (sock_errno() != NO_ERROR) {
487 APIRET rc;
488 ULONG msglen;
489 char outbuf[100];
490 int myerrorcode = sock_errno();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000491
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000492 /* Retrieve socket-related error message from MPTN.MSG file */
493 rc = DosGetMessage(NULL, 0, outbuf, sizeof(outbuf),
494 myerrorcode - SOCBASEERR + 26,
495 "mptn.msg",
496 &msglen);
497 if (rc == NO_ERROR) {
498 PyObject *v;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000499
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000500 /* OS/2 doesn't guarantee a terminator */
501 outbuf[msglen] = '\0';
502 if (strlen(outbuf) > 0) {
503 /* If non-empty msg, trim CRLF */
504 char *lastc = &outbuf[ strlen(outbuf)-1 ];
505 while (lastc > outbuf && isspace(*lastc)) {
506 /* Trim trailing whitespace (CRLF) */
507 *lastc-- = '\0';
508 }
509 }
510 v = Py_BuildValue("(is)", myerrorcode, outbuf);
511 if (v != NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000512 PyErr_SetObject(socket_error, v);
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000513 Py_DECREF(v);
514 }
515 return NULL;
516 }
517 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000518#endif
519
Martin v. Löwisa94568a2003-05-10 07:36:56 +0000520#if defined(RISCOS)
521 if (_inet_error.errnum != NULL) {
522 PyObject *v;
523 v = Py_BuildValue("(is)", errno, _inet_err());
524 if (v != NULL) {
525 PyErr_SetObject(socket_error, v);
526 Py_DECREF(v);
527 }
528 return NULL;
529 }
530#endif
531
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000532 return PyErr_SetFromErrno(socket_error);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000533}
534
Guido van Rossum30a685f1991-06-27 15:51:29 +0000535
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000536static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000537set_herror(int h_error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000538{
539 PyObject *v;
540
541#ifdef HAVE_HSTRERROR
542 v = Py_BuildValue("(is)", h_error, (char *)hstrerror(h_error));
543#else
544 v = Py_BuildValue("(is)", h_error, "host not found");
545#endif
546 if (v != NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000547 PyErr_SetObject(socket_herror, v);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000548 Py_DECREF(v);
549 }
550
551 return NULL;
552}
553
554
555static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000556set_gaierror(int error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000557{
558 PyObject *v;
559
Martin v. Löwis272cb402002-03-01 08:31:07 +0000560#ifdef EAI_SYSTEM
561 /* EAI_SYSTEM is not available on Windows XP. */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000562 if (error == EAI_SYSTEM)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000563 return set_error();
Martin v. Löwis272cb402002-03-01 08:31:07 +0000564#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000565
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000566#ifdef HAVE_GAI_STRERROR
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000567 v = Py_BuildValue("(is)", error, gai_strerror(error));
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000568#else
569 v = Py_BuildValue("(is)", error, "getaddrinfo failed");
570#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000571 if (v != NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000572 PyErr_SetObject(socket_gaierror, v);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000573 Py_DECREF(v);
574 }
575
576 return NULL;
577}
578
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000579/* Function to perform the setting of socket blocking mode
580 internally. block = (1 | 0). */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000581static int
582internal_setblocking(PySocketSockObject *s, int block)
583{
584#ifndef RISCOS
585#ifndef MS_WINDOWS
586 int delay_flag;
587#endif
588#endif
589
590 Py_BEGIN_ALLOW_THREADS
591#ifdef __BEOS__
592 block = !block;
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000593 setsockopt(s->sock_fd, SOL_SOCKET, SO_NONBLOCK,
594 (void *)(&block), sizeof(int));
Guido van Rossum67f7a382002-06-06 21:08:16 +0000595#else
596#ifndef RISCOS
597#ifndef MS_WINDOWS
598#if defined(PYOS_OS2) && !defined(PYCC_GCC)
599 block = !block;
600 ioctl(s->sock_fd, FIONBIO, (caddr_t)&block, sizeof(block));
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000601#elif defined(__VMS)
602 block = !block;
603 ioctl(s->sock_fd, FIONBIO, (char *)&block);
604#else /* !PYOS_OS2 && !_VMS */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000605 delay_flag = fcntl(s->sock_fd, F_GETFL, 0);
606 if (block)
Neal Norwitz39d22e52002-11-02 19:55:21 +0000607 delay_flag &= (~O_NONBLOCK);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000608 else
Neal Norwitz39d22e52002-11-02 19:55:21 +0000609 delay_flag |= O_NONBLOCK;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000610 fcntl(s->sock_fd, F_SETFL, delay_flag);
611#endif /* !PYOS_OS2 */
612#else /* MS_WINDOWS */
613 block = !block;
614 ioctlsocket(s->sock_fd, FIONBIO, (u_long*)&block);
615#endif /* MS_WINDOWS */
Martin v. Löwisa94568a2003-05-10 07:36:56 +0000616#else /* RISCOS */
617 block = !block;
618 socketioctl(s->sock_fd, FIONBIO, (u_long*)&block);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000619#endif /* RISCOS */
Martin v. Löwisa94568a2003-05-10 07:36:56 +0000620#endif /* __BEOS__ */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000621 Py_END_ALLOW_THREADS
622
623 /* Since these don't return anything */
624 return 1;
625}
626
Guido van Rossum11ba0942002-06-13 15:07:44 +0000627/* Do a select() on the socket, if necessary (sock_timeout > 0).
628 The argument writing indicates the direction.
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000629 This does not raise an exception; we'll let our caller do that
630 after they've reacquired the interpreter lock.
631 Returns 1 on timeout, 0 otherwise. */
632static int
Guido van Rossumb9e916a2002-06-07 01:42:47 +0000633internal_select(PySocketSockObject *s, int writing)
Guido van Rossum67f7a382002-06-06 21:08:16 +0000634{
635 fd_set fds;
636 struct timeval tv;
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000637 int n;
Guido van Rossum11ba0942002-06-13 15:07:44 +0000638
Guido van Rossumad654902002-07-19 12:44:59 +0000639 /* Nothing to do unless we're in timeout mode (not non-blocking) */
Guido van Rossum11ba0942002-06-13 15:07:44 +0000640 if (s->sock_timeout <= 0.0)
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000641 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000642
Guido van Rossumad654902002-07-19 12:44:59 +0000643 /* Guard against closed socket */
644 if (s->sock_fd < 0)
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000645 return 0;
Guido van Rossumad654902002-07-19 12:44:59 +0000646
Guido van Rossum67f7a382002-06-06 21:08:16 +0000647 /* Construct the arguments to select */
648 tv.tv_sec = (int)s->sock_timeout;
Guido van Rossumb9e916a2002-06-07 01:42:47 +0000649 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000650 FD_ZERO(&fds);
651 FD_SET(s->sock_fd, &fds);
652
653 /* See if the socket is ready */
Guido van Rossumb9e916a2002-06-07 01:42:47 +0000654 if (writing)
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000655 n = select(s->sock_fd+1, NULL, &fds, NULL, &tv);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000656 else
Raymond Hettingeref7343c2003-06-29 03:08:05 +0000657 n = select(s->sock_fd+1, &fds, NULL, NULL, &tv);
658 if (n == 0)
659 return 1;
660 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000661}
662
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000663/* Initialize a new socket object. */
664
Tim Petersa12b4cf2002-07-18 22:38:44 +0000665static double defaulttimeout = -1.0; /* Default timeout for new sockets */
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000666
Mark Hammond62b1ab12002-07-23 06:31:15 +0000667PyMODINIT_FUNC
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000668init_sockobject(PySocketSockObject *s,
669 SOCKET_T fd, int family, int type, int proto)
670{
671#ifdef RISCOS
672 int block = 1;
673#endif
674 s->sock_fd = fd;
675 s->sock_family = family;
676 s->sock_type = type;
677 s->sock_proto = proto;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000678 s->sock_timeout = defaulttimeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000679
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000680 s->errorhandler = &set_error;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000681
682 if (defaulttimeout >= 0.0)
683 internal_setblocking(s, 0);
684
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000685#ifdef RISCOS
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000686 if (taskwindow)
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000687 socketioctl(s->sock_fd, 0x80046679, (u_long*)&block);
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000688#endif
689}
690
691
Guido van Rossum30a685f1991-06-27 15:51:29 +0000692/* Create a new socket object.
693 This just creates the object and initializes it.
694 If the creation fails, return NULL and set an exception (implicit
695 in NEWOBJ()). */
696
Guido van Rossum73624e91994-10-10 17:59:00 +0000697static PySocketSockObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000698new_sockobject(SOCKET_T fd, int family, int type, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000699{
Guido van Rossum73624e91994-10-10 17:59:00 +0000700 PySocketSockObject *s;
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000701 s = (PySocketSockObject *)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000702 PyType_GenericNew(&sock_type, NULL, NULL);
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000703 if (s != NULL)
704 init_sockobject(s, fd, family, type, proto);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000705 return s;
706}
707
Guido van Rossum30a685f1991-06-27 15:51:29 +0000708
Guido van Rossum48a680c2001-03-02 06:34:14 +0000709/* Lock to allow python interpreter to continue, but only allow one
Just van Rossum1040d2c2003-05-09 07:53:18 +0000710 thread to be in gethostbyname or getaddrinfo */
711#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
712PyThread_type_lock netdb_lock;
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000713#endif
714
715
Guido van Rossum30a685f1991-06-27 15:51:29 +0000716/* Convert a string specifying a host name or one of a few symbolic
717 names to a numeric IP address. This usually calls gethostbyname()
718 to do the work; the names "" and "<broadcast>" are special.
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000719 Return the length (IPv4 should be 4 bytes), or negative if
Guido van Rossum30a685f1991-06-27 15:51:29 +0000720 an error occurred; then an exception is raised. */
721
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000722static int
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000723setipaddr(char *name, struct sockaddr *addr_ret, size_t addr_ret_size, int af)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000724{
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000725 struct addrinfo hints, *res;
726 int error;
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000727 int d1, d2, d3, d4;
728 char ch;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000729
Guido van Rossuma376cc51996-12-05 23:43:35 +0000730 memset((void *) addr_ret, '\0', sizeof(*addr_ret));
Guido van Rossum30a685f1991-06-27 15:51:29 +0000731 if (name[0] == '\0') {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000732 int siz;
733 memset(&hints, 0, sizeof(hints));
734 hints.ai_family = af;
735 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
736 hints.ai_flags = AI_PASSIVE;
Just van Rossum1040d2c2003-05-09 07:53:18 +0000737 Py_BEGIN_ALLOW_THREADS
738 ACQUIRE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000739 error = getaddrinfo(NULL, "0", &hints, &res);
Just van Rossum1040d2c2003-05-09 07:53:18 +0000740 Py_END_ALLOW_THREADS
741 /* We assume that those thread-unsafe getaddrinfo() versions
742 *are* safe regarding their return value, ie. that a
743 subsequent call to getaddrinfo() does not destroy the
744 outcome of the first call. */
745 RELEASE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000746 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000747 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000748 return -1;
749 }
750 switch (res->ai_family) {
751 case AF_INET:
752 siz = 4;
753 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000754#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000755 case AF_INET6:
756 siz = 16;
757 break;
758#endif
759 default:
760 freeaddrinfo(res);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000761 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000762 "unsupported address family");
763 return -1;
764 }
765 if (res->ai_next) {
Martin v. Löwisf0b11d22001-11-07 08:31:03 +0000766 freeaddrinfo(res);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000767 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000768 "wildcard resolved to multiple address");
769 return -1;
770 }
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000771 if (res->ai_addrlen < addr_ret_size)
772 addr_ret_size = res->ai_addrlen;
773 memcpy(addr_ret, res->ai_addr, addr_ret_size);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000774 freeaddrinfo(res);
775 return siz;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000776 }
Guido van Rossum30a685f1991-06-27 15:51:29 +0000777 if (name[0] == '<' && strcmp(name, "<broadcast>") == 0) {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000778 struct sockaddr_in *sin;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +0000779 if (af != AF_INET && af != AF_UNSPEC) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000780 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000781 "address family mismatched");
782 return -1;
783 }
784 sin = (struct sockaddr_in *)addr_ret;
785 memset((void *) sin, '\0', sizeof(*sin));
786 sin->sin_family = AF_INET;
787#ifdef HAVE_SOCKADDR_SA_LEN
788 sin->sin_len = sizeof(*sin);
789#endif
790 sin->sin_addr.s_addr = INADDR_BROADCAST;
791 return sizeof(sin->sin_addr);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000792 }
Anthony Baxter0e85f9d2003-05-02 15:40:46 +0000793 if (sscanf(name, "%d.%d.%d.%d%c", &d1, &d2, &d3, &d4, &ch) == 4 &&
794 0 <= d1 && d1 <= 255 && 0 <= d2 && d2 <= 255 &&
795 0 <= d3 && d3 <= 255 && 0 <= d4 && d4 <= 255) {
796 struct sockaddr_in *sin;
797 sin = (struct sockaddr_in *)addr_ret;
798 sin->sin_addr.s_addr = htonl(
799 ((long) d1 << 24) | ((long) d2 << 16) |
800 ((long) d3 << 8) | ((long) d4 << 0));
801 sin->sin_family = AF_INET;
802#ifdef HAVE_SOCKADDR_SA_LEN
803 sin->sin_len = sizeof(*sin);
804#endif
805 return 4;
806 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000807 memset(&hints, 0, sizeof(hints));
808 hints.ai_family = af;
Just van Rossum1040d2c2003-05-09 07:53:18 +0000809 Py_BEGIN_ALLOW_THREADS
810 ACQUIRE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000811 error = getaddrinfo(name, NULL, &hints, &res);
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000812#if defined(__digital__) && defined(__unix__)
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000813 if (error == EAI_NONAME && af == AF_UNSPEC) {
814 /* On Tru64 V5.1, numeric-to-addr conversion fails
815 if no address family is given. Assume IPv4 for now.*/
816 hints.ai_family = AF_INET;
817 error = getaddrinfo(name, NULL, &hints, &res);
818 }
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000819#endif
Just van Rossum1040d2c2003-05-09 07:53:18 +0000820 Py_END_ALLOW_THREADS
821 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000822 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000823 set_gaierror(error);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000824 return -1;
825 }
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000826 if (res->ai_addrlen < addr_ret_size)
827 addr_ret_size = res->ai_addrlen;
828 memcpy((char *) addr_ret, res->ai_addr, addr_ret_size);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000829 freeaddrinfo(res);
830 switch (addr_ret->sa_family) {
831 case AF_INET:
832 return 4;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000833#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000834 case AF_INET6:
835 return 16;
Guido van Rossum955becc1999-03-22 20:14:53 +0000836#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000837 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000838 PyErr_SetString(socket_error, "unknown address family");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000839 return -1;
840 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000841}
842
Guido van Rossum30a685f1991-06-27 15:51:29 +0000843
Guido van Rossum30a685f1991-06-27 15:51:29 +0000844/* Create a string object representing an IP address.
845 This is always a string of the form 'dd.dd.dd.dd' (with variable
846 size numbers). */
847
Guido van Rossum73624e91994-10-10 17:59:00 +0000848static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000849makeipaddr(struct sockaddr *addr, int addrlen)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000850{
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000851 char buf[NI_MAXHOST];
852 int error;
853
854 error = getnameinfo(addr, addrlen, buf, sizeof(buf), NULL, 0,
855 NI_NUMERICHOST);
856 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000857 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000858 return NULL;
859 }
Guido van Rossum73624e91994-10-10 17:59:00 +0000860 return PyString_FromString(buf);
Guido van Rossum30a685f1991-06-27 15:51:29 +0000861}
862
863
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000864#ifdef USE_BLUETOOTH
865/* Convert a string representation of a Bluetooth address into a numeric
866 address. Returns the length (6), or raises an exception and returns -1 if
867 an error occurred. */
868
869static int
870setbdaddr(char *name, bdaddr_t *bdaddr)
871{
872 unsigned int b0, b1, b2, b3, b4, b5;
873 char ch;
874 int n;
875
876 n = sscanf(name, "%X:%X:%X:%X:%X:%X%c",
877 &b5, &b4, &b3, &b2, &b1, &b0, &ch);
878 if (n == 6 && (b0 | b1 | b2 | b3 | b4 | b5) < 256) {
879 bdaddr->b[0] = b0;
880 bdaddr->b[1] = b1;
881 bdaddr->b[2] = b2;
882 bdaddr->b[3] = b3;
883 bdaddr->b[4] = b4;
884 bdaddr->b[5] = b5;
885 return 6;
886 } else {
887 PyErr_SetString(socket_error, "bad bluetooth address");
888 return -1;
889 }
890}
891
892/* Create a string representation of the Bluetooth address. This is always a
893 string of the form 'XX:XX:XX:XX:XX:XX' where XX is a two digit hexadecimal
894 value (zero padded if necessary). */
895
896static PyObject *
897makebdaddr(bdaddr_t *bdaddr)
898{
899 char buf[(6 * 2) + 5 + 1];
900
901 sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
902 bdaddr->b[5], bdaddr->b[4], bdaddr->b[3],
903 bdaddr->b[2], bdaddr->b[1], bdaddr->b[0]);
904 return PyString_FromString(buf);
905}
906#endif
907
908
Guido van Rossum30a685f1991-06-27 15:51:29 +0000909/* Create an object representing the given socket address,
910 suitable for passing it back to bind(), connect() etc.
911 The family field of the sockaddr structure is inspected
912 to determine what kind of address it really is. */
913
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000914/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +0000915static PyObject *
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000916makesockaddr(int sockfd, struct sockaddr *addr, int addrlen, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000917{
Guido van Rossum25bec8c1992-08-05 19:00:45 +0000918 if (addrlen == 0) {
919 /* No address -- may be recvfrom() from known socket */
Guido van Rossum73624e91994-10-10 17:59:00 +0000920 Py_INCREF(Py_None);
921 return Py_None;
Guido van Rossum25bec8c1992-08-05 19:00:45 +0000922 }
923
Guido van Rossumbcc20741998-08-04 22:53:56 +0000924#ifdef __BEOS__
Guido van Rossum2c8bcb82000-04-25 21:34:53 +0000925 /* XXX: BeOS version of accept() doesn't set family correctly */
Guido van Rossumbcc20741998-08-04 22:53:56 +0000926 addr->sa_family = AF_INET;
927#endif
928
Guido van Rossum30a685f1991-06-27 15:51:29 +0000929 switch (addr->sa_family) {
930
931 case AF_INET:
932 {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000933 struct sockaddr_in *a;
934 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
Barry Warsaw752300b1997-01-03 17:18:10 +0000935 PyObject *ret = NULL;
936 if (addrobj) {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000937 a = (struct sockaddr_in *)addr;
Barry Warsaw752300b1997-01-03 17:18:10 +0000938 ret = Py_BuildValue("Oi", addrobj, ntohs(a->sin_port));
939 Py_DECREF(addrobj);
940 }
Guido van Rossum6f5afc91993-02-05 09:46:15 +0000941 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000942 }
Guido van Rossum30a685f1991-06-27 15:51:29 +0000943
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +0000944#if defined(AF_UNIX)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000945 case AF_UNIX:
946 {
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000947 struct sockaddr_un *a = (struct sockaddr_un *) addr;
Guido van Rossum73624e91994-10-10 17:59:00 +0000948 return PyString_FromString(a->sun_path);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000949 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000950#endif /* AF_UNIX */
951
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000952#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000953 case AF_INET6:
954 {
955 struct sockaddr_in6 *a;
956 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
957 PyObject *ret = NULL;
958 if (addrobj) {
959 a = (struct sockaddr_in6 *)addr;
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000960 ret = Py_BuildValue("Oiii",
961 addrobj,
962 ntohs(a->sin6_port),
963 a->sin6_flowinfo,
964 a->sin6_scope_id);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000965 Py_DECREF(addrobj);
966 }
967 return ret;
968 }
Jeremy Hylton22308652001-02-02 03:23:09 +0000969#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +0000970
Martin v. Löwis558d9bf2004-06-03 09:24:42 +0000971#ifdef USE_BLUETOOTH
972 case AF_BLUETOOTH:
973 switch (proto) {
974
975 case BTPROTO_L2CAP:
976 {
977 struct sockaddr_l2 *a = (struct sockaddr_l2 *) addr;
978 PyObject *addrobj = makebdaddr(&_BT_L2_MEMB(a, bdaddr));
979 PyObject *ret = NULL;
980 if (addrobj) {
981 ret = Py_BuildValue("Oi",
982 addrobj,
983 _BT_L2_MEMB(a, psm));
984 Py_DECREF(addrobj);
985 }
986 return ret;
987 }
988
989 case BTPROTO_RFCOMM:
990 {
991 struct sockaddr_rc *a = (struct sockaddr_rc *) addr;
992 PyObject *addrobj = makebdaddr(&_BT_RC_MEMB(a, bdaddr));
993 PyObject *ret = NULL;
994 if (addrobj) {
995 ret = Py_BuildValue("Oi",
996 addrobj,
997 _BT_RC_MEMB(a, channel));
998 Py_DECREF(addrobj);
999 }
1000 return ret;
1001 }
1002
1003#if !defined(__FreeBSD__)
1004 case BTPROTO_SCO:
1005 {
1006 struct sockaddr_sco *a = (struct sockaddr_sco *) addr;
1007 return makebdaddr(&_BT_SCO_MEMB(a, bdaddr));
1008 }
1009#endif
1010
1011 }
1012#endif
1013
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001014#ifdef HAVE_NETPACKET_PACKET_H
Jeremy Hylton22308652001-02-02 03:23:09 +00001015 case AF_PACKET:
1016 {
1017 struct sockaddr_ll *a = (struct sockaddr_ll *)addr;
1018 char *ifname = "";
1019 struct ifreq ifr;
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001020 /* need to look up interface name give index */
1021 if (a->sll_ifindex) {
Jeremy Hylton22308652001-02-02 03:23:09 +00001022 ifr.ifr_ifindex = a->sll_ifindex;
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001023 if (ioctl(sockfd, SIOCGIFNAME, &ifr) == 0)
Jeremy Hylton22308652001-02-02 03:23:09 +00001024 ifname = ifr.ifr_name;
Jeremy Hylton22308652001-02-02 03:23:09 +00001025 }
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001026 return Py_BuildValue("shbhs#",
1027 ifname,
1028 ntohs(a->sll_protocol),
1029 a->sll_pkttype,
1030 a->sll_hatype,
1031 a->sll_addr,
1032 a->sll_halen);
Jeremy Hylton22308652001-02-02 03:23:09 +00001033 }
1034#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001035
Guido van Rossum30a685f1991-06-27 15:51:29 +00001036 /* More cases here... */
1037
1038 default:
Guido van Rossumaa948df1997-05-07 17:41:48 +00001039 /* If we don't know the address family, don't raise an
1040 exception -- return it as a tuple. */
1041 return Py_BuildValue("is#",
1042 addr->sa_family,
1043 addr->sa_data,
1044 sizeof(addr->sa_data));
Guido van Rossum25bec8c1992-08-05 19:00:45 +00001045
Guido van Rossum30a685f1991-06-27 15:51:29 +00001046 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001047}
1048
Guido van Rossum30a685f1991-06-27 15:51:29 +00001049
1050/* Parse a socket address argument according to the socket object's
1051 address family. Return 1 if the address was in the proper format,
1052 0 of not. The address is returned through addr_ret, its length
1053 through len_ret. */
1054
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001055static int
Guido van Rossum48a680c2001-03-02 06:34:14 +00001056getsockaddrarg(PySocketSockObject *s, PyObject *args,
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001057 struct sockaddr **addr_ret, int *len_ret)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001058{
Guido van Rossum30a685f1991-06-27 15:51:29 +00001059 switch (s->sock_family) {
1060
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001061#if defined(AF_UNIX)
Guido van Rossum30a685f1991-06-27 15:51:29 +00001062 case AF_UNIX:
1063 {
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001064 struct sockaddr_un* addr;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001065 char *path;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001066 int len;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001067 addr = (struct sockaddr_un*)&(s->sock_addr).un;
Guido van Rossum7e488981998-10-08 02:25:24 +00001068 if (!PyArg_Parse(args, "t#", &path, &len))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001069 return 0;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001070 if (len > sizeof addr->sun_path) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001071 PyErr_SetString(socket_error,
Barry Warsaw752300b1997-01-03 17:18:10 +00001072 "AF_UNIX path too long");
Guido van Rossum30a685f1991-06-27 15:51:29 +00001073 return 0;
1074 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001075 addr->sun_family = s->sock_family;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001076 memcpy(addr->sun_path, path, len);
Guido van Rossum30b6b2b1995-06-14 22:28:08 +00001077 addr->sun_path[len] = 0;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001078 *addr_ret = (struct sockaddr *) addr;
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001079#if defined(PYOS_OS2)
1080 *len_ret = sizeof(*addr);
1081#else
Guido van Rossum65af28a1996-06-11 18:36:33 +00001082 *len_ret = len + sizeof(*addr) - sizeof(addr->sun_path);
Andrew MacIntyredaedf212004-04-11 12:03:57 +00001083#endif
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001084 return 1;
1085 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001086#endif /* AF_UNIX */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001087
Guido van Rossum30a685f1991-06-27 15:51:29 +00001088 case AF_INET:
1089 {
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001090 struct sockaddr_in* addr;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001091 char *host;
Martin v. Löwis5db099a2003-08-07 11:55:15 +00001092 int port, result;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001093 addr=(struct sockaddr_in*)&(s->sock_addr).in;
Guido van Rossume4dad902000-12-01 13:13:11 +00001094 if (!PyTuple_Check(args)) {
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001095 PyErr_Format(
1096 PyExc_TypeError,
1097 "getsockaddrarg: "
1098 "AF_INET address must be tuple, not %.500s",
1099 args->ob_type->tp_name);
Guido van Rossume4dad902000-12-01 13:13:11 +00001100 return 0;
1101 }
Martin v. Löwis2548c732003-04-18 10:39:54 +00001102 if (!PyArg_ParseTuple(args, "eti:getsockaddrarg",
1103 "idna", &host, &port))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001104 return 0;
Martin v. Löwis5db099a2003-08-07 11:55:15 +00001105 result = setipaddr(host, (struct sockaddr *)addr,
1106 sizeof(*addr), AF_INET);
1107 PyMem_Free(host);
1108 if (result < 0)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001109 return 0;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001110 addr->sin_family = AF_INET;
Guido van Rossum644a12b1997-04-09 19:24:53 +00001111 addr->sin_port = htons((short)port);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00001112 *addr_ret = (struct sockaddr *) addr;
1113 *len_ret = sizeof *addr;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001114 return 1;
1115 }
1116
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001117#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001118 case AF_INET6:
1119 {
1120 struct sockaddr_in6* addr;
1121 char *host;
Martin v. Löwis5db099a2003-08-07 11:55:15 +00001122 int port, flowinfo, scope_id, result;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001123 addr = (struct sockaddr_in6*)&(s->sock_addr).in6;
1124 flowinfo = scope_id = 0;
Martin v. Löwis2548c732003-04-18 10:39:54 +00001125 if (!PyArg_ParseTuple(args, "eti|ii",
1126 "idna", &host, &port, &flowinfo,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001127 &scope_id)) {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001128 return 0;
1129 }
Martin v. Löwis5db099a2003-08-07 11:55:15 +00001130 result = setipaddr(host, (struct sockaddr *)addr,
1131 sizeof(*addr), AF_INET6);
1132 PyMem_Free(host);
1133 if (result < 0)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001134 return 0;
1135 addr->sin6_family = s->sock_family;
1136 addr->sin6_port = htons((short)port);
1137 addr->sin6_flowinfo = flowinfo;
1138 addr->sin6_scope_id = scope_id;
1139 *addr_ret = (struct sockaddr *) addr;
1140 *len_ret = sizeof *addr;
1141 return 1;
1142 }
1143#endif
1144
Hye-Shik Chang81268602004-02-02 06:05:24 +00001145#ifdef USE_BLUETOOTH
Martin v. Löwis12af0482004-01-31 12:34:17 +00001146 case AF_BLUETOOTH:
1147 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001148 switch (s->sock_proto) {
1149 case BTPROTO_L2CAP:
Martin v. Löwis12af0482004-01-31 12:34:17 +00001150 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001151 struct sockaddr_l2 *addr = (struct sockaddr_l2 *) _BT_SOCKADDR_MEMB(s, l2);
1152 char *straddr;
Martin v. Löwis12af0482004-01-31 12:34:17 +00001153
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001154 _BT_L2_MEMB(addr, family) = AF_BLUETOOTH;
1155 if (!PyArg_ParseTuple(args, "si", &straddr,
1156 &_BT_L2_MEMB(addr, psm))) {
1157 PyErr_SetString(socket_error, "getsockaddrarg: "
1158 "wrong format");
Martin v. Löwis12af0482004-01-31 12:34:17 +00001159 return 0;
1160 }
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001161 if (setbdaddr(straddr, &_BT_L2_MEMB(addr, bdaddr)) < 0)
1162 return 0;
1163
1164 *addr_ret = (struct sockaddr *) addr;
1165 *len_ret = sizeof *addr;
1166 return 1;
1167 }
1168 case BTPROTO_RFCOMM:
1169 {
1170 struct sockaddr_rc *addr = (struct sockaddr_rc *) _BT_SOCKADDR_MEMB(s, rc);
1171 char *straddr;
1172
1173 _BT_RC_MEMB(addr, family) = AF_BLUETOOTH;
1174 if (!PyArg_ParseTuple(args, "si", &straddr,
1175 &_BT_RC_MEMB(addr, channel))) {
1176 PyErr_SetString(socket_error, "getsockaddrarg: "
1177 "wrong format");
1178 return 0;
1179 }
1180 if (setbdaddr(straddr, &_BT_RC_MEMB(addr, bdaddr)) < 0)
1181 return 0;
1182
1183 *addr_ret = (struct sockaddr *) addr;
1184 *len_ret = sizeof *addr;
1185 return 1;
1186 }
1187#if !defined(__FreeBSD__)
1188 case BTPROTO_SCO:
1189 {
1190 struct sockaddr_sco *addr = (struct sockaddr_sco *) _BT_SOCKADDR_MEMB(s, sco);
1191 char *straddr;
1192
1193 _BT_SCO_MEMB(addr, family) = AF_BLUETOOTH;
1194 straddr = PyString_AsString(args);
1195 if (straddr == NULL) {
1196 PyErr_SetString(socket_error, "getsockaddrarg: "
1197 "wrong format");
1198 return 0;
1199 }
1200 if (setbdaddr(straddr, &_BT_SCO_MEMB(addr, bdaddr)) < 0)
1201 return 0;
1202
1203 *addr_ret = (struct sockaddr *) addr;
1204 *len_ret = sizeof *addr;
1205 return 1;
1206 }
1207#endif
1208 default:
1209 PyErr_SetString(socket_error, "getsockaddrarg: unknown Bluetooth protocol");
1210 return 0;
Martin v. Löwis12af0482004-01-31 12:34:17 +00001211 }
1212 }
1213#endif
1214
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001215#ifdef HAVE_NETPACKET_PACKET_H
Jeremy Hylton22308652001-02-02 03:23:09 +00001216 case AF_PACKET:
1217 {
1218 struct sockaddr_ll* addr;
1219 struct ifreq ifr;
1220 char *interfaceName;
1221 int protoNumber;
1222 int hatype = 0;
1223 int pkttype = 0;
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +00001224 char *haddr = NULL;
1225 unsigned int halen = 0;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001226
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +00001227 if (!PyArg_ParseTuple(args, "si|iis#", &interfaceName,
1228 &protoNumber, &pkttype, &hatype,
1229 &haddr, &halen))
Jeremy Hylton22308652001-02-02 03:23:09 +00001230 return 0;
1231 strncpy(ifr.ifr_name, interfaceName, sizeof(ifr.ifr_name));
1232 ifr.ifr_name[(sizeof(ifr.ifr_name))-1] = '\0';
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001233 if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001234 s->errorhandler();
Jeremy Hylton22308652001-02-02 03:23:09 +00001235 return 0;
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001236 }
Jeremy Hylton22308652001-02-02 03:23:09 +00001237 addr = &(s->sock_addr.ll);
1238 addr->sll_family = AF_PACKET;
1239 addr->sll_protocol = htons((short)protoNumber);
1240 addr->sll_ifindex = ifr.ifr_ifindex;
1241 addr->sll_pkttype = pkttype;
1242 addr->sll_hatype = hatype;
Andrew M. Kuchlingb8e17172004-07-10 23:39:35 +00001243 if (halen > 8) {
1244 PyErr_SetString(PyExc_ValueError,
1245 "Hardware address must be 8 bytes or less");
1246 return 0;
1247 }
1248 if (halen != 0) {
1249 memcpy(&addr->sll_addr, haddr, halen);
1250 }
1251 addr->sll_halen = halen;
Jeremy Hylton22308652001-02-02 03:23:09 +00001252 *addr_ret = (struct sockaddr *) addr;
1253 *len_ret = sizeof *addr;
1254 return 1;
1255 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001256#endif
1257
Guido van Rossum30a685f1991-06-27 15:51:29 +00001258 /* More cases here... */
1259
1260 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001261 PyErr_SetString(socket_error, "getsockaddrarg: bad family");
Guido van Rossum30a685f1991-06-27 15:51:29 +00001262 return 0;
1263
1264 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001265}
1266
Guido van Rossum30a685f1991-06-27 15:51:29 +00001267
Guido van Rossum48a680c2001-03-02 06:34:14 +00001268/* Get the address length according to the socket object's address family.
Guido van Rossum710e1df1992-06-12 10:39:36 +00001269 Return 1 if the family is known, 0 otherwise. The length is returned
1270 through len_ret. */
1271
1272static int
Peter Schneider-Kamp618e25d2000-07-11 23:00:12 +00001273getsockaddrlen(PySocketSockObject *s, socklen_t *len_ret)
Guido van Rossum710e1df1992-06-12 10:39:36 +00001274{
1275 switch (s->sock_family) {
1276
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00001277#if defined(AF_UNIX)
Guido van Rossum710e1df1992-06-12 10:39:36 +00001278 case AF_UNIX:
1279 {
1280 *len_ret = sizeof (struct sockaddr_un);
1281 return 1;
1282 }
Guido van Rossumb6775db1994-08-01 11:34:53 +00001283#endif /* AF_UNIX */
Guido van Rossum710e1df1992-06-12 10:39:36 +00001284
1285 case AF_INET:
1286 {
1287 *len_ret = sizeof (struct sockaddr_in);
1288 return 1;
1289 }
1290
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00001291#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00001292 case AF_INET6:
1293 {
1294 *len_ret = sizeof (struct sockaddr_in6);
1295 return 1;
1296 }
1297#endif
1298
Hye-Shik Chang81268602004-02-02 06:05:24 +00001299#ifdef USE_BLUETOOTH
Martin v. Löwis12af0482004-01-31 12:34:17 +00001300 case AF_BLUETOOTH:
1301 {
1302 switch(s->sock_proto)
1303 {
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001304
1305 case BTPROTO_L2CAP:
1306 *len_ret = sizeof (struct sockaddr_l2);
1307 return 1;
1308 case BTPROTO_RFCOMM:
1309 *len_ret = sizeof (struct sockaddr_rc);
1310 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001311#if !defined(__FreeBSD__)
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001312 case BTPROTO_SCO:
1313 *len_ret = sizeof (struct sockaddr_sco);
1314 return 1;
Hye-Shik Chang81268602004-02-02 06:05:24 +00001315#endif
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001316 default:
1317 PyErr_SetString(socket_error, "getsockaddrlen: "
1318 "unknown BT protocol");
1319 return 0;
1320
Martin v. Löwis12af0482004-01-31 12:34:17 +00001321 }
1322 }
1323#endif
1324
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00001325#ifdef HAVE_NETPACKET_PACKET_H
Jeremy Hylton22308652001-02-02 03:23:09 +00001326 case AF_PACKET:
1327 {
1328 *len_ret = sizeof (struct sockaddr_ll);
1329 return 1;
1330 }
1331#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +00001332
Guido van Rossum710e1df1992-06-12 10:39:36 +00001333 /* More cases here... */
1334
1335 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001336 PyErr_SetString(socket_error, "getsockaddrlen: bad family");
Guido van Rossum710e1df1992-06-12 10:39:36 +00001337 return 0;
1338
1339 }
1340}
1341
1342
Guido van Rossum30a685f1991-06-27 15:51:29 +00001343/* s.accept() method */
1344
Guido van Rossum73624e91994-10-10 17:59:00 +00001345static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001346sock_accept(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001347{
1348 char addrbuf[256];
Fred Drakea04eaad2000-06-30 02:46:07 +00001349 SOCKET_T newfd;
Guido van Rossumff3ab422000-04-24 15:16:03 +00001350 socklen_t addrlen;
Barry Warsaw752300b1997-01-03 17:18:10 +00001351 PyObject *sock = NULL;
1352 PyObject *addr = NULL;
1353 PyObject *res = NULL;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001354 int timeout;
Barry Warsaw752300b1997-01-03 17:18:10 +00001355
Guido van Rossum710e1df1992-06-12 10:39:36 +00001356 if (!getsockaddrlen(s, &addrlen))
1357 return NULL;
Fred Drakefd168342001-05-09 19:11:33 +00001358 memset(addrbuf, 0, addrlen);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001359
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001360#ifdef MS_WINDOWS
1361 newfd = INVALID_SOCKET;
1362#else
1363 newfd = -1;
1364#endif
1365
Guido van Rossum73624e91994-10-10 17:59:00 +00001366 Py_BEGIN_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001367 timeout = internal_select(s, 0);
1368 if (!timeout)
1369 newfd = accept(s->sock_fd, (struct sockaddr *) addrbuf,
1370 &addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001371 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001372
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001373 if (timeout) {
1374 PyErr_SetString(socket_timeout, "timed out");
1375 return NULL;
1376 }
1377
Fred Drakea04eaad2000-06-30 02:46:07 +00001378#ifdef MS_WINDOWS
1379 if (newfd == INVALID_SOCKET)
1380#else
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001381 if (newfd < 0)
Fred Drakea04eaad2000-06-30 02:46:07 +00001382#endif
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001383 return s->errorhandler();
Barry Warsaw752300b1997-01-03 17:18:10 +00001384
Guido van Rossum30a685f1991-06-27 15:51:29 +00001385 /* Create the new object with unspecified family,
1386 to avoid calls to bind() etc. on it. */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001387 sock = (PyObject *) new_sockobject(newfd,
1388 s->sock_family,
1389 s->sock_type,
1390 s->sock_proto);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001391
Barry Warsaw752300b1997-01-03 17:18:10 +00001392 if (sock == NULL) {
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +00001393 SOCKETCLOSE(newfd);
Barry Warsaw752300b1997-01-03 17:18:10 +00001394 goto finally;
1395 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001396 addr = makesockaddr(s->sock_fd, (struct sockaddr *)addrbuf,
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001397 addrlen, s->sock_proto);
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001398 if (addr == NULL)
Barry Warsaw752300b1997-01-03 17:18:10 +00001399 goto finally;
1400
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001401 res = PyTuple_Pack(2, sock, addr);
Barry Warsaw752300b1997-01-03 17:18:10 +00001402
Guido van Rossum67f7a382002-06-06 21:08:16 +00001403finally:
Guido van Rossum73624e91994-10-10 17:59:00 +00001404 Py_XDECREF(sock);
1405 Py_XDECREF(addr);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001406 return res;
1407}
1408
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001409PyDoc_STRVAR(accept_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001410"accept() -> (socket object, address info)\n\
1411\n\
1412Wait for an incoming connection. Return a new socket representing the\n\
1413connection, and the address of the client. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001414info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001415
Guido van Rossum11ba0942002-06-13 15:07:44 +00001416/* s.setblocking(flag) method. Argument:
1417 False -- non-blocking mode; same as settimeout(0)
1418 True -- blocking mode; same as settimeout(None)
1419*/
Guido van Rossume4485b01994-09-07 14:32:49 +00001420
Guido van Rossum73624e91994-10-10 17:59:00 +00001421static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001422sock_setblocking(PySocketSockObject *s, PyObject *arg)
Guido van Rossume4485b01994-09-07 14:32:49 +00001423{
1424 int block;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001425
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001426 block = PyInt_AsLong(arg);
1427 if (block == -1 && PyErr_Occurred())
Guido van Rossume4485b01994-09-07 14:32:49 +00001428 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001429
Guido van Rossum11ba0942002-06-13 15:07:44 +00001430 s->sock_timeout = block ? -1.0 : 0.0;
Guido van Rossume8008f02002-06-07 03:36:20 +00001431 internal_setblocking(s, block);
Guido van Rossume4485b01994-09-07 14:32:49 +00001432
Guido van Rossum73624e91994-10-10 17:59:00 +00001433 Py_INCREF(Py_None);
1434 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +00001435}
Guido van Rossume4485b01994-09-07 14:32:49 +00001436
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001437PyDoc_STRVAR(setblocking_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001438"setblocking(flag)\n\
1439\n\
1440Set the socket to blocking (flag is true) or non-blocking (false).\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001441setblocking(True) is equivalent to settimeout(None);\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001442setblocking(False) is equivalent to settimeout(0.0).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001443
Guido van Rossum11ba0942002-06-13 15:07:44 +00001444/* s.settimeout(timeout) method. Argument:
1445 None -- no timeout, blocking mode; same as setblocking(True)
1446 0.0 -- non-blocking mode; same as setblocking(False)
1447 > 0 -- timeout mode; operations time out after timeout seconds
1448 < 0 -- illegal; raises an exception
1449*/
Guido van Rossum67f7a382002-06-06 21:08:16 +00001450static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001451sock_settimeout(PySocketSockObject *s, PyObject *arg)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001452{
Guido van Rossum11ba0942002-06-13 15:07:44 +00001453 double timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001454
1455 if (arg == Py_None)
Guido van Rossum11ba0942002-06-13 15:07:44 +00001456 timeout = -1.0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001457 else {
Guido van Rossum11ba0942002-06-13 15:07:44 +00001458 timeout = PyFloat_AsDouble(arg);
1459 if (timeout < 0.0) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00001460 if (!PyErr_Occurred())
1461 PyErr_SetString(PyExc_ValueError,
Guido van Rossum11ba0942002-06-13 15:07:44 +00001462 "Timeout value out of range");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001463 return NULL;
1464 }
1465 }
1466
Guido van Rossum11ba0942002-06-13 15:07:44 +00001467 s->sock_timeout = timeout;
1468 internal_setblocking(s, timeout < 0.0);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001469
1470 Py_INCREF(Py_None);
1471 return Py_None;
1472}
1473
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001474PyDoc_STRVAR(settimeout_doc,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001475"settimeout(timeout)\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001476\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001477Set a timeout on socket operations. 'timeout' can be a float,\n\
1478giving in seconds, or None. Setting a timeout of None disables\n\
1479the timeout feature and is equivalent to setblocking(1).\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001480Setting a timeout of zero is the same as setblocking(0).");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001481
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001482/* s.gettimeout() method.
1483 Returns the timeout associated with a socket. */
Guido van Rossum67f7a382002-06-06 21:08:16 +00001484static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001485sock_gettimeout(PySocketSockObject *s)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001486{
1487 if (s->sock_timeout < 0.0) {
1488 Py_INCREF(Py_None);
1489 return Py_None;
1490 }
1491 else
1492 return PyFloat_FromDouble(s->sock_timeout);
1493}
1494
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001495PyDoc_STRVAR(gettimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001496"gettimeout() -> timeout\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001497\n\
1498Returns the timeout in floating seconds associated with socket \n\
1499operations. A timeout of None indicates that timeouts on socket \n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001500operations are disabled.");
Guido van Rossume4485b01994-09-07 14:32:49 +00001501
Guido van Rossum48a680c2001-03-02 06:34:14 +00001502#ifdef RISCOS
1503/* s.sleeptaskw(1 | 0) method */
1504
1505static PyObject *
Martin v. Löwisa94568a2003-05-10 07:36:56 +00001506sock_sleeptaskw(PySocketSockObject *s,PyObject *arg)
Guido van Rossum48a680c2001-03-02 06:34:14 +00001507{
Guido van Rossum67f7a382002-06-06 21:08:16 +00001508 int block;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00001509 block = PyInt_AsLong(arg);
1510 if (block == -1 && PyErr_Occurred())
Guido van Rossum67f7a382002-06-06 21:08:16 +00001511 return NULL;
1512 Py_BEGIN_ALLOW_THREADS
1513 socketioctl(s->sock_fd, 0x80046679, (u_long*)&block);
1514 Py_END_ALLOW_THREADS
Guido van Rossum48a680c2001-03-02 06:34:14 +00001515
Guido van Rossum67f7a382002-06-06 21:08:16 +00001516 Py_INCREF(Py_None);
1517 return Py_None;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001518}
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001519PyDoc_STRVAR(sleeptaskw_doc,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001520"sleeptaskw(flag)\n\
1521\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001522Allow sleeps in taskwindows.");
Guido van Rossum48a680c2001-03-02 06:34:14 +00001523#endif
1524
1525
Guido van Rossumaee08791992-09-08 09:05:33 +00001526/* s.setsockopt() method.
1527 With an integer third argument, sets an integer option.
1528 With a string third argument, sets an option from a buffer;
1529 use optional built-in module 'struct' to encode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001530
Guido van Rossum73624e91994-10-10 17:59:00 +00001531static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001532sock_setsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001533{
1534 int level;
1535 int optname;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001536 int res;
Guido van Rossumaee08791992-09-08 09:05:33 +00001537 char *buf;
1538 int buflen;
1539 int flag;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001540
Guido van Rossum27fc3c02000-03-24 20:56:56 +00001541 if (PyArg_ParseTuple(args, "iii:setsockopt",
1542 &level, &optname, &flag)) {
Guido van Rossumaee08791992-09-08 09:05:33 +00001543 buf = (char *) &flag;
1544 buflen = sizeof flag;
1545 }
1546 else {
Guido van Rossum73624e91994-10-10 17:59:00 +00001547 PyErr_Clear();
Guido van Rossum27fc3c02000-03-24 20:56:56 +00001548 if (!PyArg_ParseTuple(args, "iis#:setsockopt",
1549 &level, &optname, &buf, &buflen))
Guido van Rossumaee08791992-09-08 09:05:33 +00001550 return NULL;
1551 }
Thomas Wouters334fb892000-07-25 12:56:38 +00001552 res = setsockopt(s->sock_fd, level, optname, (void *)buf, buflen);
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001553 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001554 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001555 Py_INCREF(Py_None);
1556 return Py_None;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001557}
1558
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001559PyDoc_STRVAR(setsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001560"setsockopt(level, option, value)\n\
1561\n\
1562Set a socket option. See the Unix manual for level and option.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001563The value argument can either be an integer or a string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001564
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001565
Guido van Rossumaee08791992-09-08 09:05:33 +00001566/* s.getsockopt() method.
1567 With two arguments, retrieves an integer option.
1568 With a third integer argument, retrieves a string buffer of that size;
1569 use optional built-in module 'struct' to decode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001570
Guido van Rossum73624e91994-10-10 17:59:00 +00001571static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001572sock_getsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001573{
1574 int level;
1575 int optname;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001576 int res;
Guido van Rossum73624e91994-10-10 17:59:00 +00001577 PyObject *buf;
Guido van Rossumff3ab422000-04-24 15:16:03 +00001578 socklen_t buflen = 0;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001579
Guido van Rossumbcc20741998-08-04 22:53:56 +00001580#ifdef __BEOS__
Fred Drake564a6cc2001-05-11 20:12:26 +00001581 /* We have incomplete socket support. */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001582 PyErr_SetString(socket_error, "getsockopt not supported");
Guido van Rossumbcc20741998-08-04 22:53:56 +00001583 return NULL;
1584#else
1585
Guido van Rossum27fc3c02000-03-24 20:56:56 +00001586 if (!PyArg_ParseTuple(args, "ii|i:getsockopt",
1587 &level, &optname, &buflen))
Guido van Rossumbe32c891996-06-20 16:25:29 +00001588 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001589
Guido van Rossumbe32c891996-06-20 16:25:29 +00001590 if (buflen == 0) {
Guido van Rossumaee08791992-09-08 09:05:33 +00001591 int flag = 0;
Guido van Rossumff3ab422000-04-24 15:16:03 +00001592 socklen_t flagsize = sizeof flag;
Guido van Rossumb376a4a1993-11-23 17:53:17 +00001593 res = getsockopt(s->sock_fd, level, optname,
Thomas Wouters334fb892000-07-25 12:56:38 +00001594 (void *)&flag, &flagsize);
Guido van Rossumaee08791992-09-08 09:05:33 +00001595 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001596 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001597 return PyInt_FromLong(flag);
Guido van Rossumaee08791992-09-08 09:05:33 +00001598 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001599#ifdef __VMS
1600 if (buflen > 1024) {
1601#else
Guido van Rossumaee08791992-09-08 09:05:33 +00001602 if (buflen <= 0 || buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001603#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001604 PyErr_SetString(socket_error,
Barry Warsaw752300b1997-01-03 17:18:10 +00001605 "getsockopt buflen out of range");
Guido van Rossumaee08791992-09-08 09:05:33 +00001606 return NULL;
1607 }
Guido van Rossum73624e91994-10-10 17:59:00 +00001608 buf = PyString_FromStringAndSize((char *)NULL, buflen);
Guido van Rossumaee08791992-09-08 09:05:33 +00001609 if (buf == NULL)
1610 return NULL;
Guido van Rossumb376a4a1993-11-23 17:53:17 +00001611 res = getsockopt(s->sock_fd, level, optname,
Fred Drake564a6cc2001-05-11 20:12:26 +00001612 (void *)PyString_AS_STRING(buf), &buflen);
Guido van Rossumaee08791992-09-08 09:05:33 +00001613 if (res < 0) {
Guido van Rossum73624e91994-10-10 17:59:00 +00001614 Py_DECREF(buf);
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001615 return s->errorhandler();
Guido van Rossumaee08791992-09-08 09:05:33 +00001616 }
Guido van Rossum73624e91994-10-10 17:59:00 +00001617 _PyString_Resize(&buf, buflen);
Guido van Rossumaee08791992-09-08 09:05:33 +00001618 return buf;
Guido van Rossumbcc20741998-08-04 22:53:56 +00001619#endif /* __BEOS__ */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001620}
1621
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001622PyDoc_STRVAR(getsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001623"getsockopt(level, option[, buffersize]) -> value\n\
1624\n\
1625Get a socket option. See the Unix manual for level and option.\n\
1626If a nonzero buffersize argument is given, the return value is a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001627string of that length; otherwise it is an integer.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001628
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001629
Fred Drake728819a2000-07-01 03:40:12 +00001630/* s.bind(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001631
Guido van Rossum73624e91994-10-10 17:59:00 +00001632static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001633sock_bind(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001634{
1635 struct sockaddr *addr;
1636 int addrlen;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001637 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001638
Fred Drake728819a2000-07-01 03:40:12 +00001639 if (!getsockaddrarg(s, addro, &addr, &addrlen))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001640 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00001641 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001642 res = bind(s->sock_fd, addr, addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001643 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001644 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001645 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001646 Py_INCREF(Py_None);
1647 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001648}
1649
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001650PyDoc_STRVAR(bind_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001651"bind(address)\n\
1652\n\
1653Bind the socket to a local address. For IP sockets, the address is a\n\
Jeremy Hylton22308652001-02-02 03:23:09 +00001654pair (host, port); the host must refer to the local host. For raw packet\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001655sockets the address is a tuple (ifname, proto [,pkttype [,hatype]])");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001656
Guido van Rossum30a685f1991-06-27 15:51:29 +00001657
1658/* s.close() method.
1659 Set the file descriptor to -1 so operations tried subsequently
1660 will surely fail. */
1661
Guido van Rossum73624e91994-10-10 17:59:00 +00001662static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001663sock_close(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001664{
Guido van Rossum20d3fc02000-12-18 22:23:44 +00001665 SOCKET_T fd;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001666
Guido van Rossum20d3fc02000-12-18 22:23:44 +00001667 if ((fd = s->sock_fd) != -1) {
1668 s->sock_fd = -1;
Guido van Rossum08481461996-10-12 14:07:22 +00001669 Py_BEGIN_ALLOW_THREADS
Guido van Rossum20d3fc02000-12-18 22:23:44 +00001670 (void) SOCKETCLOSE(fd);
Guido van Rossum08481461996-10-12 14:07:22 +00001671 Py_END_ALLOW_THREADS
1672 }
Guido van Rossum73624e91994-10-10 17:59:00 +00001673 Py_INCREF(Py_None);
1674 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001675}
1676
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001677PyDoc_STRVAR(close_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001678"close()\n\
1679\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001680Close the socket. It cannot be used after this call.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001681
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001682static int
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001683internal_connect(PySocketSockObject *s, struct sockaddr *addr, int addrlen,
1684 int *timeoutp)
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001685{
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001686 int res, timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001687
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001688 timeout = 0;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001689 res = connect(s->sock_fd, addr, addrlen);
1690
1691#ifdef MS_WINDOWS
1692
1693 if (s->sock_timeout > 0.0) {
1694 if (res < 0 && WSAGetLastError() == WSAEWOULDBLOCK) {
Guido van Rossumb76bdf82003-02-19 17:50:16 +00001695 /* This is a mess. Best solution: trust select */
1696 fd_set fds;
1697 struct timeval tv;
1698 tv.tv_sec = (int)s->sock_timeout;
1699 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
1700 FD_ZERO(&fds);
1701 FD_SET(s->sock_fd, &fds);
1702 res = select(s->sock_fd+1, NULL, &fds, NULL, &tv);
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001703 if (res == 0) {
Guido van Rossumb76bdf82003-02-19 17:50:16 +00001704 res = WSAEWOULDBLOCK;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001705 timeout = 1;
1706 } else if (res > 0)
Guido van Rossumb76bdf82003-02-19 17:50:16 +00001707 res = 0;
1708 /* else if (res < 0) an error occurred */
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001709 }
1710 }
1711
1712 if (res < 0)
1713 res = WSAGetLastError();
1714
1715#else
1716
1717 if (s->sock_timeout > 0.0) {
1718 if (res < 0 && errno == EINPROGRESS) {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001719 timeout = internal_select(s, 1);
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001720 res = connect(s->sock_fd, addr, addrlen);
Guido van Rossum12e3c712002-08-08 20:39:30 +00001721 if (res < 0 && errno == EISCONN)
1722 res = 0;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001723 }
1724 }
1725
1726 if (res < 0)
1727 res = errno;
1728
1729#endif
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001730 *timeoutp = timeout;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001731
1732 return res;
1733}
Guido van Rossum30a685f1991-06-27 15:51:29 +00001734
Fred Drake728819a2000-07-01 03:40:12 +00001735/* s.connect(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001736
Guido van Rossum73624e91994-10-10 17:59:00 +00001737static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001738sock_connect(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001739{
1740 struct sockaddr *addr;
1741 int addrlen;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001742 int res;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001743 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001744
Fred Drake728819a2000-07-01 03:40:12 +00001745 if (!getsockaddrarg(s, addro, &addr, &addrlen))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001746 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001747
Guido van Rossum73624e91994-10-10 17:59:00 +00001748 Py_BEGIN_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001749 res = internal_connect(s, addr, addrlen, &timeout);
Guido van Rossum11ba0942002-06-13 15:07:44 +00001750 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001751
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001752 if (timeout) {
1753 PyErr_SetString(socket_timeout, "timed out");
1754 return NULL;
1755 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001756 if (res != 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001757 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001758 Py_INCREF(Py_None);
1759 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001760}
1761
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001762PyDoc_STRVAR(connect_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001763"connect(address)\n\
1764\n\
1765Connect the socket to a remote address. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001766is a pair (host, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001767
Guido van Rossum30a685f1991-06-27 15:51:29 +00001768
Fred Drake728819a2000-07-01 03:40:12 +00001769/* s.connect_ex(sockaddr) method */
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001770
1771static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001772sock_connect_ex(PySocketSockObject *s, PyObject *addro)
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001773{
1774 struct sockaddr *addr;
1775 int addrlen;
1776 int res;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001777 int timeout;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001778
Fred Drake728819a2000-07-01 03:40:12 +00001779 if (!getsockaddrarg(s, addro, &addr, &addrlen))
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001780 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001781
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001782 Py_BEGIN_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00001783 res = internal_connect(s, addr, addrlen, &timeout);
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001784 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001785
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001786 return PyInt_FromLong((long) res);
1787}
1788
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001789PyDoc_STRVAR(connect_ex_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001790"connect_ex(address) -> errno\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00001791\n\
1792This is like connect(address), but returns an error code (the errno value)\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001793instead of raising an exception when an error occurs.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001794
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001795
Guido van Rossumed233a51992-06-23 09:07:03 +00001796/* s.fileno() method */
1797
Guido van Rossum73624e91994-10-10 17:59:00 +00001798static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001799sock_fileno(PySocketSockObject *s)
Guido van Rossumed233a51992-06-23 09:07:03 +00001800{
Fred Drakea04eaad2000-06-30 02:46:07 +00001801#if SIZEOF_SOCKET_T <= SIZEOF_LONG
Guido van Rossum73624e91994-10-10 17:59:00 +00001802 return PyInt_FromLong((long) s->sock_fd);
Fred Drakea04eaad2000-06-30 02:46:07 +00001803#else
Martin v. Löwisb9a0f912003-03-29 10:06:18 +00001804 return PyLong_FromLongLong((PY_LONG_LONG)s->sock_fd);
Fred Drakea04eaad2000-06-30 02:46:07 +00001805#endif
Guido van Rossumed233a51992-06-23 09:07:03 +00001806}
1807
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001808PyDoc_STRVAR(fileno_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001809"fileno() -> integer\n\
1810\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001811Return the integer file descriptor of the socket.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001812
Guido van Rossumed233a51992-06-23 09:07:03 +00001813
Guido van Rossumbe32c891996-06-20 16:25:29 +00001814#ifndef NO_DUP
1815/* s.dup() method */
1816
1817static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001818sock_dup(PySocketSockObject *s)
Guido van Rossumbe32c891996-06-20 16:25:29 +00001819{
Fred Drakea04eaad2000-06-30 02:46:07 +00001820 SOCKET_T newfd;
Guido van Rossumbe32c891996-06-20 16:25:29 +00001821 PyObject *sock;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001822
Guido van Rossumbe32c891996-06-20 16:25:29 +00001823 newfd = dup(s->sock_fd);
1824 if (newfd < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001825 return s->errorhandler();
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001826 sock = (PyObject *) new_sockobject(newfd,
1827 s->sock_family,
1828 s->sock_type,
1829 s->sock_proto);
Guido van Rossumbe32c891996-06-20 16:25:29 +00001830 if (sock == NULL)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +00001831 SOCKETCLOSE(newfd);
Guido van Rossumbe32c891996-06-20 16:25:29 +00001832 return sock;
1833}
Guido van Rossum82a5c661998-07-07 20:45:43 +00001834
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001835PyDoc_STRVAR(dup_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001836"dup() -> socket object\n\
1837\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001838Return a new socket object connected to the same system resource.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001839
Guido van Rossumbe32c891996-06-20 16:25:29 +00001840#endif
1841
1842
Guido van Rossumc89705d1992-11-26 08:54:07 +00001843/* s.getsockname() method */
1844
Guido van Rossum73624e91994-10-10 17:59:00 +00001845static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001846sock_getsockname(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00001847{
1848 char addrbuf[256];
Guido van Rossumff3ab422000-04-24 15:16:03 +00001849 int res;
1850 socklen_t addrlen;
1851
Guido van Rossumc89705d1992-11-26 08:54:07 +00001852 if (!getsockaddrlen(s, &addrlen))
1853 return NULL;
Guido van Rossumca42b161996-01-12 01:36:05 +00001854 memset(addrbuf, 0, addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001855 Py_BEGIN_ALLOW_THREADS
Guido van Rossumc89705d1992-11-26 08:54:07 +00001856 res = getsockname(s->sock_fd, (struct sockaddr *) addrbuf, &addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001857 Py_END_ALLOW_THREADS
Guido van Rossumc89705d1992-11-26 08:54:07 +00001858 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001859 return s->errorhandler();
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001860 return makesockaddr(s->sock_fd, (struct sockaddr *) addrbuf, addrlen,
1861 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00001862}
1863
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001864PyDoc_STRVAR(getsockname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001865"getsockname() -> address info\n\
1866\n\
1867Return the address of the local endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001868info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001869
Guido van Rossumc89705d1992-11-26 08:54:07 +00001870
Guido van Rossumb6775db1994-08-01 11:34:53 +00001871#ifdef HAVE_GETPEERNAME /* Cray APP doesn't have this :-( */
Guido van Rossumc89705d1992-11-26 08:54:07 +00001872/* s.getpeername() method */
1873
Guido van Rossum73624e91994-10-10 17:59:00 +00001874static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001875sock_getpeername(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00001876{
1877 char addrbuf[256];
Guido van Rossumff3ab422000-04-24 15:16:03 +00001878 int res;
1879 socklen_t addrlen;
1880
Guido van Rossumc89705d1992-11-26 08:54:07 +00001881 if (!getsockaddrlen(s, &addrlen))
1882 return NULL;
Fred Drakefd168342001-05-09 19:11:33 +00001883 memset(addrbuf, 0, addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001884 Py_BEGIN_ALLOW_THREADS
Guido van Rossumc89705d1992-11-26 08:54:07 +00001885 res = getpeername(s->sock_fd, (struct sockaddr *) addrbuf, &addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001886 Py_END_ALLOW_THREADS
Guido van Rossumc89705d1992-11-26 08:54:07 +00001887 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001888 return s->errorhandler();
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00001889 return makesockaddr(s->sock_fd, (struct sockaddr *) addrbuf, addrlen,
1890 s->sock_proto);
Guido van Rossumc89705d1992-11-26 08:54:07 +00001891}
Guido van Rossum82a5c661998-07-07 20:45:43 +00001892
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001893PyDoc_STRVAR(getpeername_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001894"getpeername() -> address info\n\
1895\n\
1896Return the address of the remote endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001897info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001898
Guido van Rossumb6775db1994-08-01 11:34:53 +00001899#endif /* HAVE_GETPEERNAME */
Guido van Rossumc89705d1992-11-26 08:54:07 +00001900
1901
Guido van Rossum30a685f1991-06-27 15:51:29 +00001902/* s.listen(n) method */
1903
Guido van Rossum73624e91994-10-10 17:59:00 +00001904static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001905sock_listen(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001906{
1907 int backlog;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001908 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001909
1910 backlog = PyInt_AsLong(arg);
1911 if (backlog == -1 && PyErr_Occurred())
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001912 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00001913 Py_BEGIN_ALLOW_THREADS
Guido van Rossumb6775db1994-08-01 11:34:53 +00001914 if (backlog < 1)
1915 backlog = 1;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001916 res = listen(s->sock_fd, backlog);
Guido van Rossum73624e91994-10-10 17:59:00 +00001917 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001918 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001919 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001920 Py_INCREF(Py_None);
1921 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001922}
1923
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001924PyDoc_STRVAR(listen_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001925"listen(backlog)\n\
1926\n\
1927Enable a server to accept connections. The backlog argument must be at\n\
1928least 1; it specifies the number of unaccepted connection that the system\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001929will allow before refusing new connections.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001930
1931
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001932#ifndef NO_DUP
Guido van Rossum30a685f1991-06-27 15:51:29 +00001933/* s.makefile(mode) method.
1934 Create a new open file object referring to a dupped version of
1935 the socket's file descriptor. (The dup() call is necessary so
1936 that the open file and socket objects may be closed independent
1937 of each other.)
1938 The mode argument specifies 'r' or 'w' passed to fdopen(). */
1939
Guido van Rossum73624e91994-10-10 17:59:00 +00001940static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001941sock_makefile(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001942{
Tim Petersdbd9ba62000-07-09 03:09:57 +00001943 extern int fclose(FILE *);
Guido van Rossum6b144911995-03-14 15:05:13 +00001944 char *mode = "r";
1945 int bufsize = -1;
Fred Drakea04eaad2000-06-30 02:46:07 +00001946#ifdef MS_WIN32
Tim Peters79248aa2001-08-29 21:37:10 +00001947 Py_intptr_t fd;
Fred Drakea04eaad2000-06-30 02:46:07 +00001948#else
Guido van Rossum30a685f1991-06-27 15:51:29 +00001949 int fd;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001950#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00001951 FILE *fp;
Guido van Rossum6b144911995-03-14 15:05:13 +00001952 PyObject *f;
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001953#ifdef __VMS
1954 char *mode_r = "r";
1955 char *mode_w = "w";
1956#endif
Guido van Rossum6b144911995-03-14 15:05:13 +00001957
Guido van Rossum43713e52000-02-29 13:59:29 +00001958 if (!PyArg_ParseTuple(args, "|si:makefile", &mode, &bufsize))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001959 return NULL;
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001960#ifdef __VMS
1961 if (strcmp(mode,"rb") == 0) {
1962 mode = mode_r;
1963 }
1964 else {
1965 if (strcmp(mode,"wb") == 0) {
1966 mode = mode_w;
1967 }
1968 }
1969#endif
Guido van Rossum8d665e61996-06-26 18:22:49 +00001970#ifdef MS_WIN32
Guido van Rossum82a5c661998-07-07 20:45:43 +00001971 if (((fd = _open_osfhandle(s->sock_fd, _O_BINARY)) < 0) ||
1972 ((fd = dup(fd)) < 0) || ((fp = fdopen(fd, mode)) == NULL))
Guido van Rossum30b6b2b1995-06-14 22:28:08 +00001973#else
Guido van Rossum82a5c661998-07-07 20:45:43 +00001974 if ((fd = dup(s->sock_fd)) < 0 || (fp = fdopen(fd, mode)) == NULL)
Guido van Rossum30b6b2b1995-06-14 22:28:08 +00001975#endif
Guido van Rossum82a5c661998-07-07 20:45:43 +00001976 {
Guido van Rossum6b144911995-03-14 15:05:13 +00001977 if (fd >= 0)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +00001978 SOCKETCLOSE(fd);
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001979 return s->errorhandler();
Guido van Rossum6b144911995-03-14 15:05:13 +00001980 }
1981 f = PyFile_FromFile(fp, "<socket>", mode, fclose);
1982 if (f != NULL)
1983 PyFile_SetBufSize(f, bufsize);
1984 return f;
Guido van Rossum30a685f1991-06-27 15:51:29 +00001985}
Guido van Rossum82a5c661998-07-07 20:45:43 +00001986
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001987PyDoc_STRVAR(makefile_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001988"makefile([mode[, buffersize]]) -> file object\n\
1989\n\
1990Return a regular file object corresponding to the socket.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001991The mode and buffersize arguments are as for the built-in open() function.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001992
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001993#endif /* NO_DUP */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001994
Guido van Rossum48a680c2001-03-02 06:34:14 +00001995
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00001996/* s.recv(nbytes [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001997
Guido van Rossum73624e91994-10-10 17:59:00 +00001998static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001999sock_recv(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002000{
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002001 int len, n = 0, flags = 0, timeout;
Guido van Rossum73624e91994-10-10 17:59:00 +00002002 PyObject *buf;
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002003#ifdef __VMS
2004 int read_length;
2005 char *read_buf;
2006#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00002007
Guido van Rossum43713e52000-02-29 13:59:29 +00002008 if (!PyArg_ParseTuple(args, "i|i:recv", &len, &flags))
Guido van Rossumbe32c891996-06-20 16:25:29 +00002009 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002010
2011 if (len < 0) {
Martin v. Löwisfba64e12001-11-19 10:41:26 +00002012 PyErr_SetString(PyExc_ValueError,
Guido van Rossum11ba0942002-06-13 15:07:44 +00002013 "negative buffersize in recv");
Martin v. Löwisfba64e12001-11-19 10:41:26 +00002014 return NULL;
2015 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002016
Guido van Rossum73624e91994-10-10 17:59:00 +00002017 buf = PyString_FromStringAndSize((char *) 0, len);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002018 if (buf == NULL)
2019 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002020
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002021#ifndef __VMS
Guido van Rossum73624e91994-10-10 17:59:00 +00002022 Py_BEGIN_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002023 timeout = internal_select(s, 0);
2024 if (!timeout)
2025 n = recv(s->sock_fd, PyString_AS_STRING(buf), len, flags);
Guido van Rossum73624e91994-10-10 17:59:00 +00002026 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002027
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002028 if (timeout) {
2029 Py_DECREF(buf);
2030 PyErr_SetString(socket_timeout, "timed out");
2031 return NULL;
2032 }
Guido van Rossum7c53b771995-09-13 18:39:47 +00002033 if (n < 0) {
2034 Py_DECREF(buf);
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002035 return s->errorhandler();
Guido van Rossum7c53b771995-09-13 18:39:47 +00002036 }
Tim Peters5de98422002-04-27 18:44:32 +00002037 if (n != len)
2038 _PyString_Resize(&buf, n);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002039#else
2040 read_buf = PyString_AsString(buf);
2041 read_length = len;
2042 while (read_length != 0) {
2043 unsigned int segment;
2044
2045 segment = read_length /SEGMENT_SIZE;
2046 if (segment != 0) {
2047 segment = SEGMENT_SIZE;
2048 }
2049 else {
2050 segment = read_length;
2051 }
2052
2053 Py_BEGIN_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002054 timeout = internal_select(s, 0);
2055 if (!timeout)
2056 n = recv(s->sock_fd, read_buf, segment, flags);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002057 Py_END_ALLOW_THREADS
2058
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002059 if (timeout) {
2060 Py_DECREF(buf);
2061 PyErr_SetString(socket_timeout, "timed out");
2062 return NULL;
2063 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002064 if (n < 0) {
2065 Py_DECREF(buf);
2066 return s->errorhandler();
2067 }
2068 if (n != read_length) {
2069 read_buf += n;
2070 break;
2071 }
2072
2073 read_length -= segment;
2074 read_buf += segment;
2075 }
2076 if (_PyString_Resize(&buf, (read_buf - PyString_AsString(buf))) < 0)
2077 {
2078 return NULL;
2079 }
2080#endif /* !__VMS */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002081 return buf;
2082}
2083
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002084PyDoc_STRVAR(recv_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002085"recv(buffersize[, flags]) -> data\n\
2086\n\
2087Receive up to buffersize bytes from the socket. For the optional flags\n\
2088argument, see the Unix manual. When no data is available, block until\n\
2089at least one byte is available or until the remote end is closed. When\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002090the remote end is closed and all data is read, return the empty string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002091
Guido van Rossum30a685f1991-06-27 15:51:29 +00002092
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002093/* s.recvfrom(nbytes [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002094
Guido van Rossum73624e91994-10-10 17:59:00 +00002095static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002096sock_recvfrom(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002097{
2098 char addrbuf[256];
Barry Warsaw752300b1997-01-03 17:18:10 +00002099 PyObject *buf = NULL;
2100 PyObject *addr = NULL;
2101 PyObject *ret = NULL;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002102 int len, n = 0, flags = 0, timeout;
Guido van Rossumff3ab422000-04-24 15:16:03 +00002103 socklen_t addrlen;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002104
Guido van Rossum43713e52000-02-29 13:59:29 +00002105 if (!PyArg_ParseTuple(args, "i|i:recvfrom", &len, &flags))
Guido van Rossumbe32c891996-06-20 16:25:29 +00002106 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002107
Guido van Rossum18c9a4f1993-05-25 12:16:29 +00002108 if (!getsockaddrlen(s, &addrlen))
2109 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00002110 buf = PyString_FromStringAndSize((char *) 0, len);
Guido van Rossum18c9a4f1993-05-25 12:16:29 +00002111 if (buf == NULL)
2112 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002113
Guido van Rossum73624e91994-10-10 17:59:00 +00002114 Py_BEGIN_ALLOW_THREADS
Fred Drakefd168342001-05-09 19:11:33 +00002115 memset(addrbuf, 0, addrlen);
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002116 timeout = internal_select(s, 0);
2117 if (!timeout)
2118 n = recvfrom(s->sock_fd, PyString_AS_STRING(buf), len, flags,
Guido van Rossum8d665e61996-06-26 18:22:49 +00002119#ifndef MS_WINDOWS
Andrew MacIntyreba43e872002-03-03 03:03:52 +00002120#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002121 (struct sockaddr *)addrbuf, &addrlen
Guido van Rossum32c575d1997-12-02 20:37:32 +00002122#else
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002123 (void *)addrbuf, &addrlen
Guido van Rossum32c575d1997-12-02 20:37:32 +00002124#endif
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002125#else
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002126 (struct sockaddr *)addrbuf, &addrlen
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002127#endif
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002128 );
Guido van Rossum73624e91994-10-10 17:59:00 +00002129 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002130
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002131 if (timeout) {
2132 Py_DECREF(buf);
2133 PyErr_SetString(socket_timeout, "timed out");
2134 return NULL;
2135 }
Guido van Rossum7c53b771995-09-13 18:39:47 +00002136 if (n < 0) {
2137 Py_DECREF(buf);
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002138 return s->errorhandler();
Guido van Rossum7c53b771995-09-13 18:39:47 +00002139 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002140
Guido van Rossum53a9bf81996-06-11 18:35:24 +00002141 if (n != len && _PyString_Resize(&buf, n) < 0)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002142 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00002143
Guido van Rossum67f7a382002-06-06 21:08:16 +00002144 if (!(addr = makesockaddr(s->sock_fd, (struct sockaddr *)addrbuf,
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00002145 addrlen, s->sock_proto)))
Barry Warsaw752300b1997-01-03 17:18:10 +00002146 goto finally;
2147
Raymond Hettinger8ae46892003-10-12 19:09:37 +00002148 ret = PyTuple_Pack(2, buf, addr);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002149
2150finally:
Guido van Rossum73624e91994-10-10 17:59:00 +00002151 Py_XDECREF(addr);
2152 Py_XDECREF(buf);
Guido van Rossum6f5afc91993-02-05 09:46:15 +00002153 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002154}
2155
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002156PyDoc_STRVAR(recvfrom_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002157"recvfrom(buffersize[, flags]) -> (data, address info)\n\
2158\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002159Like recv(buffersize, flags) but also return the sender's address info.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002160
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002161/* s.send(data [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002162
Guido van Rossum73624e91994-10-10 17:59:00 +00002163static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002164sock_send(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002165{
Guido van Rossumff4949e1992-08-05 19:58:53 +00002166 char *buf;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002167 int len, n = 0, flags = 0, timeout;
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002168#ifdef __VMS
2169 int send_length;
2170#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00002171
Guido van Rossum43713e52000-02-29 13:59:29 +00002172 if (!PyArg_ParseTuple(args, "s#|i:send", &buf, &len, &flags))
Guido van Rossumbe32c891996-06-20 16:25:29 +00002173 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002174
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002175#ifndef __VMS
Guido van Rossum73624e91994-10-10 17:59:00 +00002176 Py_BEGIN_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002177 timeout = internal_select(s, 1);
2178 if (!timeout)
2179 n = send(s->sock_fd, buf, len, flags);
Guido van Rossum73624e91994-10-10 17:59:00 +00002180 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002181
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002182 if (timeout) {
2183 PyErr_SetString(socket_timeout, "timed out");
2184 return NULL;
2185 }
Guido van Rossum30a685f1991-06-27 15:51:29 +00002186 if (n < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002187 return s->errorhandler();
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002188#else
2189 /* Divide packet into smaller segments for */
2190 /* TCP/IP Services for OpenVMS */
2191 send_length = len;
2192 while (send_length != 0) {
2193 unsigned int segment;
2194
2195 segment = send_length / SEGMENT_SIZE;
2196 if (segment != 0) {
2197 segment = SEGMENT_SIZE;
2198 }
2199 else {
2200 segment = send_length;
2201 }
2202 Py_BEGIN_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002203 timeout = internal_select(s, 1);
2204 if (!timeout)
2205 n = send(s->sock_fd, buf, segment, flags);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002206 Py_END_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002207 if (timeout) {
2208 PyErr_SetString(socket_timeout, "timed out");
2209 return NULL;
2210 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00002211 if (n < 0) {
2212 return s->errorhandler();
2213 }
2214 send_length -= segment;
2215 buf += segment;
2216 } /* end while */
2217#endif /* !__VMS */
Guido van Rossum73624e91994-10-10 17:59:00 +00002218 return PyInt_FromLong((long)n);
Guido van Rossum30a685f1991-06-27 15:51:29 +00002219}
2220
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002221PyDoc_STRVAR(send_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002222"send(data[, flags]) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002223\n\
2224Send a data string to the socket. For the optional flags\n\
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002225argument, see the Unix manual. Return the number of bytes\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002226sent; this may be less than len(data) if the network is busy.");
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002227
2228
2229/* s.sendall(data [,flags]) method */
2230
2231static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002232sock_sendall(PySocketSockObject *s, PyObject *args)
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002233{
2234 char *buf;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002235 int len, n = 0, flags = 0, timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002236
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002237 if (!PyArg_ParseTuple(args, "s#|i:sendall", &buf, &len, &flags))
2238 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002239
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002240 Py_BEGIN_ALLOW_THREADS
2241 do {
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002242 timeout = internal_select(s, 1);
2243 if (timeout)
2244 break;
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002245 n = send(s->sock_fd, buf, len, flags);
2246 if (n < 0)
2247 break;
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002248 buf += n;
2249 len -= n;
2250 } while (len > 0);
2251 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002252
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002253 if (timeout) {
2254 PyErr_SetString(socket_timeout, "timed out");
2255 return NULL;
2256 }
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002257 if (n < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002258 return s->errorhandler();
Guido van Rossum67f7a382002-06-06 21:08:16 +00002259
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002260 Py_INCREF(Py_None);
2261 return Py_None;
2262}
2263
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002264PyDoc_STRVAR(sendall_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002265"sendall(data[, flags])\n\
2266\n\
2267Send a data string to the socket. For the optional flags\n\
2268argument, see the Unix manual. This calls send() repeatedly\n\
2269until all data is sent. If an error occurs, it's impossible\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002270to tell how much data has been sent.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002271
Guido van Rossum30a685f1991-06-27 15:51:29 +00002272
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002273/* s.sendto(data, [flags,] sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002274
Guido van Rossum73624e91994-10-10 17:59:00 +00002275static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002276sock_sendto(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002277{
Guido van Rossum73624e91994-10-10 17:59:00 +00002278 PyObject *addro;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002279 char *buf;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002280 struct sockaddr *addr;
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002281 int addrlen, len, n = 0, flags, timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002282
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002283 flags = 0;
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002284 if (!PyArg_ParseTuple(args, "s#O:sendto", &buf, &len, &addro)) {
Guido van Rossum73624e91994-10-10 17:59:00 +00002285 PyErr_Clear();
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002286 if (!PyArg_ParseTuple(args, "s#iO:sendto",
2287 &buf, &len, &flags, &addro))
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002288 return NULL;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002289 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002290
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00002291 if (!getsockaddrarg(s, addro, &addr, &addrlen))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002292 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002293
Guido van Rossum73624e91994-10-10 17:59:00 +00002294 Py_BEGIN_ALLOW_THREADS
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002295 timeout = internal_select(s, 1);
2296 if (!timeout)
2297 n = sendto(s->sock_fd, buf, len, flags, addr, addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00002298 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002299
Raymond Hettingeref7343c2003-06-29 03:08:05 +00002300 if (timeout) {
2301 PyErr_SetString(socket_timeout, "timed out");
2302 return NULL;
2303 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002304 if (n < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002305 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00002306 return PyInt_FromLong((long)n);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002307}
2308
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002309PyDoc_STRVAR(sendto_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002310"sendto(data[, flags], address) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002311\n\
2312Like send(data, flags) but allows specifying the destination address.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002313For IP sockets, the address is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002314
Guido van Rossum30a685f1991-06-27 15:51:29 +00002315
2316/* s.shutdown(how) method */
2317
Guido van Rossum73624e91994-10-10 17:59:00 +00002318static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002319sock_shutdown(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002320{
2321 int how;
Guido van Rossumff4949e1992-08-05 19:58:53 +00002322 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00002323
2324 how = PyInt_AsLong(arg);
2325 if (how == -1 && PyErr_Occurred())
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002326 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00002327 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002328 res = shutdown(s->sock_fd, how);
Guido van Rossum73624e91994-10-10 17:59:00 +00002329 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002330 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00002331 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00002332 Py_INCREF(Py_None);
2333 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002334}
2335
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002336PyDoc_STRVAR(shutdown_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002337"shutdown(flag)\n\
2338\n\
Martin v. Löwis94681fc2003-11-27 19:40:22 +00002339Shut down the reading side of the socket (flag == SHUT_RD), the writing side\n\
2340of the socket (flag == SHUT_WR), or both ends (flag == SHUT_RDWR).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002341
Guido van Rossum30a685f1991-06-27 15:51:29 +00002342
2343/* List of methods for socket objects */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002344
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002345static PyMethodDef sock_methods[] = {
2346 {"accept", (PyCFunction)sock_accept, METH_NOARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002347 accept_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002348 {"bind", (PyCFunction)sock_bind, METH_O,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002349 bind_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002350 {"close", (PyCFunction)sock_close, METH_NOARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002351 close_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002352 {"connect", (PyCFunction)sock_connect, METH_O,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002353 connect_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002354 {"connect_ex", (PyCFunction)sock_connect_ex, METH_O,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002355 connect_ex_doc},
Guido van Rossumbe32c891996-06-20 16:25:29 +00002356#ifndef NO_DUP
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002357 {"dup", (PyCFunction)sock_dup, METH_NOARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002358 dup_doc},
Guido van Rossumbe32c891996-06-20 16:25:29 +00002359#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002360 {"fileno", (PyCFunction)sock_fileno, METH_NOARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002361 fileno_doc},
Guido van Rossumb6775db1994-08-01 11:34:53 +00002362#ifdef HAVE_GETPEERNAME
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002363 {"getpeername", (PyCFunction)sock_getpeername,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002364 METH_NOARGS, getpeername_doc},
Guido van Rossum9575a441993-04-07 14:06:14 +00002365#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002366 {"getsockname", (PyCFunction)sock_getsockname,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002367 METH_NOARGS, getsockname_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002368 {"getsockopt", (PyCFunction)sock_getsockopt, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002369 getsockopt_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002370 {"listen", (PyCFunction)sock_listen, METH_O,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002371 listen_doc},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002372#ifndef NO_DUP
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002373 {"makefile", (PyCFunction)sock_makefile, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002374 makefile_doc},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002375#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002376 {"recv", (PyCFunction)sock_recv, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002377 recv_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002378 {"recvfrom", (PyCFunction)sock_recvfrom, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002379 recvfrom_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002380 {"send", (PyCFunction)sock_send, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002381 send_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002382 {"sendall", (PyCFunction)sock_sendall, METH_VARARGS,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00002383 sendall_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002384 {"sendto", (PyCFunction)sock_sendto, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002385 sendto_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002386 {"setblocking", (PyCFunction)sock_setblocking, METH_O,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002387 setblocking_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002388 {"settimeout", (PyCFunction)sock_settimeout, METH_O,
Guido van Rossum67f7a382002-06-06 21:08:16 +00002389 settimeout_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002390 {"gettimeout", (PyCFunction)sock_gettimeout, METH_NOARGS,
Guido van Rossum67f7a382002-06-06 21:08:16 +00002391 gettimeout_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002392 {"setsockopt", (PyCFunction)sock_setsockopt, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002393 setsockopt_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002394 {"shutdown", (PyCFunction)sock_shutdown, METH_O,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002395 shutdown_doc},
2396#ifdef RISCOS
Martin v. Löwisa94568a2003-05-10 07:36:56 +00002397 {"sleeptaskw", (PyCFunction)sock_sleeptaskw, METH_O,
Guido van Rossum48a680c2001-03-02 06:34:14 +00002398 sleeptaskw_doc},
2399#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00002400 {NULL, NULL} /* sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002401};
2402
Guido van Rossum30a685f1991-06-27 15:51:29 +00002403
Guido van Rossum73624e91994-10-10 17:59:00 +00002404/* Deallocate a socket object in response to the last Py_DECREF().
Guido van Rossum30a685f1991-06-27 15:51:29 +00002405 First close the file description. */
2406
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002407static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002408sock_dealloc(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002409{
Guido van Rossumfa972c92000-04-10 12:45:45 +00002410 if (s->sock_fd != -1)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +00002411 (void) SOCKETCLOSE(s->sock_fd);
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002412 s->ob_type->tp_free((PyObject *)s);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002413}
2414
Guido van Rossum30a685f1991-06-27 15:51:29 +00002415
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002416static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002417sock_repr(PySocketSockObject *s)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002418{
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002419 char buf[512];
Fred Drakea04eaad2000-06-30 02:46:07 +00002420#if SIZEOF_SOCKET_T > SIZEOF_LONG
2421 if (s->sock_fd > LONG_MAX) {
2422 /* this can occur on Win64, and actually there is a special
2423 ugly printf formatter for decimal pointer length integer
2424 printing, only bother if necessary*/
2425 PyErr_SetString(PyExc_OverflowError,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002426 "no printf formatter to display "
2427 "the socket descriptor in decimal");
Fred Drakea04eaad2000-06-30 02:46:07 +00002428 return NULL;
2429 }
2430#endif
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002431 PyOS_snprintf(
2432 buf, sizeof(buf),
2433 "<socket object, fd=%ld, family=%d, type=%d, protocol=%d>",
2434 (long)s->sock_fd, s->sock_family,
2435 s->sock_type,
2436 s->sock_proto);
Guido van Rossumdc1c64a1996-02-27 15:21:19 +00002437 return PyString_FromString(buf);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002438}
2439
2440
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002441/* Create a new, uninitialized socket object. */
2442
2443static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002444sock_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002445{
2446 PyObject *new;
2447
2448 new = type->tp_alloc(type, 0);
Guido van Rossume1c478f2002-06-06 20:08:25 +00002449 if (new != NULL) {
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002450 ((PySocketSockObject *)new)->sock_fd = -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002451 ((PySocketSockObject *)new)->sock_timeout = -1.0;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002452 ((PySocketSockObject *)new)->errorhandler = &set_error;
Guido van Rossume1c478f2002-06-06 20:08:25 +00002453 }
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002454 return new;
2455}
2456
2457
2458/* Initialize a new socket object. */
2459
2460/*ARGSUSED*/
2461static int
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00002462sock_initobj(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002463{
2464 PySocketSockObject *s = (PySocketSockObject *)self;
2465 SOCKET_T fd;
2466 int family = AF_INET, type = SOCK_STREAM, proto = 0;
2467 static char *keywords[] = {"family", "type", "proto", 0};
2468
2469 if (!PyArg_ParseTupleAndKeywords(args, kwds,
2470 "|iii:socket", keywords,
2471 &family, &type, &proto))
2472 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002473
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002474 Py_BEGIN_ALLOW_THREADS
2475 fd = socket(family, type, proto);
2476 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002477
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002478#ifdef MS_WINDOWS
2479 if (fd == INVALID_SOCKET)
2480#else
2481 if (fd < 0)
2482#endif
2483 {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002484 set_error();
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002485 return -1;
2486 }
2487 init_sockobject(s, fd, family, type, proto);
2488 /* From now on, ignore SIGPIPE and let the error checking
2489 do the work. */
2490#ifdef SIGPIPE
2491 (void) signal(SIGPIPE, SIG_IGN);
2492#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00002493
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002494 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002495
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002496}
2497
2498
Guido van Rossumb6775db1994-08-01 11:34:53 +00002499/* Type object for socket objects. */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002500
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002501static PyTypeObject sock_type = {
Guido van Rossum55558541996-05-23 22:54:50 +00002502 PyObject_HEAD_INIT(0) /* Must fill in type value later */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002503 0, /* ob_size */
Guido van Rossumca6dfa52001-10-28 12:31:33 +00002504 "_socket.socket", /* tp_name */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002505 sizeof(PySocketSockObject), /* tp_basicsize */
2506 0, /* tp_itemsize */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002507 (destructor)sock_dealloc, /* tp_dealloc */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002508 0, /* tp_print */
2509 0, /* tp_getattr */
2510 0, /* tp_setattr */
2511 0, /* tp_compare */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002512 (reprfunc)sock_repr, /* tp_repr */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002513 0, /* tp_as_number */
2514 0, /* tp_as_sequence */
2515 0, /* tp_as_mapping */
2516 0, /* tp_hash */
2517 0, /* tp_call */
2518 0, /* tp_str */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00002519 PyObject_GenericGetAttr, /* tp_getattro */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002520 0, /* tp_setattro */
2521 0, /* tp_as_buffer */
2522 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002523 sock_doc, /* tp_doc */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002524 0, /* tp_traverse */
2525 0, /* tp_clear */
2526 0, /* tp_richcompare */
2527 0, /* tp_weaklistoffset */
2528 0, /* tp_iter */
2529 0, /* tp_iternext */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002530 sock_methods, /* tp_methods */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002531 0, /* tp_members */
2532 0, /* tp_getset */
2533 0, /* tp_base */
2534 0, /* tp_dict */
2535 0, /* tp_descr_get */
2536 0, /* tp_descr_set */
2537 0, /* tp_dictoffset */
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00002538 sock_initobj, /* tp_init */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00002539 PyType_GenericAlloc, /* tp_alloc */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002540 sock_new, /* tp_new */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00002541 PyObject_Del, /* tp_free */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002542};
2543
Guido van Rossum30a685f1991-06-27 15:51:29 +00002544
Guido van Rossum81194471991-07-27 21:42:02 +00002545/* Python interface to gethostname(). */
2546
2547/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002548static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002549socket_gethostname(PyObject *self, PyObject *args)
Guido van Rossum81194471991-07-27 21:42:02 +00002550{
2551 char buf[1024];
Guido van Rossumff4949e1992-08-05 19:58:53 +00002552 int res;
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002553 if (!PyArg_ParseTuple(args, ":gethostname"))
Guido van Rossum81194471991-07-27 21:42:02 +00002554 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00002555 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002556 res = gethostname(buf, (int) sizeof buf - 1);
Guido van Rossum73624e91994-10-10 17:59:00 +00002557 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002558 if (res < 0)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002559 return set_error();
Guido van Rossum81194471991-07-27 21:42:02 +00002560 buf[sizeof buf - 1] = '\0';
Guido van Rossum73624e91994-10-10 17:59:00 +00002561 return PyString_FromString(buf);
Guido van Rossum81194471991-07-27 21:42:02 +00002562}
Guido van Rossumff4949e1992-08-05 19:58:53 +00002563
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002564PyDoc_STRVAR(gethostname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002565"gethostname() -> string\n\
2566\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002567Return the current host name.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002568
Guido van Rossumff4949e1992-08-05 19:58:53 +00002569
Guido van Rossum30a685f1991-06-27 15:51:29 +00002570/* Python interface to gethostbyname(name). */
2571
2572/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002573static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002574socket_gethostbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002575{
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002576 char *name;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00002577#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002578 struct sockaddr_storage addrbuf;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00002579#else
2580 struct sockaddr_in addrbuf;
2581#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002582
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002583 if (!PyArg_ParseTuple(args, "s:gethostbyname", &name))
Guido van Rossum30a685f1991-06-27 15:51:29 +00002584 return NULL;
Martin v. Löwisddc6f472002-07-28 16:10:31 +00002585 if (setipaddr(name, (struct sockaddr *)&addrbuf, sizeof(addrbuf), AF_INET) < 0)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002586 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002587 return makeipaddr((struct sockaddr *)&addrbuf,
2588 sizeof(struct sockaddr_in));
Guido van Rossum30a685f1991-06-27 15:51:29 +00002589}
2590
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002591PyDoc_STRVAR(gethostbyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002592"gethostbyname(host) -> address\n\
2593\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002594Return the IP address (a string of the form '255.255.255.255') for a host.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002595
2596
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002597/* Convenience function common to gethostbyname_ex and gethostbyaddr */
2598
2599static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002600gethost_common(struct hostent *h, struct sockaddr *addr, int alen, int af)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002601{
2602 char **pch;
2603 PyObject *rtn_tuple = (PyObject *)NULL;
2604 PyObject *name_list = (PyObject *)NULL;
2605 PyObject *addr_list = (PyObject *)NULL;
2606 PyObject *tmp;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002607
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002608 if (h == NULL) {
Guido van Rossum48a680c2001-03-02 06:34:14 +00002609 /* Let's get real error message to return */
Guido van Rossume2ae77b2001-10-24 20:42:55 +00002610#ifndef RISCOS
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002611 set_herror(h_errno);
Guido van Rossume2ae77b2001-10-24 20:42:55 +00002612#else
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002613 PyErr_SetString(socket_error, "host not found");
Guido van Rossume2ae77b2001-10-24 20:42:55 +00002614#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002615 return NULL;
2616 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002617
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002618 if (h->h_addrtype != af) {
2619#ifdef HAVE_STRERROR
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002620 /* Let's get real error message to return */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002621 PyErr_SetString(socket_error,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002622 (char *)strerror(EAFNOSUPPORT));
2623#else
2624 PyErr_SetString(
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002625 socket_error,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002626 "Address family not supported by protocol family");
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002627#endif
2628 return NULL;
2629 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002630
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002631 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00002632
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002633 case AF_INET:
2634 if (alen < sizeof(struct sockaddr_in))
2635 return NULL;
2636 break;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002637
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00002638#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002639 case AF_INET6:
2640 if (alen < sizeof(struct sockaddr_in6))
2641 return NULL;
2642 break;
2643#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00002644
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002645 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002646
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002647 if ((name_list = PyList_New(0)) == NULL)
2648 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002649
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002650 if ((addr_list = PyList_New(0)) == NULL)
2651 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002652
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002653 for (pch = h->h_aliases; *pch != NULL; pch++) {
2654 int status;
2655 tmp = PyString_FromString(*pch);
2656 if (tmp == NULL)
2657 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002658
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002659 status = PyList_Append(name_list, tmp);
2660 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002661
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002662 if (status)
2663 goto err;
2664 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002665
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002666 for (pch = h->h_addr_list; *pch != NULL; pch++) {
2667 int status;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002668
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002669 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00002670
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002671 case AF_INET:
2672 {
2673 struct sockaddr_in sin;
2674 memset(&sin, 0, sizeof(sin));
2675 sin.sin_family = af;
2676#ifdef HAVE_SOCKADDR_SA_LEN
2677 sin.sin_len = sizeof(sin);
2678#endif
2679 memcpy(&sin.sin_addr, *pch, sizeof(sin.sin_addr));
2680 tmp = makeipaddr((struct sockaddr *)&sin, sizeof(sin));
Guido van Rossum67f7a382002-06-06 21:08:16 +00002681
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002682 if (pch == h->h_addr_list && alen >= sizeof(sin))
2683 memcpy((char *) addr, &sin, sizeof(sin));
2684 break;
2685 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002686
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00002687#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002688 case AF_INET6:
2689 {
2690 struct sockaddr_in6 sin6;
2691 memset(&sin6, 0, sizeof(sin6));
2692 sin6.sin6_family = af;
2693#ifdef HAVE_SOCKADDR_SA_LEN
2694 sin6.sin6_len = sizeof(sin6);
2695#endif
2696 memcpy(&sin6.sin6_addr, *pch, sizeof(sin6.sin6_addr));
2697 tmp = makeipaddr((struct sockaddr *)&sin6,
2698 sizeof(sin6));
Guido van Rossum67f7a382002-06-06 21:08:16 +00002699
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002700 if (pch == h->h_addr_list && alen >= sizeof(sin6))
2701 memcpy((char *) addr, &sin6, sizeof(sin6));
2702 break;
2703 }
2704#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00002705
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002706 default: /* can't happen */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002707 PyErr_SetString(socket_error,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002708 "unsupported address family");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002709 return NULL;
2710 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002711
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002712 if (tmp == NULL)
2713 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002714
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002715 status = PyList_Append(addr_list, tmp);
2716 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002717
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002718 if (status)
2719 goto err;
2720 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002721
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002722 rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002723
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002724 err:
2725 Py_XDECREF(name_list);
2726 Py_XDECREF(addr_list);
2727 return rtn_tuple;
2728}
2729
2730
2731/* Python interface to gethostbyname_ex(name). */
2732
2733/*ARGSUSED*/
2734static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002735socket_gethostbyname_ex(PyObject *self, PyObject *args)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002736{
2737 char *name;
2738 struct hostent *h;
Martin v. Löwisc16f3bd2003-05-03 09:14:54 +00002739#ifdef ENABLE_IPV6
2740 struct sockaddr_storage addr;
2741#else
2742 struct sockaddr_in addr;
2743#endif
Martin v. Löwis9db2f572001-07-23 01:30:10 +00002744 struct sockaddr *sa;
Guido van Rossum955becc1999-03-22 20:14:53 +00002745 PyObject *ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002746#ifdef HAVE_GETHOSTBYNAME_R
2747 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002748#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
2749 struct hostent_data data;
2750#else
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002751 char buf[16384];
2752 int buf_len = (sizeof buf) - 1;
2753 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002754#endif
2755#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Guido van Rossume9cd07b1999-03-15 21:40:14 +00002756 int result;
2757#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002758#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002759
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002760 if (!PyArg_ParseTuple(args, "s:gethostbyname_ex", &name))
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002761 return NULL;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00002762 if (setipaddr(name, (struct sockaddr *)&addr, sizeof(addr), AF_INET) < 0)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002763 return NULL;
2764 Py_BEGIN_ALLOW_THREADS
2765#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002766#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002767 result = gethostbyname_r(name, &hp_allocated, buf, buf_len,
2768 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002769#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002770 h = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002771#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Guido van Rossume7de2061999-03-24 17:24:33 +00002772 memset((void *) &data, '\0', sizeof(data));
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002773 result = gethostbyname_r(name, &hp_allocated, &data);
2774 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00002775#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002776#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00002777#ifdef USE_GETHOSTBYNAME_LOCK
Just van Rossum1040d2c2003-05-09 07:53:18 +00002778 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002779#endif
2780 h = gethostbyname(name);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002781#endif /* HAVE_GETHOSTBYNAME_R */
2782 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002783 /* Some C libraries would require addr.__ss_family instead of
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002784 addr.ss_family.
2785 Therefore, we cast the sockaddr_storage into sockaddr to
2786 access sa_family. */
Martin v. Löwis9db2f572001-07-23 01:30:10 +00002787 sa = (struct sockaddr*)&addr;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002788 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr),
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002789 sa->sa_family);
Guido van Rossum3baaa131999-03-22 21:44:51 +00002790#ifdef USE_GETHOSTBYNAME_LOCK
Just van Rossum1040d2c2003-05-09 07:53:18 +00002791 PyThread_release_lock(netdb_lock);
Guido van Rossum955becc1999-03-22 20:14:53 +00002792#endif
2793 return ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002794}
2795
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002796PyDoc_STRVAR(ghbn_ex_doc,
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002797"gethostbyname_ex(host) -> (name, aliaslist, addresslist)\n\
2798\n\
2799Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002800for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002801
2802
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002803/* Python interface to gethostbyaddr(IP). */
2804
2805/*ARGSUSED*/
2806static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002807socket_gethostbyaddr(PyObject *self, PyObject *args)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002808{
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00002809#ifdef ENABLE_IPV6
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002810 struct sockaddr_storage addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002811#else
Guido van Rossum48a680c2001-03-02 06:34:14 +00002812 struct sockaddr_in addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002813#endif
2814 struct sockaddr *sa = (struct sockaddr *)&addr;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002815 char *ip_num;
2816 struct hostent *h;
Guido van Rossum3baaa131999-03-22 21:44:51 +00002817 PyObject *ret;
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002818#ifdef HAVE_GETHOSTBYNAME_R
2819 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002820#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
2821 struct hostent_data data;
2822#else
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002823 char buf[16384];
2824 int buf_len = (sizeof buf) - 1;
2825 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002826#endif
2827#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Guido van Rossume9cd07b1999-03-15 21:40:14 +00002828 int result;
2829#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002830#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002831 char *ap;
2832 int al;
2833 int af;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002834
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002835 if (!PyArg_ParseTuple(args, "s:gethostbyaddr", &ip_num))
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002836 return NULL;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00002837 af = AF_UNSPEC;
Martin v. Löwisddc6f472002-07-28 16:10:31 +00002838 if (setipaddr(ip_num, sa, sizeof(addr), af) < 0)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002839 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002840 af = sa->sa_family;
2841 ap = NULL;
2842 al = 0;
2843 switch (af) {
2844 case AF_INET:
2845 ap = (char *)&((struct sockaddr_in *)sa)->sin_addr;
2846 al = sizeof(((struct sockaddr_in *)sa)->sin_addr);
2847 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00002848#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002849 case AF_INET6:
2850 ap = (char *)&((struct sockaddr_in6 *)sa)->sin6_addr;
2851 al = sizeof(((struct sockaddr_in6 *)sa)->sin6_addr);
2852 break;
2853#endif
2854 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002855 PyErr_SetString(socket_error, "unsupported address family");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002856 return NULL;
2857 }
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002858 Py_BEGIN_ALLOW_THREADS
2859#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002860#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002861 result = gethostbyaddr_r(ap, al, af,
2862 &hp_allocated, buf, buf_len,
Guido van Rossume9cd07b1999-03-15 21:40:14 +00002863 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002864#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002865 h = gethostbyaddr_r(ap, al, af,
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002866 &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002867#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Guido van Rossume7de2061999-03-24 17:24:33 +00002868 memset((void *) &data, '\0', sizeof(data));
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002869 result = gethostbyaddr_r(ap, al, af, &hp_allocated, &data);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002870 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00002871#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002872#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00002873#ifdef USE_GETHOSTBYNAME_LOCK
Just van Rossum1040d2c2003-05-09 07:53:18 +00002874 PyThread_acquire_lock(netdb_lock, 1);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002875#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002876 h = gethostbyaddr(ap, al, af);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002877#endif /* HAVE_GETHOSTBYNAME_R */
2878 Py_END_ALLOW_THREADS
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002879 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr), af);
Guido van Rossum3baaa131999-03-22 21:44:51 +00002880#ifdef USE_GETHOSTBYNAME_LOCK
Just van Rossum1040d2c2003-05-09 07:53:18 +00002881 PyThread_release_lock(netdb_lock);
Guido van Rossum3baaa131999-03-22 21:44:51 +00002882#endif
2883 return ret;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002884}
2885
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002886PyDoc_STRVAR(gethostbyaddr_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002887"gethostbyaddr(host) -> (name, aliaslist, addresslist)\n\
2888\n\
2889Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002890for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002891
Guido van Rossum30a685f1991-06-27 15:51:29 +00002892
2893/* Python interface to getservbyname(name).
2894 This only returns the port number, since the other info is already
2895 known or not useful (like the list of aliases). */
2896
2897/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002898static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002899socket_getservbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002900{
Barry Warsaw11b91a02004-06-28 00:50:43 +00002901 char *name, *proto=NULL;
Guido van Rossum30a685f1991-06-27 15:51:29 +00002902 struct servent *sp;
Barry Warsaw11b91a02004-06-28 00:50:43 +00002903 if (!PyArg_ParseTuple(args, "s|s:getservbyname", &name, &proto))
Guido van Rossum30a685f1991-06-27 15:51:29 +00002904 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00002905 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002906 sp = getservbyname(name, proto);
Guido van Rossum73624e91994-10-10 17:59:00 +00002907 Py_END_ALLOW_THREADS
Guido van Rossum30a685f1991-06-27 15:51:29 +00002908 if (sp == NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002909 PyErr_SetString(socket_error, "service/proto not found");
Guido van Rossum30a685f1991-06-27 15:51:29 +00002910 return NULL;
2911 }
Guido van Rossum73624e91994-10-10 17:59:00 +00002912 return PyInt_FromLong((long) ntohs(sp->s_port));
Guido van Rossum30a685f1991-06-27 15:51:29 +00002913}
2914
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002915PyDoc_STRVAR(getservbyname_doc,
Barry Warsaw11b91a02004-06-28 00:50:43 +00002916"getservbyname(servicename[, protocolname]) -> integer\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00002917\n\
2918Return a port number from a service name and protocol name.\n\
Barry Warsaw11b91a02004-06-28 00:50:43 +00002919The optional protocol name, if given, should be 'tcp' or 'udp',\n\
2920otherwise any protocol will match.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002921
Guido van Rossum30a685f1991-06-27 15:51:29 +00002922
Barry Warsaw11b91a02004-06-28 00:50:43 +00002923/* Python interface to getservbyport(port).
2924 This only returns the service name, since the other info is already
2925 known or not useful (like the list of aliases). */
2926
2927/*ARGSUSED*/
2928static PyObject *
2929socket_getservbyport(PyObject *self, PyObject *args)
2930{
2931 int port;
2932 char *proto=NULL;
2933 struct servent *sp;
2934 if (!PyArg_ParseTuple(args, "i|s:getservbyport", &port, &proto))
2935 return NULL;
2936 Py_BEGIN_ALLOW_THREADS
2937 sp = getservbyport(htons(port), proto);
2938 Py_END_ALLOW_THREADS
2939 if (sp == NULL) {
2940 PyErr_SetString(socket_error, "port/proto not found");
2941 return NULL;
2942 }
2943 return PyString_FromString(sp->s_name);
2944}
2945
2946PyDoc_STRVAR(getservbyport_doc,
2947"getservbyport(port[, protocolname]) -> string\n\
2948\n\
2949Return the service name from a port number and protocol name.\n\
2950The optional protocol name, if given, should be 'tcp' or 'udp',\n\
2951otherwise any protocol will match.");
2952
Guido van Rossum3901d851996-12-19 16:35:04 +00002953/* Python interface to getprotobyname(name).
2954 This only returns the protocol number, since the other info is
2955 already known or not useful (like the list of aliases). */
2956
2957/*ARGSUSED*/
2958static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002959socket_getprotobyname(PyObject *self, PyObject *args)
Guido van Rossum3901d851996-12-19 16:35:04 +00002960{
2961 char *name;
2962 struct protoent *sp;
Guido van Rossumbcc20741998-08-04 22:53:56 +00002963#ifdef __BEOS__
2964/* Not available in BeOS yet. - [cjh] */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002965 PyErr_SetString(socket_error, "getprotobyname not supported");
Guido van Rossumbcc20741998-08-04 22:53:56 +00002966 return NULL;
2967#else
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002968 if (!PyArg_ParseTuple(args, "s:getprotobyname", &name))
Guido van Rossum3901d851996-12-19 16:35:04 +00002969 return NULL;
2970 Py_BEGIN_ALLOW_THREADS
2971 sp = getprotobyname(name);
2972 Py_END_ALLOW_THREADS
2973 if (sp == NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002974 PyErr_SetString(socket_error, "protocol not found");
Guido van Rossum3901d851996-12-19 16:35:04 +00002975 return NULL;
2976 }
2977 return PyInt_FromLong((long) sp->p_proto);
Guido van Rossumbcc20741998-08-04 22:53:56 +00002978#endif
Guido van Rossum3901d851996-12-19 16:35:04 +00002979}
2980
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002981PyDoc_STRVAR(getprotobyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002982"getprotobyname(name) -> integer\n\
2983\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002984Return the protocol number for the named protocol. (Rarely used.)");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002985
Guido van Rossum3901d851996-12-19 16:35:04 +00002986
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002987#ifndef NO_DUP
Guido van Rossum2a7178e1992-12-08 13:38:24 +00002988/* Create a socket object from a numeric file description.
2989 Useful e.g. if stdin is a socket.
2990 Additional arguments as for socket(). */
2991
2992/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002993static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002994socket_fromfd(PyObject *self, PyObject *args)
Guido van Rossum2a7178e1992-12-08 13:38:24 +00002995{
Guido van Rossum73624e91994-10-10 17:59:00 +00002996 PySocketSockObject *s;
Fred Drakea04eaad2000-06-30 02:46:07 +00002997 SOCKET_T fd;
2998 int family, type, proto = 0;
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002999 if (!PyArg_ParseTuple(args, "iii|i:fromfd",
3000 &fd, &family, &type, &proto))
Guido van Rossumbe32c891996-06-20 16:25:29 +00003001 return NULL;
Guido van Rossum5f59d601992-12-14 16:59:51 +00003002 /* Dup the fd so it and the socket can be closed independently */
3003 fd = dup(fd);
3004 if (fd < 0)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003005 return set_error();
3006 s = new_sockobject(fd, family, type, proto);
Guido van Rossum2a7178e1992-12-08 13:38:24 +00003007 /* From now on, ignore SIGPIPE and let the error checking
3008 do the work. */
Guido van Rossum48a680c2001-03-02 06:34:14 +00003009#ifdef SIGPIPE
Guido van Rossum2a7178e1992-12-08 13:38:24 +00003010 (void) signal(SIGPIPE, SIG_IGN);
Guido van Rossum48a680c2001-03-02 06:34:14 +00003011#endif
Guido van Rossum73624e91994-10-10 17:59:00 +00003012 return (PyObject *) s;
Guido van Rossum2a7178e1992-12-08 13:38:24 +00003013}
Guido van Rossum82a5c661998-07-07 20:45:43 +00003014
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003015PyDoc_STRVAR(fromfd_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003016"fromfd(fd, family, type[, proto]) -> socket object\n\
3017\n\
3018Create a socket object from the given file descriptor.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003019The remaining arguments are the same as for socket().");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003020
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00003021#endif /* NO_DUP */
Guido van Rossum2a7178e1992-12-08 13:38:24 +00003022
Guido van Rossum82a5c661998-07-07 20:45:43 +00003023
Guido van Rossum006bf911996-06-12 04:04:55 +00003024static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003025socket_ntohs(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003026{
3027 int x1, x2;
3028
Guido van Rossum27fc3c02000-03-24 20:56:56 +00003029 if (!PyArg_ParseTuple(args, "i:ntohs", &x1)) {
Guido van Rossum006bf911996-06-12 04:04:55 +00003030 return NULL;
3031 }
3032 x2 = (int)ntohs((short)x1);
3033 return PyInt_FromLong(x2);
3034}
3035
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003036PyDoc_STRVAR(ntohs_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003037"ntohs(integer) -> integer\n\
3038\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003039Convert a 16-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003040
3041
Guido van Rossum006bf911996-06-12 04:04:55 +00003042static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003043socket_ntohl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003044{
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003045 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003046
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003047 if (PyInt_Check(arg)) {
3048 x = PyInt_AS_LONG(arg);
Jeremy Hylton825e47b2002-07-25 16:37:51 +00003049 if (x == (unsigned long) -1 && PyErr_Occurred())
3050 return NULL;
Guido van Rossum006bf911996-06-12 04:04:55 +00003051 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003052 else if (PyLong_Check(arg)) {
3053 x = PyLong_AsUnsignedLong(arg);
Jeremy Hylton825e47b2002-07-25 16:37:51 +00003054 if (x == (unsigned long) -1 && PyErr_Occurred())
3055 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003056#if SIZEOF_LONG > 4
3057 {
3058 unsigned long y;
3059 /* only want the trailing 32 bits */
3060 y = x & 0xFFFFFFFFUL;
3061 if (y ^ x)
3062 return PyErr_Format(PyExc_OverflowError,
3063 "long int larger than 32 bits");
3064 x = y;
Tim Peters58141872002-08-06 22:25:02 +00003065 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003066#endif
3067 }
3068 else
Tim Peters58141872002-08-06 22:25:02 +00003069 return PyErr_Format(PyExc_TypeError,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003070 "expected int/long, %s found",
3071 arg->ob_type->tp_name);
3072 if (x == (unsigned long) -1 && PyErr_Occurred())
3073 return NULL;
3074 return PyInt_FromLong(ntohl(x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003075}
3076
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003077PyDoc_STRVAR(ntohl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003078"ntohl(integer) -> integer\n\
3079\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003080Convert a 32-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003081
3082
Guido van Rossum006bf911996-06-12 04:04:55 +00003083static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003084socket_htons(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00003085{
Hye-Shik Changa7b673f2003-12-17 09:40:23 +00003086 int x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00003087
Guido van Rossum27fc3c02000-03-24 20:56:56 +00003088 if (!PyArg_ParseTuple(args, "i:htons", &x1)) {
Guido van Rossum006bf911996-06-12 04:04:55 +00003089 return NULL;
3090 }
3091 x2 = (int)htons((short)x1);
3092 return PyInt_FromLong(x2);
3093}
3094
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003095PyDoc_STRVAR(htons_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003096"htons(integer) -> integer\n\
3097\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003098Convert a 16-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003099
3100
Guido van Rossum006bf911996-06-12 04:04:55 +00003101static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003102socket_htonl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00003103{
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003104 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00003105
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003106 if (PyInt_Check(arg)) {
3107 x = PyInt_AS_LONG(arg);
Jeremy Hylton825e47b2002-07-25 16:37:51 +00003108 if (x == (unsigned long) -1 && PyErr_Occurred())
3109 return NULL;
Guido van Rossum006bf911996-06-12 04:04:55 +00003110 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003111 else if (PyLong_Check(arg)) {
3112 x = PyLong_AsUnsignedLong(arg);
Jeremy Hylton825e47b2002-07-25 16:37:51 +00003113 if (x == (unsigned long) -1 && PyErr_Occurred())
3114 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003115#if SIZEOF_LONG > 4
3116 {
3117 unsigned long y;
3118 /* only want the trailing 32 bits */
3119 y = x & 0xFFFFFFFFUL;
3120 if (y ^ x)
3121 return PyErr_Format(PyExc_OverflowError,
3122 "long int larger than 32 bits");
3123 x = y;
Tim Peters58141872002-08-06 22:25:02 +00003124 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003125#endif
3126 }
3127 else
Tim Peters58141872002-08-06 22:25:02 +00003128 return PyErr_Format(PyExc_TypeError,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003129 "expected int/long, %s found",
3130 arg->ob_type->tp_name);
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003131 return PyInt_FromLong(htonl(x));
Guido van Rossum006bf911996-06-12 04:04:55 +00003132}
3133
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003134PyDoc_STRVAR(htonl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00003135"htonl(integer) -> integer\n\
3136\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003137Convert a 32-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003138
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003139/* socket.inet_aton() and socket.inet_ntoa() functions. */
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003140
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003141PyDoc_STRVAR(inet_aton_doc,
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003142"inet_aton(string) -> packed 32-bit IP representation\n\
3143\n\
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003144Convert 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 +00003145binary format used in low-level network functions.");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003146
3147static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003148socket_inet_aton(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003149{
Guido van Rossuma2e48551999-09-09 15:42:59 +00003150#ifndef INADDR_NONE
3151#define INADDR_NONE (-1)
3152#endif
Neal Norwitz88f115b2003-02-13 02:15:42 +00003153#ifdef HAVE_INET_ATON
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003154 struct in_addr buf;
Tim Peters1df9fdd2003-02-13 03:13:40 +00003155#else
3156 /* Have to use inet_addr() instead */
3157 unsigned long packed_addr;
3158#endif
3159 char *ip_addr;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003160
Tim Peters1df9fdd2003-02-13 03:13:40 +00003161 if (!PyArg_ParseTuple(args, "s:inet_aton", &ip_addr))
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003162 return NULL;
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003163
Tim Peters1df9fdd2003-02-13 03:13:40 +00003164
3165#ifdef HAVE_INET_ATON
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003166 if (inet_aton(ip_addr, &buf))
3167 return PyString_FromStringAndSize((char *)(&buf),
3168 sizeof(buf));
3169
3170 PyErr_SetString(socket_error,
3171 "illegal IP address string passed to inet_aton");
3172 return NULL;
3173
Tim Peters1df9fdd2003-02-13 03:13:40 +00003174#else /* ! HAVE_INET_ATON */
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003175 /* XXX Problem here: inet_aton('255.255.255.255') raises
3176 an exception while it should be a valid address. */
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003177 packed_addr = inet_addr(ip_addr);
3178
3179 if (packed_addr == INADDR_NONE) { /* invalid address */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003180 PyErr_SetString(socket_error,
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003181 "illegal IP address string passed to inet_aton");
3182 return NULL;
3183 }
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003184 return PyString_FromStringAndSize((char *) &packed_addr,
3185 sizeof(packed_addr));
Guido van Rossumad05cdf2003-02-12 23:08:22 +00003186#endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003187}
3188
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003189PyDoc_STRVAR(inet_ntoa_doc,
Fred Drakee0661342000-03-07 14:05:16 +00003190"inet_ntoa(packed_ip) -> ip_address_string\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003191\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003192Convert an IP address from 32-bit packed binary format to string format");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003193
3194static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003195socket_inet_ntoa(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003196{
3197 char *packed_str;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00003198 int addr_len;
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003199 struct in_addr packed_addr;
3200
Guido van Rossum27fc3c02000-03-24 20:56:56 +00003201 if (!PyArg_ParseTuple(args, "s#:inet_ntoa", &packed_str, &addr_len)) {
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003202 return NULL;
3203 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00003204
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003205 if (addr_len != sizeof(packed_addr)) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003206 PyErr_SetString(socket_error,
Guido van Rossum5c9eb211999-08-20 18:21:51 +00003207 "packed IP wrong length for inet_ntoa");
3208 return NULL;
3209 }
3210
3211 memcpy(&packed_addr, packed_str, addr_len);
3212
3213 return PyString_FromString(inet_ntoa(packed_addr));
3214}
Guido van Rossum82a5c661998-07-07 20:45:43 +00003215
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003216#ifdef HAVE_INET_PTON
3217
3218PyDoc_STRVAR(inet_pton_doc,
3219"inet_pton(af, ip) -> packed IP address string\n\
3220\n\
3221Convert an IP address from string format to a packed string suitable\n\
3222for use with low-level network functions.");
3223
3224static PyObject *
3225socket_inet_pton(PyObject *self, PyObject *args)
3226{
3227 int af;
3228 char* ip;
3229 int retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003230#ifdef ENABLE_IPV6
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003231 char packed[MAX(sizeof(struct in_addr), sizeof(struct in6_addr))];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003232#else
3233 char packed[sizeof(struct in_addr)];
3234#endif
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003235 if (!PyArg_ParseTuple(args, "is:inet_pton", &af, &ip)) {
3236 return NULL;
3237 }
3238
Martin v. Löwis04697e82004-06-02 12:35:29 +00003239#if !defined(ENABLE_IPV6) && defined(AF_INET6)
Martin v. Löwis10649092003-08-05 06:25:06 +00003240 if(af == AF_INET6) {
3241 PyErr_SetString(socket_error,
3242 "can't use AF_INET6, IPv6 is disabled");
3243 return NULL;
3244 }
3245#endif
3246
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003247 retval = inet_pton(af, ip, packed);
3248 if (retval < 0) {
3249 PyErr_SetFromErrno(socket_error);
3250 return NULL;
3251 } else if (retval == 0) {
3252 PyErr_SetString(socket_error,
3253 "illegal IP address string passed to inet_pton");
3254 return NULL;
3255 } else if (af == AF_INET) {
3256 return PyString_FromStringAndSize(packed,
3257 sizeof(struct in_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003258#ifdef ENABLE_IPV6
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003259 } else if (af == AF_INET6) {
3260 return PyString_FromStringAndSize(packed,
3261 sizeof(struct in6_addr));
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003262#endif
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003263 } else {
3264 PyErr_SetString(socket_error, "unknown address family");
3265 return NULL;
3266 }
3267}
3268
3269PyDoc_STRVAR(inet_ntop_doc,
3270"inet_ntop(af, packed_ip) -> string formatted IP address\n\
3271\n\
3272Convert a packed IP address of the given family to string format.");
3273
3274static PyObject *
3275socket_inet_ntop(PyObject *self, PyObject *args)
3276{
3277 int af;
3278 char* packed;
3279 int len;
3280 const char* retval;
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003281#ifdef ENABLE_IPV6
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003282 char ip[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN) + 1];
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003283#else
3284 char ip[INET_ADDRSTRLEN + 1];
3285#endif
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003286
3287 /* Guarantee NUL-termination for PyString_FromString() below */
3288 memset((void *) &ip[0], '\0', sizeof(ip) + 1);
3289
3290 if (!PyArg_ParseTuple(args, "is#:inet_ntop", &af, &packed, &len)) {
3291 return NULL;
3292 }
3293
3294 if (af == AF_INET) {
3295 if (len != sizeof(struct in_addr)) {
3296 PyErr_SetString(PyExc_ValueError,
3297 "invalid length of packed IP address string");
3298 return NULL;
3299 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003300#ifdef ENABLE_IPV6
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003301 } else if (af == AF_INET6) {
3302 if (len != sizeof(struct in6_addr)) {
3303 PyErr_SetString(PyExc_ValueError,
3304 "invalid length of packed IP address string");
3305 return NULL;
3306 }
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003307#endif
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003308 } else {
3309 PyErr_Format(PyExc_ValueError,
3310 "unknown address family %d", af);
3311 return NULL;
3312 }
3313
3314 retval = inet_ntop(af, packed, ip, sizeof(ip));
3315 if (!retval) {
3316 PyErr_SetFromErrno(socket_error);
3317 return NULL;
3318 } else {
3319 return PyString_FromString(retval);
3320 }
3321
3322 /* NOTREACHED */
3323 PyErr_SetString(PyExc_RuntimeError, "invalid handling of inet_ntop");
3324 return NULL;
3325}
3326
3327#endif /* HAVE_INET_PTON */
3328
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003329/* Python interface to getaddrinfo(host, port). */
3330
3331/*ARGSUSED*/
3332static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003333socket_getaddrinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003334{
Martin v. Löwisf0b11d22001-11-07 08:31:03 +00003335 struct addrinfo hints, *res;
3336 struct addrinfo *res0 = NULL;
Martin v. Löwis2548c732003-04-18 10:39:54 +00003337 PyObject *hobj = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003338 PyObject *pobj = (PyObject *)NULL;
Martin v. Löwisf65b1a12001-08-12 09:28:40 +00003339 char pbuf[30];
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003340 char *hptr, *pptr;
3341 int family, socktype, protocol, flags;
3342 int error;
3343 PyObject *all = (PyObject *)NULL;
3344 PyObject *single = (PyObject *)NULL;
Martin v. Löwis2548c732003-04-18 10:39:54 +00003345 PyObject *idna = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003346
3347 family = socktype = protocol = flags = 0;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00003348 family = AF_UNSPEC;
Martin v. Löwis2548c732003-04-18 10:39:54 +00003349 if (!PyArg_ParseTuple(args, "OO|iiii:getaddrinfo",
3350 &hobj, &pobj, &family, &socktype,
3351 &protocol, &flags)) {
3352 return NULL;
3353 }
3354 if (hobj == Py_None) {
3355 hptr = NULL;
3356 } else if (PyUnicode_Check(hobj)) {
3357 idna = PyObject_CallMethod(hobj, "encode", "s", "idna");
3358 if (!idna)
3359 return NULL;
3360 hptr = PyString_AsString(idna);
3361 } else if (PyString_Check(hobj)) {
3362 hptr = PyString_AsString(hobj);
3363 } else {
3364 PyErr_SetString(PyExc_TypeError,
3365 "getaddrinfo() argument 1 must be string or None");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003366 return NULL;
3367 }
3368 if (PyInt_Check(pobj)) {
Marc-André Lemburg03d1b182001-07-31 18:05:33 +00003369 PyOS_snprintf(pbuf, sizeof(pbuf), "%ld", PyInt_AsLong(pobj));
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003370 pptr = pbuf;
3371 } else if (PyString_Check(pobj)) {
3372 pptr = PyString_AsString(pobj);
3373 } else if (pobj == Py_None) {
3374 pptr = (char *)NULL;
3375 } else {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003376 PyErr_SetString(socket_error, "Int or String expected");
Martin v. Löwis5db099a2003-08-07 11:55:15 +00003377 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003378 }
3379 memset(&hints, 0, sizeof(hints));
3380 hints.ai_family = family;
3381 hints.ai_socktype = socktype;
3382 hints.ai_protocol = protocol;
3383 hints.ai_flags = flags;
Just van Rossum1040d2c2003-05-09 07:53:18 +00003384 Py_BEGIN_ALLOW_THREADS
3385 ACQUIRE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003386 error = getaddrinfo(hptr, pptr, &hints, &res0);
Just van Rossum1040d2c2003-05-09 07:53:18 +00003387 Py_END_ALLOW_THREADS
3388 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003389 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003390 set_gaierror(error);
Martin v. Löwis5db099a2003-08-07 11:55:15 +00003391 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003392 }
3393
3394 if ((all = PyList_New(0)) == NULL)
3395 goto err;
3396 for (res = res0; res; res = res->ai_next) {
Guido van Rossum716aac02001-10-12 18:59:27 +00003397 PyObject *addr =
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00003398 makesockaddr(-1, res->ai_addr, res->ai_addrlen, protocol);
Guido van Rossum716aac02001-10-12 18:59:27 +00003399 if (addr == NULL)
3400 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003401 single = Py_BuildValue("iiisO", res->ai_family,
3402 res->ai_socktype, res->ai_protocol,
3403 res->ai_canonname ? res->ai_canonname : "",
Guido van Rossum716aac02001-10-12 18:59:27 +00003404 addr);
3405 Py_DECREF(addr);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003406 if (single == NULL)
3407 goto err;
3408
3409 if (PyList_Append(all, single))
3410 goto err;
3411 Py_XDECREF(single);
3412 }
Martin v. Löwis2548c732003-04-18 10:39:54 +00003413 Py_XDECREF(idna);
Neal Norwitz90128ba2002-08-09 03:37:42 +00003414 if (res0)
3415 freeaddrinfo(res0);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003416 return all;
3417 err:
3418 Py_XDECREF(single);
3419 Py_XDECREF(all);
Martin v. Löwis2548c732003-04-18 10:39:54 +00003420 Py_XDECREF(idna);
Martin v. Löwisf0b11d22001-11-07 08:31:03 +00003421 if (res0)
3422 freeaddrinfo(res0);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003423 return (PyObject *)NULL;
3424}
3425
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003426PyDoc_STRVAR(getaddrinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003427"getaddrinfo(host, port [, family, socktype, proto, flags])\n\
3428 -> list of (family, socktype, proto, canonname, sockaddr)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003429\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003430Resolve host and port into addrinfo struct.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003431
3432/* Python interface to getnameinfo(sa, flags). */
3433
3434/*ARGSUSED*/
3435static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003436socket_getnameinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003437{
3438 PyObject *sa = (PyObject *)NULL;
3439 int flags;
3440 char *hostp;
Martin v. Löwis06b1d212001-11-02 23:34:52 +00003441 int port, flowinfo, scope_id;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003442 char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
3443 struct addrinfo hints, *res = NULL;
3444 int error;
3445 PyObject *ret = (PyObject *)NULL;
3446
3447 flags = flowinfo = scope_id = 0;
Martin v. Löwis06b1d212001-11-02 23:34:52 +00003448 if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags))
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003449 return NULL;
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003450 if (!PyArg_ParseTuple(sa, "si|ii",
3451 &hostp, &port, &flowinfo, &scope_id))
Martin v. Löwis06b1d212001-11-02 23:34:52 +00003452 return NULL;
Marc-André Lemburg03d1b182001-07-31 18:05:33 +00003453 PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003454 memset(&hints, 0, sizeof(hints));
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00003455 hints.ai_family = AF_UNSPEC;
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003456 hints.ai_socktype = SOCK_DGRAM; /* make numeric port happy */
Just van Rossum1040d2c2003-05-09 07:53:18 +00003457 Py_BEGIN_ALLOW_THREADS
3458 ACQUIRE_GETADDRINFO_LOCK
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003459 error = getaddrinfo(hostp, pbuf, &hints, &res);
Just van Rossum1040d2c2003-05-09 07:53:18 +00003460 Py_END_ALLOW_THREADS
3461 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003462 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003463 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003464 goto fail;
3465 }
3466 if (res->ai_next) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003467 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003468 "sockaddr resolved to multiple addresses");
3469 goto fail;
3470 }
3471 switch (res->ai_family) {
3472 case AF_INET:
3473 {
3474 char *t1;
3475 int t2;
3476 if (PyArg_ParseTuple(sa, "si", &t1, &t2) == 0) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003477 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003478 "IPv4 sockaddr must be 2 tuple");
3479 goto fail;
3480 }
3481 break;
3482 }
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00003483#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003484 case AF_INET6:
3485 {
3486 struct sockaddr_in6 *sin6;
3487 sin6 = (struct sockaddr_in6 *)res->ai_addr;
3488 sin6->sin6_flowinfo = flowinfo;
3489 sin6->sin6_scope_id = scope_id;
3490 break;
3491 }
3492#endif
3493 }
3494 error = getnameinfo(res->ai_addr, res->ai_addrlen,
3495 hbuf, sizeof(hbuf), pbuf, sizeof(pbuf), flags);
3496 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003497 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003498 goto fail;
3499 }
3500 ret = Py_BuildValue("ss", hbuf, pbuf);
3501
3502fail:
3503 if (res)
3504 freeaddrinfo(res);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003505 return ret;
3506}
3507
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003508PyDoc_STRVAR(getnameinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003509"getnameinfo(sockaddr, flags) --> (host, port)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003510\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003511Get host and port for a sockaddr.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003512
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00003513
3514/* Python API to getting and setting the default timeout value. */
3515
3516static PyObject *
3517socket_getdefaulttimeout(PyObject *self)
3518{
3519 if (defaulttimeout < 0.0) {
3520 Py_INCREF(Py_None);
3521 return Py_None;
3522 }
3523 else
3524 return PyFloat_FromDouble(defaulttimeout);
3525}
3526
3527PyDoc_STRVAR(getdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003528"getdefaulttimeout() -> timeout\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00003529\n\
3530Returns the default timeout in floating seconds for new socket objects.\n\
3531A value of None indicates that new socket objects have no timeout.\n\
3532When the socket module is first imported, the default is None.");
3533
3534static PyObject *
3535socket_setdefaulttimeout(PyObject *self, PyObject *arg)
3536{
3537 double timeout;
3538
3539 if (arg == Py_None)
3540 timeout = -1.0;
3541 else {
3542 timeout = PyFloat_AsDouble(arg);
3543 if (timeout < 0.0) {
3544 if (!PyErr_Occurred())
3545 PyErr_SetString(PyExc_ValueError,
3546 "Timeout value out of range");
3547 return NULL;
3548 }
3549 }
3550
3551 defaulttimeout = timeout;
3552
3553 Py_INCREF(Py_None);
3554 return Py_None;
3555}
3556
3557PyDoc_STRVAR(setdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003558"setdefaulttimeout(timeout)\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00003559\n\
3560Set the default timeout in floating seconds for new socket objects.\n\
3561A value of None indicates that new socket objects have no timeout.\n\
3562When the socket module is first imported, the default is None.");
3563
3564
Guido van Rossum30a685f1991-06-27 15:51:29 +00003565/* List of functions exported by this module. */
3566
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003567static PyMethodDef socket_methods[] = {
3568 {"gethostbyname", socket_gethostbyname,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003569 METH_VARARGS, gethostbyname_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003570 {"gethostbyname_ex", socket_gethostbyname_ex,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003571 METH_VARARGS, ghbn_ex_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003572 {"gethostbyaddr", socket_gethostbyaddr,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003573 METH_VARARGS, gethostbyaddr_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003574 {"gethostname", socket_gethostname,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003575 METH_VARARGS, gethostname_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003576 {"getservbyname", socket_getservbyname,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003577 METH_VARARGS, getservbyname_doc},
Barry Warsaw11b91a02004-06-28 00:50:43 +00003578 {"getservbyport", socket_getservbyport,
3579 METH_VARARGS, getservbyport_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003580 {"getprotobyname", socket_getprotobyname,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003581 METH_VARARGS,getprotobyname_doc},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00003582#ifndef NO_DUP
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003583 {"fromfd", socket_fromfd,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003584 METH_VARARGS, fromfd_doc},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00003585#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003586 {"ntohs", socket_ntohs,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003587 METH_VARARGS, ntohs_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003588 {"ntohl", socket_ntohl,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003589 METH_O, ntohl_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003590 {"htons", socket_htons,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003591 METH_VARARGS, htons_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003592 {"htonl", socket_htonl,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003593 METH_O, htonl_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003594 {"inet_aton", socket_inet_aton,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003595 METH_VARARGS, inet_aton_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003596 {"inet_ntoa", socket_inet_ntoa,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003597 METH_VARARGS, inet_ntoa_doc},
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003598#ifdef HAVE_INET_PTON
3599 {"inet_pton", socket_inet_pton,
3600 METH_VARARGS, inet_pton_doc},
3601 {"inet_ntop", socket_inet_ntop,
3602 METH_VARARGS, inet_ntop_doc},
3603#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003604 {"getaddrinfo", socket_getaddrinfo,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003605 METH_VARARGS, getaddrinfo_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003606 {"getnameinfo", socket_getnameinfo,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003607 METH_VARARGS, getnameinfo_doc},
Guido van Rossum1693ba82002-07-18 21:11:26 +00003608 {"getdefaulttimeout", (PyCFunction)socket_getdefaulttimeout,
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00003609 METH_NOARGS, getdefaulttimeout_doc},
3610 {"setdefaulttimeout", socket_setdefaulttimeout,
3611 METH_O, setdefaulttimeout_doc},
Guido van Rossum30a685f1991-06-27 15:51:29 +00003612 {NULL, NULL} /* Sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00003613};
3614
Guido van Rossum30a685f1991-06-27 15:51:29 +00003615
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003616#ifdef RISCOS
3617#define OS_INIT_DEFINED
Guido van Rossumbe32c891996-06-20 16:25:29 +00003618
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003619static int
3620os_init(void)
3621{
3622 _kernel_swi_regs r;
3623
3624 r.r[0] = 0;
3625 _kernel_swi(0x43380, &r, &r);
3626 taskwindow = r.r[0];
3627
Martin v. Löwisa94568a2003-05-10 07:36:56 +00003628 return 1;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003629}
3630
3631#endif /* RISCOS */
3632
3633
3634#ifdef MS_WINDOWS
3635#define OS_INIT_DEFINED
3636
3637/* Additional initialization and cleanup for Windows */
Guido van Rossumbe32c891996-06-20 16:25:29 +00003638
3639static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003640os_cleanup(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00003641{
3642 WSACleanup();
Guido van Rossumbe32c891996-06-20 16:25:29 +00003643}
3644
3645static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003646os_init(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00003647{
3648 WSADATA WSAData;
3649 int ret;
3650 char buf[100];
3651 ret = WSAStartup(0x0101, &WSAData);
3652 switch (ret) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003653 case 0: /* No error */
Mark Hammond14350ab2003-07-09 04:57:46 +00003654 Py_AtExit(os_cleanup);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003655 return 1; /* Success */
Guido van Rossumbe32c891996-06-20 16:25:29 +00003656 case WSASYSNOTREADY:
3657 PyErr_SetString(PyExc_ImportError,
3658 "WSAStartup failed: network not ready");
3659 break;
3660 case WSAVERNOTSUPPORTED:
3661 case WSAEINVAL:
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003662 PyErr_SetString(
3663 PyExc_ImportError,
3664 "WSAStartup failed: requested version not supported");
Guido van Rossumbe32c891996-06-20 16:25:29 +00003665 break;
3666 default:
Tim Peters885d4572001-11-28 20:27:42 +00003667 PyOS_snprintf(buf, sizeof(buf),
3668 "WSAStartup failed: error code %d", ret);
Guido van Rossumbe32c891996-06-20 16:25:29 +00003669 PyErr_SetString(PyExc_ImportError, buf);
3670 break;
3671 }
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003672 return 0; /* Failure */
Guido van Rossumbe32c891996-06-20 16:25:29 +00003673}
3674
Guido van Rossum8d665e61996-06-26 18:22:49 +00003675#endif /* MS_WINDOWS */
Guido van Rossumbe32c891996-06-20 16:25:29 +00003676
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003677
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003678#ifdef PYOS_OS2
3679#define OS_INIT_DEFINED
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003680
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003681/* Additional initialization for OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003682
3683static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003684os_init(void)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003685{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003686#ifndef PYCC_GCC
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003687 char reason[64];
3688 int rc = sock_init();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003689
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003690 if (rc == 0) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003691 return 1; /* Success */
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003692 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003693
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003694 PyOS_snprintf(reason, sizeof(reason),
3695 "OS/2 TCP/IP Error# %d", sock_errno());
3696 PyErr_SetString(PyExc_ImportError, reason);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003697
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003698 return 0; /* Failure */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00003699#else
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003700 /* No need to initialise sockets with GCC/EMX */
3701 return 1; /* Success */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00003702#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003703}
3704
3705#endif /* PYOS_OS2 */
3706
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003707
3708#ifndef OS_INIT_DEFINED
3709static int
3710os_init(void)
3711{
3712 return 1; /* Success */
3713}
3714#endif
3715
3716
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00003717/* C API table - always add new things to the end for binary
3718 compatibility. */
3719static
3720PySocketModule_APIObject PySocketModuleAPI =
3721{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003722 &sock_type,
Brett Cannon06c34792004-03-23 23:16:54 +00003723 NULL
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00003724};
3725
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003726
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003727/* Initialize the _socket module.
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003728
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003729 This module is actually called "_socket", and there's a wrapper
3730 "socket.py" which implements some additional functionality. On some
3731 platforms (e.g. Windows and OS/2), socket.py also implements a
3732 wrapper for the socket type that provides missing functionality such
3733 as makefile(), dup() and fromfd(). The import of "_socket" may fail
3734 with an ImportError exception if os-specific initialization fails.
3735 On Windows, this does WINSOCK initialization. When WINSOCK is
3736 initialized succesfully, a call to WSACleanup() is scheduled to be
3737 made at exit time.
3738*/
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003739
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003740PyDoc_STRVAR(socket_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003741"Implementation module for socket operations.\n\
3742\n\
3743See the socket module for documentation.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003744
Mark Hammondfe51c6d2002-08-02 02:27:13 +00003745PyMODINIT_FUNC
Thomas Wouters1e0c2f42000-07-24 16:06:23 +00003746init_socket(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00003747{
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003748 PyObject *m, *has_ipv6;
Fred Drake4baedc12002-04-01 14:53:37 +00003749
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003750 if (!os_init())
Guido van Rossum0cb96de1997-10-01 04:29:29 +00003751 return;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003752
3753 sock_type.ob_type = &PyType_Type;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003754 m = Py_InitModule3(PySocket_MODULE_NAME,
3755 socket_methods,
3756 socket_doc);
3757
3758 socket_error = PyErr_NewException("socket.error", NULL, NULL);
3759 if (socket_error == NULL)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003760 return;
Brett Cannon06c34792004-03-23 23:16:54 +00003761 PySocketModuleAPI.error = socket_error;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003762 Py_INCREF(socket_error);
3763 PyModule_AddObject(m, "error", socket_error);
3764 socket_herror = PyErr_NewException("socket.herror",
3765 socket_error, NULL);
3766 if (socket_herror == NULL)
3767 return;
3768 Py_INCREF(socket_herror);
3769 PyModule_AddObject(m, "herror", socket_herror);
3770 socket_gaierror = PyErr_NewException("socket.gaierror", socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003771 NULL);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003772 if (socket_gaierror == NULL)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003773 return;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003774 Py_INCREF(socket_gaierror);
3775 PyModule_AddObject(m, "gaierror", socket_gaierror);
Raymond Hettingeref7343c2003-06-29 03:08:05 +00003776 socket_timeout = PyErr_NewException("socket.timeout",
3777 socket_error, NULL);
3778 if (socket_timeout == NULL)
3779 return;
3780 Py_INCREF(socket_timeout);
3781 PyModule_AddObject(m, "timeout", socket_timeout);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003782 Py_INCREF((PyObject *)&sock_type);
Fred Drake4baedc12002-04-01 14:53:37 +00003783 if (PyModule_AddObject(m, "SocketType",
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003784 (PyObject *)&sock_type) != 0)
Guido van Rossum0cb96de1997-10-01 04:29:29 +00003785 return;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003786 Py_INCREF((PyObject *)&sock_type);
Fred Drake4baedc12002-04-01 14:53:37 +00003787 if (PyModule_AddObject(m, "socket",
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003788 (PyObject *)&sock_type) != 0)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003789 return;
Guido van Rossum09be4091999-08-09 14:40:40 +00003790
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00003791#ifdef ENABLE_IPV6
3792 has_ipv6 = Py_True;
3793#else
3794 has_ipv6 = Py_False;
3795#endif
3796 Py_INCREF(has_ipv6);
3797 PyModule_AddObject(m, "has_ipv6", has_ipv6);
3798
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00003799 /* Export C API */
Fred Drake4baedc12002-04-01 14:53:37 +00003800 if (PyModule_AddObject(m, PySocket_CAPI_NAME,
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00003801 PyCObject_FromVoidPtr((void *)&PySocketModuleAPI, NULL)
3802 ) != 0)
3803 return;
3804
Guido van Rossum09be4091999-08-09 14:40:40 +00003805 /* Address families (we only support AF_INET and AF_UNIX) */
3806#ifdef AF_UNSPEC
Fred Drake4baedc12002-04-01 14:53:37 +00003807 PyModule_AddIntConstant(m, "AF_UNSPEC", AF_UNSPEC);
Guido van Rossum09be4091999-08-09 14:40:40 +00003808#endif
Fred Drake4baedc12002-04-01 14:53:37 +00003809 PyModule_AddIntConstant(m, "AF_INET", AF_INET);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003810#ifdef AF_INET6
Fred Drake4baedc12002-04-01 14:53:37 +00003811 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003812#endif /* AF_INET6 */
Andrew MacIntyred12dfbb2004-04-04 07:13:49 +00003813#if defined(AF_UNIX)
Fred Drake4baedc12002-04-01 14:53:37 +00003814 PyModule_AddIntConstant(m, "AF_UNIX", AF_UNIX);
Guido van Rossumb6775db1994-08-01 11:34:53 +00003815#endif /* AF_UNIX */
Guido van Rossum09be4091999-08-09 14:40:40 +00003816#ifdef AF_AX25
Fred Drake4baedc12002-04-01 14:53:37 +00003817 /* Amateur Radio AX.25 */
3818 PyModule_AddIntConstant(m, "AF_AX25", AF_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00003819#endif
3820#ifdef AF_IPX
Fred Drake4baedc12002-04-01 14:53:37 +00003821 PyModule_AddIntConstant(m, "AF_IPX", AF_IPX); /* Novell IPX */
Guido van Rossum09be4091999-08-09 14:40:40 +00003822#endif
3823#ifdef AF_APPLETALK
Fred Drake4baedc12002-04-01 14:53:37 +00003824 /* Appletalk DDP */
3825 PyModule_AddIntConstant(m, "AF_APPLETALK", AF_APPLETALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00003826#endif
3827#ifdef AF_NETROM
Fred Drake4baedc12002-04-01 14:53:37 +00003828 /* Amateur radio NetROM */
3829 PyModule_AddIntConstant(m, "AF_NETROM", AF_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00003830#endif
3831#ifdef AF_BRIDGE
Fred Drake4baedc12002-04-01 14:53:37 +00003832 /* Multiprotocol bridge */
3833 PyModule_AddIntConstant(m, "AF_BRIDGE", AF_BRIDGE);
Guido van Rossum09be4091999-08-09 14:40:40 +00003834#endif
3835#ifdef AF_AAL5
Fred Drake4baedc12002-04-01 14:53:37 +00003836 /* Reserved for Werner's ATM */
3837 PyModule_AddIntConstant(m, "AF_AAL5", AF_AAL5);
Guido van Rossum09be4091999-08-09 14:40:40 +00003838#endif
3839#ifdef AF_X25
Fred Drake4baedc12002-04-01 14:53:37 +00003840 /* Reserved for X.25 project */
3841 PyModule_AddIntConstant(m, "AF_X25", AF_X25);
Guido van Rossum09be4091999-08-09 14:40:40 +00003842#endif
3843#ifdef AF_INET6
Fred Drake4baedc12002-04-01 14:53:37 +00003844 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6); /* IP version 6 */
Guido van Rossum09be4091999-08-09 14:40:40 +00003845#endif
3846#ifdef AF_ROSE
Fred Drake4baedc12002-04-01 14:53:37 +00003847 /* Amateur Radio X.25 PLP */
3848 PyModule_AddIntConstant(m, "AF_ROSE", AF_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00003849#endif
Martin v. Löwis12af0482004-01-31 12:34:17 +00003850
Hye-Shik Chang81268602004-02-02 06:05:24 +00003851#ifdef USE_BLUETOOTH
Martin v. Löwis12af0482004-01-31 12:34:17 +00003852 PyModule_AddIntConstant(m, "AF_BLUETOOTH", AF_BLUETOOTH);
3853 PyModule_AddIntConstant(m, "BTPROTO_L2CAP", BTPROTO_L2CAP);
Hye-Shik Chang81268602004-02-02 06:05:24 +00003854#if !defined(__FreeBSD__)
Martin v. Löwis12af0482004-01-31 12:34:17 +00003855 PyModule_AddIntConstant(m, "BTPROTO_SCO", BTPROTO_SCO);
Hye-Shik Chang81268602004-02-02 06:05:24 +00003856#endif
Martin v. Löwis12af0482004-01-31 12:34:17 +00003857 PyModule_AddIntConstant(m, "BTPROTO_RFCOMM", BTPROTO_RFCOMM);
Martin v. Löwis558d9bf2004-06-03 09:24:42 +00003858 PyModule_AddObject(m, "BDADDR_ANY", Py_BuildValue("s", "00:00:00:00:00:00"));
3859 PyModule_AddObject(m, "BDADDR_LOCAL", Py_BuildValue("s", "00:00:00:FF:FF:FF"));
Martin v. Löwis12af0482004-01-31 12:34:17 +00003860#endif
3861
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00003862#ifdef HAVE_NETPACKET_PACKET_H
Fred Drake4baedc12002-04-01 14:53:37 +00003863 PyModule_AddIntConstant(m, "AF_PACKET", AF_PACKET);
3864 PyModule_AddIntConstant(m, "PF_PACKET", PF_PACKET);
3865 PyModule_AddIntConstant(m, "PACKET_HOST", PACKET_HOST);
3866 PyModule_AddIntConstant(m, "PACKET_BROADCAST", PACKET_BROADCAST);
3867 PyModule_AddIntConstant(m, "PACKET_MULTICAST", PACKET_MULTICAST);
3868 PyModule_AddIntConstant(m, "PACKET_OTHERHOST", PACKET_OTHERHOST);
3869 PyModule_AddIntConstant(m, "PACKET_OUTGOING", PACKET_OUTGOING);
3870 PyModule_AddIntConstant(m, "PACKET_LOOPBACK", PACKET_LOOPBACK);
3871 PyModule_AddIntConstant(m, "PACKET_FASTROUTE", PACKET_FASTROUTE);
Guido van Rossum48a680c2001-03-02 06:34:14 +00003872#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00003873
3874 /* Socket types */
Fred Drake4baedc12002-04-01 14:53:37 +00003875 PyModule_AddIntConstant(m, "SOCK_STREAM", SOCK_STREAM);
3876 PyModule_AddIntConstant(m, "SOCK_DGRAM", SOCK_DGRAM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00003877#ifndef __BEOS__
3878/* We have incomplete socket support. */
Fred Drake4baedc12002-04-01 14:53:37 +00003879 PyModule_AddIntConstant(m, "SOCK_RAW", SOCK_RAW);
3880 PyModule_AddIntConstant(m, "SOCK_SEQPACKET", SOCK_SEQPACKET);
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00003881#if defined(SOCK_RDM)
Fred Drake4baedc12002-04-01 14:53:37 +00003882 PyModule_AddIntConstant(m, "SOCK_RDM", SOCK_RDM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00003883#endif
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00003884#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003885
3886#ifdef SO_DEBUG
Fred Drake4baedc12002-04-01 14:53:37 +00003887 PyModule_AddIntConstant(m, "SO_DEBUG", SO_DEBUG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003888#endif
3889#ifdef SO_ACCEPTCONN
Fred Drake4baedc12002-04-01 14:53:37 +00003890 PyModule_AddIntConstant(m, "SO_ACCEPTCONN", SO_ACCEPTCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003891#endif
3892#ifdef SO_REUSEADDR
Fred Drake4baedc12002-04-01 14:53:37 +00003893 PyModule_AddIntConstant(m, "SO_REUSEADDR", SO_REUSEADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003894#endif
Andrew M. Kuchling42851ab2004-07-10 14:19:21 +00003895#ifdef SO_EXCLUSIVEADDRUSE
3896 PyModule_AddIntConstant(m, "SO_EXCLUSIVEADDRUSE", SO_EXCLUSIVEADDRUSE);
3897#endif
3898
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003899#ifdef SO_KEEPALIVE
Fred Drake4baedc12002-04-01 14:53:37 +00003900 PyModule_AddIntConstant(m, "SO_KEEPALIVE", SO_KEEPALIVE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003901#endif
3902#ifdef SO_DONTROUTE
Fred Drake4baedc12002-04-01 14:53:37 +00003903 PyModule_AddIntConstant(m, "SO_DONTROUTE", SO_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003904#endif
3905#ifdef SO_BROADCAST
Fred Drake4baedc12002-04-01 14:53:37 +00003906 PyModule_AddIntConstant(m, "SO_BROADCAST", SO_BROADCAST);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003907#endif
3908#ifdef SO_USELOOPBACK
Fred Drake4baedc12002-04-01 14:53:37 +00003909 PyModule_AddIntConstant(m, "SO_USELOOPBACK", SO_USELOOPBACK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003910#endif
3911#ifdef SO_LINGER
Fred Drake4baedc12002-04-01 14:53:37 +00003912 PyModule_AddIntConstant(m, "SO_LINGER", SO_LINGER);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003913#endif
3914#ifdef SO_OOBINLINE
Fred Drake4baedc12002-04-01 14:53:37 +00003915 PyModule_AddIntConstant(m, "SO_OOBINLINE", SO_OOBINLINE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003916#endif
3917#ifdef SO_REUSEPORT
Fred Drake4baedc12002-04-01 14:53:37 +00003918 PyModule_AddIntConstant(m, "SO_REUSEPORT", SO_REUSEPORT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003919#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003920#ifdef SO_SNDBUF
Fred Drake4baedc12002-04-01 14:53:37 +00003921 PyModule_AddIntConstant(m, "SO_SNDBUF", SO_SNDBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003922#endif
3923#ifdef SO_RCVBUF
Fred Drake4baedc12002-04-01 14:53:37 +00003924 PyModule_AddIntConstant(m, "SO_RCVBUF", SO_RCVBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003925#endif
3926#ifdef SO_SNDLOWAT
Fred Drake4baedc12002-04-01 14:53:37 +00003927 PyModule_AddIntConstant(m, "SO_SNDLOWAT", SO_SNDLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003928#endif
3929#ifdef SO_RCVLOWAT
Fred Drake4baedc12002-04-01 14:53:37 +00003930 PyModule_AddIntConstant(m, "SO_RCVLOWAT", SO_RCVLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003931#endif
3932#ifdef SO_SNDTIMEO
Fred Drake4baedc12002-04-01 14:53:37 +00003933 PyModule_AddIntConstant(m, "SO_SNDTIMEO", SO_SNDTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003934#endif
3935#ifdef SO_RCVTIMEO
Fred Drake4baedc12002-04-01 14:53:37 +00003936 PyModule_AddIntConstant(m, "SO_RCVTIMEO", SO_RCVTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003937#endif
3938#ifdef SO_ERROR
Fred Drake4baedc12002-04-01 14:53:37 +00003939 PyModule_AddIntConstant(m, "SO_ERROR", SO_ERROR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003940#endif
3941#ifdef SO_TYPE
Fred Drake4baedc12002-04-01 14:53:37 +00003942 PyModule_AddIntConstant(m, "SO_TYPE", SO_TYPE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003943#endif
3944
3945 /* Maximum number of connections for "listen" */
3946#ifdef SOMAXCONN
Fred Drake4baedc12002-04-01 14:53:37 +00003947 PyModule_AddIntConstant(m, "SOMAXCONN", SOMAXCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003948#else
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003949 PyModule_AddIntConstant(m, "SOMAXCONN", 5); /* Common value */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003950#endif
3951
3952 /* Flags for send, recv */
3953#ifdef MSG_OOB
Fred Drake4baedc12002-04-01 14:53:37 +00003954 PyModule_AddIntConstant(m, "MSG_OOB", MSG_OOB);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003955#endif
3956#ifdef MSG_PEEK
Fred Drake4baedc12002-04-01 14:53:37 +00003957 PyModule_AddIntConstant(m, "MSG_PEEK", MSG_PEEK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003958#endif
3959#ifdef MSG_DONTROUTE
Fred Drake4baedc12002-04-01 14:53:37 +00003960 PyModule_AddIntConstant(m, "MSG_DONTROUTE", MSG_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003961#endif
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00003962#ifdef MSG_DONTWAIT
Fred Drake4baedc12002-04-01 14:53:37 +00003963 PyModule_AddIntConstant(m, "MSG_DONTWAIT", MSG_DONTWAIT);
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00003964#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003965#ifdef MSG_EOR
Fred Drake4baedc12002-04-01 14:53:37 +00003966 PyModule_AddIntConstant(m, "MSG_EOR", MSG_EOR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003967#endif
3968#ifdef MSG_TRUNC
Fred Drake4baedc12002-04-01 14:53:37 +00003969 PyModule_AddIntConstant(m, "MSG_TRUNC", MSG_TRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003970#endif
3971#ifdef MSG_CTRUNC
Fred Drake4baedc12002-04-01 14:53:37 +00003972 PyModule_AddIntConstant(m, "MSG_CTRUNC", MSG_CTRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003973#endif
3974#ifdef MSG_WAITALL
Fred Drake4baedc12002-04-01 14:53:37 +00003975 PyModule_AddIntConstant(m, "MSG_WAITALL", MSG_WAITALL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003976#endif
3977#ifdef MSG_BTAG
Fred Drake4baedc12002-04-01 14:53:37 +00003978 PyModule_AddIntConstant(m, "MSG_BTAG", MSG_BTAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003979#endif
3980#ifdef MSG_ETAG
Fred Drake4baedc12002-04-01 14:53:37 +00003981 PyModule_AddIntConstant(m, "MSG_ETAG", MSG_ETAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003982#endif
3983
3984 /* Protocol level and numbers, usable for [gs]etsockopt */
3985#ifdef SOL_SOCKET
Fred Drake4baedc12002-04-01 14:53:37 +00003986 PyModule_AddIntConstant(m, "SOL_SOCKET", SOL_SOCKET);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003987#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00003988#ifdef SOL_IP
Fred Drake4baedc12002-04-01 14:53:37 +00003989 PyModule_AddIntConstant(m, "SOL_IP", SOL_IP);
Guido van Rossum09be4091999-08-09 14:40:40 +00003990#else
Fred Drake4baedc12002-04-01 14:53:37 +00003991 PyModule_AddIntConstant(m, "SOL_IP", 0);
Guido van Rossum09be4091999-08-09 14:40:40 +00003992#endif
3993#ifdef SOL_IPX
Fred Drake4baedc12002-04-01 14:53:37 +00003994 PyModule_AddIntConstant(m, "SOL_IPX", SOL_IPX);
Guido van Rossum09be4091999-08-09 14:40:40 +00003995#endif
3996#ifdef SOL_AX25
Fred Drake4baedc12002-04-01 14:53:37 +00003997 PyModule_AddIntConstant(m, "SOL_AX25", SOL_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00003998#endif
3999#ifdef SOL_ATALK
Fred Drake4baedc12002-04-01 14:53:37 +00004000 PyModule_AddIntConstant(m, "SOL_ATALK", SOL_ATALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00004001#endif
4002#ifdef SOL_NETROM
Fred Drake4baedc12002-04-01 14:53:37 +00004003 PyModule_AddIntConstant(m, "SOL_NETROM", SOL_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00004004#endif
4005#ifdef SOL_ROSE
Fred Drake4baedc12002-04-01 14:53:37 +00004006 PyModule_AddIntConstant(m, "SOL_ROSE", SOL_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004007#endif
4008#ifdef SOL_TCP
Fred Drake4baedc12002-04-01 14:53:37 +00004009 PyModule_AddIntConstant(m, "SOL_TCP", SOL_TCP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004010#else
Fred Drake4baedc12002-04-01 14:53:37 +00004011 PyModule_AddIntConstant(m, "SOL_TCP", 6);
Guido van Rossum09be4091999-08-09 14:40:40 +00004012#endif
4013#ifdef SOL_UDP
Fred Drake4baedc12002-04-01 14:53:37 +00004014 PyModule_AddIntConstant(m, "SOL_UDP", SOL_UDP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004015#else
Fred Drake4baedc12002-04-01 14:53:37 +00004016 PyModule_AddIntConstant(m, "SOL_UDP", 17);
Guido van Rossum09be4091999-08-09 14:40:40 +00004017#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004018#ifdef IPPROTO_IP
Fred Drake4baedc12002-04-01 14:53:37 +00004019 PyModule_AddIntConstant(m, "IPPROTO_IP", IPPROTO_IP);
Guido van Rossum578de301998-05-28 20:18:18 +00004020#else
Fred Drake4baedc12002-04-01 14:53:37 +00004021 PyModule_AddIntConstant(m, "IPPROTO_IP", 0);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004022#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004023#ifdef IPPROTO_HOPOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004024 PyModule_AddIntConstant(m, "IPPROTO_HOPOPTS", IPPROTO_HOPOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004025#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004026#ifdef IPPROTO_ICMP
Fred Drake4baedc12002-04-01 14:53:37 +00004027 PyModule_AddIntConstant(m, "IPPROTO_ICMP", IPPROTO_ICMP);
Guido van Rossum578de301998-05-28 20:18:18 +00004028#else
Fred Drake4baedc12002-04-01 14:53:37 +00004029 PyModule_AddIntConstant(m, "IPPROTO_ICMP", 1);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004030#endif
4031#ifdef IPPROTO_IGMP
Fred Drake4baedc12002-04-01 14:53:37 +00004032 PyModule_AddIntConstant(m, "IPPROTO_IGMP", IPPROTO_IGMP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004033#endif
4034#ifdef IPPROTO_GGP
Fred Drake4baedc12002-04-01 14:53:37 +00004035 PyModule_AddIntConstant(m, "IPPROTO_GGP", IPPROTO_GGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004036#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004037#ifdef IPPROTO_IPV4
Fred Drake4baedc12002-04-01 14:53:37 +00004038 PyModule_AddIntConstant(m, "IPPROTO_IPV4", IPPROTO_IPV4);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004039#endif
Martin v. Löwisa0f17342003-10-03 13:56:20 +00004040#ifdef IPPROTO_IPV6
4041 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
4042#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004043#ifdef IPPROTO_IPIP
Fred Drake4baedc12002-04-01 14:53:37 +00004044 PyModule_AddIntConstant(m, "IPPROTO_IPIP", IPPROTO_IPIP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004045#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004046#ifdef IPPROTO_TCP
Fred Drake4baedc12002-04-01 14:53:37 +00004047 PyModule_AddIntConstant(m, "IPPROTO_TCP", IPPROTO_TCP);
Guido van Rossum578de301998-05-28 20:18:18 +00004048#else
Fred Drake4baedc12002-04-01 14:53:37 +00004049 PyModule_AddIntConstant(m, "IPPROTO_TCP", 6);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004050#endif
4051#ifdef IPPROTO_EGP
Fred Drake4baedc12002-04-01 14:53:37 +00004052 PyModule_AddIntConstant(m, "IPPROTO_EGP", IPPROTO_EGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004053#endif
4054#ifdef IPPROTO_PUP
Fred Drake4baedc12002-04-01 14:53:37 +00004055 PyModule_AddIntConstant(m, "IPPROTO_PUP", IPPROTO_PUP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004056#endif
4057#ifdef IPPROTO_UDP
Fred Drake4baedc12002-04-01 14:53:37 +00004058 PyModule_AddIntConstant(m, "IPPROTO_UDP", IPPROTO_UDP);
Guido van Rossum578de301998-05-28 20:18:18 +00004059#else
Fred Drake4baedc12002-04-01 14:53:37 +00004060 PyModule_AddIntConstant(m, "IPPROTO_UDP", 17);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004061#endif
4062#ifdef IPPROTO_IDP
Fred Drake4baedc12002-04-01 14:53:37 +00004063 PyModule_AddIntConstant(m, "IPPROTO_IDP", IPPROTO_IDP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004064#endif
4065#ifdef IPPROTO_HELLO
Fred Drake4baedc12002-04-01 14:53:37 +00004066 PyModule_AddIntConstant(m, "IPPROTO_HELLO", IPPROTO_HELLO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004067#endif
4068#ifdef IPPROTO_ND
Fred Drake4baedc12002-04-01 14:53:37 +00004069 PyModule_AddIntConstant(m, "IPPROTO_ND", IPPROTO_ND);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004070#endif
4071#ifdef IPPROTO_TP
Fred Drake4baedc12002-04-01 14:53:37 +00004072 PyModule_AddIntConstant(m, "IPPROTO_TP", IPPROTO_TP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004073#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004074#ifdef IPPROTO_IPV6
Fred Drake4baedc12002-04-01 14:53:37 +00004075 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004076#endif
4077#ifdef IPPROTO_ROUTING
Fred Drake4baedc12002-04-01 14:53:37 +00004078 PyModule_AddIntConstant(m, "IPPROTO_ROUTING", IPPROTO_ROUTING);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004079#endif
4080#ifdef IPPROTO_FRAGMENT
Fred Drake4baedc12002-04-01 14:53:37 +00004081 PyModule_AddIntConstant(m, "IPPROTO_FRAGMENT", IPPROTO_FRAGMENT);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004082#endif
4083#ifdef IPPROTO_RSVP
Fred Drake4baedc12002-04-01 14:53:37 +00004084 PyModule_AddIntConstant(m, "IPPROTO_RSVP", IPPROTO_RSVP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004085#endif
4086#ifdef IPPROTO_GRE
Fred Drake4baedc12002-04-01 14:53:37 +00004087 PyModule_AddIntConstant(m, "IPPROTO_GRE", IPPROTO_GRE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004088#endif
4089#ifdef IPPROTO_ESP
Fred Drake4baedc12002-04-01 14:53:37 +00004090 PyModule_AddIntConstant(m, "IPPROTO_ESP", IPPROTO_ESP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004091#endif
4092#ifdef IPPROTO_AH
Fred Drake4baedc12002-04-01 14:53:37 +00004093 PyModule_AddIntConstant(m, "IPPROTO_AH", IPPROTO_AH);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004094#endif
4095#ifdef IPPROTO_MOBILE
Fred Drake4baedc12002-04-01 14:53:37 +00004096 PyModule_AddIntConstant(m, "IPPROTO_MOBILE", IPPROTO_MOBILE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004097#endif
4098#ifdef IPPROTO_ICMPV6
Fred Drake4baedc12002-04-01 14:53:37 +00004099 PyModule_AddIntConstant(m, "IPPROTO_ICMPV6", IPPROTO_ICMPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004100#endif
4101#ifdef IPPROTO_NONE
Fred Drake4baedc12002-04-01 14:53:37 +00004102 PyModule_AddIntConstant(m, "IPPROTO_NONE", IPPROTO_NONE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004103#endif
4104#ifdef IPPROTO_DSTOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004105 PyModule_AddIntConstant(m, "IPPROTO_DSTOPTS", IPPROTO_DSTOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004106#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004107#ifdef IPPROTO_XTP
Fred Drake4baedc12002-04-01 14:53:37 +00004108 PyModule_AddIntConstant(m, "IPPROTO_XTP", IPPROTO_XTP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004109#endif
4110#ifdef IPPROTO_EON
Fred Drake4baedc12002-04-01 14:53:37 +00004111 PyModule_AddIntConstant(m, "IPPROTO_EON", IPPROTO_EON);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004112#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004113#ifdef IPPROTO_PIM
Fred Drake4baedc12002-04-01 14:53:37 +00004114 PyModule_AddIntConstant(m, "IPPROTO_PIM", IPPROTO_PIM);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004115#endif
4116#ifdef IPPROTO_IPCOMP
Fred Drake4baedc12002-04-01 14:53:37 +00004117 PyModule_AddIntConstant(m, "IPPROTO_IPCOMP", IPPROTO_IPCOMP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004118#endif
4119#ifdef IPPROTO_VRRP
Fred Drake4baedc12002-04-01 14:53:37 +00004120 PyModule_AddIntConstant(m, "IPPROTO_VRRP", IPPROTO_VRRP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004121#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004122#ifdef IPPROTO_BIP
Fred Drake4baedc12002-04-01 14:53:37 +00004123 PyModule_AddIntConstant(m, "IPPROTO_BIP", IPPROTO_BIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004124#endif
4125/**/
4126#ifdef IPPROTO_RAW
Fred Drake4baedc12002-04-01 14:53:37 +00004127 PyModule_AddIntConstant(m, "IPPROTO_RAW", IPPROTO_RAW);
Guido van Rossum578de301998-05-28 20:18:18 +00004128#else
Fred Drake4baedc12002-04-01 14:53:37 +00004129 PyModule_AddIntConstant(m, "IPPROTO_RAW", 255);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004130#endif
4131#ifdef IPPROTO_MAX
Fred Drake4baedc12002-04-01 14:53:37 +00004132 PyModule_AddIntConstant(m, "IPPROTO_MAX", IPPROTO_MAX);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004133#endif
4134
4135 /* Some port configuration */
4136#ifdef IPPORT_RESERVED
Fred Drake4baedc12002-04-01 14:53:37 +00004137 PyModule_AddIntConstant(m, "IPPORT_RESERVED", IPPORT_RESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004138#else
Fred Drake4baedc12002-04-01 14:53:37 +00004139 PyModule_AddIntConstant(m, "IPPORT_RESERVED", 1024);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004140#endif
4141#ifdef IPPORT_USERRESERVED
Fred Drake4baedc12002-04-01 14:53:37 +00004142 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", IPPORT_USERRESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004143#else
Fred Drake4baedc12002-04-01 14:53:37 +00004144 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", 5000);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004145#endif
4146
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004147 /* Some reserved IP v.4 addresses */
4148#ifdef INADDR_ANY
Fred Drake4baedc12002-04-01 14:53:37 +00004149 PyModule_AddIntConstant(m, "INADDR_ANY", INADDR_ANY);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004150#else
Fred Drake4baedc12002-04-01 14:53:37 +00004151 PyModule_AddIntConstant(m, "INADDR_ANY", 0x00000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004152#endif
4153#ifdef INADDR_BROADCAST
Fred Drake4baedc12002-04-01 14:53:37 +00004154 PyModule_AddIntConstant(m, "INADDR_BROADCAST", INADDR_BROADCAST);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004155#else
Fred Drake4baedc12002-04-01 14:53:37 +00004156 PyModule_AddIntConstant(m, "INADDR_BROADCAST", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004157#endif
4158#ifdef INADDR_LOOPBACK
Fred Drake4baedc12002-04-01 14:53:37 +00004159 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", INADDR_LOOPBACK);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004160#else
Fred Drake4baedc12002-04-01 14:53:37 +00004161 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", 0x7F000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004162#endif
4163#ifdef INADDR_UNSPEC_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004164 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004165#else
Fred Drake4baedc12002-04-01 14:53:37 +00004166 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", 0xe0000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004167#endif
4168#ifdef INADDR_ALLHOSTS_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004169 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP",
4170 INADDR_ALLHOSTS_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004171#else
Fred Drake4baedc12002-04-01 14:53:37 +00004172 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP", 0xe0000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004173#endif
4174#ifdef INADDR_MAX_LOCAL_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004175 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP",
4176 INADDR_MAX_LOCAL_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004177#else
Fred Drake4baedc12002-04-01 14:53:37 +00004178 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004179#endif
4180#ifdef INADDR_NONE
Fred Drake4baedc12002-04-01 14:53:37 +00004181 PyModule_AddIntConstant(m, "INADDR_NONE", INADDR_NONE);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004182#else
Fred Drake4baedc12002-04-01 14:53:37 +00004183 PyModule_AddIntConstant(m, "INADDR_NONE", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00004184#endif
4185
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004186 /* IPv4 [gs]etsockopt options */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004187#ifdef IP_OPTIONS
Fred Drake4baedc12002-04-01 14:53:37 +00004188 PyModule_AddIntConstant(m, "IP_OPTIONS", IP_OPTIONS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004189#endif
4190#ifdef IP_HDRINCL
Fred Drake4baedc12002-04-01 14:53:37 +00004191 PyModule_AddIntConstant(m, "IP_HDRINCL", IP_HDRINCL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004192#endif
4193#ifdef IP_TOS
Fred Drake4baedc12002-04-01 14:53:37 +00004194 PyModule_AddIntConstant(m, "IP_TOS", IP_TOS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004195#endif
4196#ifdef IP_TTL
Fred Drake4baedc12002-04-01 14:53:37 +00004197 PyModule_AddIntConstant(m, "IP_TTL", IP_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004198#endif
4199#ifdef IP_RECVOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004200 PyModule_AddIntConstant(m, "IP_RECVOPTS", IP_RECVOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004201#endif
4202#ifdef IP_RECVRETOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004203 PyModule_AddIntConstant(m, "IP_RECVRETOPTS", IP_RECVRETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004204#endif
4205#ifdef IP_RECVDSTADDR
Fred Drake4baedc12002-04-01 14:53:37 +00004206 PyModule_AddIntConstant(m, "IP_RECVDSTADDR", IP_RECVDSTADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004207#endif
4208#ifdef IP_RETOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00004209 PyModule_AddIntConstant(m, "IP_RETOPTS", IP_RETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004210#endif
4211#ifdef IP_MULTICAST_IF
Fred Drake4baedc12002-04-01 14:53:37 +00004212 PyModule_AddIntConstant(m, "IP_MULTICAST_IF", IP_MULTICAST_IF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004213#endif
4214#ifdef IP_MULTICAST_TTL
Fred Drake4baedc12002-04-01 14:53:37 +00004215 PyModule_AddIntConstant(m, "IP_MULTICAST_TTL", IP_MULTICAST_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004216#endif
4217#ifdef IP_MULTICAST_LOOP
Fred Drake4baedc12002-04-01 14:53:37 +00004218 PyModule_AddIntConstant(m, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004219#endif
4220#ifdef IP_ADD_MEMBERSHIP
Fred Drake4baedc12002-04-01 14:53:37 +00004221 PyModule_AddIntConstant(m, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004222#endif
4223#ifdef IP_DROP_MEMBERSHIP
Fred Drake4baedc12002-04-01 14:53:37 +00004224 PyModule_AddIntConstant(m, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00004225#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00004226#ifdef IP_DEFAULT_MULTICAST_TTL
Fred Drake4baedc12002-04-01 14:53:37 +00004227 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_TTL",
4228 IP_DEFAULT_MULTICAST_TTL);
Guido van Rossum09be4091999-08-09 14:40:40 +00004229#endif
4230#ifdef IP_DEFAULT_MULTICAST_LOOP
Fred Drake4baedc12002-04-01 14:53:37 +00004231 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_LOOP",
4232 IP_DEFAULT_MULTICAST_LOOP);
Guido van Rossum09be4091999-08-09 14:40:40 +00004233#endif
4234#ifdef IP_MAX_MEMBERSHIPS
Fred Drake4baedc12002-04-01 14:53:37 +00004235 PyModule_AddIntConstant(m, "IP_MAX_MEMBERSHIPS", IP_MAX_MEMBERSHIPS);
Guido van Rossum09be4091999-08-09 14:40:40 +00004236#endif
4237
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004238 /* IPv6 [gs]etsockopt options, defined in RFC2553 */
4239#ifdef IPV6_JOIN_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004240 PyModule_AddIntConstant(m, "IPV6_JOIN_GROUP", IPV6_JOIN_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004241#endif
4242#ifdef IPV6_LEAVE_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00004243 PyModule_AddIntConstant(m, "IPV6_LEAVE_GROUP", IPV6_LEAVE_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004244#endif
4245#ifdef IPV6_MULTICAST_HOPS
Fred Drake4baedc12002-04-01 14:53:37 +00004246 PyModule_AddIntConstant(m, "IPV6_MULTICAST_HOPS", IPV6_MULTICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004247#endif
4248#ifdef IPV6_MULTICAST_IF
Fred Drake4baedc12002-04-01 14:53:37 +00004249 PyModule_AddIntConstant(m, "IPV6_MULTICAST_IF", IPV6_MULTICAST_IF);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004250#endif
4251#ifdef IPV6_MULTICAST_LOOP
Fred Drake4baedc12002-04-01 14:53:37 +00004252 PyModule_AddIntConstant(m, "IPV6_MULTICAST_LOOP", IPV6_MULTICAST_LOOP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004253#endif
4254#ifdef IPV6_UNICAST_HOPS
Fred Drake4baedc12002-04-01 14:53:37 +00004255 PyModule_AddIntConstant(m, "IPV6_UNICAST_HOPS", IPV6_UNICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004256#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00004257 /* Additional IPV6 socket options, defined in RFC 3493 */
4258#ifdef IPV6_V6ONLY
4259 PyModule_AddIntConstant(m, "IPV6_V6ONLY", IPV6_V6ONLY);
4260#endif
4261 /* Advanced IPV6 socket options, from RFC 3542 */
4262#ifdef IPV6_CHECKSUM
4263 PyModule_AddIntConstant(m, "IPV6_CHECKSUM", IPV6_CHECKSUM);
4264#endif
4265#ifdef IPV6_DONTFRAG
4266 PyModule_AddIntConstant(m, "IPV6_DONTFRAG", IPV6_DONTFRAG);
4267#endif
4268#ifdef IPV6_DSTOPTS
4269 PyModule_AddIntConstant(m, "IPV6_DSTOPTS", IPV6_DSTOPTS);
4270#endif
4271#ifdef IPV6_HOPLIMIT
4272 PyModule_AddIntConstant(m, "IPV6_HOPLIMIT", IPV6_HOPLIMIT);
4273#endif
4274#ifdef IPV6_HOPOPTS
4275 PyModule_AddIntConstant(m, "IPV6_HOPOPTS", IPV6_HOPOPTS);
4276#endif
4277#ifdef IPV6_NEXTHOP
4278 PyModule_AddIntConstant(m, "IPV6_NEXTHOP", IPV6_NEXTHOP);
4279#endif
4280#ifdef IPV6_PATHMTU
4281 PyModule_AddIntConstant(m, "IPV6_PATHMTU", IPV6_PATHMTU);
4282#endif
4283#ifdef IPV6_PKTINFO
4284 PyModule_AddIntConstant(m, "IPV6_PKTINFO", IPV6_PKTINFO);
4285#endif
4286#ifdef IPV6_RECVDSTOPTS
4287 PyModule_AddIntConstant(m, "IPV6_RECVDSTOPTS", IPV6_RECVDSTOPTS);
4288#endif
4289#ifdef IPV6_RECVHOPLIMIT
4290 PyModule_AddIntConstant(m, "IPV6_RECVHOPLIMIT", IPV6_RECVHOPLIMIT);
4291#endif
4292#ifdef IPV6_RECVHOPOPTS
4293 PyModule_AddIntConstant(m, "IPV6_RECVHOPOPTS", IPV6_RECVHOPOPTS);
4294#endif
4295#ifdef IPV6_RECVPKTINFO
4296 PyModule_AddIntConstant(m, "IPV6_RECVPKTINFO", IPV6_RECVPKTINFO);
4297#endif
4298#ifdef IPV6_RECVRTHDR
4299 PyModule_AddIntConstant(m, "IPV6_RECVRTHDR", IPV6_RECVRTHDR);
4300#endif
4301#ifdef IPV6_RECVTCLASS
4302 PyModule_AddIntConstant(m, "IPV6_RECVTCLASS", IPV6_RECVTCLASS);
4303#endif
4304#ifdef IPV6_RTHDR
4305 PyModule_AddIntConstant(m, "IPV6_RTHDR", IPV6_RTHDR);
4306#endif
4307#ifdef IPV6_RTHDRDSTOPTS
4308 PyModule_AddIntConstant(m, "IPV6_RTHDRDSTOPTS", IPV6_RTHDRDSTOPTS);
4309#endif
4310#ifdef IPV6_RTHDR_TYPE_0
4311 PyModule_AddIntConstant(m, "IPV6_RTHDR_TYPE_0", IPV6_RTHDR_TYPE_0);
4312#endif
4313#ifdef IPV6_RECVPATHMTU
4314 PyModule_AddIntConstant(m, "IPV6_RECVPATHMTU", IPV6_RECVPATHMTU);
4315#endif
4316#ifdef IPV6_TCLASS
4317 PyModule_AddIntConstant(m, "IPV6_TCLASS", IPV6_TCLASS);
4318#endif
4319#ifdef IPV6_USE_MIN_MTU
4320 PyModule_AddIntConstant(m, "IPV6_USE_MIN_MTU", IPV6_USE_MIN_MTU);
4321#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00004322
Guido van Rossum09be4091999-08-09 14:40:40 +00004323 /* TCP options */
4324#ifdef TCP_NODELAY
Fred Drake4baedc12002-04-01 14:53:37 +00004325 PyModule_AddIntConstant(m, "TCP_NODELAY", TCP_NODELAY);
Guido van Rossum09be4091999-08-09 14:40:40 +00004326#endif
4327#ifdef TCP_MAXSEG
Fred Drake4baedc12002-04-01 14:53:37 +00004328 PyModule_AddIntConstant(m, "TCP_MAXSEG", TCP_MAXSEG);
Guido van Rossum09be4091999-08-09 14:40:40 +00004329#endif
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004330#ifdef TCP_CORK
Fred Drake4baedc12002-04-01 14:53:37 +00004331 PyModule_AddIntConstant(m, "TCP_CORK", TCP_CORK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004332#endif
4333#ifdef TCP_KEEPIDLE
Fred Drake4baedc12002-04-01 14:53:37 +00004334 PyModule_AddIntConstant(m, "TCP_KEEPIDLE", TCP_KEEPIDLE);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004335#endif
4336#ifdef TCP_KEEPINTVL
Fred Drake4baedc12002-04-01 14:53:37 +00004337 PyModule_AddIntConstant(m, "TCP_KEEPINTVL", TCP_KEEPINTVL);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004338#endif
4339#ifdef TCP_KEEPCNT
Fred Drake4baedc12002-04-01 14:53:37 +00004340 PyModule_AddIntConstant(m, "TCP_KEEPCNT", TCP_KEEPCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004341#endif
4342#ifdef TCP_SYNCNT
Fred Drake4baedc12002-04-01 14:53:37 +00004343 PyModule_AddIntConstant(m, "TCP_SYNCNT", TCP_SYNCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004344#endif
4345#ifdef TCP_LINGER2
Fred Drake4baedc12002-04-01 14:53:37 +00004346 PyModule_AddIntConstant(m, "TCP_LINGER2", TCP_LINGER2);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004347#endif
4348#ifdef TCP_DEFER_ACCEPT
Fred Drake4baedc12002-04-01 14:53:37 +00004349 PyModule_AddIntConstant(m, "TCP_DEFER_ACCEPT", TCP_DEFER_ACCEPT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004350#endif
4351#ifdef TCP_WINDOW_CLAMP
Fred Drake4baedc12002-04-01 14:53:37 +00004352 PyModule_AddIntConstant(m, "TCP_WINDOW_CLAMP", TCP_WINDOW_CLAMP);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004353#endif
4354#ifdef TCP_INFO
Fred Drake4baedc12002-04-01 14:53:37 +00004355 PyModule_AddIntConstant(m, "TCP_INFO", TCP_INFO);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004356#endif
4357#ifdef TCP_QUICKACK
Fred Drake4baedc12002-04-01 14:53:37 +00004358 PyModule_AddIntConstant(m, "TCP_QUICKACK", TCP_QUICKACK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00004359#endif
4360
Guido van Rossum09be4091999-08-09 14:40:40 +00004361
4362 /* IPX options */
4363#ifdef IPX_TYPE
Fred Drake4baedc12002-04-01 14:53:37 +00004364 PyModule_AddIntConstant(m, "IPX_TYPE", IPX_TYPE);
Guido van Rossum09be4091999-08-09 14:40:40 +00004365#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00004366
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004367 /* get{addr,name}info parameters */
4368#ifdef EAI_ADDRFAMILY
Fred Drake4baedc12002-04-01 14:53:37 +00004369 PyModule_AddIntConstant(m, "EAI_ADDRFAMILY", EAI_ADDRFAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004370#endif
4371#ifdef EAI_AGAIN
Fred Drake4baedc12002-04-01 14:53:37 +00004372 PyModule_AddIntConstant(m, "EAI_AGAIN", EAI_AGAIN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004373#endif
4374#ifdef EAI_BADFLAGS
Fred Drake4baedc12002-04-01 14:53:37 +00004375 PyModule_AddIntConstant(m, "EAI_BADFLAGS", EAI_BADFLAGS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004376#endif
4377#ifdef EAI_FAIL
Fred Drake4baedc12002-04-01 14:53:37 +00004378 PyModule_AddIntConstant(m, "EAI_FAIL", EAI_FAIL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004379#endif
4380#ifdef EAI_FAMILY
Fred Drake4baedc12002-04-01 14:53:37 +00004381 PyModule_AddIntConstant(m, "EAI_FAMILY", EAI_FAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004382#endif
4383#ifdef EAI_MEMORY
Fred Drake4baedc12002-04-01 14:53:37 +00004384 PyModule_AddIntConstant(m, "EAI_MEMORY", EAI_MEMORY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004385#endif
4386#ifdef EAI_NODATA
Fred Drake4baedc12002-04-01 14:53:37 +00004387 PyModule_AddIntConstant(m, "EAI_NODATA", EAI_NODATA);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004388#endif
4389#ifdef EAI_NONAME
Fred Drake4baedc12002-04-01 14:53:37 +00004390 PyModule_AddIntConstant(m, "EAI_NONAME", EAI_NONAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004391#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00004392#ifdef EAI_OVERFLOW
4393 PyModule_AddIntConstant(m, "EAI_OVERFLOW", EAI_OVERFLOW);
4394#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004395#ifdef EAI_SERVICE
Fred Drake4baedc12002-04-01 14:53:37 +00004396 PyModule_AddIntConstant(m, "EAI_SERVICE", EAI_SERVICE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004397#endif
4398#ifdef EAI_SOCKTYPE
Fred Drake4baedc12002-04-01 14:53:37 +00004399 PyModule_AddIntConstant(m, "EAI_SOCKTYPE", EAI_SOCKTYPE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004400#endif
4401#ifdef EAI_SYSTEM
Fred Drake4baedc12002-04-01 14:53:37 +00004402 PyModule_AddIntConstant(m, "EAI_SYSTEM", EAI_SYSTEM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004403#endif
4404#ifdef EAI_BADHINTS
Fred Drake4baedc12002-04-01 14:53:37 +00004405 PyModule_AddIntConstant(m, "EAI_BADHINTS", EAI_BADHINTS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004406#endif
4407#ifdef EAI_PROTOCOL
Fred Drake4baedc12002-04-01 14:53:37 +00004408 PyModule_AddIntConstant(m, "EAI_PROTOCOL", EAI_PROTOCOL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004409#endif
4410#ifdef EAI_MAX
Fred Drake4baedc12002-04-01 14:53:37 +00004411 PyModule_AddIntConstant(m, "EAI_MAX", EAI_MAX);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004412#endif
4413#ifdef AI_PASSIVE
Fred Drake4baedc12002-04-01 14:53:37 +00004414 PyModule_AddIntConstant(m, "AI_PASSIVE", AI_PASSIVE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004415#endif
4416#ifdef AI_CANONNAME
Fred Drake4baedc12002-04-01 14:53:37 +00004417 PyModule_AddIntConstant(m, "AI_CANONNAME", AI_CANONNAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004418#endif
4419#ifdef AI_NUMERICHOST
Fred Drake4baedc12002-04-01 14:53:37 +00004420 PyModule_AddIntConstant(m, "AI_NUMERICHOST", AI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004421#endif
Martin v. Löwisda91d022003-12-30 11:14:01 +00004422#ifdef AI_NUMERICSERV
4423 PyModule_AddIntConstant(m, "AI_NUMERICSERV", AI_NUMERICSERV);
4424#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004425#ifdef AI_MASK
Fred Drake4baedc12002-04-01 14:53:37 +00004426 PyModule_AddIntConstant(m, "AI_MASK", AI_MASK);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004427#endif
4428#ifdef AI_ALL
Fred Drake4baedc12002-04-01 14:53:37 +00004429 PyModule_AddIntConstant(m, "AI_ALL", AI_ALL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004430#endif
4431#ifdef AI_V4MAPPED_CFG
Fred Drake4baedc12002-04-01 14:53:37 +00004432 PyModule_AddIntConstant(m, "AI_V4MAPPED_CFG", AI_V4MAPPED_CFG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004433#endif
4434#ifdef AI_ADDRCONFIG
Fred Drake4baedc12002-04-01 14:53:37 +00004435 PyModule_AddIntConstant(m, "AI_ADDRCONFIG", AI_ADDRCONFIG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004436#endif
4437#ifdef AI_V4MAPPED
Fred Drake4baedc12002-04-01 14:53:37 +00004438 PyModule_AddIntConstant(m, "AI_V4MAPPED", AI_V4MAPPED);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004439#endif
4440#ifdef AI_DEFAULT
Fred Drake4baedc12002-04-01 14:53:37 +00004441 PyModule_AddIntConstant(m, "AI_DEFAULT", AI_DEFAULT);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004442#endif
4443#ifdef NI_MAXHOST
Fred Drake4baedc12002-04-01 14:53:37 +00004444 PyModule_AddIntConstant(m, "NI_MAXHOST", NI_MAXHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004445#endif
4446#ifdef NI_MAXSERV
Fred Drake4baedc12002-04-01 14:53:37 +00004447 PyModule_AddIntConstant(m, "NI_MAXSERV", NI_MAXSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004448#endif
4449#ifdef NI_NOFQDN
Fred Drake4baedc12002-04-01 14:53:37 +00004450 PyModule_AddIntConstant(m, "NI_NOFQDN", NI_NOFQDN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004451#endif
4452#ifdef NI_NUMERICHOST
Fred Drake4baedc12002-04-01 14:53:37 +00004453 PyModule_AddIntConstant(m, "NI_NUMERICHOST", NI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004454#endif
4455#ifdef NI_NAMEREQD
Fred Drake4baedc12002-04-01 14:53:37 +00004456 PyModule_AddIntConstant(m, "NI_NAMEREQD", NI_NAMEREQD);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004457#endif
4458#ifdef NI_NUMERICSERV
Fred Drake4baedc12002-04-01 14:53:37 +00004459 PyModule_AddIntConstant(m, "NI_NUMERICSERV", NI_NUMERICSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004460#endif
4461#ifdef NI_DGRAM
Fred Drake4baedc12002-04-01 14:53:37 +00004462 PyModule_AddIntConstant(m, "NI_DGRAM", NI_DGRAM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00004463#endif
4464
Martin v. Löwis94681fc2003-11-27 19:40:22 +00004465 /* shutdown() parameters */
4466#ifdef SHUT_RD
4467 PyModule_AddIntConstant(m, "SHUT_RD", SHUT_RD);
4468#elif defined(SD_RECEIVE)
4469 PyModule_AddIntConstant(m, "SHUT_RD", SD_RECEIVE);
4470#else
4471 PyModule_AddIntConstant(m, "SHUT_RD", 0);
4472#endif
4473#ifdef SHUT_WR
4474 PyModule_AddIntConstant(m, "SHUT_WR", SHUT_WR);
4475#elif defined(SD_SEND)
4476 PyModule_AddIntConstant(m, "SHUT_WR", SD_SEND);
4477#else
4478 PyModule_AddIntConstant(m, "SHUT_WR", 1);
4479#endif
4480#ifdef SHUT_RDWR
4481 PyModule_AddIntConstant(m, "SHUT_RDWR", SHUT_RDWR);
4482#elif defined(SD_BOTH)
4483 PyModule_AddIntConstant(m, "SHUT_RDWR", SD_BOTH);
4484#else
4485 PyModule_AddIntConstant(m, "SHUT_RDWR", 2);
4486#endif
4487
Guido van Rossum4f199ea1998-04-09 20:56:35 +00004488 /* Initialize gethostbyname lock */
Just van Rossum1040d2c2003-05-09 07:53:18 +00004489#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
4490 netdb_lock = PyThread_allocate_lock();
Guido van Rossum4f199ea1998-04-09 20:56:35 +00004491#endif
Guido van Rossum6574b3e1991-06-25 21:36:08 +00004492}
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00004493
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004494
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00004495#ifndef HAVE_INET_PTON
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004496
4497/* Simplistic emulation code for inet_pton that only works for IPv4 */
Guido van Rossum47dfa4a2003-04-25 05:48:32 +00004498/* These are not exposed because they do not set errno properly */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004499
Guido van Rossum3eede5a2002-06-07 02:08:35 +00004500int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004501inet_pton(int af, const char *src, void *dst)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00004502{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004503 if (af == AF_INET) {
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00004504 long packed_addr;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00004505 packed_addr = inet_addr(src);
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00004506 if (packed_addr == INADDR_NONE)
4507 return 0;
4508 memcpy(dst, &packed_addr, 4);
4509 return 1;
4510 }
4511 /* Should set errno to EAFNOSUPPORT */
4512 return -1;
4513}
4514
Martin v. Löwisc925b1532001-07-21 09:42:15 +00004515const char *
4516inet_ntop(int af, const void *src, char *dst, socklen_t size)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00004517{
4518 if (af == AF_INET) {
4519 struct in_addr packed_addr;
4520 if (size < 16)
4521 /* Should set errno to ENOSPC. */
4522 return NULL;
4523 memcpy(&packed_addr, src, sizeof(packed_addr));
4524 return strncpy(dst, inet_ntoa(packed_addr), size);
4525 }
4526 /* Should set errno to EAFNOSUPPORT */
4527 return NULL;
4528}
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00004529
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00004530#endif