blob: f7bbeae24e05d378d3f7c49ab1ebad0ba9da53d3 [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
Guido van Rossum27e177d1995-03-16 15:43:47 +000028- socket.getservbyname(servicename, protocolname) --> port number
Guido van Rossum83a072d2002-09-03 19:10:18 +000029- socket.socket([family[, type [, proto]]]) --> new socket object
Guido van Rossum006bf911996-06-12 04:04:55 +000030- socket.ntohs(16 bit value) --> new int object
31- socket.ntohl(32 bit value) --> new int object
32- socket.htons(16 bit value) --> new int object
33- socket.htonl(32 bit value) --> new int object
Martin v. Löwis2d8d4272001-07-21 18:05:31 +000034- socket.getaddrinfo(host, port [, family, socktype, proto, flags])
35 --> List of (family, socktype, proto, canonname, sockaddr)
36- socket.getnameinfo(sockaddr, flags) --> (host, port)
Guido van Rossum27e177d1995-03-16 15:43:47 +000037- socket.AF_INET, socket.SOCK_STREAM, etc.: constants from <socket.h>
Guido van Rossum5c9eb211999-08-20 18:21:51 +000038- socket.inet_aton(IP address) -> 32-bit packed IP representation
39- socket.inet_ntoa(packed IP) -> IP address string
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +000040- socket.getdefaulttimeout() -> None | float
41- socket.setdefaulttimeout(None | float)
Guido van Rossum6574b3e1991-06-25 21:36:08 +000042- an Internet socket address is a pair (hostname, port)
43 where hostname can be anything recognized by gethostbyname()
44 (including the dd.dd.dd.dd notation) and port is in host byte order
45- where a hostname is returned, the dd.dd.dd.dd notation is used
Guido van Rossum27e177d1995-03-16 15:43:47 +000046- a UNIX domain socket address is a string specifying the pathname
Jeremy Hylton22308652001-02-02 03:23:09 +000047- an AF_PACKET socket address is a tuple containing a string
48 specifying the ethernet interface and an integer specifying
49 the Ethernet protocol number to be received. For example:
Jeremy Hyltondbfb6622001-02-02 19:55:17 +000050 ("eth0",0x1234). Optional 3rd,4th,5th elements in the tuple
51 specify packet-type and ha-type/addr -- these are ignored by
Jeremy Hylton22308652001-02-02 03:23:09 +000052 networking code, but accepted since they are returned by the
53 getsockname() method.
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
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000063#include "Python.h"
64
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000065/* Socket object documentation */
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000066PyDoc_STRVAR(sock_doc,
Guido van Rossumc4fcfa32002-06-07 03:19:37 +000067"socket([family[, type[, proto]]]) -> socket object\n\
68\n\
69Open a socket of the given type. The family argument specifies the\n\
70address family; it defaults to AF_INET. The type argument specifies\n\
71whether this is a stream (SOCK_STREAM, this is the default)\n\
72or datagram (SOCK_DGRAM) socket. The protocol argument defaults to 0,\n\
73specifying the default protocol. Keyword arguments are accepted.\n\
74\n\
75A socket object represents one endpoint of a network connection.\n\
76\n\
77Methods of socket objects (keyword arguments not allowed):\n\
78\n\
79accept() -- accept a connection, returning new socket and client address\n\
80bind(addr) -- bind the socket to a local address\n\
81close() -- close the socket\n\
82connect(addr) -- connect the socket to a remote address\n\
83connect_ex(addr) -- connect, return an error code instead of an exception\n\
84dup() -- return a new socket object identical to the current one [*]\n\
85fileno() -- return underlying file descriptor\n\
86getpeername() -- return remote address [*]\n\
87getsockname() -- return local address\n\
88getsockopt(level, optname[, buflen]) -- get socket options\n\
89gettimeout() -- return timeout or None\n\
90listen(n) -- start listening for incoming connections\n\
91makefile([mode, [bufsize]]) -- return a file object for the socket [*]\n\
92recv(buflen[, flags]) -- receive data\n\
93recvfrom(buflen[, flags]) -- receive data and sender's address\n\
94sendall(data[, flags]) -- send all data\n\
95send(data[, flags]) -- send data, may not send all of it\n\
96sendto(data[, flags], addr) -- send data to a given address\n\
97setblocking(0 | 1) -- set or clear the blocking I/O flag\n\
98setsockopt(level, optname, value) -- set socket options\n\
99settimeout(None | float) -- set or clear the timeout\n\
100shutdown(how) -- shut down traffic in one or both directions\n\
101\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000102 [*] not available on all platforms!");
Guido van Rossum3baaa131999-03-22 21:44:51 +0000103
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000104/* XXX This is a terrible mess of of platform-dependent preprocessor hacks.
105 I hope some day someone can clean this up please... */
106
Guido van Rossum9376b741999-09-15 22:01:40 +0000107/* Hacks for gethostbyname_r(). On some non-Linux platforms, the configure
108 script doesn't get this right, so we hardcode some platform checks below.
109 On the other hand, not all Linux versions agree, so there the settings
110 computed by the configure script are needed! */
111
112#ifndef linux
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000113# undef HAVE_GETHOSTBYNAME_R_3_ARG
114# undef HAVE_GETHOSTBYNAME_R_5_ARG
115# undef HAVE_GETHOSTBYNAME_R_6_ARG
Guido van Rossum9376b741999-09-15 22:01:40 +0000116#endif
Guido van Rossume7de2061999-03-24 17:24:33 +0000117
Guido van Rossum7a122991999-04-13 04:07:32 +0000118#ifndef WITH_THREAD
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000119# undef HAVE_GETHOSTBYNAME_R
Guido van Rossum7a122991999-04-13 04:07:32 +0000120#endif
121
Guido van Rossume7de2061999-03-24 17:24:33 +0000122#ifdef HAVE_GETHOSTBYNAME_R
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000123# if defined(_AIX) || defined(__osf__)
124# define HAVE_GETHOSTBYNAME_R_3_ARG
125# elif defined(__sun) || defined(__sgi)
126# define HAVE_GETHOSTBYNAME_R_5_ARG
127# elif defined(linux)
Guido van Rossum9376b741999-09-15 22:01:40 +0000128/* Rely on the configure script */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000129# else
130# undef HAVE_GETHOSTBYNAME_R
131# endif
Guido van Rossume7de2061999-03-24 17:24:33 +0000132#endif
133
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000134#if !defined(HAVE_GETHOSTBYNAME_R) && defined(WITH_THREAD) && \
135 !defined(MS_WINDOWS)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000136# define USE_GETHOSTBYNAME_LOCK
Guido van Rossum3baaa131999-03-22 21:44:51 +0000137#endif
138
139#ifdef USE_GETHOSTBYNAME_LOCK
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000140# include "pythread.h"
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000141#endif
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000142
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000143#if defined(PYCC_VACPP)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000144# include <types.h>
145# include <io.h>
146# include <sys/ioctl.h>
147# include <utils.h>
148# include <ctype.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000149#endif
150
Martin v. Löwis9e437302002-12-06 12:57:26 +0000151#if defined(__VMS)
152#if ! defined(_SOCKADDR_LEN)
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000153# ifdef getaddrinfo
154# undef getaddrinfo
155# endif
156# include "TCPIP_IOCTL_ROUTINE"
157#else
158# include <ioctl.h>
159#endif
Martin v. Löwis9e437302002-12-06 12:57:26 +0000160#endif
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000161
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000162#if defined(PYOS_OS2)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000163# define INCL_DOS
164# define INCL_DOSERRORS
165# define INCL_NOPMAPI
166# include <os2.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000167#endif
168
Martin v. Löwis8eb92a02002-09-19 08:03:21 +0000169#if defined(__sgi)&&_COMPILER_VERSION>700 && !_SGIAPI
170/* make sure that the reentrant (gethostbyaddr_r etc)
171 functions are declared correctly if compiling with
172 MIPSPro 7.x in ANSI C mode (default) */
173#define _SGIAPI 1
174#include "netdb.h"
175#endif
176
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000177/* Generic includes */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000178#include <sys/types.h>
Guido van Rossum81194471991-07-27 21:42:02 +0000179#include <signal.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000180
Marc-André Lemburg976ade62002-02-16 18:47:07 +0000181/* Generic socket object definitions and includes */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000182#define PySocket_BUILDING_SOCKET
Marc-André Lemburgbb8b78b2002-02-16 18:44:52 +0000183#include "socketmodule.h"
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000184
185/* Addressing includes */
186
Guido van Rossum6f489d91996-06-28 20:15:15 +0000187#ifndef MS_WINDOWS
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000188
189/* Non-MS WINDOWS includes */
190# include <netdb.h>
Guido van Rossum5c9eb211999-08-20 18:21:51 +0000191
Guido van Rossum9376b741999-09-15 22:01:40 +0000192/* Headers needed for inet_ntoa() and inet_addr() */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000193# ifdef __BEOS__
194# include <net/netdb.h>
195# elif defined(PYOS_OS2) && defined(PYCC_VACPP)
196# include <netdb.h>
Tim Peters603c6832001-11-05 02:45:59 +0000197typedef size_t socklen_t;
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000198# else
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000199# include <arpa/inet.h>
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000200# endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +0000201
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000202# ifndef RISCOS
203# include <fcntl.h>
204# else
205# include <sys/fcntl.h>
206# define NO_DUP
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000207int h_errno; /* not used */
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000208# endif
209
Guido van Rossume2ae77b2001-10-24 20:42:55 +0000210#else
Guido van Rossum48a680c2001-03-02 06:34:14 +0000211
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000212/* MS_WINDOWS includes */
213# include <fcntl.h>
Guido van Rossum48a680c2001-03-02 06:34:14 +0000214
Jeremy Hylton22308652001-02-02 03:23:09 +0000215#endif
216
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000217#ifdef HAVE_STDDEF_H
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000218# include <stddef.h>
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000219#endif
220
221#ifndef offsetof
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000222# define offsetof(type, member) ((size_t)(&((type *)0)->member))
Martin v. Löwisa45ecae2001-06-24 21:28:42 +0000223#endif
224
Neal Norwitz39d22e52002-11-02 19:55:21 +0000225#ifndef O_NONBLOCK
226# define O_NONBLOCK O_NDELAY
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000227#endif
228
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000229#include "addrinfo.h"
230
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000231#ifndef HAVE_INET_PTON
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000232int inet_pton(int af, const char *src, void *dst);
Martin v. Löwisc925b1532001-07-21 09:42:15 +0000233const char *inet_ntop(int af, const void *src, char *dst, socklen_t size);
Martin v. Löwisb9ab1592001-06-24 21:18:26 +0000234#endif
235
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000236#ifdef __APPLE__
237/* On OS X, getaddrinfo returns no error indication of lookup
238 failure, so we must use the emulation instead of the libinfo
239 implementation. Unfortunately, performing an autoconf test
240 for this bug would require DNS access for the machine performing
241 the configuration, which is not acceptable. Therefore, we
242 determine the bug just by checking for __APPLE__. If this bug
243 gets ever fixed, perhaps checking for sys/version.h would be
244 appropriate, which is 10/0 on the system with the bug. */
Jack Jansen84262fb2002-07-02 14:40:42 +0000245#ifndef HAVE_GETNAMEINFO
246/* This bug seems to be fixed in Jaguar. Ths easiest way I could
247 Find to check for Jaguar is that it has getnameinfo(), which
248 older releases don't have */
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000249#undef HAVE_GETADDRINFO
250/* avoid clashes with the C library definition of the symbol. */
251#define getaddrinfo fake_getaddrinfo
252#endif
Jack Jansen84262fb2002-07-02 14:40:42 +0000253#endif
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000254
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000255/* I know this is a bad practice, but it is the easiest... */
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000256#if !defined(HAVE_GETADDRINFO)
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000257#include "getaddrinfo.c"
258#endif
Martin v. Löwisae26dc22001-08-03 10:02:29 +0000259#if !defined(HAVE_GETNAMEINFO)
Martin v. Löwis01dfdb32001-06-23 16:30:13 +0000260#include "getnameinfo.c"
261#endif
262
Guido van Rossumbcc20741998-08-04 22:53:56 +0000263#if defined(MS_WINDOWS) || defined(__BEOS__)
264/* BeOS suffers from the same socket dichotomy as Win32... - [cjh] */
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000265/* seem to be a few differences in the API */
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000266#define SOCKETCLOSE closesocket
Guido van Rossumbe32c891996-06-20 16:25:29 +0000267#define NO_DUP /* Actually it exists on NT 3.5, but what the heck... */
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +0000268#endif
269
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000270#ifdef MS_WIN32
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000271#define EAFNOSUPPORT WSAEAFNOSUPPORT
272#define snprintf _snprintf
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000273#endif
Fred Drakea04eaad2000-06-30 02:46:07 +0000274
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000275#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000276#define SOCKETCLOSE soclose
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000277#define NO_DUP /* Sockets are Not Actual File Handles under OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000278#endif
279
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +0000280#ifndef SOCKETCLOSE
281#define SOCKETCLOSE close
282#endif
283
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000284#ifdef __VMS
285/* TCP/IP Services for VMS uses a maximum send/revc buffer length of 65535 */
286#define SEGMENT_SIZE 65535
287#endif
288
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000289/* XXX There's a problem here: *static* functions are not supposed to have
290 a Py prefix (or use CapitalizedWords). Later... */
291
Guido van Rossum30a685f1991-06-27 15:51:29 +0000292/* Global variable holding the exception type for errors detected
293 by this module (but not argument type or memory errors, etc.). */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000294static PyObject *socket_error;
295static PyObject *socket_herror;
296static PyObject *socket_gaierror;
Guido van Rossum30a685f1991-06-27 15:51:29 +0000297
Guido van Rossum48a680c2001-03-02 06:34:14 +0000298#ifdef RISCOS
299/* Global variable which is !=0 if Python is running in a RISC OS taskwindow */
300static int taskwindow;
301#endif
302
Tim Peters643a7fc2002-02-17 04:13:21 +0000303/* A forward reference to the socket type object.
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000304 The sock_type variable contains pointers to various functions,
305 some of which call new_sockobject(), which uses sock_type, so
Tim Peters643a7fc2002-02-17 04:13:21 +0000306 there has to be a circular reference. */
Jeremy Hylton938ace62002-07-17 16:30:39 +0000307static PyTypeObject sock_type;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000308
Guido van Rossum30a685f1991-06-27 15:51:29 +0000309/* Convenience function to raise an error according to errno
310 and return a NULL pointer from a function. */
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000311
Guido van Rossum73624e91994-10-10 17:59:00 +0000312static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000313set_error(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000314{
Guido van Rossum8d665e61996-06-26 18:22:49 +0000315#ifdef MS_WINDOWS
Mark Hammond46a733d2000-07-24 01:45:11 +0000316 int err_no = WSAGetLastError();
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000317 static struct {
318 int no;
319 const char *msg;
320 } *msgp, msgs[] = {
321 {WSAEINTR, "Interrupted system call"},
322 {WSAEBADF, "Bad file descriptor"},
323 {WSAEACCES, "Permission denied"},
324 {WSAEFAULT, "Bad address"},
325 {WSAEINVAL, "Invalid argument"},
326 {WSAEMFILE, "Too many open files"},
327 {WSAEWOULDBLOCK,
328 "The socket operation could not complete "
329 "without blocking"},
330 {WSAEINPROGRESS, "Operation now in progress"},
331 {WSAEALREADY, "Operation already in progress"},
332 {WSAENOTSOCK, "Socket operation on non-socket"},
333 {WSAEDESTADDRREQ, "Destination address required"},
334 {WSAEMSGSIZE, "Message too long"},
335 {WSAEPROTOTYPE, "Protocol wrong type for socket"},
336 {WSAENOPROTOOPT, "Protocol not available"},
337 {WSAEPROTONOSUPPORT, "Protocol not supported"},
338 {WSAESOCKTNOSUPPORT, "Socket type not supported"},
339 {WSAEOPNOTSUPP, "Operation not supported"},
340 {WSAEPFNOSUPPORT, "Protocol family not supported"},
341 {WSAEAFNOSUPPORT, "Address family not supported"},
342 {WSAEADDRINUSE, "Address already in use"},
343 {WSAEADDRNOTAVAIL, "Can't assign requested address"},
344 {WSAENETDOWN, "Network is down"},
345 {WSAENETUNREACH, "Network is unreachable"},
346 {WSAENETRESET, "Network dropped connection on reset"},
347 {WSAECONNABORTED, "Software caused connection abort"},
348 {WSAECONNRESET, "Connection reset by peer"},
349 {WSAENOBUFS, "No buffer space available"},
350 {WSAEISCONN, "Socket is already connected"},
351 {WSAENOTCONN, "Socket is not connected"},
352 {WSAESHUTDOWN, "Can't send after socket shutdown"},
353 {WSAETOOMANYREFS, "Too many references: can't splice"},
354 {WSAETIMEDOUT, "Operation timed out"},
355 {WSAECONNREFUSED, "Connection refused"},
356 {WSAELOOP, "Too many levels of symbolic links"},
357 {WSAENAMETOOLONG, "File name too long"},
358 {WSAEHOSTDOWN, "Host is down"},
359 {WSAEHOSTUNREACH, "No route to host"},
360 {WSAENOTEMPTY, "Directory not empty"},
361 {WSAEPROCLIM, "Too many processes"},
362 {WSAEUSERS, "Too many users"},
363 {WSAEDQUOT, "Disc quota exceeded"},
364 {WSAESTALE, "Stale NFS file handle"},
365 {WSAEREMOTE, "Too many levels of remote in path"},
366 {WSASYSNOTREADY, "Network subsystem is unvailable"},
367 {WSAVERNOTSUPPORTED, "WinSock version is not supported"},
368 {WSANOTINITIALISED,
369 "Successful WSAStartup() not yet performed"},
370 {WSAEDISCON, "Graceful shutdown in progress"},
371 /* Resolver errors */
372 {WSAHOST_NOT_FOUND, "No such host is known"},
373 {WSATRY_AGAIN, "Host not found, or server failed"},
374 {WSANO_RECOVERY, "Unexpected server error encountered"},
375 {WSANO_DATA, "Valid name without requested data"},
376 {WSANO_ADDRESS, "No address, look for MX record"},
377 {0, NULL}
378 };
Mark Hammond46a733d2000-07-24 01:45:11 +0000379 if (err_no) {
Guido van Rossum73624e91994-10-10 17:59:00 +0000380 PyObject *v;
Mark Hammond46a733d2000-07-24 01:45:11 +0000381 const char *msg = "winsock error";
Guido van Rossum48a680c2001-03-02 06:34:14 +0000382
Mark Hammond46a733d2000-07-24 01:45:11 +0000383 for (msgp = msgs; msgp->msg; msgp++) {
384 if (err_no == msgp->no) {
385 msg = msgp->msg;
386 break;
387 }
388 }
389
390 v = Py_BuildValue("(is)", err_no, msg);
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000391 if (v != NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000392 PyErr_SetObject(socket_error, v);
Guido van Rossum73624e91994-10-10 17:59:00 +0000393 Py_DECREF(v);
Guido van Rossum14ed0b21994-09-29 09:50:09 +0000394 }
395 return NULL;
396 }
397 else
398#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000399
Andrew MacIntyreba43e872002-03-03 03:03:52 +0000400#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000401 if (sock_errno() != NO_ERROR) {
402 APIRET rc;
403 ULONG msglen;
404 char outbuf[100];
405 int myerrorcode = sock_errno();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000406
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000407 /* Retrieve socket-related error message from MPTN.MSG file */
408 rc = DosGetMessage(NULL, 0, outbuf, sizeof(outbuf),
409 myerrorcode - SOCBASEERR + 26,
410 "mptn.msg",
411 &msglen);
412 if (rc == NO_ERROR) {
413 PyObject *v;
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000414
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000415 /* OS/2 doesn't guarantee a terminator */
416 outbuf[msglen] = '\0';
417 if (strlen(outbuf) > 0) {
418 /* If non-empty msg, trim CRLF */
419 char *lastc = &outbuf[ strlen(outbuf)-1 ];
420 while (lastc > outbuf && isspace(*lastc)) {
421 /* Trim trailing whitespace (CRLF) */
422 *lastc-- = '\0';
423 }
424 }
425 v = Py_BuildValue("(is)", myerrorcode, outbuf);
426 if (v != NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000427 PyErr_SetObject(socket_error, v);
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000428 Py_DECREF(v);
429 }
430 return NULL;
431 }
432 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000433#endif
434
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000435 return PyErr_SetFromErrno(socket_error);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000436}
437
Guido van Rossum30a685f1991-06-27 15:51:29 +0000438
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000439static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000440set_herror(int h_error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000441{
442 PyObject *v;
443
444#ifdef HAVE_HSTRERROR
445 v = Py_BuildValue("(is)", h_error, (char *)hstrerror(h_error));
446#else
447 v = Py_BuildValue("(is)", h_error, "host not found");
448#endif
449 if (v != NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000450 PyErr_SetObject(socket_herror, v);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000451 Py_DECREF(v);
452 }
453
454 return NULL;
455}
456
457
458static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000459set_gaierror(int error)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000460{
461 PyObject *v;
462
Martin v. Löwis272cb402002-03-01 08:31:07 +0000463#ifdef EAI_SYSTEM
464 /* EAI_SYSTEM is not available on Windows XP. */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000465 if (error == EAI_SYSTEM)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000466 return set_error();
Martin v. Löwis272cb402002-03-01 08:31:07 +0000467#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000468
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000469#ifdef HAVE_GAI_STRERROR
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000470 v = Py_BuildValue("(is)", error, gai_strerror(error));
Martin v. Löwisf95dd0a2001-08-15 17:14:33 +0000471#else
472 v = Py_BuildValue("(is)", error, "getaddrinfo failed");
473#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000474 if (v != NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000475 PyErr_SetObject(socket_gaierror, v);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000476 Py_DECREF(v);
477 }
478
479 return NULL;
480}
481
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000482/* Function to perform the setting of socket blocking mode
483 internally. block = (1 | 0). */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000484static int
485internal_setblocking(PySocketSockObject *s, int block)
486{
487#ifndef RISCOS
488#ifndef MS_WINDOWS
489 int delay_flag;
490#endif
491#endif
492
493 Py_BEGIN_ALLOW_THREADS
494#ifdef __BEOS__
495 block = !block;
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000496 setsockopt(s->sock_fd, SOL_SOCKET, SO_NONBLOCK,
497 (void *)(&block), sizeof(int));
Guido van Rossum67f7a382002-06-06 21:08:16 +0000498#else
499#ifndef RISCOS
500#ifndef MS_WINDOWS
501#if defined(PYOS_OS2) && !defined(PYCC_GCC)
502 block = !block;
503 ioctl(s->sock_fd, FIONBIO, (caddr_t)&block, sizeof(block));
Martin v. Löwis79acb9e2002-12-06 12:48:53 +0000504#elif defined(__VMS)
505 block = !block;
506 ioctl(s->sock_fd, FIONBIO, (char *)&block);
507#else /* !PYOS_OS2 && !_VMS */
Guido van Rossum67f7a382002-06-06 21:08:16 +0000508 delay_flag = fcntl(s->sock_fd, F_GETFL, 0);
509 if (block)
Neal Norwitz39d22e52002-11-02 19:55:21 +0000510 delay_flag &= (~O_NONBLOCK);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000511 else
Neal Norwitz39d22e52002-11-02 19:55:21 +0000512 delay_flag |= O_NONBLOCK;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000513 fcntl(s->sock_fd, F_SETFL, delay_flag);
514#endif /* !PYOS_OS2 */
515#else /* MS_WINDOWS */
516 block = !block;
517 ioctlsocket(s->sock_fd, FIONBIO, (u_long*)&block);
518#endif /* MS_WINDOWS */
519#endif /* __BEOS__ */
520#endif /* RISCOS */
521 Py_END_ALLOW_THREADS
522
523 /* Since these don't return anything */
524 return 1;
525}
526
Guido van Rossum11ba0942002-06-13 15:07:44 +0000527/* Do a select() on the socket, if necessary (sock_timeout > 0).
528 The argument writing indicates the direction.
529 This does not raise an exception or return a success indicator;
530 we'll let the actual socket call do that. */
531static void
Guido van Rossumb9e916a2002-06-07 01:42:47 +0000532internal_select(PySocketSockObject *s, int writing)
Guido van Rossum67f7a382002-06-06 21:08:16 +0000533{
534 fd_set fds;
535 struct timeval tv;
Guido van Rossum11ba0942002-06-13 15:07:44 +0000536
Guido van Rossumad654902002-07-19 12:44:59 +0000537 /* Nothing to do unless we're in timeout mode (not non-blocking) */
Guido van Rossum11ba0942002-06-13 15:07:44 +0000538 if (s->sock_timeout <= 0.0)
539 return;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000540
Guido van Rossumad654902002-07-19 12:44:59 +0000541 /* Guard against closed socket */
542 if (s->sock_fd < 0)
543 return;
544
Guido van Rossum67f7a382002-06-06 21:08:16 +0000545 /* Construct the arguments to select */
546 tv.tv_sec = (int)s->sock_timeout;
Guido van Rossumb9e916a2002-06-07 01:42:47 +0000547 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000548 FD_ZERO(&fds);
549 FD_SET(s->sock_fd, &fds);
550
551 /* See if the socket is ready */
Guido van Rossumb9e916a2002-06-07 01:42:47 +0000552 if (writing)
Guido van Rossum11ba0942002-06-13 15:07:44 +0000553 select(s->sock_fd+1, NULL, &fds, NULL, &tv);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000554 else
Guido van Rossum11ba0942002-06-13 15:07:44 +0000555 select(s->sock_fd+1, &fds, NULL, NULL, &tv);
Guido van Rossum67f7a382002-06-06 21:08:16 +0000556}
557
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000558/* Initialize a new socket object. */
559
Tim Petersa12b4cf2002-07-18 22:38:44 +0000560static double defaulttimeout = -1.0; /* Default timeout for new sockets */
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000561
Mark Hammond62b1ab12002-07-23 06:31:15 +0000562PyMODINIT_FUNC
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000563init_sockobject(PySocketSockObject *s,
564 SOCKET_T fd, int family, int type, int proto)
565{
566#ifdef RISCOS
567 int block = 1;
568#endif
569 s->sock_fd = fd;
570 s->sock_family = family;
571 s->sock_type = type;
572 s->sock_proto = proto;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000573 s->sock_timeout = defaulttimeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000574
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000575 s->errorhandler = &set_error;
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +0000576
577 if (defaulttimeout >= 0.0)
578 internal_setblocking(s, 0);
579
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000580#ifdef RISCOS
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000581 if (taskwindow)
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000582 socketioctl(s->sock_fd, 0x80046679, (u_long*)&block);
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000583#endif
584}
585
586
Guido van Rossum30a685f1991-06-27 15:51:29 +0000587/* Create a new socket object.
588 This just creates the object and initializes it.
589 If the creation fails, return NULL and set an exception (implicit
590 in NEWOBJ()). */
591
Guido van Rossum73624e91994-10-10 17:59:00 +0000592static PySocketSockObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000593new_sockobject(SOCKET_T fd, int family, int type, int proto)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000594{
Guido van Rossum73624e91994-10-10 17:59:00 +0000595 PySocketSockObject *s;
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000596 s = (PySocketSockObject *)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000597 PyType_GenericNew(&sock_type, NULL, NULL);
Guido van Rossum384ca9c2001-10-27 22:20:47 +0000598 if (s != NULL)
599 init_sockobject(s, fd, family, type, proto);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000600 return s;
601}
602
Guido van Rossum30a685f1991-06-27 15:51:29 +0000603
Guido van Rossum48a680c2001-03-02 06:34:14 +0000604/* Lock to allow python interpreter to continue, but only allow one
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000605 thread to be in gethostbyname */
Guido van Rossum3baaa131999-03-22 21:44:51 +0000606#ifdef USE_GETHOSTBYNAME_LOCK
Guido van Rossum65d5b571998-12-21 19:32:43 +0000607PyThread_type_lock gethostbyname_lock;
Guido van Rossum4f199ea1998-04-09 20:56:35 +0000608#endif
609
610
Guido van Rossum30a685f1991-06-27 15:51:29 +0000611/* Convert a string specifying a host name or one of a few symbolic
612 names to a numeric IP address. This usually calls gethostbyname()
613 to do the work; the names "" and "<broadcast>" are special.
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000614 Return the length (IPv4 should be 4 bytes), or negative if
Guido van Rossum30a685f1991-06-27 15:51:29 +0000615 an error occurred; then an exception is raised. */
616
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000617static int
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000618setipaddr(char *name, struct sockaddr *addr_ret, size_t addr_ret_size, int af)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000619{
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000620 struct addrinfo hints, *res;
621 int error;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000622
Guido van Rossuma376cc51996-12-05 23:43:35 +0000623 memset((void *) addr_ret, '\0', sizeof(*addr_ret));
Guido van Rossum30a685f1991-06-27 15:51:29 +0000624 if (name[0] == '\0') {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000625 int siz;
626 memset(&hints, 0, sizeof(hints));
627 hints.ai_family = af;
628 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
629 hints.ai_flags = AI_PASSIVE;
630 error = getaddrinfo(NULL, "0", &hints, &res);
631 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000632 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000633 return -1;
634 }
635 switch (res->ai_family) {
636 case AF_INET:
637 siz = 4;
638 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000639#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000640 case AF_INET6:
641 siz = 16;
642 break;
643#endif
644 default:
645 freeaddrinfo(res);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000646 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000647 "unsupported address family");
648 return -1;
649 }
650 if (res->ai_next) {
Martin v. Löwisf0b11d22001-11-07 08:31:03 +0000651 freeaddrinfo(res);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000652 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000653 "wildcard resolved to multiple address");
654 return -1;
655 }
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000656 if (res->ai_addrlen < addr_ret_size)
657 addr_ret_size = res->ai_addrlen;
658 memcpy(addr_ret, res->ai_addr, addr_ret_size);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000659 freeaddrinfo(res);
660 return siz;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000661 }
Guido van Rossum30a685f1991-06-27 15:51:29 +0000662 if (name[0] == '<' && strcmp(name, "<broadcast>") == 0) {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000663 struct sockaddr_in *sin;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +0000664 if (af != AF_INET && af != AF_UNSPEC) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000665 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000666 "address family mismatched");
667 return -1;
668 }
669 sin = (struct sockaddr_in *)addr_ret;
670 memset((void *) sin, '\0', sizeof(*sin));
671 sin->sin_family = AF_INET;
672#ifdef HAVE_SOCKADDR_SA_LEN
673 sin->sin_len = sizeof(*sin);
674#endif
675 sin->sin_addr.s_addr = INADDR_BROADCAST;
676 return sizeof(sin->sin_addr);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000677 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000678 memset(&hints, 0, sizeof(hints));
679 hints.ai_family = af;
680 error = getaddrinfo(name, NULL, &hints, &res);
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000681#if defined(__digital__) && defined(__unix__)
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000682 if (error == EAI_NONAME && af == AF_UNSPEC) {
683 /* On Tru64 V5.1, numeric-to-addr conversion fails
684 if no address family is given. Assume IPv4 for now.*/
685 hints.ai_family = AF_INET;
686 error = getaddrinfo(name, NULL, &hints, &res);
687 }
Martin v. Löwis7c4b5fa2001-10-25 09:04:03 +0000688#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000689 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000690 set_gaierror(error);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000691 return -1;
692 }
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000693 if (res->ai_addrlen < addr_ret_size)
694 addr_ret_size = res->ai_addrlen;
695 memcpy((char *) addr_ret, res->ai_addr, addr_ret_size);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000696 freeaddrinfo(res);
697 switch (addr_ret->sa_family) {
698 case AF_INET:
699 return 4;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000700#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000701 case AF_INET6:
702 return 16;
Guido van Rossum955becc1999-03-22 20:14:53 +0000703#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000704 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000705 PyErr_SetString(socket_error, "unknown address family");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000706 return -1;
707 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000708}
709
Guido van Rossum30a685f1991-06-27 15:51:29 +0000710
Guido van Rossum30a685f1991-06-27 15:51:29 +0000711/* Create a string object representing an IP address.
712 This is always a string of the form 'dd.dd.dd.dd' (with variable
713 size numbers). */
714
Guido van Rossum73624e91994-10-10 17:59:00 +0000715static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000716makeipaddr(struct sockaddr *addr, int addrlen)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000717{
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000718 char buf[NI_MAXHOST];
719 int error;
720
721 error = getnameinfo(addr, addrlen, buf, sizeof(buf), NULL, 0,
722 NI_NUMERICHOST);
723 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000724 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000725 return NULL;
726 }
Guido van Rossum73624e91994-10-10 17:59:00 +0000727 return PyString_FromString(buf);
Guido van Rossum30a685f1991-06-27 15:51:29 +0000728}
729
730
731/* Create an object representing the given socket address,
732 suitable for passing it back to bind(), connect() etc.
733 The family field of the sockaddr structure is inspected
734 to determine what kind of address it really is. */
735
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000736/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +0000737static PyObject *
Jeremy Hyltondbfb6622001-02-02 19:55:17 +0000738makesockaddr(int sockfd, struct sockaddr *addr, int addrlen)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000739{
Guido van Rossum25bec8c1992-08-05 19:00:45 +0000740 if (addrlen == 0) {
741 /* No address -- may be recvfrom() from known socket */
Guido van Rossum73624e91994-10-10 17:59:00 +0000742 Py_INCREF(Py_None);
743 return Py_None;
Guido van Rossum25bec8c1992-08-05 19:00:45 +0000744 }
745
Guido van Rossumbcc20741998-08-04 22:53:56 +0000746#ifdef __BEOS__
Guido van Rossum2c8bcb82000-04-25 21:34:53 +0000747 /* XXX: BeOS version of accept() doesn't set family correctly */
Guido van Rossumbcc20741998-08-04 22:53:56 +0000748 addr->sa_family = AF_INET;
749#endif
750
Guido van Rossum30a685f1991-06-27 15:51:29 +0000751 switch (addr->sa_family) {
752
753 case AF_INET:
754 {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000755 struct sockaddr_in *a;
756 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
Barry Warsaw752300b1997-01-03 17:18:10 +0000757 PyObject *ret = NULL;
758 if (addrobj) {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000759 a = (struct sockaddr_in *)addr;
Barry Warsaw752300b1997-01-03 17:18:10 +0000760 ret = Py_BuildValue("Oi", addrobj, ntohs(a->sin_port));
761 Py_DECREF(addrobj);
762 }
Guido van Rossum6f5afc91993-02-05 09:46:15 +0000763 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000764 }
Guido van Rossum30a685f1991-06-27 15:51:29 +0000765
Andrew MacIntyre34d3e2d2003-01-02 12:45:34 +0000766#if defined(AF_UNIX) && !defined(PYOS_OS2)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000767 case AF_UNIX:
768 {
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000769 struct sockaddr_un *a = (struct sockaddr_un *) addr;
Guido van Rossum73624e91994-10-10 17:59:00 +0000770 return PyString_FromString(a->sun_path);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000771 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000772#endif /* AF_UNIX */
773
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000774#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000775 case AF_INET6:
776 {
777 struct sockaddr_in6 *a;
778 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
779 PyObject *ret = NULL;
780 if (addrobj) {
781 a = (struct sockaddr_in6 *)addr;
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000782 ret = Py_BuildValue("Oiii",
783 addrobj,
784 ntohs(a->sin6_port),
785 a->sin6_flowinfo,
786 a->sin6_scope_id);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000787 Py_DECREF(addrobj);
788 }
789 return ret;
790 }
Jeremy Hylton22308652001-02-02 03:23:09 +0000791#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +0000792
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +0000793#ifdef HAVE_NETPACKET_PACKET_H
Jeremy Hylton22308652001-02-02 03:23:09 +0000794 case AF_PACKET:
795 {
796 struct sockaddr_ll *a = (struct sockaddr_ll *)addr;
797 char *ifname = "";
798 struct ifreq ifr;
Jeremy Hyltondbfb6622001-02-02 19:55:17 +0000799 /* need to look up interface name give index */
800 if (a->sll_ifindex) {
Jeremy Hylton22308652001-02-02 03:23:09 +0000801 ifr.ifr_ifindex = a->sll_ifindex;
Jeremy Hyltondbfb6622001-02-02 19:55:17 +0000802 if (ioctl(sockfd, SIOCGIFNAME, &ifr) == 0)
Jeremy Hylton22308652001-02-02 03:23:09 +0000803 ifname = ifr.ifr_name;
Jeremy Hylton22308652001-02-02 03:23:09 +0000804 }
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000805 return Py_BuildValue("shbhs#",
806 ifname,
807 ntohs(a->sll_protocol),
808 a->sll_pkttype,
809 a->sll_hatype,
810 a->sll_addr,
811 a->sll_halen);
Jeremy Hylton22308652001-02-02 03:23:09 +0000812 }
813#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +0000814
Guido van Rossum30a685f1991-06-27 15:51:29 +0000815 /* More cases here... */
816
817 default:
Guido van Rossumaa948df1997-05-07 17:41:48 +0000818 /* If we don't know the address family, don't raise an
819 exception -- return it as a tuple. */
820 return Py_BuildValue("is#",
821 addr->sa_family,
822 addr->sa_data,
823 sizeof(addr->sa_data));
Guido van Rossum25bec8c1992-08-05 19:00:45 +0000824
Guido van Rossum30a685f1991-06-27 15:51:29 +0000825 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000826}
827
Guido van Rossum30a685f1991-06-27 15:51:29 +0000828
829/* Parse a socket address argument according to the socket object's
830 address family. Return 1 if the address was in the proper format,
831 0 of not. The address is returned through addr_ret, its length
832 through len_ret. */
833
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000834static int
Guido van Rossum48a680c2001-03-02 06:34:14 +0000835getsockaddrarg(PySocketSockObject *s, PyObject *args,
Jeremy Hyltondbfb6622001-02-02 19:55:17 +0000836 struct sockaddr **addr_ret, int *len_ret)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000837{
Guido van Rossum30a685f1991-06-27 15:51:29 +0000838 switch (s->sock_family) {
839
Andrew MacIntyre34d3e2d2003-01-02 12:45:34 +0000840#if defined(AF_UNIX) && !defined(PYOS_OS2)
Guido van Rossum30a685f1991-06-27 15:51:29 +0000841 case AF_UNIX:
842 {
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000843 struct sockaddr_un* addr;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000844 char *path;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000845 int len;
Guido van Rossum67f7a382002-06-06 21:08:16 +0000846 addr = (struct sockaddr_un*)&(s->sock_addr).un;
Guido van Rossum7e488981998-10-08 02:25:24 +0000847 if (!PyArg_Parse(args, "t#", &path, &len))
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000848 return 0;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000849 if (len > sizeof addr->sun_path) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000850 PyErr_SetString(socket_error,
Barry Warsaw752300b1997-01-03 17:18:10 +0000851 "AF_UNIX path too long");
Guido van Rossum30a685f1991-06-27 15:51:29 +0000852 return 0;
853 }
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000854 addr->sun_family = s->sock_family;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000855 memcpy(addr->sun_path, path, len);
Guido van Rossum30b6b2b1995-06-14 22:28:08 +0000856 addr->sun_path[len] = 0;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000857 *addr_ret = (struct sockaddr *) addr;
Guido van Rossum65af28a1996-06-11 18:36:33 +0000858 *len_ret = len + sizeof(*addr) - sizeof(addr->sun_path);
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000859 return 1;
860 }
Guido van Rossumb6775db1994-08-01 11:34:53 +0000861#endif /* AF_UNIX */
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000862
Guido van Rossum30a685f1991-06-27 15:51:29 +0000863 case AF_INET:
864 {
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000865 struct sockaddr_in* addr;
Guido van Rossumff4949e1992-08-05 19:58:53 +0000866 char *host;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000867 int port;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000868 addr=(struct sockaddr_in*)&(s->sock_addr).in;
Guido van Rossume4dad902000-12-01 13:13:11 +0000869 if (!PyTuple_Check(args)) {
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000870 PyErr_Format(
871 PyExc_TypeError,
872 "getsockaddrarg: "
873 "AF_INET address must be tuple, not %.500s",
874 args->ob_type->tp_name);
Guido van Rossume4dad902000-12-01 13:13:11 +0000875 return 0;
876 }
877 if (!PyArg_ParseTuple(args, "si:getsockaddrarg", &host, &port))
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000878 return 0;
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000879 if (setipaddr(host, (struct sockaddr *)addr, sizeof(*addr), AF_INET) < 0)
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000880 return 0;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000881 addr->sin_family = AF_INET;
Guido van Rossum644a12b1997-04-09 19:24:53 +0000882 addr->sin_port = htons((short)port);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000883 *addr_ret = (struct sockaddr *) addr;
884 *len_ret = sizeof *addr;
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000885 return 1;
886 }
887
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000888#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000889 case AF_INET6:
890 {
891 struct sockaddr_in6* addr;
892 char *host;
893 int port, flowinfo, scope_id;
894 addr = (struct sockaddr_in6*)&(s->sock_addr).in6;
895 flowinfo = scope_id = 0;
896 if (!PyArg_ParseTuple(args, "si|ii", &host, &port, &flowinfo,
Guido van Rossum3eede5a2002-06-07 02:08:35 +0000897 &scope_id)) {
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000898 return 0;
899 }
Martin v. Löwisddc6f472002-07-28 16:10:31 +0000900 if (setipaddr(host, (struct sockaddr *)addr, sizeof(*addr), AF_INET6) < 0)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000901 return 0;
902 addr->sin6_family = s->sock_family;
903 addr->sin6_port = htons((short)port);
904 addr->sin6_flowinfo = flowinfo;
905 addr->sin6_scope_id = scope_id;
906 *addr_ret = (struct sockaddr *) addr;
907 *len_ret = sizeof *addr;
908 return 1;
909 }
910#endif
911
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +0000912#ifdef HAVE_NETPACKET_PACKET_H
Jeremy Hylton22308652001-02-02 03:23:09 +0000913 case AF_PACKET:
914 {
915 struct sockaddr_ll* addr;
916 struct ifreq ifr;
917 char *interfaceName;
918 int protoNumber;
919 int hatype = 0;
920 int pkttype = 0;
Jeremy Hyltondbfb6622001-02-02 19:55:17 +0000921 char *haddr;
Guido van Rossum48a680c2001-03-02 06:34:14 +0000922
Jeremy Hyltondbfb6622001-02-02 19:55:17 +0000923 if (!PyArg_ParseTuple(args, "si|iis", &interfaceName,
924 &protoNumber, &pkttype, &hatype, &haddr))
Jeremy Hylton22308652001-02-02 03:23:09 +0000925 return 0;
926 strncpy(ifr.ifr_name, interfaceName, sizeof(ifr.ifr_name));
927 ifr.ifr_name[(sizeof(ifr.ifr_name))-1] = '\0';
Jeremy Hyltondbfb6622001-02-02 19:55:17 +0000928 if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +0000929 s->errorhandler();
Jeremy Hylton22308652001-02-02 03:23:09 +0000930 return 0;
Jeremy Hyltondbfb6622001-02-02 19:55:17 +0000931 }
Jeremy Hylton22308652001-02-02 03:23:09 +0000932 addr = &(s->sock_addr.ll);
933 addr->sll_family = AF_PACKET;
934 addr->sll_protocol = htons((short)protoNumber);
935 addr->sll_ifindex = ifr.ifr_ifindex;
936 addr->sll_pkttype = pkttype;
937 addr->sll_hatype = hatype;
938 *addr_ret = (struct sockaddr *) addr;
939 *len_ret = sizeof *addr;
940 return 1;
941 }
Guido van Rossum48a680c2001-03-02 06:34:14 +0000942#endif
943
Guido van Rossum30a685f1991-06-27 15:51:29 +0000944 /* More cases here... */
945
946 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000947 PyErr_SetString(socket_error, "getsockaddrarg: bad family");
Guido van Rossum30a685f1991-06-27 15:51:29 +0000948 return 0;
949
950 }
Guido van Rossum6574b3e1991-06-25 21:36:08 +0000951}
952
Guido van Rossum30a685f1991-06-27 15:51:29 +0000953
Guido van Rossum48a680c2001-03-02 06:34:14 +0000954/* Get the address length according to the socket object's address family.
Guido van Rossum710e1df1992-06-12 10:39:36 +0000955 Return 1 if the family is known, 0 otherwise. The length is returned
956 through len_ret. */
957
958static int
Peter Schneider-Kamp618e25d2000-07-11 23:00:12 +0000959getsockaddrlen(PySocketSockObject *s, socklen_t *len_ret)
Guido van Rossum710e1df1992-06-12 10:39:36 +0000960{
961 switch (s->sock_family) {
962
Andrew MacIntyre34d3e2d2003-01-02 12:45:34 +0000963#if defined(AF_UNIX) && !defined(PYOS_OS2)
Guido van Rossum710e1df1992-06-12 10:39:36 +0000964 case AF_UNIX:
965 {
966 *len_ret = sizeof (struct sockaddr_un);
967 return 1;
968 }
Guido van Rossumb6775db1994-08-01 11:34:53 +0000969#endif /* AF_UNIX */
Guido van Rossum710e1df1992-06-12 10:39:36 +0000970
971 case AF_INET:
972 {
973 *len_ret = sizeof (struct sockaddr_in);
974 return 1;
975 }
976
Martin v. Löwis44ddbde2001-12-02 10:15:37 +0000977#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +0000978 case AF_INET6:
979 {
980 *len_ret = sizeof (struct sockaddr_in6);
981 return 1;
982 }
983#endif
984
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +0000985#ifdef HAVE_NETPACKET_PACKET_H
Jeremy Hylton22308652001-02-02 03:23:09 +0000986 case AF_PACKET:
987 {
988 *len_ret = sizeof (struct sockaddr_ll);
989 return 1;
990 }
991#endif
Guido van Rossum48a680c2001-03-02 06:34:14 +0000992
Guido van Rossum710e1df1992-06-12 10:39:36 +0000993 /* More cases here... */
994
995 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +0000996 PyErr_SetString(socket_error, "getsockaddrlen: bad family");
Guido van Rossum710e1df1992-06-12 10:39:36 +0000997 return 0;
998
999 }
1000}
1001
1002
Guido van Rossum30a685f1991-06-27 15:51:29 +00001003/* s.accept() method */
1004
Guido van Rossum73624e91994-10-10 17:59:00 +00001005static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001006sock_accept(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001007{
1008 char addrbuf[256];
Fred Drakea04eaad2000-06-30 02:46:07 +00001009 SOCKET_T newfd;
Guido van Rossumff3ab422000-04-24 15:16:03 +00001010 socklen_t addrlen;
Barry Warsaw752300b1997-01-03 17:18:10 +00001011 PyObject *sock = NULL;
1012 PyObject *addr = NULL;
1013 PyObject *res = NULL;
1014
Guido van Rossum710e1df1992-06-12 10:39:36 +00001015 if (!getsockaddrlen(s, &addrlen))
1016 return NULL;
Fred Drakefd168342001-05-09 19:11:33 +00001017 memset(addrbuf, 0, addrlen);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001018
Guido van Rossum73624e91994-10-10 17:59:00 +00001019 Py_BEGIN_ALLOW_THREADS
Guido van Rossum11ba0942002-06-13 15:07:44 +00001020 internal_select(s, 0);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001021 newfd = accept(s->sock_fd, (struct sockaddr *) addrbuf, &addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001022 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001023
Fred Drakea04eaad2000-06-30 02:46:07 +00001024#ifdef MS_WINDOWS
1025 if (newfd == INVALID_SOCKET)
1026#else
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001027 if (newfd < 0)
Fred Drakea04eaad2000-06-30 02:46:07 +00001028#endif
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001029 return s->errorhandler();
Barry Warsaw752300b1997-01-03 17:18:10 +00001030
Guido van Rossum30a685f1991-06-27 15:51:29 +00001031 /* Create the new object with unspecified family,
1032 to avoid calls to bind() etc. on it. */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001033 sock = (PyObject *) new_sockobject(newfd,
1034 s->sock_family,
1035 s->sock_type,
1036 s->sock_proto);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001037
Barry Warsaw752300b1997-01-03 17:18:10 +00001038 if (sock == NULL) {
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +00001039 SOCKETCLOSE(newfd);
Barry Warsaw752300b1997-01-03 17:18:10 +00001040 goto finally;
1041 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00001042 addr = makesockaddr(s->sock_fd, (struct sockaddr *)addrbuf,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001043 addrlen);
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001044 if (addr == NULL)
Barry Warsaw752300b1997-01-03 17:18:10 +00001045 goto finally;
1046
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001047 res = Py_BuildValue("OO", sock, addr);
Barry Warsaw752300b1997-01-03 17:18:10 +00001048
Guido van Rossum67f7a382002-06-06 21:08:16 +00001049finally:
Guido van Rossum73624e91994-10-10 17:59:00 +00001050 Py_XDECREF(sock);
1051 Py_XDECREF(addr);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001052 return res;
1053}
1054
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001055PyDoc_STRVAR(accept_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001056"accept() -> (socket object, address info)\n\
1057\n\
1058Wait for an incoming connection. Return a new socket representing the\n\
1059connection, and the address of the client. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001060info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001061
Guido van Rossum11ba0942002-06-13 15:07:44 +00001062/* s.setblocking(flag) method. Argument:
1063 False -- non-blocking mode; same as settimeout(0)
1064 True -- blocking mode; same as settimeout(None)
1065*/
Guido van Rossume4485b01994-09-07 14:32:49 +00001066
Guido van Rossum73624e91994-10-10 17:59:00 +00001067static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001068sock_setblocking(PySocketSockObject *s, PyObject *arg)
Guido van Rossume4485b01994-09-07 14:32:49 +00001069{
1070 int block;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001071
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001072 block = PyInt_AsLong(arg);
1073 if (block == -1 && PyErr_Occurred())
Guido van Rossume4485b01994-09-07 14:32:49 +00001074 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001075
Guido van Rossum11ba0942002-06-13 15:07:44 +00001076 s->sock_timeout = block ? -1.0 : 0.0;
Guido van Rossume8008f02002-06-07 03:36:20 +00001077 internal_setblocking(s, block);
Guido van Rossume4485b01994-09-07 14:32:49 +00001078
Guido van Rossum73624e91994-10-10 17:59:00 +00001079 Py_INCREF(Py_None);
1080 return Py_None;
Guido van Rossume4485b01994-09-07 14:32:49 +00001081}
Guido van Rossume4485b01994-09-07 14:32:49 +00001082
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001083PyDoc_STRVAR(setblocking_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001084"setblocking(flag)\n\
1085\n\
1086Set the socket to blocking (flag is true) or non-blocking (false).\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001087setblocking(True) is equivalent to settimeout(None);\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001088setblocking(False) is equivalent to settimeout(0.0).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001089
Guido van Rossum11ba0942002-06-13 15:07:44 +00001090/* s.settimeout(timeout) method. Argument:
1091 None -- no timeout, blocking mode; same as setblocking(True)
1092 0.0 -- non-blocking mode; same as setblocking(False)
1093 > 0 -- timeout mode; operations time out after timeout seconds
1094 < 0 -- illegal; raises an exception
1095*/
Guido van Rossum67f7a382002-06-06 21:08:16 +00001096static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001097sock_settimeout(PySocketSockObject *s, PyObject *arg)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001098{
Guido van Rossum11ba0942002-06-13 15:07:44 +00001099 double timeout;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001100
1101 if (arg == Py_None)
Guido van Rossum11ba0942002-06-13 15:07:44 +00001102 timeout = -1.0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001103 else {
Guido van Rossum11ba0942002-06-13 15:07:44 +00001104 timeout = PyFloat_AsDouble(arg);
1105 if (timeout < 0.0) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00001106 if (!PyErr_Occurred())
1107 PyErr_SetString(PyExc_ValueError,
Guido van Rossum11ba0942002-06-13 15:07:44 +00001108 "Timeout value out of range");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001109 return NULL;
1110 }
1111 }
1112
Guido van Rossum11ba0942002-06-13 15:07:44 +00001113 s->sock_timeout = timeout;
1114 internal_setblocking(s, timeout < 0.0);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001115
1116 Py_INCREF(Py_None);
1117 return Py_None;
1118}
1119
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001120PyDoc_STRVAR(settimeout_doc,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001121"settimeout(timeout)\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001122\n\
Guido van Rossum11ba0942002-06-13 15:07:44 +00001123Set a timeout on socket operations. 'timeout' can be a float,\n\
1124giving in seconds, or None. Setting a timeout of None disables\n\
1125the timeout feature and is equivalent to setblocking(1).\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001126Setting a timeout of zero is the same as setblocking(0).");
Guido van Rossum67f7a382002-06-06 21:08:16 +00001127
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001128/* s.gettimeout() method.
1129 Returns the timeout associated with a socket. */
Guido van Rossum67f7a382002-06-06 21:08:16 +00001130static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001131sock_gettimeout(PySocketSockObject *s)
Guido van Rossum67f7a382002-06-06 21:08:16 +00001132{
1133 if (s->sock_timeout < 0.0) {
1134 Py_INCREF(Py_None);
1135 return Py_None;
1136 }
1137 else
1138 return PyFloat_FromDouble(s->sock_timeout);
1139}
1140
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001141PyDoc_STRVAR(gettimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001142"gettimeout() -> timeout\n\
Guido van Rossum67f7a382002-06-06 21:08:16 +00001143\n\
1144Returns the timeout in floating seconds associated with socket \n\
1145operations. A timeout of None indicates that timeouts on socket \n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001146operations are disabled.");
Guido van Rossume4485b01994-09-07 14:32:49 +00001147
Guido van Rossum48a680c2001-03-02 06:34:14 +00001148#ifdef RISCOS
1149/* s.sleeptaskw(1 | 0) method */
1150
1151static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001152sock_sleeptaskw(PySocketSockObject *s,PyObject *args)
Guido van Rossum48a680c2001-03-02 06:34:14 +00001153{
Guido van Rossum67f7a382002-06-06 21:08:16 +00001154 int block;
1155 int delay_flag;
1156 if (!PyArg_Parse(args, "i", &block))
1157 return NULL;
1158 Py_BEGIN_ALLOW_THREADS
1159 socketioctl(s->sock_fd, 0x80046679, (u_long*)&block);
1160 Py_END_ALLOW_THREADS
Guido van Rossum48a680c2001-03-02 06:34:14 +00001161
Guido van Rossum67f7a382002-06-06 21:08:16 +00001162 Py_INCREF(Py_None);
1163 return Py_None;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001164}
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001165PyDoc_STRVAR(sleeptaskw_doc,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001166"sleeptaskw(flag)\n\
1167\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001168Allow sleeps in taskwindows.");
Guido van Rossum48a680c2001-03-02 06:34:14 +00001169#endif
1170
1171
Guido van Rossumaee08791992-09-08 09:05:33 +00001172/* s.setsockopt() method.
1173 With an integer third argument, sets an integer option.
1174 With a string third argument, sets an option from a buffer;
1175 use optional built-in module 'struct' to encode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001176
Guido van Rossum73624e91994-10-10 17:59:00 +00001177static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001178sock_setsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001179{
1180 int level;
1181 int optname;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001182 int res;
Guido van Rossumaee08791992-09-08 09:05:33 +00001183 char *buf;
1184 int buflen;
1185 int flag;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001186
Guido van Rossum27fc3c02000-03-24 20:56:56 +00001187 if (PyArg_ParseTuple(args, "iii:setsockopt",
1188 &level, &optname, &flag)) {
Guido van Rossumaee08791992-09-08 09:05:33 +00001189 buf = (char *) &flag;
1190 buflen = sizeof flag;
1191 }
1192 else {
Guido van Rossum73624e91994-10-10 17:59:00 +00001193 PyErr_Clear();
Guido van Rossum27fc3c02000-03-24 20:56:56 +00001194 if (!PyArg_ParseTuple(args, "iis#:setsockopt",
1195 &level, &optname, &buf, &buflen))
Guido van Rossumaee08791992-09-08 09:05:33 +00001196 return NULL;
1197 }
Thomas Wouters334fb892000-07-25 12:56:38 +00001198 res = setsockopt(s->sock_fd, level, optname, (void *)buf, buflen);
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001199 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001200 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001201 Py_INCREF(Py_None);
1202 return Py_None;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001203}
1204
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001205PyDoc_STRVAR(setsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001206"setsockopt(level, option, value)\n\
1207\n\
1208Set a socket option. See the Unix manual for level and option.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001209The value argument can either be an integer or a string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001210
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001211
Guido van Rossumaee08791992-09-08 09:05:33 +00001212/* s.getsockopt() method.
1213 With two arguments, retrieves an integer option.
1214 With a third integer argument, retrieves a string buffer of that size;
1215 use optional built-in module 'struct' to decode the string. */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001216
Guido van Rossum73624e91994-10-10 17:59:00 +00001217static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001218sock_getsockopt(PySocketSockObject *s, PyObject *args)
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001219{
1220 int level;
1221 int optname;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001222 int res;
Guido van Rossum73624e91994-10-10 17:59:00 +00001223 PyObject *buf;
Guido van Rossumff3ab422000-04-24 15:16:03 +00001224 socklen_t buflen = 0;
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001225
Guido van Rossumbcc20741998-08-04 22:53:56 +00001226#ifdef __BEOS__
Fred Drake564a6cc2001-05-11 20:12:26 +00001227 /* We have incomplete socket support. */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001228 PyErr_SetString(socket_error, "getsockopt not supported");
Guido van Rossumbcc20741998-08-04 22:53:56 +00001229 return NULL;
1230#else
1231
Guido van Rossum27fc3c02000-03-24 20:56:56 +00001232 if (!PyArg_ParseTuple(args, "ii|i:getsockopt",
1233 &level, &optname, &buflen))
Guido van Rossumbe32c891996-06-20 16:25:29 +00001234 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001235
Guido van Rossumbe32c891996-06-20 16:25:29 +00001236 if (buflen == 0) {
Guido van Rossumaee08791992-09-08 09:05:33 +00001237 int flag = 0;
Guido van Rossumff3ab422000-04-24 15:16:03 +00001238 socklen_t flagsize = sizeof flag;
Guido van Rossumb376a4a1993-11-23 17:53:17 +00001239 res = getsockopt(s->sock_fd, level, optname,
Thomas Wouters334fb892000-07-25 12:56:38 +00001240 (void *)&flag, &flagsize);
Guido van Rossumaee08791992-09-08 09:05:33 +00001241 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001242 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001243 return PyInt_FromLong(flag);
Guido van Rossumaee08791992-09-08 09:05:33 +00001244 }
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001245#ifdef __VMS
1246 if (buflen > 1024) {
1247#else
Guido van Rossumaee08791992-09-08 09:05:33 +00001248 if (buflen <= 0 || buflen > 1024) {
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001249#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001250 PyErr_SetString(socket_error,
Barry Warsaw752300b1997-01-03 17:18:10 +00001251 "getsockopt buflen out of range");
Guido van Rossumaee08791992-09-08 09:05:33 +00001252 return NULL;
1253 }
Guido van Rossum73624e91994-10-10 17:59:00 +00001254 buf = PyString_FromStringAndSize((char *)NULL, buflen);
Guido van Rossumaee08791992-09-08 09:05:33 +00001255 if (buf == NULL)
1256 return NULL;
Guido van Rossumb376a4a1993-11-23 17:53:17 +00001257 res = getsockopt(s->sock_fd, level, optname,
Fred Drake564a6cc2001-05-11 20:12:26 +00001258 (void *)PyString_AS_STRING(buf), &buflen);
Guido van Rossumaee08791992-09-08 09:05:33 +00001259 if (res < 0) {
Guido van Rossum73624e91994-10-10 17:59:00 +00001260 Py_DECREF(buf);
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001261 return s->errorhandler();
Guido van Rossumaee08791992-09-08 09:05:33 +00001262 }
Guido van Rossum73624e91994-10-10 17:59:00 +00001263 _PyString_Resize(&buf, buflen);
Guido van Rossumaee08791992-09-08 09:05:33 +00001264 return buf;
Guido van Rossumbcc20741998-08-04 22:53:56 +00001265#endif /* __BEOS__ */
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001266}
1267
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001268PyDoc_STRVAR(getsockopt_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001269"getsockopt(level, option[, buffersize]) -> value\n\
1270\n\
1271Get a socket option. See the Unix manual for level and option.\n\
1272If a nonzero buffersize argument is given, the return value is a\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001273string of that length; otherwise it is an integer.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001274
Guido van Rossum0e69587d1992-06-05 15:11:30 +00001275
Fred Drake728819a2000-07-01 03:40:12 +00001276/* s.bind(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001277
Guido van Rossum73624e91994-10-10 17:59:00 +00001278static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001279sock_bind(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001280{
1281 struct sockaddr *addr;
1282 int addrlen;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001283 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001284
Fred Drake728819a2000-07-01 03:40:12 +00001285 if (!getsockaddrarg(s, addro, &addr, &addrlen))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001286 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00001287 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001288 res = bind(s->sock_fd, addr, addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001289 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001290 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001291 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001292 Py_INCREF(Py_None);
1293 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001294}
1295
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001296PyDoc_STRVAR(bind_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001297"bind(address)\n\
1298\n\
1299Bind the socket to a local address. For IP sockets, the address is a\n\
Jeremy Hylton22308652001-02-02 03:23:09 +00001300pair (host, port); the host must refer to the local host. For raw packet\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001301sockets the address is a tuple (ifname, proto [,pkttype [,hatype]])");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001302
Guido van Rossum30a685f1991-06-27 15:51:29 +00001303
1304/* s.close() method.
1305 Set the file descriptor to -1 so operations tried subsequently
1306 will surely fail. */
1307
Guido van Rossum73624e91994-10-10 17:59:00 +00001308static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001309sock_close(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001310{
Guido van Rossum20d3fc02000-12-18 22:23:44 +00001311 SOCKET_T fd;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001312
Guido van Rossum20d3fc02000-12-18 22:23:44 +00001313 if ((fd = s->sock_fd) != -1) {
1314 s->sock_fd = -1;
Guido van Rossum08481461996-10-12 14:07:22 +00001315 Py_BEGIN_ALLOW_THREADS
Guido van Rossum20d3fc02000-12-18 22:23:44 +00001316 (void) SOCKETCLOSE(fd);
Guido van Rossum08481461996-10-12 14:07:22 +00001317 Py_END_ALLOW_THREADS
1318 }
Guido van Rossum73624e91994-10-10 17:59:00 +00001319 Py_INCREF(Py_None);
1320 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001321}
1322
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001323PyDoc_STRVAR(close_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001324"close()\n\
1325\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001326Close the socket. It cannot be used after this call.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001327
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001328static int
1329internal_connect(PySocketSockObject *s, struct sockaddr *addr, int addrlen)
1330{
1331 int res;
1332
1333 res = connect(s->sock_fd, addr, addrlen);
1334
1335#ifdef MS_WINDOWS
1336
1337 if (s->sock_timeout > 0.0) {
1338 if (res < 0 && WSAGetLastError() == WSAEWOULDBLOCK) {
1339 internal_select(s, 1);
1340 res = connect(s->sock_fd, addr, addrlen);
Tim Peters58141872002-08-06 22:25:02 +00001341 if (res < 0) {
1342 /* On Win98, WSAEISCONN was seen here. But
1343 * on Win2K, WSAEINVAL. So accept both as
1344 * meaning "fine".
1345 */
1346 int code = WSAGetLastError();
1347 if (code == WSAEISCONN ||
1348 code == WSAEINVAL)
1349 res = 0;
1350 }
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001351 }
1352 }
1353
1354 if (res < 0)
1355 res = WSAGetLastError();
1356
1357#else
1358
1359 if (s->sock_timeout > 0.0) {
1360 if (res < 0 && errno == EINPROGRESS) {
1361 internal_select(s, 1);
1362 res = connect(s->sock_fd, addr, addrlen);
Guido van Rossum12e3c712002-08-08 20:39:30 +00001363 if (res < 0 && errno == EISCONN)
1364 res = 0;
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001365 }
1366 }
1367
1368 if (res < 0)
1369 res = errno;
1370
1371#endif
1372
1373 return res;
1374}
Guido van Rossum30a685f1991-06-27 15:51:29 +00001375
Fred Drake728819a2000-07-01 03:40:12 +00001376/* s.connect(sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001377
Guido van Rossum73624e91994-10-10 17:59:00 +00001378static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001379sock_connect(PySocketSockObject *s, PyObject *addro)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001380{
1381 struct sockaddr *addr;
1382 int addrlen;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001383 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001384
Fred Drake728819a2000-07-01 03:40:12 +00001385 if (!getsockaddrarg(s, addro, &addr, &addrlen))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001386 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001387
Guido van Rossum73624e91994-10-10 17:59:00 +00001388 Py_BEGIN_ALLOW_THREADS
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001389 res = internal_connect(s, addr, addrlen);
Guido van Rossum11ba0942002-06-13 15:07:44 +00001390 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001391
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001392 if (res != 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001393 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001394 Py_INCREF(Py_None);
1395 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001396}
1397
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001398PyDoc_STRVAR(connect_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001399"connect(address)\n\
1400\n\
1401Connect the socket to a remote address. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001402is a pair (host, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001403
Guido van Rossum30a685f1991-06-27 15:51:29 +00001404
Fred Drake728819a2000-07-01 03:40:12 +00001405/* s.connect_ex(sockaddr) method */
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001406
1407static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001408sock_connect_ex(PySocketSockObject *s, PyObject *addro)
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001409{
1410 struct sockaddr *addr;
1411 int addrlen;
1412 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001413
Fred Drake728819a2000-07-01 03:40:12 +00001414 if (!getsockaddrarg(s, addro, &addr, &addrlen))
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001415 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001416
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001417 Py_BEGIN_ALLOW_THREADS
Guido van Rossum7b8bac12002-06-13 16:07:04 +00001418 res = internal_connect(s, addr, addrlen);
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001419 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001420
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001421 return PyInt_FromLong((long) res);
1422}
1423
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001424PyDoc_STRVAR(connect_ex_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001425"connect_ex(address) -> errno\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00001426\n\
1427This is like connect(address), but returns an error code (the errno value)\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001428instead of raising an exception when an error occurs.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001429
Guido van Rossumfc4255d1997-11-19 18:57:13 +00001430
Guido van Rossumed233a51992-06-23 09:07:03 +00001431/* s.fileno() method */
1432
Guido van Rossum73624e91994-10-10 17:59:00 +00001433static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001434sock_fileno(PySocketSockObject *s)
Guido van Rossumed233a51992-06-23 09:07:03 +00001435{
Fred Drakea04eaad2000-06-30 02:46:07 +00001436#if SIZEOF_SOCKET_T <= SIZEOF_LONG
Guido van Rossum73624e91994-10-10 17:59:00 +00001437 return PyInt_FromLong((long) s->sock_fd);
Fred Drakea04eaad2000-06-30 02:46:07 +00001438#else
1439 return PyLong_FromLongLong((LONG_LONG)s->sock_fd);
1440#endif
Guido van Rossumed233a51992-06-23 09:07:03 +00001441}
1442
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001443PyDoc_STRVAR(fileno_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001444"fileno() -> integer\n\
1445\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001446Return the integer file descriptor of the socket.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001447
Guido van Rossumed233a51992-06-23 09:07:03 +00001448
Guido van Rossumbe32c891996-06-20 16:25:29 +00001449#ifndef NO_DUP
1450/* s.dup() method */
1451
1452static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001453sock_dup(PySocketSockObject *s)
Guido van Rossumbe32c891996-06-20 16:25:29 +00001454{
Fred Drakea04eaad2000-06-30 02:46:07 +00001455 SOCKET_T newfd;
Guido van Rossumbe32c891996-06-20 16:25:29 +00001456 PyObject *sock;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001457
Guido van Rossumbe32c891996-06-20 16:25:29 +00001458 newfd = dup(s->sock_fd);
1459 if (newfd < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001460 return s->errorhandler();
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001461 sock = (PyObject *) new_sockobject(newfd,
1462 s->sock_family,
1463 s->sock_type,
1464 s->sock_proto);
Guido van Rossumbe32c891996-06-20 16:25:29 +00001465 if (sock == NULL)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +00001466 SOCKETCLOSE(newfd);
Guido van Rossumbe32c891996-06-20 16:25:29 +00001467 return sock;
1468}
Guido van Rossum82a5c661998-07-07 20:45:43 +00001469
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001470PyDoc_STRVAR(dup_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001471"dup() -> socket object\n\
1472\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001473Return a new socket object connected to the same system resource.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001474
Guido van Rossumbe32c891996-06-20 16:25:29 +00001475#endif
1476
1477
Guido van Rossumc89705d1992-11-26 08:54:07 +00001478/* s.getsockname() method */
1479
Guido van Rossum73624e91994-10-10 17:59:00 +00001480static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001481sock_getsockname(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00001482{
1483 char addrbuf[256];
Guido van Rossumff3ab422000-04-24 15:16:03 +00001484 int res;
1485 socklen_t addrlen;
1486
Guido van Rossumc89705d1992-11-26 08:54:07 +00001487 if (!getsockaddrlen(s, &addrlen))
1488 return NULL;
Guido van Rossumca42b161996-01-12 01:36:05 +00001489 memset(addrbuf, 0, addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001490 Py_BEGIN_ALLOW_THREADS
Guido van Rossumc89705d1992-11-26 08:54:07 +00001491 res = getsockname(s->sock_fd, (struct sockaddr *) addrbuf, &addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001492 Py_END_ALLOW_THREADS
Guido van Rossumc89705d1992-11-26 08:54:07 +00001493 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001494 return s->errorhandler();
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001495 return makesockaddr(s->sock_fd, (struct sockaddr *) addrbuf, addrlen);
Guido van Rossumc89705d1992-11-26 08:54:07 +00001496}
1497
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001498PyDoc_STRVAR(getsockname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001499"getsockname() -> address info\n\
1500\n\
1501Return the address of the local endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001502info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001503
Guido van Rossumc89705d1992-11-26 08:54:07 +00001504
Guido van Rossumb6775db1994-08-01 11:34:53 +00001505#ifdef HAVE_GETPEERNAME /* Cray APP doesn't have this :-( */
Guido van Rossumc89705d1992-11-26 08:54:07 +00001506/* s.getpeername() method */
1507
Guido van Rossum73624e91994-10-10 17:59:00 +00001508static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001509sock_getpeername(PySocketSockObject *s)
Guido van Rossumc89705d1992-11-26 08:54:07 +00001510{
1511 char addrbuf[256];
Guido van Rossumff3ab422000-04-24 15:16:03 +00001512 int res;
1513 socklen_t addrlen;
1514
Guido van Rossumc89705d1992-11-26 08:54:07 +00001515 if (!getsockaddrlen(s, &addrlen))
1516 return NULL;
Fred Drakefd168342001-05-09 19:11:33 +00001517 memset(addrbuf, 0, addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001518 Py_BEGIN_ALLOW_THREADS
Guido van Rossumc89705d1992-11-26 08:54:07 +00001519 res = getpeername(s->sock_fd, (struct sockaddr *) addrbuf, &addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001520 Py_END_ALLOW_THREADS
Guido van Rossumc89705d1992-11-26 08:54:07 +00001521 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001522 return s->errorhandler();
Jeremy Hyltondbfb6622001-02-02 19:55:17 +00001523 return makesockaddr(s->sock_fd, (struct sockaddr *) addrbuf, addrlen);
Guido van Rossumc89705d1992-11-26 08:54:07 +00001524}
Guido van Rossum82a5c661998-07-07 20:45:43 +00001525
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001526PyDoc_STRVAR(getpeername_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001527"getpeername() -> address info\n\
1528\n\
1529Return the address of the remote endpoint. For IP sockets, the address\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001530info is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001531
Guido van Rossumb6775db1994-08-01 11:34:53 +00001532#endif /* HAVE_GETPEERNAME */
Guido van Rossumc89705d1992-11-26 08:54:07 +00001533
1534
Guido van Rossum30a685f1991-06-27 15:51:29 +00001535/* s.listen(n) method */
1536
Guido van Rossum73624e91994-10-10 17:59:00 +00001537static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001538sock_listen(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001539{
1540 int backlog;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001541 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001542
1543 backlog = PyInt_AsLong(arg);
1544 if (backlog == -1 && PyErr_Occurred())
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001545 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00001546 Py_BEGIN_ALLOW_THREADS
Guido van Rossumb6775db1994-08-01 11:34:53 +00001547 if (backlog < 1)
1548 backlog = 1;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001549 res = listen(s->sock_fd, backlog);
Guido van Rossum73624e91994-10-10 17:59:00 +00001550 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001551 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001552 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001553 Py_INCREF(Py_None);
1554 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001555}
1556
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001557PyDoc_STRVAR(listen_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001558"listen(backlog)\n\
1559\n\
1560Enable a server to accept connections. The backlog argument must be at\n\
1561least 1; it specifies the number of unaccepted connection that the system\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001562will allow before refusing new connections.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001563
1564
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001565#ifndef NO_DUP
Guido van Rossum30a685f1991-06-27 15:51:29 +00001566/* s.makefile(mode) method.
1567 Create a new open file object referring to a dupped version of
1568 the socket's file descriptor. (The dup() call is necessary so
1569 that the open file and socket objects may be closed independent
1570 of each other.)
1571 The mode argument specifies 'r' or 'w' passed to fdopen(). */
1572
Guido van Rossum73624e91994-10-10 17:59:00 +00001573static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001574sock_makefile(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001575{
Tim Petersdbd9ba62000-07-09 03:09:57 +00001576 extern int fclose(FILE *);
Guido van Rossum6b144911995-03-14 15:05:13 +00001577 char *mode = "r";
1578 int bufsize = -1;
Fred Drakea04eaad2000-06-30 02:46:07 +00001579#ifdef MS_WIN32
Tim Peters79248aa2001-08-29 21:37:10 +00001580 Py_intptr_t fd;
Fred Drakea04eaad2000-06-30 02:46:07 +00001581#else
Guido van Rossum30a685f1991-06-27 15:51:29 +00001582 int fd;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001583#endif
Guido van Rossum30a685f1991-06-27 15:51:29 +00001584 FILE *fp;
Guido van Rossum6b144911995-03-14 15:05:13 +00001585 PyObject *f;
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001586#ifdef __VMS
1587 char *mode_r = "r";
1588 char *mode_w = "w";
1589#endif
Guido van Rossum6b144911995-03-14 15:05:13 +00001590
Guido van Rossum43713e52000-02-29 13:59:29 +00001591 if (!PyArg_ParseTuple(args, "|si:makefile", &mode, &bufsize))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001592 return NULL;
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001593#ifdef __VMS
1594 if (strcmp(mode,"rb") == 0) {
1595 mode = mode_r;
1596 }
1597 else {
1598 if (strcmp(mode,"wb") == 0) {
1599 mode = mode_w;
1600 }
1601 }
1602#endif
Guido van Rossum8d665e61996-06-26 18:22:49 +00001603#ifdef MS_WIN32
Guido van Rossum82a5c661998-07-07 20:45:43 +00001604 if (((fd = _open_osfhandle(s->sock_fd, _O_BINARY)) < 0) ||
1605 ((fd = dup(fd)) < 0) || ((fp = fdopen(fd, mode)) == NULL))
Guido van Rossum30b6b2b1995-06-14 22:28:08 +00001606#else
Guido van Rossum82a5c661998-07-07 20:45:43 +00001607 if ((fd = dup(s->sock_fd)) < 0 || (fp = fdopen(fd, mode)) == NULL)
Guido van Rossum30b6b2b1995-06-14 22:28:08 +00001608#endif
Guido van Rossum82a5c661998-07-07 20:45:43 +00001609 {
Guido van Rossum6b144911995-03-14 15:05:13 +00001610 if (fd >= 0)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +00001611 SOCKETCLOSE(fd);
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001612 return s->errorhandler();
Guido van Rossum6b144911995-03-14 15:05:13 +00001613 }
Jack Jansen6d8898b2002-03-25 15:32:01 +00001614#ifdef USE_GUSI2
1615 /* Workaround for bug in Metrowerks MSL vs. GUSI I/O library */
Guido van Rossum67f7a382002-06-06 21:08:16 +00001616 if (strchr(mode, 'b') != NULL)
Jack Jansen6d8898b2002-03-25 15:32:01 +00001617 bufsize = 0;
1618#endif
Guido van Rossum6b144911995-03-14 15:05:13 +00001619 f = PyFile_FromFile(fp, "<socket>", mode, fclose);
1620 if (f != NULL)
1621 PyFile_SetBufSize(f, bufsize);
1622 return f;
Guido van Rossum30a685f1991-06-27 15:51:29 +00001623}
Guido van Rossum82a5c661998-07-07 20:45:43 +00001624
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001625PyDoc_STRVAR(makefile_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001626"makefile([mode[, buffersize]]) -> file object\n\
1627\n\
1628Return a regular file object corresponding to the socket.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001629The mode and buffersize arguments are as for the built-in open() function.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001630
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001631#endif /* NO_DUP */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001632
Guido van Rossum48a680c2001-03-02 06:34:14 +00001633
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00001634/* s.recv(nbytes [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001635
Guido van Rossum73624e91994-10-10 17:59:00 +00001636static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001637sock_recv(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00001638{
Guido van Rossumbe32c891996-06-20 16:25:29 +00001639 int len, n, flags = 0;
Guido van Rossum73624e91994-10-10 17:59:00 +00001640 PyObject *buf;
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001641#ifdef __VMS
1642 int read_length;
1643 char *read_buf;
1644#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00001645
Guido van Rossum43713e52000-02-29 13:59:29 +00001646 if (!PyArg_ParseTuple(args, "i|i:recv", &len, &flags))
Guido van Rossumbe32c891996-06-20 16:25:29 +00001647 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001648
1649 if (len < 0) {
Martin v. Löwisfba64e12001-11-19 10:41:26 +00001650 PyErr_SetString(PyExc_ValueError,
Guido van Rossum11ba0942002-06-13 15:07:44 +00001651 "negative buffersize in recv");
Martin v. Löwisfba64e12001-11-19 10:41:26 +00001652 return NULL;
1653 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00001654
Guido van Rossum73624e91994-10-10 17:59:00 +00001655 buf = PyString_FromStringAndSize((char *) 0, len);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001656 if (buf == NULL)
1657 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001658
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001659#ifndef __VMS
Guido van Rossum73624e91994-10-10 17:59:00 +00001660 Py_BEGIN_ALLOW_THREADS
Guido van Rossum11ba0942002-06-13 15:07:44 +00001661 internal_select(s, 0);
Jeremy Hylton1ba6bad2001-08-20 19:06:36 +00001662 n = recv(s->sock_fd, PyString_AS_STRING(buf), len, flags);
Guido van Rossum73624e91994-10-10 17:59:00 +00001663 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001664
Guido van Rossum7c53b771995-09-13 18:39:47 +00001665 if (n < 0) {
1666 Py_DECREF(buf);
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001667 return s->errorhandler();
Guido van Rossum7c53b771995-09-13 18:39:47 +00001668 }
Tim Peters5de98422002-04-27 18:44:32 +00001669 if (n != len)
1670 _PyString_Resize(&buf, n);
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001671#else
1672 read_buf = PyString_AsString(buf);
1673 read_length = len;
1674 while (read_length != 0) {
1675 unsigned int segment;
1676
1677 segment = read_length /SEGMENT_SIZE;
1678 if (segment != 0) {
1679 segment = SEGMENT_SIZE;
1680 }
1681 else {
1682 segment = read_length;
1683 }
1684
1685 Py_BEGIN_ALLOW_THREADS
1686 internal_select(s, 0);
1687 n = recv(s->sock_fd, read_buf, segment, flags);
1688 Py_END_ALLOW_THREADS
1689
1690 if (n < 0) {
1691 Py_DECREF(buf);
1692 return s->errorhandler();
1693 }
1694 if (n != read_length) {
1695 read_buf += n;
1696 break;
1697 }
1698
1699 read_length -= segment;
1700 read_buf += segment;
1701 }
1702 if (_PyString_Resize(&buf, (read_buf - PyString_AsString(buf))) < 0)
1703 {
1704 return NULL;
1705 }
1706#endif /* !__VMS */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001707 return buf;
1708}
1709
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001710PyDoc_STRVAR(recv_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001711"recv(buffersize[, flags]) -> data\n\
1712\n\
1713Receive up to buffersize bytes from the socket. For the optional flags\n\
1714argument, see the Unix manual. When no data is available, block until\n\
1715at least one byte is available or until the remote end is closed. When\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001716the remote end is closed and all data is read, return the empty string.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001717
Guido van Rossum30a685f1991-06-27 15:51:29 +00001718
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00001719/* s.recvfrom(nbytes [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001720
Guido van Rossum73624e91994-10-10 17:59:00 +00001721static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001722sock_recvfrom(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001723{
1724 char addrbuf[256];
Barry Warsaw752300b1997-01-03 17:18:10 +00001725 PyObject *buf = NULL;
1726 PyObject *addr = NULL;
1727 PyObject *ret = NULL;
Guido van Rossumff3ab422000-04-24 15:16:03 +00001728 int len, n, flags = 0;
1729 socklen_t addrlen;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001730
Guido van Rossum43713e52000-02-29 13:59:29 +00001731 if (!PyArg_ParseTuple(args, "i|i:recvfrom", &len, &flags))
Guido van Rossumbe32c891996-06-20 16:25:29 +00001732 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001733
Guido van Rossum18c9a4f1993-05-25 12:16:29 +00001734 if (!getsockaddrlen(s, &addrlen))
1735 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00001736 buf = PyString_FromStringAndSize((char *) 0, len);
Guido van Rossum18c9a4f1993-05-25 12:16:29 +00001737 if (buf == NULL)
1738 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001739
Guido van Rossum73624e91994-10-10 17:59:00 +00001740 Py_BEGIN_ALLOW_THREADS
Fred Drakefd168342001-05-09 19:11:33 +00001741 memset(addrbuf, 0, addrlen);
Guido van Rossum11ba0942002-06-13 15:07:44 +00001742 internal_select(s, 0);
Fred Drakefd168342001-05-09 19:11:33 +00001743 n = recvfrom(s->sock_fd, PyString_AS_STRING(buf), len, flags,
Guido van Rossum8d665e61996-06-26 18:22:49 +00001744#ifndef MS_WINDOWS
Andrew MacIntyreba43e872002-03-03 03:03:52 +00001745#if defined(PYOS_OS2) && !defined(PYCC_GCC)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00001746 (struct sockaddr *)addrbuf, &addrlen
Guido van Rossum32c575d1997-12-02 20:37:32 +00001747#else
Thomas Wouters334fb892000-07-25 12:56:38 +00001748 (void *)addrbuf, &addrlen
Guido van Rossum32c575d1997-12-02 20:37:32 +00001749#endif
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001750#else
Barry Warsaw752300b1997-01-03 17:18:10 +00001751 (struct sockaddr *)addrbuf, &addrlen
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001752#endif
Barry Warsaw752300b1997-01-03 17:18:10 +00001753 );
Guido van Rossum73624e91994-10-10 17:59:00 +00001754 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001755
Guido van Rossum7c53b771995-09-13 18:39:47 +00001756 if (n < 0) {
1757 Py_DECREF(buf);
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001758 return s->errorhandler();
Guido van Rossum7c53b771995-09-13 18:39:47 +00001759 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00001760
Guido van Rossum53a9bf81996-06-11 18:35:24 +00001761 if (n != len && _PyString_Resize(&buf, n) < 0)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001762 return NULL;
Guido van Rossum48a680c2001-03-02 06:34:14 +00001763
Guido van Rossum67f7a382002-06-06 21:08:16 +00001764 if (!(addr = makesockaddr(s->sock_fd, (struct sockaddr *)addrbuf,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001765 addrlen)))
Barry Warsaw752300b1997-01-03 17:18:10 +00001766 goto finally;
1767
Guido van Rossum73624e91994-10-10 17:59:00 +00001768 ret = Py_BuildValue("OO", buf, addr);
Guido van Rossum67f7a382002-06-06 21:08:16 +00001769
1770finally:
Guido van Rossum73624e91994-10-10 17:59:00 +00001771 Py_XDECREF(addr);
1772 Py_XDECREF(buf);
Guido van Rossum6f5afc91993-02-05 09:46:15 +00001773 return ret;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001774}
1775
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001776PyDoc_STRVAR(recvfrom_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001777"recvfrom(buffersize[, flags]) -> (data, address info)\n\
1778\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001779Like recv(buffersize, flags) but also return the sender's address info.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001780
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00001781/* s.send(data [,flags]) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001782
Guido van Rossum73624e91994-10-10 17:59:00 +00001783static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001784sock_send(PySocketSockObject *s, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00001785{
Guido van Rossumff4949e1992-08-05 19:58:53 +00001786 char *buf;
Guido van Rossumbe32c891996-06-20 16:25:29 +00001787 int len, n, flags = 0;
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001788#ifdef __VMS
1789 int send_length;
1790#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00001791
Guido van Rossum43713e52000-02-29 13:59:29 +00001792 if (!PyArg_ParseTuple(args, "s#|i:send", &buf, &len, &flags))
Guido van Rossumbe32c891996-06-20 16:25:29 +00001793 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001794
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001795#ifndef __VMS
Guido van Rossum73624e91994-10-10 17:59:00 +00001796 Py_BEGIN_ALLOW_THREADS
Guido van Rossum11ba0942002-06-13 15:07:44 +00001797 internal_select(s, 1);
Guido van Rossumff4949e1992-08-05 19:58:53 +00001798 n = send(s->sock_fd, buf, len, flags);
Guido van Rossum73624e91994-10-10 17:59:00 +00001799 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001800
Guido van Rossum30a685f1991-06-27 15:51:29 +00001801 if (n < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001802 return s->errorhandler();
Martin v. Löwis79acb9e2002-12-06 12:48:53 +00001803#else
1804 /* Divide packet into smaller segments for */
1805 /* TCP/IP Services for OpenVMS */
1806 send_length = len;
1807 while (send_length != 0) {
1808 unsigned int segment;
1809
1810 segment = send_length / SEGMENT_SIZE;
1811 if (segment != 0) {
1812 segment = SEGMENT_SIZE;
1813 }
1814 else {
1815 segment = send_length;
1816 }
1817 Py_BEGIN_ALLOW_THREADS
1818 internal_select(s, 1);
1819 n = send(s->sock_fd, buf, segment, flags);
1820 Py_END_ALLOW_THREADS
1821 if (n < 0) {
1822 return s->errorhandler();
1823 }
1824 send_length -= segment;
1825 buf += segment;
1826 } /* end while */
1827#endif /* !__VMS */
Guido van Rossum73624e91994-10-10 17:59:00 +00001828 return PyInt_FromLong((long)n);
Guido van Rossum30a685f1991-06-27 15:51:29 +00001829}
1830
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001831PyDoc_STRVAR(send_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00001832"send(data[, flags]) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00001833\n\
1834Send a data string to the socket. For the optional flags\n\
Guido van Rossum9f7a5392001-10-26 03:25:00 +00001835argument, see the Unix manual. Return the number of bytes\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001836sent; this may be less than len(data) if the network is busy.");
Guido van Rossum9f7a5392001-10-26 03:25:00 +00001837
1838
1839/* s.sendall(data [,flags]) method */
1840
1841static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001842sock_sendall(PySocketSockObject *s, PyObject *args)
Guido van Rossum9f7a5392001-10-26 03:25:00 +00001843{
1844 char *buf;
Martin v. Löwise6cc5b62002-02-16 23:13:54 +00001845 int len, n, flags = 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001846
Guido van Rossum9f7a5392001-10-26 03:25:00 +00001847 if (!PyArg_ParseTuple(args, "s#|i:sendall", &buf, &len, &flags))
1848 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001849
Guido van Rossum9f7a5392001-10-26 03:25:00 +00001850 Py_BEGIN_ALLOW_THREADS
1851 do {
Guido van Rossum8f24cdc2003-01-31 18:15:58 +00001852 internal_select(s, 1);
Guido van Rossum9f7a5392001-10-26 03:25:00 +00001853 n = send(s->sock_fd, buf, len, flags);
1854 if (n < 0)
1855 break;
Guido van Rossum9f7a5392001-10-26 03:25:00 +00001856 buf += n;
1857 len -= n;
1858 } while (len > 0);
1859 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001860
Guido van Rossum9f7a5392001-10-26 03:25:00 +00001861 if (n < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001862 return s->errorhandler();
Guido van Rossum67f7a382002-06-06 21:08:16 +00001863
Guido van Rossum9f7a5392001-10-26 03:25:00 +00001864 Py_INCREF(Py_None);
1865 return Py_None;
1866}
1867
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001868PyDoc_STRVAR(sendall_doc,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00001869"sendall(data[, flags])\n\
1870\n\
1871Send a data string to the socket. For the optional flags\n\
1872argument, see the Unix manual. This calls send() repeatedly\n\
1873until all data is sent. If an error occurs, it's impossible\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001874to tell how much data has been sent.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001875
Guido van Rossum30a685f1991-06-27 15:51:29 +00001876
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00001877/* s.sendto(data, [flags,] sockaddr) method */
Guido van Rossum30a685f1991-06-27 15:51:29 +00001878
Guido van Rossum73624e91994-10-10 17:59:00 +00001879static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001880sock_sendto(PySocketSockObject *s, PyObject *args)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001881{
Guido van Rossum73624e91994-10-10 17:59:00 +00001882 PyObject *addro;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001883 char *buf;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001884 struct sockaddr *addr;
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00001885 int addrlen, len, n, flags;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001886
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00001887 flags = 0;
Guido van Rossum27fc3c02000-03-24 20:56:56 +00001888 if (!PyArg_ParseTuple(args, "s#O:sendto", &buf, &len, &addro)) {
Guido van Rossum73624e91994-10-10 17:59:00 +00001889 PyErr_Clear();
Guido van Rossum27fc3c02000-03-24 20:56:56 +00001890 if (!PyArg_ParseTuple(args, "s#iO:sendto",
1891 &buf, &len, &flags, &addro))
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00001892 return NULL;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001893 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00001894
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00001895 if (!getsockaddrarg(s, addro, &addr, &addrlen))
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001896 return NULL;
Guido van Rossum67f7a382002-06-06 21:08:16 +00001897
Guido van Rossum73624e91994-10-10 17:59:00 +00001898 Py_BEGIN_ALLOW_THREADS
Guido van Rossum11ba0942002-06-13 15:07:44 +00001899 internal_select(s, 1);
Guido van Rossumeb6b33a1993-05-25 09:38:27 +00001900 n = sendto(s->sock_fd, buf, len, flags, addr, addrlen);
Guido van Rossum73624e91994-10-10 17:59:00 +00001901 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00001902
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001903 if (n < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001904 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001905 return PyInt_FromLong((long)n);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001906}
1907
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001908PyDoc_STRVAR(sendto_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00001909"sendto(data[, flags], address) -> count\n\
Guido van Rossum82a5c661998-07-07 20:45:43 +00001910\n\
1911Like send(data, flags) but allows specifying the destination address.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001912For IP sockets, the address is a pair (hostaddr, port).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001913
Guido van Rossum30a685f1991-06-27 15:51:29 +00001914
1915/* s.shutdown(how) method */
1916
Guido van Rossum73624e91994-10-10 17:59:00 +00001917static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001918sock_shutdown(PySocketSockObject *s, PyObject *arg)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001919{
1920 int how;
Guido van Rossumff4949e1992-08-05 19:58:53 +00001921 int res;
Jeremy Hyltonae0013d2001-10-11 16:36:35 +00001922
1923 how = PyInt_AsLong(arg);
1924 if (how == -1 && PyErr_Occurred())
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001925 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00001926 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001927 res = shutdown(s->sock_fd, how);
Guido van Rossum73624e91994-10-10 17:59:00 +00001928 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00001929 if (res < 0)
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00001930 return s->errorhandler();
Guido van Rossum73624e91994-10-10 17:59:00 +00001931 Py_INCREF(Py_None);
1932 return Py_None;
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001933}
1934
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001935PyDoc_STRVAR(shutdown_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00001936"shutdown(flag)\n\
1937\n\
1938Shut down the reading side of the socket (flag == 0), the writing side\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001939of the socket (flag == 1), or both ends (flag == 2).");
Guido van Rossum82a5c661998-07-07 20:45:43 +00001940
Guido van Rossum30a685f1991-06-27 15:51:29 +00001941
1942/* List of methods for socket objects */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00001943
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001944static PyMethodDef sock_methods[] = {
1945 {"accept", (PyCFunction)sock_accept, METH_NOARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001946 accept_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001947 {"bind", (PyCFunction)sock_bind, METH_O,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001948 bind_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001949 {"close", (PyCFunction)sock_close, METH_NOARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001950 close_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001951 {"connect", (PyCFunction)sock_connect, METH_O,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001952 connect_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001953 {"connect_ex", (PyCFunction)sock_connect_ex, METH_O,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001954 connect_ex_doc},
Guido van Rossumbe32c891996-06-20 16:25:29 +00001955#ifndef NO_DUP
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001956 {"dup", (PyCFunction)sock_dup, METH_NOARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001957 dup_doc},
Guido van Rossumbe32c891996-06-20 16:25:29 +00001958#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001959 {"fileno", (PyCFunction)sock_fileno, METH_NOARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001960 fileno_doc},
Guido van Rossumb6775db1994-08-01 11:34:53 +00001961#ifdef HAVE_GETPEERNAME
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001962 {"getpeername", (PyCFunction)sock_getpeername,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001963 METH_NOARGS, getpeername_doc},
Guido van Rossum9575a441993-04-07 14:06:14 +00001964#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001965 {"getsockname", (PyCFunction)sock_getsockname,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00001966 METH_NOARGS, getsockname_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001967 {"getsockopt", (PyCFunction)sock_getsockopt, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001968 getsockopt_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001969 {"listen", (PyCFunction)sock_listen, METH_O,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001970 listen_doc},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001971#ifndef NO_DUP
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001972 {"makefile", (PyCFunction)sock_makefile, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001973 makefile_doc},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00001974#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001975 {"recv", (PyCFunction)sock_recv, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001976 recv_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001977 {"recvfrom", (PyCFunction)sock_recvfrom, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001978 recvfrom_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001979 {"send", (PyCFunction)sock_send, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001980 send_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001981 {"sendall", (PyCFunction)sock_sendall, METH_VARARGS,
Guido van Rossum9f7a5392001-10-26 03:25:00 +00001982 sendall_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001983 {"sendto", (PyCFunction)sock_sendto, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001984 sendto_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001985 {"setblocking", (PyCFunction)sock_setblocking, METH_O,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001986 setblocking_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001987 {"settimeout", (PyCFunction)sock_settimeout, METH_O,
Guido van Rossum67f7a382002-06-06 21:08:16 +00001988 settimeout_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001989 {"gettimeout", (PyCFunction)sock_gettimeout, METH_NOARGS,
Guido van Rossum67f7a382002-06-06 21:08:16 +00001990 gettimeout_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001991 {"setsockopt", (PyCFunction)sock_setsockopt, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001992 setsockopt_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001993 {"shutdown", (PyCFunction)sock_shutdown, METH_O,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001994 shutdown_doc},
1995#ifdef RISCOS
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00001996 {"sleeptaskw", (PyCFunction)sock_sleeptaskw, METH_VARARGS,
Guido van Rossum48a680c2001-03-02 06:34:14 +00001997 sleeptaskw_doc},
1998#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +00001999 {NULL, NULL} /* sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002000};
2001
Guido van Rossum30a685f1991-06-27 15:51:29 +00002002
Guido van Rossum73624e91994-10-10 17:59:00 +00002003/* Deallocate a socket object in response to the last Py_DECREF().
Guido van Rossum30a685f1991-06-27 15:51:29 +00002004 First close the file description. */
2005
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002006static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002007sock_dealloc(PySocketSockObject *s)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002008{
Guido van Rossumfa972c92000-04-10 12:45:45 +00002009 if (s->sock_fd != -1)
Guido van Rossum2dd8ddd2000-04-21 20:33:00 +00002010 (void) SOCKETCLOSE(s->sock_fd);
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002011 s->ob_type->tp_free((PyObject *)s);
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002012}
2013
Guido van Rossum30a685f1991-06-27 15:51:29 +00002014
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002015static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002016sock_repr(PySocketSockObject *s)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002017{
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002018 char buf[512];
Fred Drakea04eaad2000-06-30 02:46:07 +00002019#if SIZEOF_SOCKET_T > SIZEOF_LONG
2020 if (s->sock_fd > LONG_MAX) {
2021 /* this can occur on Win64, and actually there is a special
2022 ugly printf formatter for decimal pointer length integer
2023 printing, only bother if necessary*/
2024 PyErr_SetString(PyExc_OverflowError,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002025 "no printf formatter to display "
2026 "the socket descriptor in decimal");
Fred Drakea04eaad2000-06-30 02:46:07 +00002027 return NULL;
2028 }
2029#endif
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002030 PyOS_snprintf(
2031 buf, sizeof(buf),
2032 "<socket object, fd=%ld, family=%d, type=%d, protocol=%d>",
2033 (long)s->sock_fd, s->sock_family,
2034 s->sock_type,
2035 s->sock_proto);
Guido van Rossumdc1c64a1996-02-27 15:21:19 +00002036 return PyString_FromString(buf);
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002037}
2038
2039
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002040/* Create a new, uninitialized socket object. */
2041
2042static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002043sock_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002044{
2045 PyObject *new;
2046
2047 new = type->tp_alloc(type, 0);
Guido van Rossume1c478f2002-06-06 20:08:25 +00002048 if (new != NULL) {
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002049 ((PySocketSockObject *)new)->sock_fd = -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002050 ((PySocketSockObject *)new)->sock_timeout = -1.0;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002051 ((PySocketSockObject *)new)->errorhandler = &set_error;
Guido van Rossume1c478f2002-06-06 20:08:25 +00002052 }
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002053 return new;
2054}
2055
2056
2057/* Initialize a new socket object. */
2058
2059/*ARGSUSED*/
2060static int
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00002061sock_initobj(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002062{
2063 PySocketSockObject *s = (PySocketSockObject *)self;
2064 SOCKET_T fd;
2065 int family = AF_INET, type = SOCK_STREAM, proto = 0;
2066 static char *keywords[] = {"family", "type", "proto", 0};
2067
2068 if (!PyArg_ParseTupleAndKeywords(args, kwds,
2069 "|iii:socket", keywords,
2070 &family, &type, &proto))
2071 return -1;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002072
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002073 Py_BEGIN_ALLOW_THREADS
2074 fd = socket(family, type, proto);
2075 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002076
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002077#ifdef MS_WINDOWS
2078 if (fd == INVALID_SOCKET)
2079#else
2080 if (fd < 0)
2081#endif
2082 {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002083 set_error();
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002084 return -1;
2085 }
2086 init_sockobject(s, fd, family, type, proto);
2087 /* From now on, ignore SIGPIPE and let the error checking
2088 do the work. */
2089#ifdef SIGPIPE
2090 (void) signal(SIGPIPE, SIG_IGN);
2091#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00002092
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002093 return 0;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002094
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002095}
2096
2097
Guido van Rossumb6775db1994-08-01 11:34:53 +00002098/* Type object for socket objects. */
Guido van Rossum30a685f1991-06-27 15:51:29 +00002099
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002100static PyTypeObject sock_type = {
Guido van Rossum55558541996-05-23 22:54:50 +00002101 PyObject_HEAD_INIT(0) /* Must fill in type value later */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002102 0, /* ob_size */
Guido van Rossumca6dfa52001-10-28 12:31:33 +00002103 "_socket.socket", /* tp_name */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002104 sizeof(PySocketSockObject), /* tp_basicsize */
2105 0, /* tp_itemsize */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002106 (destructor)sock_dealloc, /* tp_dealloc */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002107 0, /* tp_print */
2108 0, /* tp_getattr */
2109 0, /* tp_setattr */
2110 0, /* tp_compare */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002111 (reprfunc)sock_repr, /* tp_repr */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002112 0, /* tp_as_number */
2113 0, /* tp_as_sequence */
2114 0, /* tp_as_mapping */
2115 0, /* tp_hash */
2116 0, /* tp_call */
2117 0, /* tp_str */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00002118 PyObject_GenericGetAttr, /* tp_getattro */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002119 0, /* tp_setattro */
2120 0, /* tp_as_buffer */
2121 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002122 sock_doc, /* tp_doc */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002123 0, /* tp_traverse */
2124 0, /* tp_clear */
2125 0, /* tp_richcompare */
2126 0, /* tp_weaklistoffset */
2127 0, /* tp_iter */
2128 0, /* tp_iternext */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002129 sock_methods, /* tp_methods */
Guido van Rossum384ca9c2001-10-27 22:20:47 +00002130 0, /* tp_members */
2131 0, /* tp_getset */
2132 0, /* tp_base */
2133 0, /* tp_dict */
2134 0, /* tp_descr_get */
2135 0, /* tp_descr_set */
2136 0, /* tp_dictoffset */
Andrew MacIntyre7aec4a22002-06-13 11:53:52 +00002137 sock_initobj, /* tp_init */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00002138 PyType_GenericAlloc, /* tp_alloc */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002139 sock_new, /* tp_new */
Jason Tishlerfb8595d2003-01-06 12:41:26 +00002140 PyObject_Del, /* tp_free */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00002141};
2142
Guido van Rossum30a685f1991-06-27 15:51:29 +00002143
Guido van Rossum81194471991-07-27 21:42:02 +00002144/* Python interface to gethostname(). */
2145
2146/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002147static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002148socket_gethostname(PyObject *self, PyObject *args)
Guido van Rossum81194471991-07-27 21:42:02 +00002149{
2150 char buf[1024];
Guido van Rossumff4949e1992-08-05 19:58:53 +00002151 int res;
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002152 if (!PyArg_ParseTuple(args, ":gethostname"))
Guido van Rossum81194471991-07-27 21:42:02 +00002153 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00002154 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002155 res = gethostname(buf, (int) sizeof buf - 1);
Guido van Rossum73624e91994-10-10 17:59:00 +00002156 Py_END_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002157 if (res < 0)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002158 return set_error();
Guido van Rossum81194471991-07-27 21:42:02 +00002159 buf[sizeof buf - 1] = '\0';
Guido van Rossum73624e91994-10-10 17:59:00 +00002160 return PyString_FromString(buf);
Guido van Rossum81194471991-07-27 21:42:02 +00002161}
Guido van Rossumff4949e1992-08-05 19:58:53 +00002162
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002163PyDoc_STRVAR(gethostname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002164"gethostname() -> string\n\
2165\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002166Return the current host name.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002167
Guido van Rossumff4949e1992-08-05 19:58:53 +00002168
Guido van Rossum30a685f1991-06-27 15:51:29 +00002169/* Python interface to gethostbyname(name). */
2170
2171/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002172static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002173socket_gethostbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002174{
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002175 char *name;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002176 struct sockaddr_storage addrbuf;
2177
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002178 if (!PyArg_ParseTuple(args, "s:gethostbyname", &name))
Guido van Rossum30a685f1991-06-27 15:51:29 +00002179 return NULL;
Martin v. Löwisddc6f472002-07-28 16:10:31 +00002180 if (setipaddr(name, (struct sockaddr *)&addrbuf, sizeof(addrbuf), AF_INET) < 0)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002181 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002182 return makeipaddr((struct sockaddr *)&addrbuf,
2183 sizeof(struct sockaddr_in));
Guido van Rossum30a685f1991-06-27 15:51:29 +00002184}
2185
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002186PyDoc_STRVAR(gethostbyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002187"gethostbyname(host) -> address\n\
2188\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002189Return the IP address (a string of the form '255.255.255.255') for a host.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002190
2191
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002192/* Convenience function common to gethostbyname_ex and gethostbyaddr */
2193
2194static PyObject *
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002195gethost_common(struct hostent *h, struct sockaddr *addr, int alen, int af)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002196{
2197 char **pch;
2198 PyObject *rtn_tuple = (PyObject *)NULL;
2199 PyObject *name_list = (PyObject *)NULL;
2200 PyObject *addr_list = (PyObject *)NULL;
2201 PyObject *tmp;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002202
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002203 if (h == NULL) {
Guido van Rossum48a680c2001-03-02 06:34:14 +00002204 /* Let's get real error message to return */
Guido van Rossume2ae77b2001-10-24 20:42:55 +00002205#ifndef RISCOS
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002206 set_herror(h_errno);
Guido van Rossume2ae77b2001-10-24 20:42:55 +00002207#else
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002208 PyErr_SetString(socket_error, "host not found");
Guido van Rossume2ae77b2001-10-24 20:42:55 +00002209#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002210 return NULL;
2211 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002212
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002213 if (h->h_addrtype != af) {
2214#ifdef HAVE_STRERROR
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002215 /* Let's get real error message to return */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002216 PyErr_SetString(socket_error,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002217 (char *)strerror(EAFNOSUPPORT));
2218#else
2219 PyErr_SetString(
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002220 socket_error,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002221 "Address family not supported by protocol family");
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002222#endif
2223 return NULL;
2224 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002225
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002226 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00002227
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002228 case AF_INET:
2229 if (alen < sizeof(struct sockaddr_in))
2230 return NULL;
2231 break;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002232
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00002233#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002234 case AF_INET6:
2235 if (alen < sizeof(struct sockaddr_in6))
2236 return NULL;
2237 break;
2238#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00002239
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002240 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002241
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002242 if ((name_list = PyList_New(0)) == NULL)
2243 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002244
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002245 if ((addr_list = PyList_New(0)) == NULL)
2246 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002247
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002248 for (pch = h->h_aliases; *pch != NULL; pch++) {
2249 int status;
2250 tmp = PyString_FromString(*pch);
2251 if (tmp == NULL)
2252 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002253
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002254 status = PyList_Append(name_list, tmp);
2255 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002256
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002257 if (status)
2258 goto err;
2259 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002260
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002261 for (pch = h->h_addr_list; *pch != NULL; pch++) {
2262 int status;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002263
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002264 switch (af) {
Guido van Rossum67f7a382002-06-06 21:08:16 +00002265
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002266 case AF_INET:
2267 {
2268 struct sockaddr_in sin;
2269 memset(&sin, 0, sizeof(sin));
2270 sin.sin_family = af;
2271#ifdef HAVE_SOCKADDR_SA_LEN
2272 sin.sin_len = sizeof(sin);
2273#endif
2274 memcpy(&sin.sin_addr, *pch, sizeof(sin.sin_addr));
2275 tmp = makeipaddr((struct sockaddr *)&sin, sizeof(sin));
Guido van Rossum67f7a382002-06-06 21:08:16 +00002276
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002277 if (pch == h->h_addr_list && alen >= sizeof(sin))
2278 memcpy((char *) addr, &sin, sizeof(sin));
2279 break;
2280 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002281
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00002282#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002283 case AF_INET6:
2284 {
2285 struct sockaddr_in6 sin6;
2286 memset(&sin6, 0, sizeof(sin6));
2287 sin6.sin6_family = af;
2288#ifdef HAVE_SOCKADDR_SA_LEN
2289 sin6.sin6_len = sizeof(sin6);
2290#endif
2291 memcpy(&sin6.sin6_addr, *pch, sizeof(sin6.sin6_addr));
2292 tmp = makeipaddr((struct sockaddr *)&sin6,
2293 sizeof(sin6));
Guido van Rossum67f7a382002-06-06 21:08:16 +00002294
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002295 if (pch == h->h_addr_list && alen >= sizeof(sin6))
2296 memcpy((char *) addr, &sin6, sizeof(sin6));
2297 break;
2298 }
2299#endif
Guido van Rossum67f7a382002-06-06 21:08:16 +00002300
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002301 default: /* can't happen */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002302 PyErr_SetString(socket_error,
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002303 "unsupported address family");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002304 return NULL;
2305 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002306
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002307 if (tmp == NULL)
2308 goto err;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002309
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002310 status = PyList_Append(addr_list, tmp);
2311 Py_DECREF(tmp);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002312
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002313 if (status)
2314 goto err;
2315 }
Guido van Rossum67f7a382002-06-06 21:08:16 +00002316
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002317 rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list);
Guido van Rossum67f7a382002-06-06 21:08:16 +00002318
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002319 err:
2320 Py_XDECREF(name_list);
2321 Py_XDECREF(addr_list);
2322 return rtn_tuple;
2323}
2324
2325
2326/* Python interface to gethostbyname_ex(name). */
2327
2328/*ARGSUSED*/
2329static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002330socket_gethostbyname_ex(PyObject *self, PyObject *args)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002331{
2332 char *name;
2333 struct hostent *h;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002334 struct sockaddr_storage addr;
Martin v. Löwis9db2f572001-07-23 01:30:10 +00002335 struct sockaddr *sa;
Guido van Rossum955becc1999-03-22 20:14:53 +00002336 PyObject *ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002337#ifdef HAVE_GETHOSTBYNAME_R
2338 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002339#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
2340 struct hostent_data data;
2341#else
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002342 char buf[16384];
2343 int buf_len = (sizeof buf) - 1;
2344 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002345#endif
2346#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Guido van Rossume9cd07b1999-03-15 21:40:14 +00002347 int result;
2348#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002349#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002350
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002351 if (!PyArg_ParseTuple(args, "s:gethostbyname_ex", &name))
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002352 return NULL;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00002353 if (setipaddr(name, (struct sockaddr *)&addr, sizeof(addr), AF_INET) < 0)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002354 return NULL;
2355 Py_BEGIN_ALLOW_THREADS
2356#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002357#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002358 result = gethostbyname_r(name, &hp_allocated, buf, buf_len,
2359 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002360#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002361 h = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002362#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Guido van Rossume7de2061999-03-24 17:24:33 +00002363 memset((void *) &data, '\0', sizeof(data));
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002364 result = gethostbyname_r(name, &hp_allocated, &data);
2365 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00002366#endif
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002367#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00002368#ifdef USE_GETHOSTBYNAME_LOCK
2369 PyThread_acquire_lock(gethostbyname_lock, 1);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002370#endif
2371 h = gethostbyname(name);
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002372#endif /* HAVE_GETHOSTBYNAME_R */
2373 Py_END_ALLOW_THREADS
Guido van Rossum67f7a382002-06-06 21:08:16 +00002374 /* Some C libraries would require addr.__ss_family instead of
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002375 addr.ss_family.
2376 Therefore, we cast the sockaddr_storage into sockaddr to
2377 access sa_family. */
Martin v. Löwis9db2f572001-07-23 01:30:10 +00002378 sa = (struct sockaddr*)&addr;
Guido van Rossum67f7a382002-06-06 21:08:16 +00002379 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr),
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002380 sa->sa_family);
Guido van Rossum3baaa131999-03-22 21:44:51 +00002381#ifdef USE_GETHOSTBYNAME_LOCK
Guido van Rossum955becc1999-03-22 20:14:53 +00002382 PyThread_release_lock(gethostbyname_lock);
2383#endif
2384 return ret;
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002385}
2386
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002387PyDoc_STRVAR(ghbn_ex_doc,
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002388"gethostbyname_ex(host) -> (name, aliaslist, addresslist)\n\
2389\n\
2390Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002391for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum7d896ab1998-08-04 22:16:43 +00002392
2393
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002394/* Python interface to gethostbyaddr(IP). */
2395
2396/*ARGSUSED*/
2397static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002398socket_gethostbyaddr(PyObject *self, PyObject *args)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002399{
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00002400#ifdef ENABLE_IPV6
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002401 struct sockaddr_storage addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002402#else
Guido van Rossum48a680c2001-03-02 06:34:14 +00002403 struct sockaddr_in addr;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002404#endif
2405 struct sockaddr *sa = (struct sockaddr *)&addr;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002406 char *ip_num;
2407 struct hostent *h;
Guido van Rossum3baaa131999-03-22 21:44:51 +00002408 PyObject *ret;
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002409#ifdef HAVE_GETHOSTBYNAME_R
2410 struct hostent hp_allocated;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002411#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
2412 struct hostent_data data;
2413#else
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002414 char buf[16384];
2415 int buf_len = (sizeof buf) - 1;
2416 int errnop;
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002417#endif
2418#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Guido van Rossume9cd07b1999-03-15 21:40:14 +00002419 int result;
2420#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002421#endif /* HAVE_GETHOSTBYNAME_R */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002422 char *ap;
2423 int al;
2424 int af;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002425
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002426 if (!PyArg_ParseTuple(args, "s:gethostbyaddr", &ip_num))
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002427 return NULL;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00002428 af = AF_UNSPEC;
Martin v. Löwisddc6f472002-07-28 16:10:31 +00002429 if (setipaddr(ip_num, sa, sizeof(addr), af) < 0)
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002430 return NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002431 af = sa->sa_family;
2432 ap = NULL;
2433 al = 0;
2434 switch (af) {
2435 case AF_INET:
2436 ap = (char *)&((struct sockaddr_in *)sa)->sin_addr;
2437 al = sizeof(((struct sockaddr_in *)sa)->sin_addr);
2438 break;
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00002439#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002440 case AF_INET6:
2441 ap = (char *)&((struct sockaddr_in6 *)sa)->sin6_addr;
2442 al = sizeof(((struct sockaddr_in6 *)sa)->sin6_addr);
2443 break;
2444#endif
2445 default:
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002446 PyErr_SetString(socket_error, "unsupported address family");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002447 return NULL;
2448 }
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002449 Py_BEGIN_ALLOW_THREADS
2450#ifdef HAVE_GETHOSTBYNAME_R
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002451#if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002452 result = gethostbyaddr_r(ap, al, af,
2453 &hp_allocated, buf, buf_len,
Guido van Rossume9cd07b1999-03-15 21:40:14 +00002454 &h, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002455#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002456 h = gethostbyaddr_r(ap, al, af,
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002457 &hp_allocated, buf, buf_len, &errnop);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002458#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
Guido van Rossume7de2061999-03-24 17:24:33 +00002459 memset((void *) &data, '\0', sizeof(data));
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002460 result = gethostbyaddr_r(ap, al, af, &hp_allocated, &data);
Guido van Rossum7b6c71f1999-03-24 17:20:40 +00002461 h = (result != 0) ? NULL : &hp_allocated;
Guido van Rossume9cd07b1999-03-15 21:40:14 +00002462#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002463#else /* not HAVE_GETHOSTBYNAME_R */
Guido van Rossum3baaa131999-03-22 21:44:51 +00002464#ifdef USE_GETHOSTBYNAME_LOCK
2465 PyThread_acquire_lock(gethostbyname_lock, 1);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002466#endif
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002467 h = gethostbyaddr(ap, al, af);
Guido van Rossum4f199ea1998-04-09 20:56:35 +00002468#endif /* HAVE_GETHOSTBYNAME_R */
2469 Py_END_ALLOW_THREADS
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002470 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr), af);
Guido van Rossum3baaa131999-03-22 21:44:51 +00002471#ifdef USE_GETHOSTBYNAME_LOCK
2472 PyThread_release_lock(gethostbyname_lock);
2473#endif
2474 return ret;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +00002475}
2476
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002477PyDoc_STRVAR(gethostbyaddr_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002478"gethostbyaddr(host) -> (name, aliaslist, addresslist)\n\
2479\n\
2480Return the true host name, a list of aliases, and a list of IP addresses,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002481for a host. The host argument is a string giving a host name or IP number.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002482
Guido van Rossum30a685f1991-06-27 15:51:29 +00002483
2484/* Python interface to getservbyname(name).
2485 This only returns the port number, since the other info is already
2486 known or not useful (like the list of aliases). */
2487
2488/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002489static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002490socket_getservbyname(PyObject *self, PyObject *args)
Guido van Rossum30a685f1991-06-27 15:51:29 +00002491{
Guido van Rossumff4949e1992-08-05 19:58:53 +00002492 char *name, *proto;
Guido van Rossum30a685f1991-06-27 15:51:29 +00002493 struct servent *sp;
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002494 if (!PyArg_ParseTuple(args, "ss:getservbyname", &name, &proto))
Guido van Rossum30a685f1991-06-27 15:51:29 +00002495 return NULL;
Guido van Rossum73624e91994-10-10 17:59:00 +00002496 Py_BEGIN_ALLOW_THREADS
Guido van Rossumff4949e1992-08-05 19:58:53 +00002497 sp = getservbyname(name, proto);
Guido van Rossum73624e91994-10-10 17:59:00 +00002498 Py_END_ALLOW_THREADS
Guido van Rossum30a685f1991-06-27 15:51:29 +00002499 if (sp == NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002500 PyErr_SetString(socket_error, "service/proto not found");
Guido van Rossum30a685f1991-06-27 15:51:29 +00002501 return NULL;
2502 }
Guido van Rossum73624e91994-10-10 17:59:00 +00002503 return PyInt_FromLong((long) ntohs(sp->s_port));
Guido van Rossum30a685f1991-06-27 15:51:29 +00002504}
2505
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002506PyDoc_STRVAR(getservbyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002507"getservbyname(servicename, protocolname) -> integer\n\
2508\n\
2509Return a port number from a service name and protocol name.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002510The protocol name should be 'tcp' or 'udp'.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002511
Guido van Rossum30a685f1991-06-27 15:51:29 +00002512
Guido van Rossum3901d851996-12-19 16:35:04 +00002513/* Python interface to getprotobyname(name).
2514 This only returns the protocol number, since the other info is
2515 already known or not useful (like the list of aliases). */
2516
2517/*ARGSUSED*/
2518static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002519socket_getprotobyname(PyObject *self, PyObject *args)
Guido van Rossum3901d851996-12-19 16:35:04 +00002520{
2521 char *name;
2522 struct protoent *sp;
Guido van Rossumbcc20741998-08-04 22:53:56 +00002523#ifdef __BEOS__
2524/* Not available in BeOS yet. - [cjh] */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002525 PyErr_SetString(socket_error, "getprotobyname not supported");
Guido van Rossumbcc20741998-08-04 22:53:56 +00002526 return NULL;
2527#else
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002528 if (!PyArg_ParseTuple(args, "s:getprotobyname", &name))
Guido van Rossum3901d851996-12-19 16:35:04 +00002529 return NULL;
2530 Py_BEGIN_ALLOW_THREADS
2531 sp = getprotobyname(name);
2532 Py_END_ALLOW_THREADS
2533 if (sp == NULL) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002534 PyErr_SetString(socket_error, "protocol not found");
Guido van Rossum3901d851996-12-19 16:35:04 +00002535 return NULL;
2536 }
2537 return PyInt_FromLong((long) sp->p_proto);
Guido van Rossumbcc20741998-08-04 22:53:56 +00002538#endif
Guido van Rossum3901d851996-12-19 16:35:04 +00002539}
2540
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002541PyDoc_STRVAR(getprotobyname_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002542"getprotobyname(name) -> integer\n\
2543\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002544Return the protocol number for the named protocol. (Rarely used.)");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002545
Guido van Rossum3901d851996-12-19 16:35:04 +00002546
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002547#ifndef NO_DUP
Guido van Rossum2a7178e1992-12-08 13:38:24 +00002548/* Create a socket object from a numeric file description.
2549 Useful e.g. if stdin is a socket.
2550 Additional arguments as for socket(). */
2551
2552/*ARGSUSED*/
Guido van Rossum73624e91994-10-10 17:59:00 +00002553static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002554socket_fromfd(PyObject *self, PyObject *args)
Guido van Rossum2a7178e1992-12-08 13:38:24 +00002555{
Guido van Rossum73624e91994-10-10 17:59:00 +00002556 PySocketSockObject *s;
Fred Drakea04eaad2000-06-30 02:46:07 +00002557 SOCKET_T fd;
2558 int family, type, proto = 0;
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002559 if (!PyArg_ParseTuple(args, "iii|i:fromfd",
2560 &fd, &family, &type, &proto))
Guido van Rossumbe32c891996-06-20 16:25:29 +00002561 return NULL;
Guido van Rossum5f59d601992-12-14 16:59:51 +00002562 /* Dup the fd so it and the socket can be closed independently */
2563 fd = dup(fd);
2564 if (fd < 0)
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002565 return set_error();
2566 s = new_sockobject(fd, family, type, proto);
Guido van Rossum2a7178e1992-12-08 13:38:24 +00002567 /* From now on, ignore SIGPIPE and let the error checking
2568 do the work. */
Guido van Rossum48a680c2001-03-02 06:34:14 +00002569#ifdef SIGPIPE
Guido van Rossum2a7178e1992-12-08 13:38:24 +00002570 (void) signal(SIGPIPE, SIG_IGN);
Guido van Rossum48a680c2001-03-02 06:34:14 +00002571#endif
Guido van Rossum73624e91994-10-10 17:59:00 +00002572 return (PyObject *) s;
Guido van Rossum2a7178e1992-12-08 13:38:24 +00002573}
Guido van Rossum82a5c661998-07-07 20:45:43 +00002574
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002575PyDoc_STRVAR(fromfd_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002576"fromfd(fd, family, type[, proto]) -> socket object\n\
2577\n\
2578Create a socket object from the given file descriptor.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002579The remaining arguments are the same as for socket().");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002580
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00002581#endif /* NO_DUP */
Guido van Rossum2a7178e1992-12-08 13:38:24 +00002582
Guido van Rossum82a5c661998-07-07 20:45:43 +00002583
Guido van Rossum006bf911996-06-12 04:04:55 +00002584static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002585socket_ntohs(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00002586{
2587 int x1, x2;
2588
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002589 if (!PyArg_ParseTuple(args, "i:ntohs", &x1)) {
Guido van Rossum006bf911996-06-12 04:04:55 +00002590 return NULL;
2591 }
2592 x2 = (int)ntohs((short)x1);
2593 return PyInt_FromLong(x2);
2594}
2595
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002596PyDoc_STRVAR(ntohs_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002597"ntohs(integer) -> integer\n\
2598\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002599Convert a 16-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002600
2601
Guido van Rossum006bf911996-06-12 04:04:55 +00002602static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002603socket_ntohl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00002604{
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002605 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00002606
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002607 if (PyInt_Check(arg)) {
2608 x = PyInt_AS_LONG(arg);
Jeremy Hylton825e47b2002-07-25 16:37:51 +00002609 if (x == (unsigned long) -1 && PyErr_Occurred())
2610 return NULL;
Guido van Rossum006bf911996-06-12 04:04:55 +00002611 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002612 else if (PyLong_Check(arg)) {
2613 x = PyLong_AsUnsignedLong(arg);
Jeremy Hylton825e47b2002-07-25 16:37:51 +00002614 if (x == (unsigned long) -1 && PyErr_Occurred())
2615 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002616#if SIZEOF_LONG > 4
2617 {
2618 unsigned long y;
2619 /* only want the trailing 32 bits */
2620 y = x & 0xFFFFFFFFUL;
2621 if (y ^ x)
2622 return PyErr_Format(PyExc_OverflowError,
2623 "long int larger than 32 bits");
2624 x = y;
Tim Peters58141872002-08-06 22:25:02 +00002625 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002626#endif
2627 }
2628 else
Tim Peters58141872002-08-06 22:25:02 +00002629 return PyErr_Format(PyExc_TypeError,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002630 "expected int/long, %s found",
2631 arg->ob_type->tp_name);
2632 if (x == (unsigned long) -1 && PyErr_Occurred())
2633 return NULL;
2634 return PyInt_FromLong(ntohl(x));
Guido van Rossum006bf911996-06-12 04:04:55 +00002635}
2636
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002637PyDoc_STRVAR(ntohl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002638"ntohl(integer) -> integer\n\
2639\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002640Convert a 32-bit integer from network to host byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002641
2642
Guido van Rossum006bf911996-06-12 04:04:55 +00002643static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002644socket_htons(PyObject *self, PyObject *args)
Guido van Rossum006bf911996-06-12 04:04:55 +00002645{
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002646 unsigned long x1, x2;
Guido van Rossum006bf911996-06-12 04:04:55 +00002647
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002648 if (!PyArg_ParseTuple(args, "i:htons", &x1)) {
Guido van Rossum006bf911996-06-12 04:04:55 +00002649 return NULL;
2650 }
2651 x2 = (int)htons((short)x1);
2652 return PyInt_FromLong(x2);
2653}
2654
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002655PyDoc_STRVAR(htons_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002656"htons(integer) -> integer\n\
2657\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002658Convert a 16-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002659
2660
Guido van Rossum006bf911996-06-12 04:04:55 +00002661static PyObject *
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002662socket_htonl(PyObject *self, PyObject *arg)
Guido van Rossum006bf911996-06-12 04:04:55 +00002663{
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002664 unsigned long x;
Guido van Rossum006bf911996-06-12 04:04:55 +00002665
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002666 if (PyInt_Check(arg)) {
2667 x = PyInt_AS_LONG(arg);
Jeremy Hylton825e47b2002-07-25 16:37:51 +00002668 if (x == (unsigned long) -1 && PyErr_Occurred())
2669 return NULL;
Guido van Rossum006bf911996-06-12 04:04:55 +00002670 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002671 else if (PyLong_Check(arg)) {
2672 x = PyLong_AsUnsignedLong(arg);
Jeremy Hylton825e47b2002-07-25 16:37:51 +00002673 if (x == (unsigned long) -1 && PyErr_Occurred())
2674 return NULL;
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002675#if SIZEOF_LONG > 4
2676 {
2677 unsigned long y;
2678 /* only want the trailing 32 bits */
2679 y = x & 0xFFFFFFFFUL;
2680 if (y ^ x)
2681 return PyErr_Format(PyExc_OverflowError,
2682 "long int larger than 32 bits");
2683 x = y;
Tim Peters58141872002-08-06 22:25:02 +00002684 }
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002685#endif
2686 }
2687 else
Tim Peters58141872002-08-06 22:25:02 +00002688 return PyErr_Format(PyExc_TypeError,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002689 "expected int/long, %s found",
2690 arg->ob_type->tp_name);
Jeremy Hyltonc075e192002-07-25 16:01:12 +00002691 return PyInt_FromLong(htonl(x));
Guido van Rossum006bf911996-06-12 04:04:55 +00002692}
2693
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002694PyDoc_STRVAR(htonl_doc,
Guido van Rossum82a5c661998-07-07 20:45:43 +00002695"htonl(integer) -> integer\n\
2696\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002697Convert a 32-bit integer from host to network byte order.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00002698
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002699/* socket.inet_aton() and socket.inet_ntoa() functions. */
Guido van Rossum5c9eb211999-08-20 18:21:51 +00002700
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002701PyDoc_STRVAR(inet_aton_doc,
Guido van Rossum5c9eb211999-08-20 18:21:51 +00002702"inet_aton(string) -> packed 32-bit IP representation\n\
2703\n\
Guido van Rossumc6a164b1999-08-20 19:11:27 +00002704Convert 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 +00002705binary format used in low-level network functions.");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00002706
2707static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002708socket_inet_aton(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00002709{
Guido van Rossuma2e48551999-09-09 15:42:59 +00002710#ifndef INADDR_NONE
2711#define INADDR_NONE (-1)
2712#endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +00002713
Guido van Rossumc6a164b1999-08-20 19:11:27 +00002714 /* Have to use inet_addr() instead */
2715 char *ip_addr;
Neal Norwitz88f115b2003-02-13 02:15:42 +00002716#ifdef HAVE_INET_ATON
Guido van Rossumad05cdf2003-02-12 23:08:22 +00002717 struct in_addr buf;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00002718
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002719 if (!PyArg_ParseTuple(args, "s:inet_aton", &ip_addr)) {
Guido van Rossumc6a164b1999-08-20 19:11:27 +00002720 return NULL;
2721 }
Guido van Rossumad05cdf2003-02-12 23:08:22 +00002722
2723 if (inet_aton(ip_addr, &buf))
2724 return PyString_FromStringAndSize((char *)(&buf),
2725 sizeof(buf));
2726
2727 PyErr_SetString(socket_error,
2728 "illegal IP address string passed to inet_aton");
2729 return NULL;
2730
2731#else /* In case you don't have inet_aton() */
2732 /* XXX Problem here: inet_aton('255.255.255.255') raises
2733 an exception while it should be a valid address. */
2734 unsigned long packed_addr;
2735
Guido van Rossumc6a164b1999-08-20 19:11:27 +00002736 packed_addr = inet_addr(ip_addr);
2737
2738 if (packed_addr == INADDR_NONE) { /* invalid address */
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002739 PyErr_SetString(socket_error,
Guido van Rossumc6a164b1999-08-20 19:11:27 +00002740 "illegal IP address string passed to inet_aton");
2741 return NULL;
2742 }
2743
2744 return PyString_FromStringAndSize((char *) &packed_addr,
2745 sizeof(packed_addr));
Guido van Rossumad05cdf2003-02-12 23:08:22 +00002746#endif
Guido van Rossum5c9eb211999-08-20 18:21:51 +00002747}
2748
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002749PyDoc_STRVAR(inet_ntoa_doc,
Fred Drakee0661342000-03-07 14:05:16 +00002750"inet_ntoa(packed_ip) -> ip_address_string\n\
Guido van Rossum5c9eb211999-08-20 18:21:51 +00002751\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002752Convert an IP address from 32-bit packed binary format to string format");
Guido van Rossum5c9eb211999-08-20 18:21:51 +00002753
2754static PyObject*
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002755socket_inet_ntoa(PyObject *self, PyObject *args)
Guido van Rossum5c9eb211999-08-20 18:21:51 +00002756{
2757 char *packed_str;
Guido van Rossumc6a164b1999-08-20 19:11:27 +00002758 int addr_len;
Guido van Rossum5c9eb211999-08-20 18:21:51 +00002759 struct in_addr packed_addr;
2760
Guido van Rossum27fc3c02000-03-24 20:56:56 +00002761 if (!PyArg_ParseTuple(args, "s#:inet_ntoa", &packed_str, &addr_len)) {
Guido van Rossum5c9eb211999-08-20 18:21:51 +00002762 return NULL;
2763 }
Guido van Rossum48a680c2001-03-02 06:34:14 +00002764
Guido van Rossum5c9eb211999-08-20 18:21:51 +00002765 if (addr_len != sizeof(packed_addr)) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002766 PyErr_SetString(socket_error,
Guido van Rossum5c9eb211999-08-20 18:21:51 +00002767 "packed IP wrong length for inet_ntoa");
2768 return NULL;
2769 }
2770
2771 memcpy(&packed_addr, packed_str, addr_len);
2772
2773 return PyString_FromString(inet_ntoa(packed_addr));
2774}
Guido van Rossum82a5c661998-07-07 20:45:43 +00002775
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002776/* Python interface to getaddrinfo(host, port). */
2777
2778/*ARGSUSED*/
2779static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002780socket_getaddrinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002781{
Martin v. Löwisf0b11d22001-11-07 08:31:03 +00002782 struct addrinfo hints, *res;
2783 struct addrinfo *res0 = NULL;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002784 PyObject *pobj = (PyObject *)NULL;
Martin v. Löwisf65b1a12001-08-12 09:28:40 +00002785 char pbuf[30];
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002786 char *hptr, *pptr;
2787 int family, socktype, protocol, flags;
2788 int error;
2789 PyObject *all = (PyObject *)NULL;
2790 PyObject *single = (PyObject *)NULL;
2791
2792 family = socktype = protocol = flags = 0;
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00002793 family = AF_UNSPEC;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002794 if (!PyArg_ParseTuple(args, "zO|iiii:getaddrinfo",
2795 &hptr, &pobj, &family, &socktype,
2796 &protocol, &flags)) {
2797 return NULL;
2798 }
2799 if (PyInt_Check(pobj)) {
Marc-André Lemburg03d1b182001-07-31 18:05:33 +00002800 PyOS_snprintf(pbuf, sizeof(pbuf), "%ld", PyInt_AsLong(pobj));
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002801 pptr = pbuf;
2802 } else if (PyString_Check(pobj)) {
2803 pptr = PyString_AsString(pobj);
2804 } else if (pobj == Py_None) {
2805 pptr = (char *)NULL;
2806 } else {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002807 PyErr_SetString(socket_error, "Int or String expected");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002808 return NULL;
2809 }
2810 memset(&hints, 0, sizeof(hints));
2811 hints.ai_family = family;
2812 hints.ai_socktype = socktype;
2813 hints.ai_protocol = protocol;
2814 hints.ai_flags = flags;
2815 error = getaddrinfo(hptr, pptr, &hints, &res0);
2816 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002817 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002818 return NULL;
2819 }
2820
2821 if ((all = PyList_New(0)) == NULL)
2822 goto err;
2823 for (res = res0; res; res = res->ai_next) {
Guido van Rossum716aac02001-10-12 18:59:27 +00002824 PyObject *addr =
2825 makesockaddr(-1, res->ai_addr, res->ai_addrlen);
2826 if (addr == NULL)
2827 goto err;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002828 single = Py_BuildValue("iiisO", res->ai_family,
2829 res->ai_socktype, res->ai_protocol,
2830 res->ai_canonname ? res->ai_canonname : "",
Guido van Rossum716aac02001-10-12 18:59:27 +00002831 addr);
2832 Py_DECREF(addr);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002833 if (single == NULL)
2834 goto err;
2835
2836 if (PyList_Append(all, single))
2837 goto err;
2838 Py_XDECREF(single);
2839 }
Neal Norwitz90128ba2002-08-09 03:37:42 +00002840 if (res0)
2841 freeaddrinfo(res0);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002842 return all;
2843 err:
2844 Py_XDECREF(single);
2845 Py_XDECREF(all);
Martin v. Löwisf0b11d22001-11-07 08:31:03 +00002846 if (res0)
2847 freeaddrinfo(res0);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002848 return (PyObject *)NULL;
2849}
2850
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002851PyDoc_STRVAR(getaddrinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002852"getaddrinfo(host, port [, family, socktype, proto, flags])\n\
2853 -> list of (family, socktype, proto, canonname, sockaddr)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002854\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002855Resolve host and port into addrinfo struct.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002856
2857/* Python interface to getnameinfo(sa, flags). */
2858
2859/*ARGSUSED*/
2860static PyObject *
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002861socket_getnameinfo(PyObject *self, PyObject *args)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002862{
2863 PyObject *sa = (PyObject *)NULL;
2864 int flags;
2865 char *hostp;
Martin v. Löwis06b1d212001-11-02 23:34:52 +00002866 int port, flowinfo, scope_id;
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002867 char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
2868 struct addrinfo hints, *res = NULL;
2869 int error;
2870 PyObject *ret = (PyObject *)NULL;
2871
2872 flags = flowinfo = scope_id = 0;
Martin v. Löwis06b1d212001-11-02 23:34:52 +00002873 if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags))
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002874 return NULL;
Guido van Rossum3eede5a2002-06-07 02:08:35 +00002875 if (!PyArg_ParseTuple(sa, "si|ii",
2876 &hostp, &port, &flowinfo, &scope_id))
Martin v. Löwis06b1d212001-11-02 23:34:52 +00002877 return NULL;
Marc-André Lemburg03d1b182001-07-31 18:05:33 +00002878 PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002879 memset(&hints, 0, sizeof(hints));
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00002880 hints.ai_family = AF_UNSPEC;
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00002881 hints.ai_socktype = SOCK_DGRAM; /* make numeric port happy */
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002882 error = getaddrinfo(hostp, pbuf, &hints, &res);
2883 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002884 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002885 goto fail;
2886 }
2887 if (res->ai_next) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002888 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002889 "sockaddr resolved to multiple addresses");
2890 goto fail;
2891 }
2892 switch (res->ai_family) {
2893 case AF_INET:
2894 {
2895 char *t1;
2896 int t2;
2897 if (PyArg_ParseTuple(sa, "si", &t1, &t2) == 0) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002898 PyErr_SetString(socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002899 "IPv4 sockaddr must be 2 tuple");
2900 goto fail;
2901 }
2902 break;
2903 }
Martin v. Löwis44ddbde2001-12-02 10:15:37 +00002904#ifdef ENABLE_IPV6
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002905 case AF_INET6:
2906 {
2907 struct sockaddr_in6 *sin6;
2908 sin6 = (struct sockaddr_in6 *)res->ai_addr;
2909 sin6->sin6_flowinfo = flowinfo;
2910 sin6->sin6_scope_id = scope_id;
2911 break;
2912 }
2913#endif
2914 }
2915 error = getnameinfo(res->ai_addr, res->ai_addrlen,
2916 hbuf, sizeof(hbuf), pbuf, sizeof(pbuf), flags);
2917 if (error) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002918 set_gaierror(error);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002919 goto fail;
2920 }
2921 ret = Py_BuildValue("ss", hbuf, pbuf);
2922
2923fail:
2924 if (res)
2925 freeaddrinfo(res);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002926 return ret;
2927}
2928
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002929PyDoc_STRVAR(getnameinfo_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002930"getnameinfo(sockaddr, flags) --> (host, port)\n\
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002931\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002932Get host and port for a sockaddr.");
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00002933
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00002934
2935/* Python API to getting and setting the default timeout value. */
2936
2937static PyObject *
2938socket_getdefaulttimeout(PyObject *self)
2939{
2940 if (defaulttimeout < 0.0) {
2941 Py_INCREF(Py_None);
2942 return Py_None;
2943 }
2944 else
2945 return PyFloat_FromDouble(defaulttimeout);
2946}
2947
2948PyDoc_STRVAR(getdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002949"getdefaulttimeout() -> timeout\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00002950\n\
2951Returns the default timeout in floating seconds for new socket objects.\n\
2952A value of None indicates that new socket objects have no timeout.\n\
2953When the socket module is first imported, the default is None.");
2954
2955static PyObject *
2956socket_setdefaulttimeout(PyObject *self, PyObject *arg)
2957{
2958 double timeout;
2959
2960 if (arg == Py_None)
2961 timeout = -1.0;
2962 else {
2963 timeout = PyFloat_AsDouble(arg);
2964 if (timeout < 0.0) {
2965 if (!PyErr_Occurred())
2966 PyErr_SetString(PyExc_ValueError,
2967 "Timeout value out of range");
2968 return NULL;
2969 }
2970 }
2971
2972 defaulttimeout = timeout;
2973
2974 Py_INCREF(Py_None);
2975 return Py_None;
2976}
2977
2978PyDoc_STRVAR(setdefaulttimeout_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00002979"setdefaulttimeout(timeout)\n\
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00002980\n\
2981Set the default timeout in floating seconds for new socket objects.\n\
2982A value of None indicates that new socket objects have no timeout.\n\
2983When the socket module is first imported, the default is None.");
2984
2985
Guido van Rossum30a685f1991-06-27 15:51:29 +00002986/* List of functions exported by this module. */
2987
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002988static PyMethodDef socket_methods[] = {
2989 {"gethostbyname", socket_gethostbyname,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00002990 METH_VARARGS, gethostbyname_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002991 {"gethostbyname_ex", socket_gethostbyname_ex,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00002992 METH_VARARGS, ghbn_ex_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002993 {"gethostbyaddr", socket_gethostbyaddr,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00002994 METH_VARARGS, gethostbyaddr_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002995 {"gethostname", socket_gethostname,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00002996 METH_VARARGS, gethostname_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002997 {"getservbyname", socket_getservbyname,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00002998 METH_VARARGS, getservbyname_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00002999 {"getprotobyname", socket_getprotobyname,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003000 METH_VARARGS,getprotobyname_doc},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00003001#ifndef NO_DUP
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003002 {"fromfd", socket_fromfd,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003003 METH_VARARGS, fromfd_doc},
Guido van Rossum6a3eb5f1994-08-18 15:42:46 +00003004#endif
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003005 {"ntohs", socket_ntohs,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003006 METH_VARARGS, ntohs_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003007 {"ntohl", socket_ntohl,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003008 METH_O, ntohl_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003009 {"htons", socket_htons,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003010 METH_VARARGS, htons_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003011 {"htonl", socket_htonl,
Jeremy Hyltonc075e192002-07-25 16:01:12 +00003012 METH_O, htonl_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003013 {"inet_aton", socket_inet_aton,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003014 METH_VARARGS, inet_aton_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003015 {"inet_ntoa", socket_inet_ntoa,
Andrew M. Kuchlinge365fb82000-08-03 02:06:16 +00003016 METH_VARARGS, inet_ntoa_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003017 {"getaddrinfo", socket_getaddrinfo,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003018 METH_VARARGS, getaddrinfo_doc},
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003019 {"getnameinfo", socket_getnameinfo,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003020 METH_VARARGS, getnameinfo_doc},
Guido van Rossum1693ba82002-07-18 21:11:26 +00003021 {"getdefaulttimeout", (PyCFunction)socket_getdefaulttimeout,
Guido van Rossum9d0c8ce2002-07-18 17:08:35 +00003022 METH_NOARGS, getdefaulttimeout_doc},
3023 {"setdefaulttimeout", socket_setdefaulttimeout,
3024 METH_O, setdefaulttimeout_doc},
Guido van Rossum30a685f1991-06-27 15:51:29 +00003025 {NULL, NULL} /* Sentinel */
Guido van Rossum6574b3e1991-06-25 21:36:08 +00003026};
3027
Guido van Rossum30a685f1991-06-27 15:51:29 +00003028
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003029#ifdef RISCOS
3030#define OS_INIT_DEFINED
Guido van Rossumbe32c891996-06-20 16:25:29 +00003031
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003032static int
3033os_init(void)
3034{
3035 _kernel_swi_regs r;
3036
3037 r.r[0] = 0;
3038 _kernel_swi(0x43380, &r, &r);
3039 taskwindow = r.r[0];
3040
3041 return 0;
3042}
3043
3044#endif /* RISCOS */
3045
3046
3047#ifdef MS_WINDOWS
3048#define OS_INIT_DEFINED
3049
3050/* Additional initialization and cleanup for Windows */
Guido van Rossumbe32c891996-06-20 16:25:29 +00003051
3052static void
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003053os_cleanup(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00003054{
3055 WSACleanup();
Guido van Rossumbe32c891996-06-20 16:25:29 +00003056}
3057
3058static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003059os_init(void)
Guido van Rossumbe32c891996-06-20 16:25:29 +00003060{
3061 WSADATA WSAData;
3062 int ret;
3063 char buf[100];
3064 ret = WSAStartup(0x0101, &WSAData);
3065 switch (ret) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003066 case 0: /* No error */
3067 atexit(os_cleanup);
3068 return 1; /* Success */
Guido van Rossumbe32c891996-06-20 16:25:29 +00003069 case WSASYSNOTREADY:
3070 PyErr_SetString(PyExc_ImportError,
3071 "WSAStartup failed: network not ready");
3072 break;
3073 case WSAVERNOTSUPPORTED:
3074 case WSAEINVAL:
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003075 PyErr_SetString(
3076 PyExc_ImportError,
3077 "WSAStartup failed: requested version not supported");
Guido van Rossumbe32c891996-06-20 16:25:29 +00003078 break;
3079 default:
Tim Peters885d4572001-11-28 20:27:42 +00003080 PyOS_snprintf(buf, sizeof(buf),
3081 "WSAStartup failed: error code %d", ret);
Guido van Rossumbe32c891996-06-20 16:25:29 +00003082 PyErr_SetString(PyExc_ImportError, buf);
3083 break;
3084 }
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003085 return 0; /* Failure */
Guido van Rossumbe32c891996-06-20 16:25:29 +00003086}
3087
Guido van Rossum8d665e61996-06-26 18:22:49 +00003088#endif /* MS_WINDOWS */
Guido van Rossumbe32c891996-06-20 16:25:29 +00003089
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003090
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003091#ifdef PYOS_OS2
3092#define OS_INIT_DEFINED
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003093
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003094/* Additional initialization for OS/2 */
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003095
3096static int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003097os_init(void)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003098{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003099#ifndef PYCC_GCC
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003100 char reason[64];
3101 int rc = sock_init();
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003102
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003103 if (rc == 0) {
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003104 return 1; /* Success */
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003105 }
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003106
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003107 PyOS_snprintf(reason, sizeof(reason),
3108 "OS/2 TCP/IP Error# %d", sock_errno());
3109 PyErr_SetString(PyExc_ImportError, reason);
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003110
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003111 return 0; /* Failure */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00003112#else
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003113 /* No need to initialise sockets with GCC/EMX */
3114 return 1; /* Success */
Andrew MacIntyreba43e872002-03-03 03:03:52 +00003115#endif
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +00003116}
3117
3118#endif /* PYOS_OS2 */
3119
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003120
3121#ifndef OS_INIT_DEFINED
3122static int
3123os_init(void)
3124{
3125 return 1; /* Success */
3126}
3127#endif
3128
3129
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00003130/* C API table - always add new things to the end for binary
3131 compatibility. */
3132static
3133PySocketModule_APIObject PySocketModuleAPI =
3134{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003135 &sock_type,
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00003136};
3137
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003138
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003139/* Initialize the _socket module.
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003140
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003141 This module is actually called "_socket", and there's a wrapper
3142 "socket.py" which implements some additional functionality. On some
3143 platforms (e.g. Windows and OS/2), socket.py also implements a
3144 wrapper for the socket type that provides missing functionality such
3145 as makefile(), dup() and fromfd(). The import of "_socket" may fail
3146 with an ImportError exception if os-specific initialization fails.
3147 On Windows, this does WINSOCK initialization. When WINSOCK is
3148 initialized succesfully, a call to WSACleanup() is scheduled to be
3149 made at exit time.
3150*/
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003151
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00003152PyDoc_STRVAR(socket_doc,
Guido van Rossum22a97152002-08-08 20:37:08 +00003153"Implementation module for socket operations.\n\
3154\n\
3155See the socket module for documentation.");
Guido van Rossum82a5c661998-07-07 20:45:43 +00003156
Mark Hammondfe51c6d2002-08-02 02:27:13 +00003157PyMODINIT_FUNC
Thomas Wouters1e0c2f42000-07-24 16:06:23 +00003158init_socket(void)
Guido van Rossum6574b3e1991-06-25 21:36:08 +00003159{
Fred Drake4baedc12002-04-01 14:53:37 +00003160 PyObject *m;
Fred Drake4baedc12002-04-01 14:53:37 +00003161
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003162 if (!os_init())
Guido van Rossum0cb96de1997-10-01 04:29:29 +00003163 return;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003164
3165 sock_type.ob_type = &PyType_Type;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003166 m = Py_InitModule3(PySocket_MODULE_NAME,
3167 socket_methods,
3168 socket_doc);
3169
3170 socket_error = PyErr_NewException("socket.error", NULL, NULL);
3171 if (socket_error == NULL)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003172 return;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003173 Py_INCREF(socket_error);
3174 PyModule_AddObject(m, "error", socket_error);
3175 socket_herror = PyErr_NewException("socket.herror",
3176 socket_error, NULL);
3177 if (socket_herror == NULL)
3178 return;
3179 Py_INCREF(socket_herror);
3180 PyModule_AddObject(m, "herror", socket_herror);
3181 socket_gaierror = PyErr_NewException("socket.gaierror", socket_error,
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003182 NULL);
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003183 if (socket_gaierror == NULL)
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003184 return;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003185 Py_INCREF(socket_gaierror);
3186 PyModule_AddObject(m, "gaierror", socket_gaierror);
3187 Py_INCREF((PyObject *)&sock_type);
Fred Drake4baedc12002-04-01 14:53:37 +00003188 if (PyModule_AddObject(m, "SocketType",
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003189 (PyObject *)&sock_type) != 0)
Guido van Rossum0cb96de1997-10-01 04:29:29 +00003190 return;
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003191 Py_INCREF((PyObject *)&sock_type);
Fred Drake4baedc12002-04-01 14:53:37 +00003192 if (PyModule_AddObject(m, "socket",
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003193 (PyObject *)&sock_type) != 0)
Guido van Rossum384ca9c2001-10-27 22:20:47 +00003194 return;
Guido van Rossum09be4091999-08-09 14:40:40 +00003195
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00003196 /* Export C API */
Fred Drake4baedc12002-04-01 14:53:37 +00003197 if (PyModule_AddObject(m, PySocket_CAPI_NAME,
Marc-André Lemburga5d2b4c2002-02-16 18:23:30 +00003198 PyCObject_FromVoidPtr((void *)&PySocketModuleAPI, NULL)
3199 ) != 0)
3200 return;
3201
Guido van Rossum09be4091999-08-09 14:40:40 +00003202 /* Address families (we only support AF_INET and AF_UNIX) */
3203#ifdef AF_UNSPEC
Fred Drake4baedc12002-04-01 14:53:37 +00003204 PyModule_AddIntConstant(m, "AF_UNSPEC", AF_UNSPEC);
Guido van Rossum09be4091999-08-09 14:40:40 +00003205#endif
Fred Drake4baedc12002-04-01 14:53:37 +00003206 PyModule_AddIntConstant(m, "AF_INET", AF_INET);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003207#ifdef AF_INET6
Fred Drake4baedc12002-04-01 14:53:37 +00003208 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003209#endif /* AF_INET6 */
Andrew MacIntyre34d3e2d2003-01-02 12:45:34 +00003210#if defined(AF_UNIX) && !defined(PYOS_OS2)
Fred Drake4baedc12002-04-01 14:53:37 +00003211 PyModule_AddIntConstant(m, "AF_UNIX", AF_UNIX);
Guido van Rossumb6775db1994-08-01 11:34:53 +00003212#endif /* AF_UNIX */
Guido van Rossum09be4091999-08-09 14:40:40 +00003213#ifdef AF_AX25
Fred Drake4baedc12002-04-01 14:53:37 +00003214 /* Amateur Radio AX.25 */
3215 PyModule_AddIntConstant(m, "AF_AX25", AF_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00003216#endif
3217#ifdef AF_IPX
Fred Drake4baedc12002-04-01 14:53:37 +00003218 PyModule_AddIntConstant(m, "AF_IPX", AF_IPX); /* Novell IPX */
Guido van Rossum09be4091999-08-09 14:40:40 +00003219#endif
3220#ifdef AF_APPLETALK
Fred Drake4baedc12002-04-01 14:53:37 +00003221 /* Appletalk DDP */
3222 PyModule_AddIntConstant(m, "AF_APPLETALK", AF_APPLETALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00003223#endif
3224#ifdef AF_NETROM
Fred Drake4baedc12002-04-01 14:53:37 +00003225 /* Amateur radio NetROM */
3226 PyModule_AddIntConstant(m, "AF_NETROM", AF_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00003227#endif
3228#ifdef AF_BRIDGE
Fred Drake4baedc12002-04-01 14:53:37 +00003229 /* Multiprotocol bridge */
3230 PyModule_AddIntConstant(m, "AF_BRIDGE", AF_BRIDGE);
Guido van Rossum09be4091999-08-09 14:40:40 +00003231#endif
3232#ifdef AF_AAL5
Fred Drake4baedc12002-04-01 14:53:37 +00003233 /* Reserved for Werner's ATM */
3234 PyModule_AddIntConstant(m, "AF_AAL5", AF_AAL5);
Guido van Rossum09be4091999-08-09 14:40:40 +00003235#endif
3236#ifdef AF_X25
Fred Drake4baedc12002-04-01 14:53:37 +00003237 /* Reserved for X.25 project */
3238 PyModule_AddIntConstant(m, "AF_X25", AF_X25);
Guido van Rossum09be4091999-08-09 14:40:40 +00003239#endif
3240#ifdef AF_INET6
Fred Drake4baedc12002-04-01 14:53:37 +00003241 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6); /* IP version 6 */
Guido van Rossum09be4091999-08-09 14:40:40 +00003242#endif
3243#ifdef AF_ROSE
Fred Drake4baedc12002-04-01 14:53:37 +00003244 /* Amateur Radio X.25 PLP */
3245 PyModule_AddIntConstant(m, "AF_ROSE", AF_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00003246#endif
Martin v. Löwis1ba3fd52001-08-10 20:29:40 +00003247#ifdef HAVE_NETPACKET_PACKET_H
Fred Drake4baedc12002-04-01 14:53:37 +00003248 PyModule_AddIntConstant(m, "AF_PACKET", AF_PACKET);
3249 PyModule_AddIntConstant(m, "PF_PACKET", PF_PACKET);
3250 PyModule_AddIntConstant(m, "PACKET_HOST", PACKET_HOST);
3251 PyModule_AddIntConstant(m, "PACKET_BROADCAST", PACKET_BROADCAST);
3252 PyModule_AddIntConstant(m, "PACKET_MULTICAST", PACKET_MULTICAST);
3253 PyModule_AddIntConstant(m, "PACKET_OTHERHOST", PACKET_OTHERHOST);
3254 PyModule_AddIntConstant(m, "PACKET_OUTGOING", PACKET_OUTGOING);
3255 PyModule_AddIntConstant(m, "PACKET_LOOPBACK", PACKET_LOOPBACK);
3256 PyModule_AddIntConstant(m, "PACKET_FASTROUTE", PACKET_FASTROUTE);
Guido van Rossum48a680c2001-03-02 06:34:14 +00003257#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00003258
3259 /* Socket types */
Fred Drake4baedc12002-04-01 14:53:37 +00003260 PyModule_AddIntConstant(m, "SOCK_STREAM", SOCK_STREAM);
3261 PyModule_AddIntConstant(m, "SOCK_DGRAM", SOCK_DGRAM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00003262#ifndef __BEOS__
3263/* We have incomplete socket support. */
Fred Drake4baedc12002-04-01 14:53:37 +00003264 PyModule_AddIntConstant(m, "SOCK_RAW", SOCK_RAW);
3265 PyModule_AddIntConstant(m, "SOCK_SEQPACKET", SOCK_SEQPACKET);
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00003266#if defined(SOCK_RDM)
Fred Drake4baedc12002-04-01 14:53:37 +00003267 PyModule_AddIntConstant(m, "SOCK_RDM", SOCK_RDM);
Guido van Rossumbcc20741998-08-04 22:53:56 +00003268#endif
Martin v. Löwiscf8f47e2002-12-11 13:10:57 +00003269#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003270
3271#ifdef SO_DEBUG
Fred Drake4baedc12002-04-01 14:53:37 +00003272 PyModule_AddIntConstant(m, "SO_DEBUG", SO_DEBUG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003273#endif
3274#ifdef SO_ACCEPTCONN
Fred Drake4baedc12002-04-01 14:53:37 +00003275 PyModule_AddIntConstant(m, "SO_ACCEPTCONN", SO_ACCEPTCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003276#endif
3277#ifdef SO_REUSEADDR
Fred Drake4baedc12002-04-01 14:53:37 +00003278 PyModule_AddIntConstant(m, "SO_REUSEADDR", SO_REUSEADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003279#endif
3280#ifdef SO_KEEPALIVE
Fred Drake4baedc12002-04-01 14:53:37 +00003281 PyModule_AddIntConstant(m, "SO_KEEPALIVE", SO_KEEPALIVE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003282#endif
3283#ifdef SO_DONTROUTE
Fred Drake4baedc12002-04-01 14:53:37 +00003284 PyModule_AddIntConstant(m, "SO_DONTROUTE", SO_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003285#endif
3286#ifdef SO_BROADCAST
Fred Drake4baedc12002-04-01 14:53:37 +00003287 PyModule_AddIntConstant(m, "SO_BROADCAST", SO_BROADCAST);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003288#endif
3289#ifdef SO_USELOOPBACK
Fred Drake4baedc12002-04-01 14:53:37 +00003290 PyModule_AddIntConstant(m, "SO_USELOOPBACK", SO_USELOOPBACK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003291#endif
3292#ifdef SO_LINGER
Fred Drake4baedc12002-04-01 14:53:37 +00003293 PyModule_AddIntConstant(m, "SO_LINGER", SO_LINGER);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003294#endif
3295#ifdef SO_OOBINLINE
Fred Drake4baedc12002-04-01 14:53:37 +00003296 PyModule_AddIntConstant(m, "SO_OOBINLINE", SO_OOBINLINE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003297#endif
3298#ifdef SO_REUSEPORT
Fred Drake4baedc12002-04-01 14:53:37 +00003299 PyModule_AddIntConstant(m, "SO_REUSEPORT", SO_REUSEPORT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003300#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003301#ifdef SO_SNDBUF
Fred Drake4baedc12002-04-01 14:53:37 +00003302 PyModule_AddIntConstant(m, "SO_SNDBUF", SO_SNDBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003303#endif
3304#ifdef SO_RCVBUF
Fred Drake4baedc12002-04-01 14:53:37 +00003305 PyModule_AddIntConstant(m, "SO_RCVBUF", SO_RCVBUF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003306#endif
3307#ifdef SO_SNDLOWAT
Fred Drake4baedc12002-04-01 14:53:37 +00003308 PyModule_AddIntConstant(m, "SO_SNDLOWAT", SO_SNDLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003309#endif
3310#ifdef SO_RCVLOWAT
Fred Drake4baedc12002-04-01 14:53:37 +00003311 PyModule_AddIntConstant(m, "SO_RCVLOWAT", SO_RCVLOWAT);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003312#endif
3313#ifdef SO_SNDTIMEO
Fred Drake4baedc12002-04-01 14:53:37 +00003314 PyModule_AddIntConstant(m, "SO_SNDTIMEO", SO_SNDTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003315#endif
3316#ifdef SO_RCVTIMEO
Fred Drake4baedc12002-04-01 14:53:37 +00003317 PyModule_AddIntConstant(m, "SO_RCVTIMEO", SO_RCVTIMEO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003318#endif
3319#ifdef SO_ERROR
Fred Drake4baedc12002-04-01 14:53:37 +00003320 PyModule_AddIntConstant(m, "SO_ERROR", SO_ERROR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003321#endif
3322#ifdef SO_TYPE
Fred Drake4baedc12002-04-01 14:53:37 +00003323 PyModule_AddIntConstant(m, "SO_TYPE", SO_TYPE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003324#endif
3325
3326 /* Maximum number of connections for "listen" */
3327#ifdef SOMAXCONN
Fred Drake4baedc12002-04-01 14:53:37 +00003328 PyModule_AddIntConstant(m, "SOMAXCONN", SOMAXCONN);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003329#else
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003330 PyModule_AddIntConstant(m, "SOMAXCONN", 5); /* Common value */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003331#endif
3332
3333 /* Flags for send, recv */
3334#ifdef MSG_OOB
Fred Drake4baedc12002-04-01 14:53:37 +00003335 PyModule_AddIntConstant(m, "MSG_OOB", MSG_OOB);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003336#endif
3337#ifdef MSG_PEEK
Fred Drake4baedc12002-04-01 14:53:37 +00003338 PyModule_AddIntConstant(m, "MSG_PEEK", MSG_PEEK);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003339#endif
3340#ifdef MSG_DONTROUTE
Fred Drake4baedc12002-04-01 14:53:37 +00003341 PyModule_AddIntConstant(m, "MSG_DONTROUTE", MSG_DONTROUTE);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003342#endif
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00003343#ifdef MSG_DONTWAIT
Fred Drake4baedc12002-04-01 14:53:37 +00003344 PyModule_AddIntConstant(m, "MSG_DONTWAIT", MSG_DONTWAIT);
Guido van Rossum2c8bcb82000-04-25 21:34:53 +00003345#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003346#ifdef MSG_EOR
Fred Drake4baedc12002-04-01 14:53:37 +00003347 PyModule_AddIntConstant(m, "MSG_EOR", MSG_EOR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003348#endif
3349#ifdef MSG_TRUNC
Fred Drake4baedc12002-04-01 14:53:37 +00003350 PyModule_AddIntConstant(m, "MSG_TRUNC", MSG_TRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003351#endif
3352#ifdef MSG_CTRUNC
Fred Drake4baedc12002-04-01 14:53:37 +00003353 PyModule_AddIntConstant(m, "MSG_CTRUNC", MSG_CTRUNC);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003354#endif
3355#ifdef MSG_WAITALL
Fred Drake4baedc12002-04-01 14:53:37 +00003356 PyModule_AddIntConstant(m, "MSG_WAITALL", MSG_WAITALL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003357#endif
3358#ifdef MSG_BTAG
Fred Drake4baedc12002-04-01 14:53:37 +00003359 PyModule_AddIntConstant(m, "MSG_BTAG", MSG_BTAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003360#endif
3361#ifdef MSG_ETAG
Fred Drake4baedc12002-04-01 14:53:37 +00003362 PyModule_AddIntConstant(m, "MSG_ETAG", MSG_ETAG);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003363#endif
3364
3365 /* Protocol level and numbers, usable for [gs]etsockopt */
3366#ifdef SOL_SOCKET
Fred Drake4baedc12002-04-01 14:53:37 +00003367 PyModule_AddIntConstant(m, "SOL_SOCKET", SOL_SOCKET);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003368#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00003369#ifdef SOL_IP
Fred Drake4baedc12002-04-01 14:53:37 +00003370 PyModule_AddIntConstant(m, "SOL_IP", SOL_IP);
Guido van Rossum09be4091999-08-09 14:40:40 +00003371#else
Fred Drake4baedc12002-04-01 14:53:37 +00003372 PyModule_AddIntConstant(m, "SOL_IP", 0);
Guido van Rossum09be4091999-08-09 14:40:40 +00003373#endif
3374#ifdef SOL_IPX
Fred Drake4baedc12002-04-01 14:53:37 +00003375 PyModule_AddIntConstant(m, "SOL_IPX", SOL_IPX);
Guido van Rossum09be4091999-08-09 14:40:40 +00003376#endif
3377#ifdef SOL_AX25
Fred Drake4baedc12002-04-01 14:53:37 +00003378 PyModule_AddIntConstant(m, "SOL_AX25", SOL_AX25);
Guido van Rossum09be4091999-08-09 14:40:40 +00003379#endif
3380#ifdef SOL_ATALK
Fred Drake4baedc12002-04-01 14:53:37 +00003381 PyModule_AddIntConstant(m, "SOL_ATALK", SOL_ATALK);
Guido van Rossum09be4091999-08-09 14:40:40 +00003382#endif
3383#ifdef SOL_NETROM
Fred Drake4baedc12002-04-01 14:53:37 +00003384 PyModule_AddIntConstant(m, "SOL_NETROM", SOL_NETROM);
Guido van Rossum09be4091999-08-09 14:40:40 +00003385#endif
3386#ifdef SOL_ROSE
Fred Drake4baedc12002-04-01 14:53:37 +00003387 PyModule_AddIntConstant(m, "SOL_ROSE", SOL_ROSE);
Guido van Rossum09be4091999-08-09 14:40:40 +00003388#endif
3389#ifdef SOL_TCP
Fred Drake4baedc12002-04-01 14:53:37 +00003390 PyModule_AddIntConstant(m, "SOL_TCP", SOL_TCP);
Guido van Rossum09be4091999-08-09 14:40:40 +00003391#else
Fred Drake4baedc12002-04-01 14:53:37 +00003392 PyModule_AddIntConstant(m, "SOL_TCP", 6);
Guido van Rossum09be4091999-08-09 14:40:40 +00003393#endif
3394#ifdef SOL_UDP
Fred Drake4baedc12002-04-01 14:53:37 +00003395 PyModule_AddIntConstant(m, "SOL_UDP", SOL_UDP);
Guido van Rossum09be4091999-08-09 14:40:40 +00003396#else
Fred Drake4baedc12002-04-01 14:53:37 +00003397 PyModule_AddIntConstant(m, "SOL_UDP", 17);
Guido van Rossum09be4091999-08-09 14:40:40 +00003398#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003399#ifdef IPPROTO_IP
Fred Drake4baedc12002-04-01 14:53:37 +00003400 PyModule_AddIntConstant(m, "IPPROTO_IP", IPPROTO_IP);
Guido van Rossum578de301998-05-28 20:18:18 +00003401#else
Fred Drake4baedc12002-04-01 14:53:37 +00003402 PyModule_AddIntConstant(m, "IPPROTO_IP", 0);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003403#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003404#ifdef IPPROTO_HOPOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00003405 PyModule_AddIntConstant(m, "IPPROTO_HOPOPTS", IPPROTO_HOPOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003406#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003407#ifdef IPPROTO_ICMP
Fred Drake4baedc12002-04-01 14:53:37 +00003408 PyModule_AddIntConstant(m, "IPPROTO_ICMP", IPPROTO_ICMP);
Guido van Rossum578de301998-05-28 20:18:18 +00003409#else
Fred Drake4baedc12002-04-01 14:53:37 +00003410 PyModule_AddIntConstant(m, "IPPROTO_ICMP", 1);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003411#endif
3412#ifdef IPPROTO_IGMP
Fred Drake4baedc12002-04-01 14:53:37 +00003413 PyModule_AddIntConstant(m, "IPPROTO_IGMP", IPPROTO_IGMP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003414#endif
3415#ifdef IPPROTO_GGP
Fred Drake4baedc12002-04-01 14:53:37 +00003416 PyModule_AddIntConstant(m, "IPPROTO_GGP", IPPROTO_GGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003417#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003418#ifdef IPPROTO_IPV4
Fred Drake4baedc12002-04-01 14:53:37 +00003419 PyModule_AddIntConstant(m, "IPPROTO_IPV4", IPPROTO_IPV4);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003420#endif
3421#ifdef IPPROTO_IPIP
Fred Drake4baedc12002-04-01 14:53:37 +00003422 PyModule_AddIntConstant(m, "IPPROTO_IPIP", IPPROTO_IPIP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003423#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003424#ifdef IPPROTO_TCP
Fred Drake4baedc12002-04-01 14:53:37 +00003425 PyModule_AddIntConstant(m, "IPPROTO_TCP", IPPROTO_TCP);
Guido van Rossum578de301998-05-28 20:18:18 +00003426#else
Fred Drake4baedc12002-04-01 14:53:37 +00003427 PyModule_AddIntConstant(m, "IPPROTO_TCP", 6);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003428#endif
3429#ifdef IPPROTO_EGP
Fred Drake4baedc12002-04-01 14:53:37 +00003430 PyModule_AddIntConstant(m, "IPPROTO_EGP", IPPROTO_EGP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003431#endif
3432#ifdef IPPROTO_PUP
Fred Drake4baedc12002-04-01 14:53:37 +00003433 PyModule_AddIntConstant(m, "IPPROTO_PUP", IPPROTO_PUP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003434#endif
3435#ifdef IPPROTO_UDP
Fred Drake4baedc12002-04-01 14:53:37 +00003436 PyModule_AddIntConstant(m, "IPPROTO_UDP", IPPROTO_UDP);
Guido van Rossum578de301998-05-28 20:18:18 +00003437#else
Fred Drake4baedc12002-04-01 14:53:37 +00003438 PyModule_AddIntConstant(m, "IPPROTO_UDP", 17);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003439#endif
3440#ifdef IPPROTO_IDP
Fred Drake4baedc12002-04-01 14:53:37 +00003441 PyModule_AddIntConstant(m, "IPPROTO_IDP", IPPROTO_IDP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003442#endif
3443#ifdef IPPROTO_HELLO
Fred Drake4baedc12002-04-01 14:53:37 +00003444 PyModule_AddIntConstant(m, "IPPROTO_HELLO", IPPROTO_HELLO);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003445#endif
3446#ifdef IPPROTO_ND
Fred Drake4baedc12002-04-01 14:53:37 +00003447 PyModule_AddIntConstant(m, "IPPROTO_ND", IPPROTO_ND);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003448#endif
3449#ifdef IPPROTO_TP
Fred Drake4baedc12002-04-01 14:53:37 +00003450 PyModule_AddIntConstant(m, "IPPROTO_TP", IPPROTO_TP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003451#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003452#ifdef IPPROTO_IPV6
Fred Drake4baedc12002-04-01 14:53:37 +00003453 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003454#endif
3455#ifdef IPPROTO_ROUTING
Fred Drake4baedc12002-04-01 14:53:37 +00003456 PyModule_AddIntConstant(m, "IPPROTO_ROUTING", IPPROTO_ROUTING);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003457#endif
3458#ifdef IPPROTO_FRAGMENT
Fred Drake4baedc12002-04-01 14:53:37 +00003459 PyModule_AddIntConstant(m, "IPPROTO_FRAGMENT", IPPROTO_FRAGMENT);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003460#endif
3461#ifdef IPPROTO_RSVP
Fred Drake4baedc12002-04-01 14:53:37 +00003462 PyModule_AddIntConstant(m, "IPPROTO_RSVP", IPPROTO_RSVP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003463#endif
3464#ifdef IPPROTO_GRE
Fred Drake4baedc12002-04-01 14:53:37 +00003465 PyModule_AddIntConstant(m, "IPPROTO_GRE", IPPROTO_GRE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003466#endif
3467#ifdef IPPROTO_ESP
Fred Drake4baedc12002-04-01 14:53:37 +00003468 PyModule_AddIntConstant(m, "IPPROTO_ESP", IPPROTO_ESP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003469#endif
3470#ifdef IPPROTO_AH
Fred Drake4baedc12002-04-01 14:53:37 +00003471 PyModule_AddIntConstant(m, "IPPROTO_AH", IPPROTO_AH);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003472#endif
3473#ifdef IPPROTO_MOBILE
Fred Drake4baedc12002-04-01 14:53:37 +00003474 PyModule_AddIntConstant(m, "IPPROTO_MOBILE", IPPROTO_MOBILE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003475#endif
3476#ifdef IPPROTO_ICMPV6
Fred Drake4baedc12002-04-01 14:53:37 +00003477 PyModule_AddIntConstant(m, "IPPROTO_ICMPV6", IPPROTO_ICMPV6);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003478#endif
3479#ifdef IPPROTO_NONE
Fred Drake4baedc12002-04-01 14:53:37 +00003480 PyModule_AddIntConstant(m, "IPPROTO_NONE", IPPROTO_NONE);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003481#endif
3482#ifdef IPPROTO_DSTOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00003483 PyModule_AddIntConstant(m, "IPPROTO_DSTOPTS", IPPROTO_DSTOPTS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003484#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003485#ifdef IPPROTO_XTP
Fred Drake4baedc12002-04-01 14:53:37 +00003486 PyModule_AddIntConstant(m, "IPPROTO_XTP", IPPROTO_XTP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003487#endif
3488#ifdef IPPROTO_EON
Fred Drake4baedc12002-04-01 14:53:37 +00003489 PyModule_AddIntConstant(m, "IPPROTO_EON", IPPROTO_EON);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003490#endif
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003491#ifdef IPPROTO_PIM
Fred Drake4baedc12002-04-01 14:53:37 +00003492 PyModule_AddIntConstant(m, "IPPROTO_PIM", IPPROTO_PIM);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003493#endif
3494#ifdef IPPROTO_IPCOMP
Fred Drake4baedc12002-04-01 14:53:37 +00003495 PyModule_AddIntConstant(m, "IPPROTO_IPCOMP", IPPROTO_IPCOMP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003496#endif
3497#ifdef IPPROTO_VRRP
Fred Drake4baedc12002-04-01 14:53:37 +00003498 PyModule_AddIntConstant(m, "IPPROTO_VRRP", IPPROTO_VRRP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003499#endif
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003500#ifdef IPPROTO_BIP
Fred Drake4baedc12002-04-01 14:53:37 +00003501 PyModule_AddIntConstant(m, "IPPROTO_BIP", IPPROTO_BIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003502#endif
3503/**/
3504#ifdef IPPROTO_RAW
Fred Drake4baedc12002-04-01 14:53:37 +00003505 PyModule_AddIntConstant(m, "IPPROTO_RAW", IPPROTO_RAW);
Guido van Rossum578de301998-05-28 20:18:18 +00003506#else
Fred Drake4baedc12002-04-01 14:53:37 +00003507 PyModule_AddIntConstant(m, "IPPROTO_RAW", 255);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003508#endif
3509#ifdef IPPROTO_MAX
Fred Drake4baedc12002-04-01 14:53:37 +00003510 PyModule_AddIntConstant(m, "IPPROTO_MAX", IPPROTO_MAX);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003511#endif
3512
3513 /* Some port configuration */
3514#ifdef IPPORT_RESERVED
Fred Drake4baedc12002-04-01 14:53:37 +00003515 PyModule_AddIntConstant(m, "IPPORT_RESERVED", IPPORT_RESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003516#else
Fred Drake4baedc12002-04-01 14:53:37 +00003517 PyModule_AddIntConstant(m, "IPPORT_RESERVED", 1024);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003518#endif
3519#ifdef IPPORT_USERRESERVED
Fred Drake4baedc12002-04-01 14:53:37 +00003520 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", IPPORT_USERRESERVED);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003521#else
Fred Drake4baedc12002-04-01 14:53:37 +00003522 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", 5000);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003523#endif
3524
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003525 /* Some reserved IP v.4 addresses */
3526#ifdef INADDR_ANY
Fred Drake4baedc12002-04-01 14:53:37 +00003527 PyModule_AddIntConstant(m, "INADDR_ANY", INADDR_ANY);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003528#else
Fred Drake4baedc12002-04-01 14:53:37 +00003529 PyModule_AddIntConstant(m, "INADDR_ANY", 0x00000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003530#endif
3531#ifdef INADDR_BROADCAST
Fred Drake4baedc12002-04-01 14:53:37 +00003532 PyModule_AddIntConstant(m, "INADDR_BROADCAST", INADDR_BROADCAST);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003533#else
Fred Drake4baedc12002-04-01 14:53:37 +00003534 PyModule_AddIntConstant(m, "INADDR_BROADCAST", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003535#endif
3536#ifdef INADDR_LOOPBACK
Fred Drake4baedc12002-04-01 14:53:37 +00003537 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", INADDR_LOOPBACK);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003538#else
Fred Drake4baedc12002-04-01 14:53:37 +00003539 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", 0x7F000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003540#endif
3541#ifdef INADDR_UNSPEC_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00003542 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003543#else
Fred Drake4baedc12002-04-01 14:53:37 +00003544 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", 0xe0000000);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003545#endif
3546#ifdef INADDR_ALLHOSTS_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00003547 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP",
3548 INADDR_ALLHOSTS_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003549#else
Fred Drake4baedc12002-04-01 14:53:37 +00003550 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP", 0xe0000001);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003551#endif
3552#ifdef INADDR_MAX_LOCAL_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00003553 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP",
3554 INADDR_MAX_LOCAL_GROUP);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003555#else
Fred Drake4baedc12002-04-01 14:53:37 +00003556 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003557#endif
3558#ifdef INADDR_NONE
Fred Drake4baedc12002-04-01 14:53:37 +00003559 PyModule_AddIntConstant(m, "INADDR_NONE", INADDR_NONE);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003560#else
Fred Drake4baedc12002-04-01 14:53:37 +00003561 PyModule_AddIntConstant(m, "INADDR_NONE", 0xffffffff);
Guido van Rossum5f05eb41995-02-17 15:11:07 +00003562#endif
3563
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003564 /* IPv4 [gs]etsockopt options */
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003565#ifdef IP_OPTIONS
Fred Drake4baedc12002-04-01 14:53:37 +00003566 PyModule_AddIntConstant(m, "IP_OPTIONS", IP_OPTIONS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003567#endif
3568#ifdef IP_HDRINCL
Fred Drake4baedc12002-04-01 14:53:37 +00003569 PyModule_AddIntConstant(m, "IP_HDRINCL", IP_HDRINCL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003570#endif
3571#ifdef IP_TOS
Fred Drake4baedc12002-04-01 14:53:37 +00003572 PyModule_AddIntConstant(m, "IP_TOS", IP_TOS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003573#endif
3574#ifdef IP_TTL
Fred Drake4baedc12002-04-01 14:53:37 +00003575 PyModule_AddIntConstant(m, "IP_TTL", IP_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003576#endif
3577#ifdef IP_RECVOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00003578 PyModule_AddIntConstant(m, "IP_RECVOPTS", IP_RECVOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003579#endif
3580#ifdef IP_RECVRETOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00003581 PyModule_AddIntConstant(m, "IP_RECVRETOPTS", IP_RECVRETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003582#endif
3583#ifdef IP_RECVDSTADDR
Fred Drake4baedc12002-04-01 14:53:37 +00003584 PyModule_AddIntConstant(m, "IP_RECVDSTADDR", IP_RECVDSTADDR);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003585#endif
3586#ifdef IP_RETOPTS
Fred Drake4baedc12002-04-01 14:53:37 +00003587 PyModule_AddIntConstant(m, "IP_RETOPTS", IP_RETOPTS);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003588#endif
3589#ifdef IP_MULTICAST_IF
Fred Drake4baedc12002-04-01 14:53:37 +00003590 PyModule_AddIntConstant(m, "IP_MULTICAST_IF", IP_MULTICAST_IF);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003591#endif
3592#ifdef IP_MULTICAST_TTL
Fred Drake4baedc12002-04-01 14:53:37 +00003593 PyModule_AddIntConstant(m, "IP_MULTICAST_TTL", IP_MULTICAST_TTL);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003594#endif
3595#ifdef IP_MULTICAST_LOOP
Fred Drake4baedc12002-04-01 14:53:37 +00003596 PyModule_AddIntConstant(m, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003597#endif
3598#ifdef IP_ADD_MEMBERSHIP
Fred Drake4baedc12002-04-01 14:53:37 +00003599 PyModule_AddIntConstant(m, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003600#endif
3601#ifdef IP_DROP_MEMBERSHIP
Fred Drake4baedc12002-04-01 14:53:37 +00003602 PyModule_AddIntConstant(m, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP);
Guido van Rossum3fdf58b1995-02-07 15:39:52 +00003603#endif
Guido van Rossum09be4091999-08-09 14:40:40 +00003604#ifdef IP_DEFAULT_MULTICAST_TTL
Fred Drake4baedc12002-04-01 14:53:37 +00003605 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_TTL",
3606 IP_DEFAULT_MULTICAST_TTL);
Guido van Rossum09be4091999-08-09 14:40:40 +00003607#endif
3608#ifdef IP_DEFAULT_MULTICAST_LOOP
Fred Drake4baedc12002-04-01 14:53:37 +00003609 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_LOOP",
3610 IP_DEFAULT_MULTICAST_LOOP);
Guido van Rossum09be4091999-08-09 14:40:40 +00003611#endif
3612#ifdef IP_MAX_MEMBERSHIPS
Fred Drake4baedc12002-04-01 14:53:37 +00003613 PyModule_AddIntConstant(m, "IP_MAX_MEMBERSHIPS", IP_MAX_MEMBERSHIPS);
Guido van Rossum09be4091999-08-09 14:40:40 +00003614#endif
3615
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003616 /* IPv6 [gs]etsockopt options, defined in RFC2553 */
3617#ifdef IPV6_JOIN_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00003618 PyModule_AddIntConstant(m, "IPV6_JOIN_GROUP", IPV6_JOIN_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003619#endif
3620#ifdef IPV6_LEAVE_GROUP
Fred Drake4baedc12002-04-01 14:53:37 +00003621 PyModule_AddIntConstant(m, "IPV6_LEAVE_GROUP", IPV6_LEAVE_GROUP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003622#endif
3623#ifdef IPV6_MULTICAST_HOPS
Fred Drake4baedc12002-04-01 14:53:37 +00003624 PyModule_AddIntConstant(m, "IPV6_MULTICAST_HOPS", IPV6_MULTICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003625#endif
3626#ifdef IPV6_MULTICAST_IF
Fred Drake4baedc12002-04-01 14:53:37 +00003627 PyModule_AddIntConstant(m, "IPV6_MULTICAST_IF", IPV6_MULTICAST_IF);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003628#endif
3629#ifdef IPV6_MULTICAST_LOOP
Fred Drake4baedc12002-04-01 14:53:37 +00003630 PyModule_AddIntConstant(m, "IPV6_MULTICAST_LOOP", IPV6_MULTICAST_LOOP);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003631#endif
3632#ifdef IPV6_UNICAST_HOPS
Fred Drake4baedc12002-04-01 14:53:37 +00003633 PyModule_AddIntConstant(m, "IPV6_UNICAST_HOPS", IPV6_UNICAST_HOPS);
Martin v. Löwisbcf4b352001-08-04 22:37:23 +00003634#endif
3635
Guido van Rossum09be4091999-08-09 14:40:40 +00003636 /* TCP options */
3637#ifdef TCP_NODELAY
Fred Drake4baedc12002-04-01 14:53:37 +00003638 PyModule_AddIntConstant(m, "TCP_NODELAY", TCP_NODELAY);
Guido van Rossum09be4091999-08-09 14:40:40 +00003639#endif
3640#ifdef TCP_MAXSEG
Fred Drake4baedc12002-04-01 14:53:37 +00003641 PyModule_AddIntConstant(m, "TCP_MAXSEG", TCP_MAXSEG);
Guido van Rossum09be4091999-08-09 14:40:40 +00003642#endif
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00003643#ifdef TCP_CORK
Fred Drake4baedc12002-04-01 14:53:37 +00003644 PyModule_AddIntConstant(m, "TCP_CORK", TCP_CORK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00003645#endif
3646#ifdef TCP_KEEPIDLE
Fred Drake4baedc12002-04-01 14:53:37 +00003647 PyModule_AddIntConstant(m, "TCP_KEEPIDLE", TCP_KEEPIDLE);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00003648#endif
3649#ifdef TCP_KEEPINTVL
Fred Drake4baedc12002-04-01 14:53:37 +00003650 PyModule_AddIntConstant(m, "TCP_KEEPINTVL", TCP_KEEPINTVL);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00003651#endif
3652#ifdef TCP_KEEPCNT
Fred Drake4baedc12002-04-01 14:53:37 +00003653 PyModule_AddIntConstant(m, "TCP_KEEPCNT", TCP_KEEPCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00003654#endif
3655#ifdef TCP_SYNCNT
Fred Drake4baedc12002-04-01 14:53:37 +00003656 PyModule_AddIntConstant(m, "TCP_SYNCNT", TCP_SYNCNT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00003657#endif
3658#ifdef TCP_LINGER2
Fred Drake4baedc12002-04-01 14:53:37 +00003659 PyModule_AddIntConstant(m, "TCP_LINGER2", TCP_LINGER2);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00003660#endif
3661#ifdef TCP_DEFER_ACCEPT
Fred Drake4baedc12002-04-01 14:53:37 +00003662 PyModule_AddIntConstant(m, "TCP_DEFER_ACCEPT", TCP_DEFER_ACCEPT);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00003663#endif
3664#ifdef TCP_WINDOW_CLAMP
Fred Drake4baedc12002-04-01 14:53:37 +00003665 PyModule_AddIntConstant(m, "TCP_WINDOW_CLAMP", TCP_WINDOW_CLAMP);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00003666#endif
3667#ifdef TCP_INFO
Fred Drake4baedc12002-04-01 14:53:37 +00003668 PyModule_AddIntConstant(m, "TCP_INFO", TCP_INFO);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00003669#endif
3670#ifdef TCP_QUICKACK
Fred Drake4baedc12002-04-01 14:53:37 +00003671 PyModule_AddIntConstant(m, "TCP_QUICKACK", TCP_QUICKACK);
Martin v. Löwis3cde2cb2001-12-22 15:05:32 +00003672#endif
3673
Guido van Rossum09be4091999-08-09 14:40:40 +00003674
3675 /* IPX options */
3676#ifdef IPX_TYPE
Fred Drake4baedc12002-04-01 14:53:37 +00003677 PyModule_AddIntConstant(m, "IPX_TYPE", IPX_TYPE);
Guido van Rossum09be4091999-08-09 14:40:40 +00003678#endif
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003679
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003680 /* get{addr,name}info parameters */
3681#ifdef EAI_ADDRFAMILY
Fred Drake4baedc12002-04-01 14:53:37 +00003682 PyModule_AddIntConstant(m, "EAI_ADDRFAMILY", EAI_ADDRFAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003683#endif
3684#ifdef EAI_AGAIN
Fred Drake4baedc12002-04-01 14:53:37 +00003685 PyModule_AddIntConstant(m, "EAI_AGAIN", EAI_AGAIN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003686#endif
3687#ifdef EAI_BADFLAGS
Fred Drake4baedc12002-04-01 14:53:37 +00003688 PyModule_AddIntConstant(m, "EAI_BADFLAGS", EAI_BADFLAGS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003689#endif
3690#ifdef EAI_FAIL
Fred Drake4baedc12002-04-01 14:53:37 +00003691 PyModule_AddIntConstant(m, "EAI_FAIL", EAI_FAIL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003692#endif
3693#ifdef EAI_FAMILY
Fred Drake4baedc12002-04-01 14:53:37 +00003694 PyModule_AddIntConstant(m, "EAI_FAMILY", EAI_FAMILY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003695#endif
3696#ifdef EAI_MEMORY
Fred Drake4baedc12002-04-01 14:53:37 +00003697 PyModule_AddIntConstant(m, "EAI_MEMORY", EAI_MEMORY);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003698#endif
3699#ifdef EAI_NODATA
Fred Drake4baedc12002-04-01 14:53:37 +00003700 PyModule_AddIntConstant(m, "EAI_NODATA", EAI_NODATA);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003701#endif
3702#ifdef EAI_NONAME
Fred Drake4baedc12002-04-01 14:53:37 +00003703 PyModule_AddIntConstant(m, "EAI_NONAME", EAI_NONAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003704#endif
3705#ifdef EAI_SERVICE
Fred Drake4baedc12002-04-01 14:53:37 +00003706 PyModule_AddIntConstant(m, "EAI_SERVICE", EAI_SERVICE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003707#endif
3708#ifdef EAI_SOCKTYPE
Fred Drake4baedc12002-04-01 14:53:37 +00003709 PyModule_AddIntConstant(m, "EAI_SOCKTYPE", EAI_SOCKTYPE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003710#endif
3711#ifdef EAI_SYSTEM
Fred Drake4baedc12002-04-01 14:53:37 +00003712 PyModule_AddIntConstant(m, "EAI_SYSTEM", EAI_SYSTEM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003713#endif
3714#ifdef EAI_BADHINTS
Fred Drake4baedc12002-04-01 14:53:37 +00003715 PyModule_AddIntConstant(m, "EAI_BADHINTS", EAI_BADHINTS);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003716#endif
3717#ifdef EAI_PROTOCOL
Fred Drake4baedc12002-04-01 14:53:37 +00003718 PyModule_AddIntConstant(m, "EAI_PROTOCOL", EAI_PROTOCOL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003719#endif
3720#ifdef EAI_MAX
Fred Drake4baedc12002-04-01 14:53:37 +00003721 PyModule_AddIntConstant(m, "EAI_MAX", EAI_MAX);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003722#endif
3723#ifdef AI_PASSIVE
Fred Drake4baedc12002-04-01 14:53:37 +00003724 PyModule_AddIntConstant(m, "AI_PASSIVE", AI_PASSIVE);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003725#endif
3726#ifdef AI_CANONNAME
Fred Drake4baedc12002-04-01 14:53:37 +00003727 PyModule_AddIntConstant(m, "AI_CANONNAME", AI_CANONNAME);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003728#endif
3729#ifdef AI_NUMERICHOST
Fred Drake4baedc12002-04-01 14:53:37 +00003730 PyModule_AddIntConstant(m, "AI_NUMERICHOST", AI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003731#endif
3732#ifdef AI_MASK
Fred Drake4baedc12002-04-01 14:53:37 +00003733 PyModule_AddIntConstant(m, "AI_MASK", AI_MASK);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003734#endif
3735#ifdef AI_ALL
Fred Drake4baedc12002-04-01 14:53:37 +00003736 PyModule_AddIntConstant(m, "AI_ALL", AI_ALL);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003737#endif
3738#ifdef AI_V4MAPPED_CFG
Fred Drake4baedc12002-04-01 14:53:37 +00003739 PyModule_AddIntConstant(m, "AI_V4MAPPED_CFG", AI_V4MAPPED_CFG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003740#endif
3741#ifdef AI_ADDRCONFIG
Fred Drake4baedc12002-04-01 14:53:37 +00003742 PyModule_AddIntConstant(m, "AI_ADDRCONFIG", AI_ADDRCONFIG);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003743#endif
3744#ifdef AI_V4MAPPED
Fred Drake4baedc12002-04-01 14:53:37 +00003745 PyModule_AddIntConstant(m, "AI_V4MAPPED", AI_V4MAPPED);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003746#endif
3747#ifdef AI_DEFAULT
Fred Drake4baedc12002-04-01 14:53:37 +00003748 PyModule_AddIntConstant(m, "AI_DEFAULT", AI_DEFAULT);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003749#endif
3750#ifdef NI_MAXHOST
Fred Drake4baedc12002-04-01 14:53:37 +00003751 PyModule_AddIntConstant(m, "NI_MAXHOST", NI_MAXHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003752#endif
3753#ifdef NI_MAXSERV
Fred Drake4baedc12002-04-01 14:53:37 +00003754 PyModule_AddIntConstant(m, "NI_MAXSERV", NI_MAXSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003755#endif
3756#ifdef NI_NOFQDN
Fred Drake4baedc12002-04-01 14:53:37 +00003757 PyModule_AddIntConstant(m, "NI_NOFQDN", NI_NOFQDN);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003758#endif
3759#ifdef NI_NUMERICHOST
Fred Drake4baedc12002-04-01 14:53:37 +00003760 PyModule_AddIntConstant(m, "NI_NUMERICHOST", NI_NUMERICHOST);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003761#endif
3762#ifdef NI_NAMEREQD
Fred Drake4baedc12002-04-01 14:53:37 +00003763 PyModule_AddIntConstant(m, "NI_NAMEREQD", NI_NAMEREQD);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003764#endif
3765#ifdef NI_NUMERICSERV
Fred Drake4baedc12002-04-01 14:53:37 +00003766 PyModule_AddIntConstant(m, "NI_NUMERICSERV", NI_NUMERICSERV);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003767#endif
3768#ifdef NI_DGRAM
Fred Drake4baedc12002-04-01 14:53:37 +00003769 PyModule_AddIntConstant(m, "NI_DGRAM", NI_DGRAM);
Martin v. Löwis2d8d4272001-07-21 18:05:31 +00003770#endif
3771
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003772 /* Initialize gethostbyname lock */
Guido van Rossum3baaa131999-03-22 21:44:51 +00003773#ifdef USE_GETHOSTBYNAME_LOCK
Guido van Rossum65d5b571998-12-21 19:32:43 +00003774 gethostbyname_lock = PyThread_allocate_lock();
Guido van Rossum4f199ea1998-04-09 20:56:35 +00003775#endif
Guido van Rossum6574b3e1991-06-25 21:36:08 +00003776}
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00003777
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003778
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00003779#ifndef HAVE_INET_PTON
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003780
3781/* Simplistic emulation code for inet_pton that only works for IPv4 */
3782
Guido van Rossum3eede5a2002-06-07 02:08:35 +00003783int
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003784inet_pton(int af, const char *src, void *dst)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00003785{
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003786 if (af == AF_INET) {
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00003787 long packed_addr;
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00003788 packed_addr = inet_addr(src);
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00003789 if (packed_addr == INADDR_NONE)
3790 return 0;
3791 memcpy(dst, &packed_addr, 4);
3792 return 1;
3793 }
3794 /* Should set errno to EAFNOSUPPORT */
3795 return -1;
3796}
3797
Martin v. Löwisc925b1532001-07-21 09:42:15 +00003798const char *
3799inet_ntop(int af, const void *src, char *dst, socklen_t size)
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00003800{
3801 if (af == AF_INET) {
3802 struct in_addr packed_addr;
3803 if (size < 16)
3804 /* Should set errno to ENOSPC. */
3805 return NULL;
3806 memcpy(&packed_addr, src, sizeof(packed_addr));
3807 return strncpy(dst, inet_ntoa(packed_addr), size);
3808 }
3809 /* Should set errno to EAFNOSUPPORT */
3810 return NULL;
3811}
Guido van Rossumc4fcfa32002-06-07 03:19:37 +00003812
Martin v. Löwisb9ab1592001-06-24 21:18:26 +00003813#endif