Note in the intro to Extending... that ctypes can be a simpler, more portable solutio...
[python.git] / Modules / socketmodule.c
blob5575855ef7d5d2f8aaccd528edac705fa10e85ea
1 /* Socket module */
3 /*
5 This module provides an interface to Berkeley socket IPC.
7 Limitations:
9 - Only AF_INET, AF_INET6 and AF_UNIX address families are supported in a
10 portable manner, though AF_PACKET, AF_NETLINK and AF_TIPC are supported
11 under Linux.
12 - No read/write operations (use sendall/recv or makefile instead).
13 - Additional restrictions apply on some non-Unix platforms (compensated
14 for by socket.py).
16 Module interface:
18 - socket.error: exception raised for socket specific errors
19 - socket.gaierror: exception raised for getaddrinfo/getnameinfo errors,
20 a subclass of socket.error
21 - socket.herror: exception raised for gethostby* errors,
22 a subclass of socket.error
23 - socket.fromfd(fd, family, type[, proto]) --> new socket object (created
24 from an existing file descriptor)
25 - socket.gethostbyname(hostname) --> host IP address (string: 'dd.dd.dd.dd')
26 - socket.gethostbyaddr(IP address) --> (hostname, [alias, ...], [IP addr, ...])
27 - socket.gethostname() --> host name (string: 'spam' or 'spam.domain.com')
28 - socket.getprotobyname(protocolname) --> protocol number
29 - socket.getservbyname(servicename[, protocolname]) --> port number
30 - socket.getservbyport(portnumber[, protocolname]) --> service name
31 - socket.socket([family[, type [, proto]]]) --> new socket object
32 - socket.socketpair([family[, type [, proto]]]) --> (socket, socket)
33 - socket.ntohs(16 bit value) --> new int object
34 - socket.ntohl(32 bit value) --> new int object
35 - socket.htons(16 bit value) --> new int object
36 - socket.htonl(32 bit value) --> new int object
37 - socket.getaddrinfo(host, port [, family, socktype, proto, flags])
38 --> List of (family, socktype, proto, canonname, sockaddr)
39 - socket.getnameinfo(sockaddr, flags) --> (host, port)
40 - socket.AF_INET, socket.SOCK_STREAM, etc.: constants from <socket.h>
41 - socket.has_ipv6: boolean value indicating if IPv6 is supported
42 - socket.inet_aton(IP address) -> 32-bit packed IP representation
43 - socket.inet_ntoa(packed IP) -> IP address string
44 - socket.getdefaulttimeout() -> None | float
45 - socket.setdefaulttimeout(None | float)
46 - an Internet socket address is a pair (hostname, port)
47 where hostname can be anything recognized by gethostbyname()
48 (including the dd.dd.dd.dd notation) and port is in host byte order
49 - where a hostname is returned, the dd.dd.dd.dd notation is used
50 - a UNIX domain socket address is a string specifying the pathname
51 - an AF_PACKET socket address is a tuple containing a string
52 specifying the ethernet interface and an integer specifying
53 the Ethernet protocol number to be received. For example:
54 ("eth0",0x1234). Optional 3rd,4th,5th elements in the tuple
55 specify packet-type and ha-type/addr.
56 - an AF_TIPC socket address is expressed as
57 (addr_type, v1, v2, v3 [, scope]); where addr_type can be one of:
58 TIPC_ADDR_NAMESEQ, TIPC_ADDR_NAME, and TIPC_ADDR_ID;
59 and scope can be one of:
60 TIPC_ZONE_SCOPE, TIPC_CLUSTER_SCOPE, and TIPC_NODE_SCOPE.
61 The meaning of v1, v2 and v3 depends on the value of addr_type:
62 if addr_type is TIPC_ADDR_NAME:
63 v1 is the server type
64 v2 is the port identifier
65 v3 is ignored
66 if addr_type is TIPC_ADDR_NAMESEQ:
67 v1 is the server type
68 v2 is the lower port number
69 v3 is the upper port number
70 if addr_type is TIPC_ADDR_ID:
71 v1 is the node
72 v2 is the ref
73 v3 is ignored
76 Local naming conventions:
78 - names starting with sock_ are socket object methods
79 - names starting with socket_ are module-level functions
80 - names starting with PySocket are exported through socketmodule.h
84 #ifdef __APPLE__
86 * inet_aton is not available on OSX 10.3, yet we want to use a binary
87 * that was build on 10.4 or later to work on that release, weak linking
88 * comes to the rescue.
90 # pragma weak inet_aton
91 #endif
93 #include "Python.h"
94 #include "structmember.h"
96 #undef MAX
97 #define MAX(x, y) ((x) < (y) ? (y) : (x))
99 /* Socket object documentation */
100 PyDoc_STRVAR(sock_doc,
101 "socket([family[, type[, proto]]]) -> socket object\n\
103 Open a socket of the given type. The family argument specifies the\n\
104 address family; it defaults to AF_INET. The type argument specifies\n\
105 whether this is a stream (SOCK_STREAM, this is the default)\n\
106 or datagram (SOCK_DGRAM) socket. The protocol argument defaults to 0,\n\
107 specifying the default protocol. Keyword arguments are accepted.\n\
109 A socket object represents one endpoint of a network connection.\n\
111 Methods of socket objects (keyword arguments not allowed):\n\
113 accept() -- accept a connection, returning new socket and client address\n\
114 bind(addr) -- bind the socket to a local address\n\
115 close() -- close the socket\n\
116 connect(addr) -- connect the socket to a remote address\n\
117 connect_ex(addr) -- connect, return an error code instead of an exception\n\
118 dup() -- return a new socket object identical to the current one [*]\n\
119 fileno() -- return underlying file descriptor\n\
120 getpeername() -- return remote address [*]\n\
121 getsockname() -- return local address\n\
122 getsockopt(level, optname[, buflen]) -- get socket options\n\
123 gettimeout() -- return timeout or None\n\
124 listen(n) -- start listening for incoming connections\n\
125 makefile([mode, [bufsize]]) -- return a file object for the socket [*]\n\
126 recv(buflen[, flags]) -- receive data\n\
127 recv_into(buffer[, nbytes[, flags]]) -- receive data (into a buffer)\n\
128 recvfrom(buflen[, flags]) -- receive data and sender\'s address\n\
129 recvfrom_into(buffer[, nbytes, [, flags])\n\
130 -- receive data and sender\'s address (into a buffer)\n\
131 sendall(data[, flags]) -- send all data\n\
132 send(data[, flags]) -- send data, may not send all of it\n\
133 sendto(data[, flags], addr) -- send data to a given address\n\
134 setblocking(0 | 1) -- set or clear the blocking I/O flag\n\
135 setsockopt(level, optname, value) -- set socket options\n\
136 settimeout(None | float) -- set or clear the timeout\n\
137 shutdown(how) -- shut down traffic in one or both directions\n\
139 [*] not available on all platforms!");
141 /* XXX This is a terrible mess of platform-dependent preprocessor hacks.
142 I hope some day someone can clean this up please... */
144 /* Hacks for gethostbyname_r(). On some non-Linux platforms, the configure
145 script doesn't get this right, so we hardcode some platform checks below.
146 On the other hand, not all Linux versions agree, so there the settings
147 computed by the configure script are needed! */
149 #ifndef linux
150 # undef HAVE_GETHOSTBYNAME_R_3_ARG
151 # undef HAVE_GETHOSTBYNAME_R_5_ARG
152 # undef HAVE_GETHOSTBYNAME_R_6_ARG
153 #endif
155 #ifndef WITH_THREAD
156 # undef HAVE_GETHOSTBYNAME_R
157 #endif
159 #ifdef HAVE_GETHOSTBYNAME_R
160 # if defined(_AIX) || defined(__osf__)
161 # define HAVE_GETHOSTBYNAME_R_3_ARG
162 # elif defined(__sun) || defined(__sgi)
163 # define HAVE_GETHOSTBYNAME_R_5_ARG
164 # elif defined(linux)
165 /* Rely on the configure script */
166 # else
167 # undef HAVE_GETHOSTBYNAME_R
168 # endif
169 #endif
171 #if !defined(HAVE_GETHOSTBYNAME_R) && defined(WITH_THREAD) && \
172 !defined(MS_WINDOWS)
173 # define USE_GETHOSTBYNAME_LOCK
174 #endif
176 /* To use __FreeBSD_version */
177 #ifdef HAVE_SYS_PARAM_H
178 #include <sys/param.h>
179 #endif
180 /* On systems on which getaddrinfo() is believed to not be thread-safe,
181 (this includes the getaddrinfo emulation) protect access with a lock. */
182 #if defined(WITH_THREAD) && (defined(__APPLE__) || \
183 (defined(__FreeBSD__) && __FreeBSD_version+0 < 503000) || \
184 defined(__OpenBSD__) || defined(__NetBSD__) || \
185 defined(__VMS) || !defined(HAVE_GETADDRINFO))
186 #define USE_GETADDRINFO_LOCK
187 #endif
189 #ifdef USE_GETADDRINFO_LOCK
190 #define ACQUIRE_GETADDRINFO_LOCK PyThread_acquire_lock(netdb_lock, 1);
191 #define RELEASE_GETADDRINFO_LOCK PyThread_release_lock(netdb_lock);
192 #else
193 #define ACQUIRE_GETADDRINFO_LOCK
194 #define RELEASE_GETADDRINFO_LOCK
195 #endif
197 #if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
198 # include "pythread.h"
199 #endif
201 #if defined(PYCC_VACPP)
202 # include <types.h>
203 # include <io.h>
204 # include <sys/ioctl.h>
205 # include <utils.h>
206 # include <ctype.h>
207 #endif
209 #if defined(__VMS)
210 # include <ioctl.h>
211 #endif
213 #if defined(PYOS_OS2)
214 # define INCL_DOS
215 # define INCL_DOSERRORS
216 # define INCL_NOPMAPI
217 # include <os2.h>
218 #endif
220 #if defined(__sgi) && _COMPILER_VERSION>700 && !_SGIAPI
221 /* make sure that the reentrant (gethostbyaddr_r etc)
222 functions are declared correctly if compiling with
223 MIPSPro 7.x in ANSI C mode (default) */
225 /* XXX Using _SGIAPI is the wrong thing,
226 but I don't know what the right thing is. */
227 #undef _SGIAPI /* to avoid warning */
228 #define _SGIAPI 1
230 #undef _XOPEN_SOURCE
231 #include <sys/socket.h>
232 #include <sys/types.h>
233 #include <netinet/in.h>
234 #ifdef _SS_ALIGNSIZE
235 #define HAVE_GETADDRINFO 1
236 #define HAVE_GETNAMEINFO 1
237 #endif
239 #define HAVE_INET_PTON
240 #include <netdb.h>
241 #endif
243 /* Irix 6.5 fails to define this variable at all. This is needed
244 for both GCC and SGI's compiler. I'd say that the SGI headers
245 are just busted. Same thing for Solaris. */
246 #if (defined(__sgi) || defined(sun)) && !defined(INET_ADDRSTRLEN)
247 #define INET_ADDRSTRLEN 16
248 #endif
250 /* Generic includes */
251 #ifdef HAVE_SYS_TYPES_H
252 #include <sys/types.h>
253 #endif
255 /* Generic socket object definitions and includes */
256 #define PySocket_BUILDING_SOCKET
257 #include "socketmodule.h"
259 /* Addressing includes */
261 #ifndef MS_WINDOWS
263 /* Non-MS WINDOWS includes */
264 # include <netdb.h>
266 /* Headers needed for inet_ntoa() and inet_addr() */
267 # ifdef __BEOS__
268 # include <net/netdb.h>
269 # elif defined(PYOS_OS2) && defined(PYCC_VACPP)
270 # include <netdb.h>
271 typedef size_t socklen_t;
272 # else
273 # include <arpa/inet.h>
274 # endif
276 # ifndef RISCOS
277 # include <fcntl.h>
278 # else
279 # include <sys/ioctl.h>
280 # include <socklib.h>
281 # define NO_DUP
282 int h_errno; /* not used */
283 # define INET_ADDRSTRLEN 16
284 # endif
286 #else
288 /* MS_WINDOWS includes */
289 # ifdef HAVE_FCNTL_H
290 # include <fcntl.h>
291 # endif
293 #endif
295 #include <stddef.h>
297 #ifndef offsetof
298 # define offsetof(type, member) ((size_t)(&((type *)0)->member))
299 #endif
301 #ifndef O_NONBLOCK
302 # define O_NONBLOCK O_NDELAY
303 #endif
305 /* include Python's addrinfo.h unless it causes trouble */
306 #if defined(__sgi) && _COMPILER_VERSION>700 && defined(_SS_ALIGNSIZE)
307 /* Do not include addinfo.h on some newer IRIX versions.
308 * _SS_ALIGNSIZE is defined in sys/socket.h by 6.5.21,
309 * for example, but not by 6.5.10.
311 #elif defined(_MSC_VER) && _MSC_VER>1201
312 /* Do not include addrinfo.h for MSVC7 or greater. 'addrinfo' and
313 * EAI_* constants are defined in (the already included) ws2tcpip.h.
315 #else
316 # include "addrinfo.h"
317 #endif
319 #ifndef HAVE_INET_PTON
320 #if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
321 int inet_pton(int af, const char *src, void *dst);
322 const char *inet_ntop(int af, const void *src, char *dst, socklen_t size);
323 #endif
324 #endif
326 #ifdef __APPLE__
327 /* On OS X, getaddrinfo returns no error indication of lookup
328 failure, so we must use the emulation instead of the libinfo
329 implementation. Unfortunately, performing an autoconf test
330 for this bug would require DNS access for the machine performing
331 the configuration, which is not acceptable. Therefore, we
332 determine the bug just by checking for __APPLE__. If this bug
333 gets ever fixed, perhaps checking for sys/version.h would be
334 appropriate, which is 10/0 on the system with the bug. */
335 #ifndef HAVE_GETNAMEINFO
336 /* This bug seems to be fixed in Jaguar. Ths easiest way I could
337 Find to check for Jaguar is that it has getnameinfo(), which
338 older releases don't have */
339 #undef HAVE_GETADDRINFO
340 #endif
342 #ifdef HAVE_INET_ATON
343 #define USE_INET_ATON_WEAKLINK
344 #endif
346 #endif
348 /* I know this is a bad practice, but it is the easiest... */
349 #if !defined(HAVE_GETADDRINFO)
350 /* avoid clashes with the C library definition of the symbol. */
351 #define getaddrinfo fake_getaddrinfo
352 #define gai_strerror fake_gai_strerror
353 #define freeaddrinfo fake_freeaddrinfo
354 #include "getaddrinfo.c"
355 #endif
356 #if !defined(HAVE_GETNAMEINFO)
357 #define getnameinfo fake_getnameinfo
358 #include "getnameinfo.c"
359 #endif
361 #if defined(MS_WINDOWS) || defined(__BEOS__)
362 /* BeOS suffers from the same socket dichotomy as Win32... - [cjh] */
363 /* seem to be a few differences in the API */
364 #define SOCKETCLOSE closesocket
365 #define NO_DUP /* Actually it exists on NT 3.5, but what the heck... */
366 #endif
368 #ifdef MS_WIN32
369 #define EAFNOSUPPORT WSAEAFNOSUPPORT
370 #define snprintf _snprintf
371 #endif
373 #if defined(PYOS_OS2) && !defined(PYCC_GCC)
374 #define SOCKETCLOSE soclose
375 #define NO_DUP /* Sockets are Not Actual File Handles under OS/2 */
376 #endif
378 #ifndef SOCKETCLOSE
379 #define SOCKETCLOSE close
380 #endif
382 #if defined(HAVE_BLUETOOTH_H) || defined(HAVE_BLUETOOTH_BLUETOOTH_H) && !defined(__NetBSD__)
383 #define USE_BLUETOOTH 1
384 #if defined(__FreeBSD__)
385 #define BTPROTO_L2CAP BLUETOOTH_PROTO_L2CAP
386 #define BTPROTO_RFCOMM BLUETOOTH_PROTO_RFCOMM
387 #define BTPROTO_HCI BLUETOOTH_PROTO_HCI
388 #define SOL_HCI SOL_HCI_RAW
389 #define HCI_FILTER SO_HCI_RAW_FILTER
390 #define sockaddr_l2 sockaddr_l2cap
391 #define sockaddr_rc sockaddr_rfcomm
392 #define hci_dev hci_node
393 #define _BT_L2_MEMB(sa, memb) ((sa)->l2cap_##memb)
394 #define _BT_RC_MEMB(sa, memb) ((sa)->rfcomm_##memb)
395 #define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
396 #elif defined(__NetBSD__)
397 #define sockaddr_l2 sockaddr_bt
398 #define sockaddr_rc sockaddr_bt
399 #define sockaddr_hci sockaddr_bt
400 #define sockaddr_sco sockaddr_bt
401 #define _BT_L2_MEMB(sa, memb) ((sa)->bt_##memb)
402 #define _BT_RC_MEMB(sa, memb) ((sa)->bt_##memb)
403 #define _BT_HCI_MEMB(sa, memb) ((sa)->bt_##memb)
404 #define _BT_SCO_MEMB(sa, memb) ((sa)->bt_##memb)
405 #else
406 #define _BT_L2_MEMB(sa, memb) ((sa)->l2_##memb)
407 #define _BT_RC_MEMB(sa, memb) ((sa)->rc_##memb)
408 #define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
409 #define _BT_SCO_MEMB(sa, memb) ((sa)->sco_##memb)
410 #endif
411 #endif
413 #ifdef __VMS
414 /* TCP/IP Services for VMS uses a maximum send/recv buffer length */
415 #define SEGMENT_SIZE (32 * 1024 -1)
416 #endif
418 #define SAS2SA(x) ((struct sockaddr *)(x))
421 * Constants for getnameinfo()
423 #if !defined(NI_MAXHOST)
424 #define NI_MAXHOST 1025
425 #endif
426 #if !defined(NI_MAXSERV)
427 #define NI_MAXSERV 32
428 #endif
430 /* XXX There's a problem here: *static* functions are not supposed to have
431 a Py prefix (or use CapitalizedWords). Later... */
433 /* Global variable holding the exception type for errors detected
434 by this module (but not argument type or memory errors, etc.). */
435 static PyObject *socket_error;
436 static PyObject *socket_herror;
437 static PyObject *socket_gaierror;
438 static PyObject *socket_timeout;
440 #ifdef RISCOS
441 /* Global variable which is !=0 if Python is running in a RISC OS taskwindow */
442 static int taskwindow;
443 #endif
445 /* A forward reference to the socket type object.
446 The sock_type variable contains pointers to various functions,
447 some of which call new_sockobject(), which uses sock_type, so
448 there has to be a circular reference. */
449 static PyTypeObject sock_type;
451 #if defined(HAVE_POLL_H)
452 #include <poll.h>
453 #elif defined(HAVE_SYS_POLL_H)
454 #include <sys/poll.h>
455 #endif
457 #ifdef Py_SOCKET_FD_CAN_BE_GE_FD_SETSIZE
458 /* Platform can select file descriptors beyond FD_SETSIZE */
459 #define IS_SELECTABLE(s) 1
460 #elif defined(HAVE_POLL)
461 /* Instead of select(), we'll use poll() since poll() works on any fd. */
462 #define IS_SELECTABLE(s) 1
463 /* Can we call select() with this socket without a buffer overrun? */
464 #else
465 /* POSIX says selecting file descriptors beyond FD_SETSIZE
466 has undefined behaviour. If there's no timeout left, we don't have to
467 call select, so it's a safe, little white lie. */
468 #define IS_SELECTABLE(s) ((s)->sock_fd < FD_SETSIZE || s->sock_timeout <= 0.0)
469 #endif
471 static PyObject*
472 select_error(void)
474 PyErr_SetString(socket_error, "unable to select on socket");
475 return NULL;
478 /* Convenience function to raise an error according to errno
479 and return a NULL pointer from a function. */
481 static PyObject *
482 set_error(void)
484 #ifdef MS_WINDOWS
485 int err_no = WSAGetLastError();
486 /* PyErr_SetExcFromWindowsErr() invokes FormatMessage() which
487 recognizes the error codes used by both GetLastError() and
488 WSAGetLastError */
489 if (err_no)
490 return PyErr_SetExcFromWindowsErr(socket_error, err_no);
491 #endif
493 #if defined(PYOS_OS2) && !defined(PYCC_GCC)
494 if (sock_errno() != NO_ERROR) {
495 APIRET rc;
496 ULONG msglen;
497 char outbuf[100];
498 int myerrorcode = sock_errno();
500 /* Retrieve socket-related error message from MPTN.MSG file */
501 rc = DosGetMessage(NULL, 0, outbuf, sizeof(outbuf),
502 myerrorcode - SOCBASEERR + 26,
503 "mptn.msg",
504 &msglen);
505 if (rc == NO_ERROR) {
506 PyObject *v;
508 /* OS/2 doesn't guarantee a terminator */
509 outbuf[msglen] = '\0';
510 if (strlen(outbuf) > 0) {
511 /* If non-empty msg, trim CRLF */
512 char *lastc = &outbuf[ strlen(outbuf)-1 ];
513 while (lastc > outbuf &&
514 isspace(Py_CHARMASK(*lastc))) {
515 /* Trim trailing whitespace (CRLF) */
516 *lastc-- = '\0';
519 v = Py_BuildValue("(is)", myerrorcode, outbuf);
520 if (v != NULL) {
521 PyErr_SetObject(socket_error, v);
522 Py_DECREF(v);
524 return NULL;
527 #endif
529 #if defined(RISCOS)
530 if (_inet_error.errnum != NULL) {
531 PyObject *v;
532 v = Py_BuildValue("(is)", errno, _inet_err());
533 if (v != NULL) {
534 PyErr_SetObject(socket_error, v);
535 Py_DECREF(v);
537 return NULL;
539 #endif
541 return PyErr_SetFromErrno(socket_error);
545 static PyObject *
546 set_herror(int h_error)
548 PyObject *v;
550 #ifdef HAVE_HSTRERROR
551 v = Py_BuildValue("(is)", h_error, (char *)hstrerror(h_error));
552 #else
553 v = Py_BuildValue("(is)", h_error, "host not found");
554 #endif
555 if (v != NULL) {
556 PyErr_SetObject(socket_herror, v);
557 Py_DECREF(v);
560 return NULL;
564 static PyObject *
565 set_gaierror(int error)
567 PyObject *v;
569 #ifdef EAI_SYSTEM
570 /* EAI_SYSTEM is not available on Windows XP. */
571 if (error == EAI_SYSTEM)
572 return set_error();
573 #endif
575 #ifdef HAVE_GAI_STRERROR
576 v = Py_BuildValue("(is)", error, gai_strerror(error));
577 #else
578 v = Py_BuildValue("(is)", error, "getaddrinfo failed");
579 #endif
580 if (v != NULL) {
581 PyErr_SetObject(socket_gaierror, v);
582 Py_DECREF(v);
585 return NULL;
588 #ifdef __VMS
589 /* Function to send in segments */
590 static int
591 sendsegmented(int sock_fd, char *buf, int len, int flags)
593 int n = 0;
594 int remaining = len;
596 while (remaining > 0) {
597 unsigned int segment;
599 segment = (remaining >= SEGMENT_SIZE ? SEGMENT_SIZE : remaining);
600 n = send(sock_fd, buf, segment, flags);
601 if (n < 0) {
602 return n;
604 remaining -= segment;
605 buf += segment;
606 } /* end while */
608 return len;
610 #endif
612 /* Function to perform the setting of socket blocking mode
613 internally. block = (1 | 0). */
614 static int
615 internal_setblocking(PySocketSockObject *s, int block)
617 #ifndef RISCOS
618 #ifndef MS_WINDOWS
619 int delay_flag;
620 #endif
621 #endif
623 Py_BEGIN_ALLOW_THREADS
624 #ifdef __BEOS__
625 block = !block;
626 setsockopt(s->sock_fd, SOL_SOCKET, SO_NONBLOCK,
627 (void *)(&block), sizeof(int));
628 #else
629 #ifndef RISCOS
630 #ifndef MS_WINDOWS
631 #if defined(PYOS_OS2) && !defined(PYCC_GCC)
632 block = !block;
633 ioctl(s->sock_fd, FIONBIO, (caddr_t)&block, sizeof(block));
634 #elif defined(__VMS)
635 block = !block;
636 ioctl(s->sock_fd, FIONBIO, (unsigned int *)&block);
637 #else /* !PYOS_OS2 && !__VMS */
638 delay_flag = fcntl(s->sock_fd, F_GETFL, 0);
639 if (block)
640 delay_flag &= (~O_NONBLOCK);
641 else
642 delay_flag |= O_NONBLOCK;
643 fcntl(s->sock_fd, F_SETFL, delay_flag);
644 #endif /* !PYOS_OS2 */
645 #else /* MS_WINDOWS */
646 block = !block;
647 ioctlsocket(s->sock_fd, FIONBIO, (u_long*)&block);
648 #endif /* MS_WINDOWS */
649 #else /* RISCOS */
650 block = !block;
651 socketioctl(s->sock_fd, FIONBIO, (u_long*)&block);
652 #endif /* RISCOS */
653 #endif /* __BEOS__ */
654 Py_END_ALLOW_THREADS
656 /* Since these don't return anything */
657 return 1;
660 /* Do a select()/poll() on the socket, if necessary (sock_timeout > 0).
661 The argument writing indicates the direction.
662 This does not raise an exception; we'll let our caller do that
663 after they've reacquired the interpreter lock.
664 Returns 1 on timeout, -1 on error, 0 otherwise. */
665 static int
666 internal_select(PySocketSockObject *s, int writing)
668 int n;
670 /* Nothing to do unless we're in timeout mode (not non-blocking) */
671 if (s->sock_timeout <= 0.0)
672 return 0;
674 /* Guard against closed socket */
675 if (s->sock_fd < 0)
676 return 0;
678 /* Prefer poll, if available, since you can poll() any fd
679 * which can't be done with select(). */
680 #ifdef HAVE_POLL
682 struct pollfd pollfd;
683 int timeout;
685 pollfd.fd = s->sock_fd;
686 pollfd.events = writing ? POLLOUT : POLLIN;
688 /* s->sock_timeout is in seconds, timeout in ms */
689 timeout = (int)(s->sock_timeout * 1000 + 0.5);
690 n = poll(&pollfd, 1, timeout);
692 #else
694 /* Construct the arguments to select */
695 fd_set fds;
696 struct timeval tv;
697 tv.tv_sec = (int)s->sock_timeout;
698 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
699 FD_ZERO(&fds);
700 FD_SET(s->sock_fd, &fds);
702 /* See if the socket is ready */
703 if (writing)
704 n = select(s->sock_fd+1, NULL, &fds, NULL, &tv);
705 else
706 n = select(s->sock_fd+1, &fds, NULL, NULL, &tv);
708 #endif
710 if (n < 0)
711 return -1;
712 if (n == 0)
713 return 1;
714 return 0;
717 /* Initialize a new socket object. */
719 static double defaulttimeout = -1.0; /* Default timeout for new sockets */
721 PyMODINIT_FUNC
722 init_sockobject(PySocketSockObject *s,
723 SOCKET_T fd, int family, int type, int proto)
725 #ifdef RISCOS
726 int block = 1;
727 #endif
728 s->sock_fd = fd;
729 s->sock_family = family;
730 s->sock_type = type;
731 s->sock_proto = proto;
732 s->sock_timeout = defaulttimeout;
734 s->errorhandler = &set_error;
736 if (defaulttimeout >= 0.0)
737 internal_setblocking(s, 0);
739 #ifdef RISCOS
740 if (taskwindow)
741 socketioctl(s->sock_fd, 0x80046679, (u_long*)&block);
742 #endif
746 /* Create a new socket object.
747 This just creates the object and initializes it.
748 If the creation fails, return NULL and set an exception (implicit
749 in NEWOBJ()). */
751 static PySocketSockObject *
752 new_sockobject(SOCKET_T fd, int family, int type, int proto)
754 PySocketSockObject *s;
755 s = (PySocketSockObject *)
756 PyType_GenericNew(&sock_type, NULL, NULL);
757 if (s != NULL)
758 init_sockobject(s, fd, family, type, proto);
759 return s;
763 /* Lock to allow python interpreter to continue, but only allow one
764 thread to be in gethostbyname or getaddrinfo */
765 #if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
766 PyThread_type_lock netdb_lock;
767 #endif
770 /* Convert a string specifying a host name or one of a few symbolic
771 names to a numeric IP address. This usually calls gethostbyname()
772 to do the work; the names "" and "<broadcast>" are special.
773 Return the length (IPv4 should be 4 bytes), or negative if
774 an error occurred; then an exception is raised. */
776 static int
777 setipaddr(char *name, struct sockaddr *addr_ret, size_t addr_ret_size, int af)
779 struct addrinfo hints, *res;
780 int error;
781 int d1, d2, d3, d4;
782 char ch;
784 memset((void *) addr_ret, '\0', sizeof(*addr_ret));
785 if (name[0] == '\0') {
786 int siz;
787 memset(&hints, 0, sizeof(hints));
788 hints.ai_family = af;
789 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
790 hints.ai_flags = AI_PASSIVE;
791 Py_BEGIN_ALLOW_THREADS
792 ACQUIRE_GETADDRINFO_LOCK
793 error = getaddrinfo(NULL, "0", &hints, &res);
794 Py_END_ALLOW_THREADS
795 /* We assume that those thread-unsafe getaddrinfo() versions
796 *are* safe regarding their return value, ie. that a
797 subsequent call to getaddrinfo() does not destroy the
798 outcome of the first call. */
799 RELEASE_GETADDRINFO_LOCK
800 if (error) {
801 set_gaierror(error);
802 return -1;
804 switch (res->ai_family) {
805 case AF_INET:
806 siz = 4;
807 break;
808 #ifdef ENABLE_IPV6
809 case AF_INET6:
810 siz = 16;
811 break;
812 #endif
813 default:
814 freeaddrinfo(res);
815 PyErr_SetString(socket_error,
816 "unsupported address family");
817 return -1;
819 if (res->ai_next) {
820 freeaddrinfo(res);
821 PyErr_SetString(socket_error,
822 "wildcard resolved to multiple address");
823 return -1;
825 if (res->ai_addrlen < addr_ret_size)
826 addr_ret_size = res->ai_addrlen;
827 memcpy(addr_ret, res->ai_addr, addr_ret_size);
828 freeaddrinfo(res);
829 return siz;
831 if (name[0] == '<' && strcmp(name, "<broadcast>") == 0) {
832 struct sockaddr_in *sin;
833 if (af != AF_INET && af != AF_UNSPEC) {
834 PyErr_SetString(socket_error,
835 "address family mismatched");
836 return -1;
838 sin = (struct sockaddr_in *)addr_ret;
839 memset((void *) sin, '\0', sizeof(*sin));
840 sin->sin_family = AF_INET;
841 #ifdef HAVE_SOCKADDR_SA_LEN
842 sin->sin_len = sizeof(*sin);
843 #endif
844 sin->sin_addr.s_addr = INADDR_BROADCAST;
845 return sizeof(sin->sin_addr);
847 if (sscanf(name, "%d.%d.%d.%d%c", &d1, &d2, &d3, &d4, &ch) == 4 &&
848 0 <= d1 && d1 <= 255 && 0 <= d2 && d2 <= 255 &&
849 0 <= d3 && d3 <= 255 && 0 <= d4 && d4 <= 255) {
850 struct sockaddr_in *sin;
851 sin = (struct sockaddr_in *)addr_ret;
852 sin->sin_addr.s_addr = htonl(
853 ((long) d1 << 24) | ((long) d2 << 16) |
854 ((long) d3 << 8) | ((long) d4 << 0));
855 sin->sin_family = AF_INET;
856 #ifdef HAVE_SOCKADDR_SA_LEN
857 sin->sin_len = sizeof(*sin);
858 #endif
859 return 4;
861 memset(&hints, 0, sizeof(hints));
862 hints.ai_family = af;
863 Py_BEGIN_ALLOW_THREADS
864 ACQUIRE_GETADDRINFO_LOCK
865 error = getaddrinfo(name, NULL, &hints, &res);
866 #if defined(__digital__) && defined(__unix__)
867 if (error == EAI_NONAME && af == AF_UNSPEC) {
868 /* On Tru64 V5.1, numeric-to-addr conversion fails
869 if no address family is given. Assume IPv4 for now.*/
870 hints.ai_family = AF_INET;
871 error = getaddrinfo(name, NULL, &hints, &res);
873 #endif
874 Py_END_ALLOW_THREADS
875 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
876 if (error) {
877 set_gaierror(error);
878 return -1;
880 if (res->ai_addrlen < addr_ret_size)
881 addr_ret_size = res->ai_addrlen;
882 memcpy((char *) addr_ret, res->ai_addr, addr_ret_size);
883 freeaddrinfo(res);
884 switch (addr_ret->sa_family) {
885 case AF_INET:
886 return 4;
887 #ifdef ENABLE_IPV6
888 case AF_INET6:
889 return 16;
890 #endif
891 default:
892 PyErr_SetString(socket_error, "unknown address family");
893 return -1;
898 /* Create a string object representing an IP address.
899 This is always a string of the form 'dd.dd.dd.dd' (with variable
900 size numbers). */
902 static PyObject *
903 makeipaddr(struct sockaddr *addr, int addrlen)
905 char buf[NI_MAXHOST];
906 int error;
908 error = getnameinfo(addr, addrlen, buf, sizeof(buf), NULL, 0,
909 NI_NUMERICHOST);
910 if (error) {
911 set_gaierror(error);
912 return NULL;
914 return PyString_FromString(buf);
918 #ifdef USE_BLUETOOTH
919 /* Convert a string representation of a Bluetooth address into a numeric
920 address. Returns the length (6), or raises an exception and returns -1 if
921 an error occurred. */
923 static int
924 setbdaddr(char *name, bdaddr_t *bdaddr)
926 unsigned int b0, b1, b2, b3, b4, b5;
927 char ch;
928 int n;
930 n = sscanf(name, "%X:%X:%X:%X:%X:%X%c",
931 &b5, &b4, &b3, &b2, &b1, &b0, &ch);
932 if (n == 6 && (b0 | b1 | b2 | b3 | b4 | b5) < 256) {
933 bdaddr->b[0] = b0;
934 bdaddr->b[1] = b1;
935 bdaddr->b[2] = b2;
936 bdaddr->b[3] = b3;
937 bdaddr->b[4] = b4;
938 bdaddr->b[5] = b5;
939 return 6;
940 } else {
941 PyErr_SetString(socket_error, "bad bluetooth address");
942 return -1;
946 /* Create a string representation of the Bluetooth address. This is always a
947 string of the form 'XX:XX:XX:XX:XX:XX' where XX is a two digit hexadecimal
948 value (zero padded if necessary). */
950 static PyObject *
951 makebdaddr(bdaddr_t *bdaddr)
953 char buf[(6 * 2) + 5 + 1];
955 sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
956 bdaddr->b[5], bdaddr->b[4], bdaddr->b[3],
957 bdaddr->b[2], bdaddr->b[1], bdaddr->b[0]);
958 return PyString_FromString(buf);
960 #endif
963 /* Create an object representing the given socket address,
964 suitable for passing it back to bind(), connect() etc.
965 The family field of the sockaddr structure is inspected
966 to determine what kind of address it really is. */
968 /*ARGSUSED*/
969 static PyObject *
970 makesockaddr(int sockfd, struct sockaddr *addr, int addrlen, int proto)
972 if (addrlen == 0) {
973 /* No address -- may be recvfrom() from known socket */
974 Py_INCREF(Py_None);
975 return Py_None;
978 #ifdef __BEOS__
979 /* XXX: BeOS version of accept() doesn't set family correctly */
980 addr->sa_family = AF_INET;
981 #endif
983 switch (addr->sa_family) {
985 case AF_INET:
987 struct sockaddr_in *a;
988 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
989 PyObject *ret = NULL;
990 if (addrobj) {
991 a = (struct sockaddr_in *)addr;
992 ret = Py_BuildValue("Oi", addrobj, ntohs(a->sin_port));
993 Py_DECREF(addrobj);
995 return ret;
998 #if defined(AF_UNIX)
999 case AF_UNIX:
1001 struct sockaddr_un *a = (struct sockaddr_un *) addr;
1002 #ifdef linux
1003 if (a->sun_path[0] == 0) { /* Linux abstract namespace */
1004 addrlen -= offsetof(struct sockaddr_un, sun_path);
1005 return PyString_FromStringAndSize(a->sun_path,
1006 addrlen);
1008 else
1009 #endif /* linux */
1011 /* regular NULL-terminated string */
1012 return PyString_FromString(a->sun_path);
1015 #endif /* AF_UNIX */
1017 #if defined(AF_NETLINK)
1018 case AF_NETLINK:
1020 struct sockaddr_nl *a = (struct sockaddr_nl *) addr;
1021 return Py_BuildValue("II", a->nl_pid, a->nl_groups);
1023 #endif /* AF_NETLINK */
1025 #ifdef ENABLE_IPV6
1026 case AF_INET6:
1028 struct sockaddr_in6 *a;
1029 PyObject *addrobj = makeipaddr(addr, sizeof(*a));
1030 PyObject *ret = NULL;
1031 if (addrobj) {
1032 a = (struct sockaddr_in6 *)addr;
1033 ret = Py_BuildValue("Oiii",
1034 addrobj,
1035 ntohs(a->sin6_port),
1036 a->sin6_flowinfo,
1037 a->sin6_scope_id);
1038 Py_DECREF(addrobj);
1040 return ret;
1042 #endif
1044 #ifdef USE_BLUETOOTH
1045 case AF_BLUETOOTH:
1046 switch (proto) {
1048 case BTPROTO_L2CAP:
1050 struct sockaddr_l2 *a = (struct sockaddr_l2 *) addr;
1051 PyObject *addrobj = makebdaddr(&_BT_L2_MEMB(a, bdaddr));
1052 PyObject *ret = NULL;
1053 if (addrobj) {
1054 ret = Py_BuildValue("Oi",
1055 addrobj,
1056 _BT_L2_MEMB(a, psm));
1057 Py_DECREF(addrobj);
1059 return ret;
1062 case BTPROTO_RFCOMM:
1064 struct sockaddr_rc *a = (struct sockaddr_rc *) addr;
1065 PyObject *addrobj = makebdaddr(&_BT_RC_MEMB(a, bdaddr));
1066 PyObject *ret = NULL;
1067 if (addrobj) {
1068 ret = Py_BuildValue("Oi",
1069 addrobj,
1070 _BT_RC_MEMB(a, channel));
1071 Py_DECREF(addrobj);
1073 return ret;
1076 case BTPROTO_HCI:
1078 struct sockaddr_hci *a = (struct sockaddr_hci *) addr;
1079 PyObject *ret = NULL;
1080 ret = Py_BuildValue("i", _BT_HCI_MEMB(a, dev));
1081 return ret;
1084 #if !defined(__FreeBSD__)
1085 case BTPROTO_SCO:
1087 struct sockaddr_sco *a = (struct sockaddr_sco *) addr;
1088 return makebdaddr(&_BT_SCO_MEMB(a, bdaddr));
1090 #endif
1093 #endif
1095 #ifdef HAVE_NETPACKET_PACKET_H
1096 case AF_PACKET:
1098 struct sockaddr_ll *a = (struct sockaddr_ll *)addr;
1099 char *ifname = "";
1100 struct ifreq ifr;
1101 /* need to look up interface name give index */
1102 if (a->sll_ifindex) {
1103 ifr.ifr_ifindex = a->sll_ifindex;
1104 if (ioctl(sockfd, SIOCGIFNAME, &ifr) == 0)
1105 ifname = ifr.ifr_name;
1107 return Py_BuildValue("shbhs#",
1108 ifname,
1109 ntohs(a->sll_protocol),
1110 a->sll_pkttype,
1111 a->sll_hatype,
1112 a->sll_addr,
1113 a->sll_halen);
1115 #endif
1117 #ifdef HAVE_LINUX_TIPC_H
1118 case AF_TIPC:
1120 struct sockaddr_tipc *a = (struct sockaddr_tipc *) addr;
1121 if (a->addrtype == TIPC_ADDR_NAMESEQ) {
1122 return Py_BuildValue("IIIII",
1123 a->addrtype,
1124 a->addr.nameseq.type,
1125 a->addr.nameseq.lower,
1126 a->addr.nameseq.upper,
1127 a->scope);
1128 } else if (a->addrtype == TIPC_ADDR_NAME) {
1129 return Py_BuildValue("IIIII",
1130 a->addrtype,
1131 a->addr.name.name.type,
1132 a->addr.name.name.instance,
1133 a->addr.name.name.instance,
1134 a->scope);
1135 } else if (a->addrtype == TIPC_ADDR_ID) {
1136 return Py_BuildValue("IIIII",
1137 a->addrtype,
1138 a->addr.id.node,
1139 a->addr.id.ref,
1141 a->scope);
1142 } else {
1143 PyErr_SetString(PyExc_TypeError,
1144 "Invalid address type");
1145 return NULL;
1148 #endif
1150 /* More cases here... */
1152 default:
1153 /* If we don't know the address family, don't raise an
1154 exception -- return it as a tuple. */
1155 return Py_BuildValue("is#",
1156 addr->sa_family,
1157 addr->sa_data,
1158 sizeof(addr->sa_data));
1164 /* Parse a socket address argument according to the socket object's
1165 address family. Return 1 if the address was in the proper format,
1166 0 of not. The address is returned through addr_ret, its length
1167 through len_ret. */
1169 static int
1170 getsockaddrarg(PySocketSockObject *s, PyObject *args,
1171 struct sockaddr *addr_ret, int *len_ret)
1173 switch (s->sock_family) {
1175 #if defined(AF_UNIX)
1176 case AF_UNIX:
1178 struct sockaddr_un* addr;
1179 char *path;
1180 int len;
1181 if (!PyArg_Parse(args, "t#", &path, &len))
1182 return 0;
1184 addr = (struct sockaddr_un*)addr_ret;
1185 #ifdef linux
1186 if (len > 0 && path[0] == 0) {
1187 /* Linux abstract namespace extension */
1188 if (len > sizeof addr->sun_path) {
1189 PyErr_SetString(socket_error,
1190 "AF_UNIX path too long");
1191 return 0;
1194 else
1195 #endif /* linux */
1197 /* regular NULL-terminated string */
1198 if (len >= sizeof addr->sun_path) {
1199 PyErr_SetString(socket_error,
1200 "AF_UNIX path too long");
1201 return 0;
1203 addr->sun_path[len] = 0;
1205 addr->sun_family = s->sock_family;
1206 memcpy(addr->sun_path, path, len);
1207 #if defined(PYOS_OS2)
1208 *len_ret = sizeof(*addr);
1209 #else
1210 *len_ret = len + offsetof(struct sockaddr_un, sun_path);
1211 #endif
1212 return 1;
1214 #endif /* AF_UNIX */
1216 #if defined(AF_NETLINK)
1217 case AF_NETLINK:
1219 struct sockaddr_nl* addr;
1220 int pid, groups;
1221 addr = (struct sockaddr_nl *)addr_ret;
1222 if (!PyTuple_Check(args)) {
1223 PyErr_Format(
1224 PyExc_TypeError,
1225 "getsockaddrarg: "
1226 "AF_NETLINK address must be tuple, not %.500s",
1227 Py_TYPE(args)->tp_name);
1228 return 0;
1230 if (!PyArg_ParseTuple(args, "II:getsockaddrarg", &pid, &groups))
1231 return 0;
1232 addr->nl_family = AF_NETLINK;
1233 addr->nl_pid = pid;
1234 addr->nl_groups = groups;
1235 *len_ret = sizeof(*addr);
1236 return 1;
1238 #endif
1240 case AF_INET:
1242 struct sockaddr_in* addr;
1243 char *host;
1244 int port, result;
1245 if (!PyTuple_Check(args)) {
1246 PyErr_Format(
1247 PyExc_TypeError,
1248 "getsockaddrarg: "
1249 "AF_INET address must be tuple, not %.500s",
1250 Py_TYPE(args)->tp_name);
1251 return 0;
1253 if (!PyArg_ParseTuple(args, "eti:getsockaddrarg",
1254 "idna", &host, &port))
1255 return 0;
1256 addr=(struct sockaddr_in*)addr_ret;
1257 result = setipaddr(host, (struct sockaddr *)addr,
1258 sizeof(*addr), AF_INET);
1259 PyMem_Free(host);
1260 if (result < 0)
1261 return 0;
1262 if (port < 0 || port > 0xffff) {
1263 PyErr_SetString(
1264 PyExc_OverflowError,
1265 "getsockaddrarg: port must be 0-65535.");
1266 return 0;
1268 addr->sin_family = AF_INET;
1269 addr->sin_port = htons((short)port);
1270 *len_ret = sizeof *addr;
1271 return 1;
1274 #ifdef ENABLE_IPV6
1275 case AF_INET6:
1277 struct sockaddr_in6* addr;
1278 char *host;
1279 int port, flowinfo, scope_id, result;
1280 flowinfo = scope_id = 0;
1281 if (!PyTuple_Check(args)) {
1282 PyErr_Format(
1283 PyExc_TypeError,
1284 "getsockaddrarg: "
1285 "AF_INET6 address must be tuple, not %.500s",
1286 Py_TYPE(args)->tp_name);
1287 return 0;
1289 if (!PyArg_ParseTuple(args, "eti|ii",
1290 "idna", &host, &port, &flowinfo,
1291 &scope_id)) {
1292 return 0;
1294 addr = (struct sockaddr_in6*)addr_ret;
1295 result = setipaddr(host, (struct sockaddr *)addr,
1296 sizeof(*addr), AF_INET6);
1297 PyMem_Free(host);
1298 if (result < 0)
1299 return 0;
1300 if (port < 0 || port > 0xffff) {
1301 PyErr_SetString(
1302 PyExc_OverflowError,
1303 "getsockaddrarg: port must be 0-65535.");
1304 return 0;
1306 addr->sin6_family = s->sock_family;
1307 addr->sin6_port = htons((short)port);
1308 addr->sin6_flowinfo = flowinfo;
1309 addr->sin6_scope_id = scope_id;
1310 *len_ret = sizeof *addr;
1311 return 1;
1313 #endif
1315 #ifdef USE_BLUETOOTH
1316 case AF_BLUETOOTH:
1318 switch (s->sock_proto) {
1319 case BTPROTO_L2CAP:
1321 struct sockaddr_l2 *addr;
1322 char *straddr;
1324 addr = (struct sockaddr_l2 *)addr_ret;
1325 _BT_L2_MEMB(addr, family) = AF_BLUETOOTH;
1326 if (!PyArg_ParseTuple(args, "si", &straddr,
1327 &_BT_L2_MEMB(addr, psm))) {
1328 PyErr_SetString(socket_error, "getsockaddrarg: "
1329 "wrong format");
1330 return 0;
1332 if (setbdaddr(straddr, &_BT_L2_MEMB(addr, bdaddr)) < 0)
1333 return 0;
1335 *len_ret = sizeof *addr;
1336 return 1;
1338 case BTPROTO_RFCOMM:
1340 struct sockaddr_rc *addr;
1341 char *straddr;
1343 addr = (struct sockaddr_rc *)addr_ret;
1344 _BT_RC_MEMB(addr, family) = AF_BLUETOOTH;
1345 if (!PyArg_ParseTuple(args, "si", &straddr,
1346 &_BT_RC_MEMB(addr, channel))) {
1347 PyErr_SetString(socket_error, "getsockaddrarg: "
1348 "wrong format");
1349 return 0;
1351 if (setbdaddr(straddr, &_BT_RC_MEMB(addr, bdaddr)) < 0)
1352 return 0;
1354 *len_ret = sizeof *addr;
1355 return 1;
1357 case BTPROTO_HCI:
1359 struct sockaddr_hci *addr = (struct sockaddr_hci *)addr_ret;
1360 _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
1361 if (!PyArg_ParseTuple(args, "i", &_BT_HCI_MEMB(addr, dev))) {
1362 PyErr_SetString(socket_error, "getsockaddrarg: "
1363 "wrong format");
1364 return 0;
1366 *len_ret = sizeof *addr;
1367 return 1;
1369 #if !defined(__FreeBSD__)
1370 case BTPROTO_SCO:
1372 struct sockaddr_sco *addr;
1373 char *straddr;
1375 addr = (struct sockaddr_sco *)addr_ret;
1376 _BT_SCO_MEMB(addr, family) = AF_BLUETOOTH;
1377 straddr = PyString_AsString(args);
1378 if (straddr == NULL) {
1379 PyErr_SetString(socket_error, "getsockaddrarg: "
1380 "wrong format");
1381 return 0;
1383 if (setbdaddr(straddr, &_BT_SCO_MEMB(addr, bdaddr)) < 0)
1384 return 0;
1386 *len_ret = sizeof *addr;
1387 return 1;
1389 #endif
1390 default:
1391 PyErr_SetString(socket_error, "getsockaddrarg: unknown Bluetooth protocol");
1392 return 0;
1395 #endif
1397 #ifdef HAVE_NETPACKET_PACKET_H
1398 case AF_PACKET:
1400 struct sockaddr_ll* addr;
1401 struct ifreq ifr;
1402 char *interfaceName;
1403 int protoNumber;
1404 int hatype = 0;
1405 int pkttype = 0;
1406 char *haddr = NULL;
1407 unsigned int halen = 0;
1409 if (!PyTuple_Check(args)) {
1410 PyErr_Format(
1411 PyExc_TypeError,
1412 "getsockaddrarg: "
1413 "AF_PACKET address must be tuple, not %.500s",
1414 Py_TYPE(args)->tp_name);
1415 return 0;
1417 if (!PyArg_ParseTuple(args, "si|iis#", &interfaceName,
1418 &protoNumber, &pkttype, &hatype,
1419 &haddr, &halen))
1420 return 0;
1421 strncpy(ifr.ifr_name, interfaceName, sizeof(ifr.ifr_name));
1422 ifr.ifr_name[(sizeof(ifr.ifr_name))-1] = '\0';
1423 if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
1424 s->errorhandler();
1425 return 0;
1427 if (halen > 8) {
1428 PyErr_SetString(PyExc_ValueError,
1429 "Hardware address must be 8 bytes or less");
1430 return 0;
1432 if (protoNumber < 0 || protoNumber > 0xffff) {
1433 PyErr_SetString(
1434 PyExc_OverflowError,
1435 "getsockaddrarg: protoNumber must be 0-65535.");
1436 return 0;
1438 addr = (struct sockaddr_ll*)addr_ret;
1439 addr->sll_family = AF_PACKET;
1440 addr->sll_protocol = htons((short)protoNumber);
1441 addr->sll_ifindex = ifr.ifr_ifindex;
1442 addr->sll_pkttype = pkttype;
1443 addr->sll_hatype = hatype;
1444 if (halen != 0) {
1445 memcpy(&addr->sll_addr, haddr, halen);
1447 addr->sll_halen = halen;
1448 *len_ret = sizeof *addr;
1449 return 1;
1451 #endif
1453 #ifdef HAVE_LINUX_TIPC_H
1454 case AF_TIPC:
1456 unsigned int atype, v1, v2, v3;
1457 unsigned int scope = TIPC_CLUSTER_SCOPE;
1458 struct sockaddr_tipc *addr;
1460 if (!PyTuple_Check(args)) {
1461 PyErr_Format(
1462 PyExc_TypeError,
1463 "getsockaddrarg: "
1464 "AF_TIPC address must be tuple, not %.500s",
1465 Py_TYPE(args)->tp_name);
1466 return 0;
1469 if (!PyArg_ParseTuple(args,
1470 "IIII|I;Invalid TIPC address format",
1471 &atype, &v1, &v2, &v3, &scope))
1472 return 0;
1474 addr = (struct sockaddr_tipc *) addr_ret;
1475 memset(addr, 0, sizeof(struct sockaddr_tipc));
1477 addr->family = AF_TIPC;
1478 addr->scope = scope;
1479 addr->addrtype = atype;
1481 if (atype == TIPC_ADDR_NAMESEQ) {
1482 addr->addr.nameseq.type = v1;
1483 addr->addr.nameseq.lower = v2;
1484 addr->addr.nameseq.upper = v3;
1485 } else if (atype == TIPC_ADDR_NAME) {
1486 addr->addr.name.name.type = v1;
1487 addr->addr.name.name.instance = v2;
1488 } else if (atype == TIPC_ADDR_ID) {
1489 addr->addr.id.node = v1;
1490 addr->addr.id.ref = v2;
1491 } else {
1492 /* Shouldn't happen */
1493 PyErr_SetString(PyExc_TypeError, "Invalid address type");
1494 return 0;
1497 *len_ret = sizeof(*addr);
1499 return 1;
1501 #endif
1503 /* More cases here... */
1505 default:
1506 PyErr_SetString(socket_error, "getsockaddrarg: bad family");
1507 return 0;
1513 /* Get the address length according to the socket object's address family.
1514 Return 1 if the family is known, 0 otherwise. The length is returned
1515 through len_ret. */
1517 static int
1518 getsockaddrlen(PySocketSockObject *s, socklen_t *len_ret)
1520 switch (s->sock_family) {
1522 #if defined(AF_UNIX)
1523 case AF_UNIX:
1525 *len_ret = sizeof (struct sockaddr_un);
1526 return 1;
1528 #endif /* AF_UNIX */
1529 #if defined(AF_NETLINK)
1530 case AF_NETLINK:
1532 *len_ret = sizeof (struct sockaddr_nl);
1533 return 1;
1535 #endif
1537 case AF_INET:
1539 *len_ret = sizeof (struct sockaddr_in);
1540 return 1;
1543 #ifdef ENABLE_IPV6
1544 case AF_INET6:
1546 *len_ret = sizeof (struct sockaddr_in6);
1547 return 1;
1549 #endif
1551 #ifdef USE_BLUETOOTH
1552 case AF_BLUETOOTH:
1554 switch(s->sock_proto)
1557 case BTPROTO_L2CAP:
1558 *len_ret = sizeof (struct sockaddr_l2);
1559 return 1;
1560 case BTPROTO_RFCOMM:
1561 *len_ret = sizeof (struct sockaddr_rc);
1562 return 1;
1563 case BTPROTO_HCI:
1564 *len_ret = sizeof (struct sockaddr_hci);
1565 return 1;
1566 #if !defined(__FreeBSD__)
1567 case BTPROTO_SCO:
1568 *len_ret = sizeof (struct sockaddr_sco);
1569 return 1;
1570 #endif
1571 default:
1572 PyErr_SetString(socket_error, "getsockaddrlen: "
1573 "unknown BT protocol");
1574 return 0;
1578 #endif
1580 #ifdef HAVE_NETPACKET_PACKET_H
1581 case AF_PACKET:
1583 *len_ret = sizeof (struct sockaddr_ll);
1584 return 1;
1586 #endif
1588 #ifdef HAVE_LINUX_TIPC_H
1589 case AF_TIPC:
1591 *len_ret = sizeof (struct sockaddr_tipc);
1592 return 1;
1594 #endif
1596 /* More cases here... */
1598 default:
1599 PyErr_SetString(socket_error, "getsockaddrlen: bad family");
1600 return 0;
1606 /* s.accept() method */
1608 static PyObject *
1609 sock_accept(PySocketSockObject *s)
1611 sock_addr_t addrbuf;
1612 SOCKET_T newfd;
1613 socklen_t addrlen;
1614 PyObject *sock = NULL;
1615 PyObject *addr = NULL;
1616 PyObject *res = NULL;
1617 int timeout;
1619 if (!getsockaddrlen(s, &addrlen))
1620 return NULL;
1621 memset(&addrbuf, 0, addrlen);
1623 #ifdef MS_WINDOWS
1624 newfd = INVALID_SOCKET;
1625 #else
1626 newfd = -1;
1627 #endif
1629 if (!IS_SELECTABLE(s))
1630 return select_error();
1632 Py_BEGIN_ALLOW_THREADS
1633 timeout = internal_select(s, 0);
1634 if (!timeout)
1635 newfd = accept(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
1636 Py_END_ALLOW_THREADS
1638 if (timeout == 1) {
1639 PyErr_SetString(socket_timeout, "timed out");
1640 return NULL;
1643 #ifdef MS_WINDOWS
1644 if (newfd == INVALID_SOCKET)
1645 #else
1646 if (newfd < 0)
1647 #endif
1648 return s->errorhandler();
1650 /* Create the new object with unspecified family,
1651 to avoid calls to bind() etc. on it. */
1652 sock = (PyObject *) new_sockobject(newfd,
1653 s->sock_family,
1654 s->sock_type,
1655 s->sock_proto);
1657 if (sock == NULL) {
1658 SOCKETCLOSE(newfd);
1659 goto finally;
1661 addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
1662 addrlen, s->sock_proto);
1663 if (addr == NULL)
1664 goto finally;
1666 res = PyTuple_Pack(2, sock, addr);
1668 finally:
1669 Py_XDECREF(sock);
1670 Py_XDECREF(addr);
1671 return res;
1674 PyDoc_STRVAR(accept_doc,
1675 "accept() -> (socket object, address info)\n\
1677 Wait for an incoming connection. Return a new socket representing the\n\
1678 connection, and the address of the client. For IP sockets, the address\n\
1679 info is a pair (hostaddr, port).");
1681 /* s.setblocking(flag) method. Argument:
1682 False -- non-blocking mode; same as settimeout(0)
1683 True -- blocking mode; same as settimeout(None)
1686 static PyObject *
1687 sock_setblocking(PySocketSockObject *s, PyObject *arg)
1689 int block;
1691 block = PyInt_AsLong(arg);
1692 if (block == -1 && PyErr_Occurred())
1693 return NULL;
1695 s->sock_timeout = block ? -1.0 : 0.0;
1696 internal_setblocking(s, block);
1698 Py_INCREF(Py_None);
1699 return Py_None;
1702 PyDoc_STRVAR(setblocking_doc,
1703 "setblocking(flag)\n\
1705 Set the socket to blocking (flag is true) or non-blocking (false).\n\
1706 setblocking(True) is equivalent to settimeout(None);\n\
1707 setblocking(False) is equivalent to settimeout(0.0).");
1709 /* s.settimeout(timeout) method. Argument:
1710 None -- no timeout, blocking mode; same as setblocking(True)
1711 0.0 -- non-blocking mode; same as setblocking(False)
1712 > 0 -- timeout mode; operations time out after timeout seconds
1713 < 0 -- illegal; raises an exception
1715 static PyObject *
1716 sock_settimeout(PySocketSockObject *s, PyObject *arg)
1718 double timeout;
1720 if (arg == Py_None)
1721 timeout = -1.0;
1722 else {
1723 timeout = PyFloat_AsDouble(arg);
1724 if (timeout < 0.0) {
1725 if (!PyErr_Occurred())
1726 PyErr_SetString(PyExc_ValueError,
1727 "Timeout value out of range");
1728 return NULL;
1732 s->sock_timeout = timeout;
1733 internal_setblocking(s, timeout < 0.0);
1735 Py_INCREF(Py_None);
1736 return Py_None;
1739 PyDoc_STRVAR(settimeout_doc,
1740 "settimeout(timeout)\n\
1742 Set a timeout on socket operations. 'timeout' can be a float,\n\
1743 giving in seconds, or None. Setting a timeout of None disables\n\
1744 the timeout feature and is equivalent to setblocking(1).\n\
1745 Setting a timeout of zero is the same as setblocking(0).");
1747 /* s.gettimeout() method.
1748 Returns the timeout associated with a socket. */
1749 static PyObject *
1750 sock_gettimeout(PySocketSockObject *s)
1752 if (s->sock_timeout < 0.0) {
1753 Py_INCREF(Py_None);
1754 return Py_None;
1756 else
1757 return PyFloat_FromDouble(s->sock_timeout);
1760 PyDoc_STRVAR(gettimeout_doc,
1761 "gettimeout() -> timeout\n\
1763 Returns the timeout in floating seconds associated with socket \n\
1764 operations. A timeout of None indicates that timeouts on socket \n\
1765 operations are disabled.");
1767 #ifdef RISCOS
1768 /* s.sleeptaskw(1 | 0) method */
1770 static PyObject *
1771 sock_sleeptaskw(PySocketSockObject *s,PyObject *arg)
1773 int block;
1774 block = PyInt_AsLong(arg);
1775 if (block == -1 && PyErr_Occurred())
1776 return NULL;
1777 Py_BEGIN_ALLOW_THREADS
1778 socketioctl(s->sock_fd, 0x80046679, (u_long*)&block);
1779 Py_END_ALLOW_THREADS
1781 Py_INCREF(Py_None);
1782 return Py_None;
1784 PyDoc_STRVAR(sleeptaskw_doc,
1785 "sleeptaskw(flag)\n\
1787 Allow sleeps in taskwindows.");
1788 #endif
1791 /* s.setsockopt() method.
1792 With an integer third argument, sets an integer option.
1793 With a string third argument, sets an option from a buffer;
1794 use optional built-in module 'struct' to encode the string. */
1796 static PyObject *
1797 sock_setsockopt(PySocketSockObject *s, PyObject *args)
1799 int level;
1800 int optname;
1801 int res;
1802 char *buf;
1803 int buflen;
1804 int flag;
1806 if (PyArg_ParseTuple(args, "iii:setsockopt",
1807 &level, &optname, &flag)) {
1808 buf = (char *) &flag;
1809 buflen = sizeof flag;
1811 else {
1812 PyErr_Clear();
1813 if (!PyArg_ParseTuple(args, "iis#:setsockopt",
1814 &level, &optname, &buf, &buflen))
1815 return NULL;
1817 res = setsockopt(s->sock_fd, level, optname, (void *)buf, buflen);
1818 if (res < 0)
1819 return s->errorhandler();
1820 Py_INCREF(Py_None);
1821 return Py_None;
1824 PyDoc_STRVAR(setsockopt_doc,
1825 "setsockopt(level, option, value)\n\
1827 Set a socket option. See the Unix manual for level and option.\n\
1828 The value argument can either be an integer or a string.");
1831 /* s.getsockopt() method.
1832 With two arguments, retrieves an integer option.
1833 With a third integer argument, retrieves a string buffer of that size;
1834 use optional built-in module 'struct' to decode the string. */
1836 static PyObject *
1837 sock_getsockopt(PySocketSockObject *s, PyObject *args)
1839 int level;
1840 int optname;
1841 int res;
1842 PyObject *buf;
1843 socklen_t buflen = 0;
1845 #ifdef __BEOS__
1846 /* We have incomplete socket support. */
1847 PyErr_SetString(socket_error, "getsockopt not supported");
1848 return NULL;
1849 #else
1851 if (!PyArg_ParseTuple(args, "ii|i:getsockopt",
1852 &level, &optname, &buflen))
1853 return NULL;
1855 if (buflen == 0) {
1856 int flag = 0;
1857 socklen_t flagsize = sizeof flag;
1858 res = getsockopt(s->sock_fd, level, optname,
1859 (void *)&flag, &flagsize);
1860 if (res < 0)
1861 return s->errorhandler();
1862 return PyInt_FromLong(flag);
1864 #ifdef __VMS
1865 /* socklen_t is unsigned so no negative test is needed,
1866 test buflen == 0 is previously done */
1867 if (buflen > 1024) {
1868 #else
1869 if (buflen <= 0 || buflen > 1024) {
1870 #endif
1871 PyErr_SetString(socket_error,
1872 "getsockopt buflen out of range");
1873 return NULL;
1875 buf = PyString_FromStringAndSize((char *)NULL, buflen);
1876 if (buf == NULL)
1877 return NULL;
1878 res = getsockopt(s->sock_fd, level, optname,
1879 (void *)PyString_AS_STRING(buf), &buflen);
1880 if (res < 0) {
1881 Py_DECREF(buf);
1882 return s->errorhandler();
1884 _PyString_Resize(&buf, buflen);
1885 return buf;
1886 #endif /* __BEOS__ */
1889 PyDoc_STRVAR(getsockopt_doc,
1890 "getsockopt(level, option[, buffersize]) -> value\n\
1892 Get a socket option. See the Unix manual for level and option.\n\
1893 If a nonzero buffersize argument is given, the return value is a\n\
1894 string of that length; otherwise it is an integer.");
1897 /* s.bind(sockaddr) method */
1899 static PyObject *
1900 sock_bind(PySocketSockObject *s, PyObject *addro)
1902 sock_addr_t addrbuf;
1903 int addrlen;
1904 int res;
1906 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
1907 return NULL;
1908 Py_BEGIN_ALLOW_THREADS
1909 res = bind(s->sock_fd, SAS2SA(&addrbuf), addrlen);
1910 Py_END_ALLOW_THREADS
1911 if (res < 0)
1912 return s->errorhandler();
1913 Py_INCREF(Py_None);
1914 return Py_None;
1917 PyDoc_STRVAR(bind_doc,
1918 "bind(address)\n\
1920 Bind the socket to a local address. For IP sockets, the address is a\n\
1921 pair (host, port); the host must refer to the local host. For raw packet\n\
1922 sockets the address is a tuple (ifname, proto [,pkttype [,hatype]])");
1925 /* s.close() method.
1926 Set the file descriptor to -1 so operations tried subsequently
1927 will surely fail. */
1929 static PyObject *
1930 sock_close(PySocketSockObject *s)
1932 SOCKET_T fd;
1934 if ((fd = s->sock_fd) != -1) {
1935 s->sock_fd = -1;
1936 Py_BEGIN_ALLOW_THREADS
1937 (void) SOCKETCLOSE(fd);
1938 Py_END_ALLOW_THREADS
1940 Py_INCREF(Py_None);
1941 return Py_None;
1944 PyDoc_STRVAR(close_doc,
1945 "close()\n\
1947 Close the socket. It cannot be used after this call.");
1949 static int
1950 internal_connect(PySocketSockObject *s, struct sockaddr *addr, int addrlen,
1951 int *timeoutp)
1953 int res, timeout;
1955 timeout = 0;
1956 res = connect(s->sock_fd, addr, addrlen);
1958 #ifdef MS_WINDOWS
1960 if (s->sock_timeout > 0.0) {
1961 if (res < 0 && WSAGetLastError() == WSAEWOULDBLOCK &&
1962 IS_SELECTABLE(s)) {
1963 /* This is a mess. Best solution: trust select */
1964 fd_set fds;
1965 fd_set fds_exc;
1966 struct timeval tv;
1967 tv.tv_sec = (int)s->sock_timeout;
1968 tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
1969 FD_ZERO(&fds);
1970 FD_SET(s->sock_fd, &fds);
1971 FD_ZERO(&fds_exc);
1972 FD_SET(s->sock_fd, &fds_exc);
1973 res = select(s->sock_fd+1, NULL, &fds, &fds_exc, &tv);
1974 if (res == 0) {
1975 res = WSAEWOULDBLOCK;
1976 timeout = 1;
1977 } else if (res > 0) {
1978 if (FD_ISSET(s->sock_fd, &fds))
1979 /* The socket is in the writeable set - this
1980 means connected */
1981 res = 0;
1982 else {
1983 /* As per MS docs, we need to call getsockopt()
1984 to get the underlying error */
1985 int res_size = sizeof res;
1986 /* It must be in the exception set */
1987 assert(FD_ISSET(s->sock_fd, &fds_exc));
1988 if (0 == getsockopt(s->sock_fd, SOL_SOCKET, SO_ERROR,
1989 (char *)&res, &res_size))
1990 /* getsockopt also clears WSAGetLastError,
1991 so reset it back. */
1992 WSASetLastError(res);
1993 else
1994 res = WSAGetLastError();
1997 /* else if (res < 0) an error occurred */
2001 if (res < 0)
2002 res = WSAGetLastError();
2004 #else
2006 if (s->sock_timeout > 0.0) {
2007 if (res < 0 && errno == EINPROGRESS && IS_SELECTABLE(s)) {
2008 timeout = internal_select(s, 1);
2009 if (timeout == 0) {
2010 /* Bug #1019808: in case of an EINPROGRESS,
2011 use getsockopt(SO_ERROR) to get the real
2012 error. */
2013 socklen_t res_size = sizeof res;
2014 (void)getsockopt(s->sock_fd, SOL_SOCKET,
2015 SO_ERROR, &res, &res_size);
2016 if (res == EISCONN)
2017 res = 0;
2018 errno = res;
2020 else if (timeout == -1) {
2021 res = errno; /* had error */
2023 else
2024 res = EWOULDBLOCK; /* timed out */
2028 if (res < 0)
2029 res = errno;
2031 #endif
2032 *timeoutp = timeout;
2034 return res;
2037 /* s.connect(sockaddr) method */
2039 static PyObject *
2040 sock_connect(PySocketSockObject *s, PyObject *addro)
2042 sock_addr_t addrbuf;
2043 int addrlen;
2044 int res;
2045 int timeout;
2047 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2048 return NULL;
2050 Py_BEGIN_ALLOW_THREADS
2051 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2052 Py_END_ALLOW_THREADS
2054 if (timeout == 1) {
2055 PyErr_SetString(socket_timeout, "timed out");
2056 return NULL;
2058 if (res != 0)
2059 return s->errorhandler();
2060 Py_INCREF(Py_None);
2061 return Py_None;
2064 PyDoc_STRVAR(connect_doc,
2065 "connect(address)\n\
2067 Connect the socket to a remote address. For IP sockets, the address\n\
2068 is a pair (host, port).");
2071 /* s.connect_ex(sockaddr) method */
2073 static PyObject *
2074 sock_connect_ex(PySocketSockObject *s, PyObject *addro)
2076 sock_addr_t addrbuf;
2077 int addrlen;
2078 int res;
2079 int timeout;
2081 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
2082 return NULL;
2084 Py_BEGIN_ALLOW_THREADS
2085 res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
2086 Py_END_ALLOW_THREADS
2088 /* Signals are not errors (though they may raise exceptions). Adapted
2089 from PyErr_SetFromErrnoWithFilenameObject(). */
2090 #ifdef EINTR
2091 if (res == EINTR && PyErr_CheckSignals())
2092 return NULL;
2093 #endif
2095 return PyInt_FromLong((long) res);
2098 PyDoc_STRVAR(connect_ex_doc,
2099 "connect_ex(address) -> errno\n\
2101 This is like connect(address), but returns an error code (the errno value)\n\
2102 instead of raising an exception when an error occurs.");
2105 /* s.fileno() method */
2107 static PyObject *
2108 sock_fileno(PySocketSockObject *s)
2110 #if SIZEOF_SOCKET_T <= SIZEOF_LONG
2111 return PyInt_FromLong((long) s->sock_fd);
2112 #else
2113 return PyLong_FromLongLong((PY_LONG_LONG)s->sock_fd);
2114 #endif
2117 PyDoc_STRVAR(fileno_doc,
2118 "fileno() -> integer\n\
2120 Return the integer file descriptor of the socket.");
2123 #ifndef NO_DUP
2124 /* s.dup() method */
2126 static PyObject *
2127 sock_dup(PySocketSockObject *s)
2129 SOCKET_T newfd;
2130 PyObject *sock;
2132 newfd = dup(s->sock_fd);
2133 if (newfd < 0)
2134 return s->errorhandler();
2135 sock = (PyObject *) new_sockobject(newfd,
2136 s->sock_family,
2137 s->sock_type,
2138 s->sock_proto);
2139 if (sock == NULL)
2140 SOCKETCLOSE(newfd);
2141 return sock;
2144 PyDoc_STRVAR(dup_doc,
2145 "dup() -> socket object\n\
2147 Return a new socket object connected to the same system resource.");
2149 #endif
2152 /* s.getsockname() method */
2154 static PyObject *
2155 sock_getsockname(PySocketSockObject *s)
2157 sock_addr_t addrbuf;
2158 int res;
2159 socklen_t addrlen;
2161 if (!getsockaddrlen(s, &addrlen))
2162 return NULL;
2163 memset(&addrbuf, 0, addrlen);
2164 Py_BEGIN_ALLOW_THREADS
2165 res = getsockname(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2166 Py_END_ALLOW_THREADS
2167 if (res < 0)
2168 return s->errorhandler();
2169 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2170 s->sock_proto);
2173 PyDoc_STRVAR(getsockname_doc,
2174 "getsockname() -> address info\n\
2176 Return the address of the local endpoint. For IP sockets, the address\n\
2177 info is a pair (hostaddr, port).");
2180 #ifdef HAVE_GETPEERNAME /* Cray APP doesn't have this :-( */
2181 /* s.getpeername() method */
2183 static PyObject *
2184 sock_getpeername(PySocketSockObject *s)
2186 sock_addr_t addrbuf;
2187 int res;
2188 socklen_t addrlen;
2190 if (!getsockaddrlen(s, &addrlen))
2191 return NULL;
2192 memset(&addrbuf, 0, addrlen);
2193 Py_BEGIN_ALLOW_THREADS
2194 res = getpeername(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
2195 Py_END_ALLOW_THREADS
2196 if (res < 0)
2197 return s->errorhandler();
2198 return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
2199 s->sock_proto);
2202 PyDoc_STRVAR(getpeername_doc,
2203 "getpeername() -> address info\n\
2205 Return the address of the remote endpoint. For IP sockets, the address\n\
2206 info is a pair (hostaddr, port).");
2208 #endif /* HAVE_GETPEERNAME */
2211 /* s.listen(n) method */
2213 static PyObject *
2214 sock_listen(PySocketSockObject *s, PyObject *arg)
2216 int backlog;
2217 int res;
2219 backlog = PyInt_AsLong(arg);
2220 if (backlog == -1 && PyErr_Occurred())
2221 return NULL;
2222 Py_BEGIN_ALLOW_THREADS
2223 if (backlog < 1)
2224 backlog = 1;
2225 res = listen(s->sock_fd, backlog);
2226 Py_END_ALLOW_THREADS
2227 if (res < 0)
2228 return s->errorhandler();
2229 Py_INCREF(Py_None);
2230 return Py_None;
2233 PyDoc_STRVAR(listen_doc,
2234 "listen(backlog)\n\
2236 Enable a server to accept connections. The backlog argument must be at\n\
2237 least 1; it specifies the number of unaccepted connection that the system\n\
2238 will allow before refusing new connections.");
2241 #ifndef NO_DUP
2242 /* s.makefile(mode) method.
2243 Create a new open file object referring to a dupped version of
2244 the socket's file descriptor. (The dup() call is necessary so
2245 that the open file and socket objects may be closed independent
2246 of each other.)
2247 The mode argument specifies 'r' or 'w' passed to fdopen(). */
2249 static PyObject *
2250 sock_makefile(PySocketSockObject *s, PyObject *args)
2252 extern int fclose(FILE *);
2253 char *mode = "r";
2254 int bufsize = -1;
2255 #ifdef MS_WIN32
2256 Py_intptr_t fd;
2257 #else
2258 int fd;
2259 #endif
2260 FILE *fp;
2261 PyObject *f;
2262 #ifdef __VMS
2263 char *mode_r = "r";
2264 char *mode_w = "w";
2265 #endif
2267 if (!PyArg_ParseTuple(args, "|si:makefile", &mode, &bufsize))
2268 return NULL;
2269 #ifdef __VMS
2270 if (strcmp(mode,"rb") == 0) {
2271 mode = mode_r;
2273 else {
2274 if (strcmp(mode,"wb") == 0) {
2275 mode = mode_w;
2278 #endif
2279 #ifdef MS_WIN32
2280 if (((fd = _open_osfhandle(s->sock_fd, _O_BINARY)) < 0) ||
2281 ((fd = dup(fd)) < 0) || ((fp = fdopen(fd, mode)) == NULL))
2282 #else
2283 if ((fd = dup(s->sock_fd)) < 0 || (fp = fdopen(fd, mode)) == NULL)
2284 #endif
2286 if (fd >= 0)
2287 SOCKETCLOSE(fd);
2288 return s->errorhandler();
2290 f = PyFile_FromFile(fp, "<socket>", mode, fclose);
2291 if (f != NULL)
2292 PyFile_SetBufSize(f, bufsize);
2293 return f;
2296 PyDoc_STRVAR(makefile_doc,
2297 "makefile([mode[, buffersize]]) -> file object\n\
2299 Return a regular file object corresponding to the socket.\n\
2300 The mode and buffersize arguments are as for the built-in open() function.");
2302 #endif /* NO_DUP */
2305 * This is the guts of the recv() and recv_into() methods, which reads into a
2306 * char buffer. If you have any inc/dec ref to do to the objects that contain
2307 * the buffer, do it in the caller. This function returns the number of bytes
2308 * succesfully read. If there was an error, it returns -1. Note that it is
2309 * also possible that we return a number of bytes smaller than the request
2310 * bytes.
2312 static ssize_t
2313 sock_recv_guts(PySocketSockObject *s, char* cbuf, int len, int flags)
2315 ssize_t outlen = -1;
2316 int timeout;
2317 #ifdef __VMS
2318 int remaining;
2319 char *read_buf;
2320 #endif
2322 if (!IS_SELECTABLE(s)) {
2323 select_error();
2324 return -1;
2327 #ifndef __VMS
2328 Py_BEGIN_ALLOW_THREADS
2329 timeout = internal_select(s, 0);
2330 if (!timeout)
2331 outlen = recv(s->sock_fd, cbuf, len, flags);
2332 Py_END_ALLOW_THREADS
2334 if (timeout == 1) {
2335 PyErr_SetString(socket_timeout, "timed out");
2336 return -1;
2338 if (outlen < 0) {
2339 /* Note: the call to errorhandler() ALWAYS indirectly returned
2340 NULL, so ignore its return value */
2341 s->errorhandler();
2342 return -1;
2344 #else
2345 read_buf = cbuf;
2346 remaining = len;
2347 while (remaining != 0) {
2348 unsigned int segment;
2349 int nread = -1;
2351 segment = remaining /SEGMENT_SIZE;
2352 if (segment != 0) {
2353 segment = SEGMENT_SIZE;
2355 else {
2356 segment = remaining;
2359 Py_BEGIN_ALLOW_THREADS
2360 timeout = internal_select(s, 0);
2361 if (!timeout)
2362 nread = recv(s->sock_fd, read_buf, segment, flags);
2363 Py_END_ALLOW_THREADS
2365 if (timeout == 1) {
2366 PyErr_SetString(socket_timeout, "timed out");
2367 return -1;
2369 if (nread < 0) {
2370 s->errorhandler();
2371 return -1;
2373 if (nread != remaining) {
2374 read_buf += nread;
2375 break;
2378 remaining -= segment;
2379 read_buf += segment;
2381 outlen = read_buf - cbuf;
2382 #endif /* !__VMS */
2384 return outlen;
2388 /* s.recv(nbytes [,flags]) method */
2390 static PyObject *
2391 sock_recv(PySocketSockObject *s, PyObject *args)
2393 int recvlen, flags = 0;
2394 ssize_t outlen;
2395 PyObject *buf;
2397 if (!PyArg_ParseTuple(args, "i|i:recv", &recvlen, &flags))
2398 return NULL;
2400 if (recvlen < 0) {
2401 PyErr_SetString(PyExc_ValueError,
2402 "negative buffersize in recv");
2403 return NULL;
2406 /* Allocate a new string. */
2407 buf = PyString_FromStringAndSize((char *) 0, recvlen);
2408 if (buf == NULL)
2409 return NULL;
2411 /* Call the guts */
2412 outlen = sock_recv_guts(s, PyString_AS_STRING(buf), recvlen, flags);
2413 if (outlen < 0) {
2414 /* An error occurred, release the string and return an
2415 error. */
2416 Py_DECREF(buf);
2417 return NULL;
2419 if (outlen != recvlen) {
2420 /* We did not read as many bytes as we anticipated, resize the
2421 string if possible and be succesful. */
2422 if (_PyString_Resize(&buf, outlen) < 0)
2423 /* Oopsy, not so succesful after all. */
2424 return NULL;
2427 return buf;
2430 PyDoc_STRVAR(recv_doc,
2431 "recv(buffersize[, flags]) -> data\n\
2433 Receive up to buffersize bytes from the socket. For the optional flags\n\
2434 argument, see the Unix manual. When no data is available, block until\n\
2435 at least one byte is available or until the remote end is closed. When\n\
2436 the remote end is closed and all data is read, return the empty string.");
2439 /* s.recv_into(buffer, [nbytes [,flags]]) method */
2441 static PyObject*
2442 sock_recv_into(PySocketSockObject *s, PyObject *args, PyObject *kwds)
2444 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
2446 int recvlen = 0, flags = 0;
2447 ssize_t readlen;
2448 char *buf;
2449 int buflen;
2451 /* Get the buffer's memory */
2452 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w#|ii:recv_into", kwlist,
2453 &buf, &buflen, &recvlen, &flags))
2454 return NULL;
2455 assert(buf != 0 && buflen > 0);
2457 if (recvlen < 0) {
2458 PyErr_SetString(PyExc_ValueError,
2459 "negative buffersize in recv_into");
2460 return NULL;
2462 if (recvlen == 0) {
2463 /* If nbytes was not specified, use the buffer's length */
2464 recvlen = buflen;
2467 /* Check if the buffer is large enough */
2468 if (buflen < recvlen) {
2469 PyErr_SetString(PyExc_ValueError,
2470 "buffer too small for requested bytes");
2471 return NULL;
2474 /* Call the guts */
2475 readlen = sock_recv_guts(s, buf, recvlen, flags);
2476 if (readlen < 0) {
2477 /* Return an error. */
2478 return NULL;
2481 /* Return the number of bytes read. Note that we do not do anything
2482 special here in the case that readlen < recvlen. */
2483 return PyInt_FromSsize_t(readlen);
2486 PyDoc_STRVAR(recv_into_doc,
2487 "recv_into(buffer, [nbytes[, flags]]) -> nbytes_read\n\
2489 A version of recv() that stores its data into a buffer rather than creating \n\
2490 a new string. Receive up to buffersize bytes from the socket. If buffersize \n\
2491 is not specified (or 0), receive up to the size available in the given buffer.\n\
2493 See recv() for documentation about the flags.");
2497 * This is the guts of the recvfrom() and recvfrom_into() methods, which reads
2498 * into a char buffer. If you have any inc/def ref to do to the objects that
2499 * contain the buffer, do it in the caller. This function returns the number
2500 * of bytes succesfully read. If there was an error, it returns -1. Note
2501 * that it is also possible that we return a number of bytes smaller than the
2502 * request bytes.
2504 * 'addr' is a return value for the address object. Note that you must decref
2505 * it yourself.
2507 static ssize_t
2508 sock_recvfrom_guts(PySocketSockObject *s, char* cbuf, int len, int flags,
2509 PyObject** addr)
2511 sock_addr_t addrbuf;
2512 int timeout;
2513 ssize_t n = -1;
2514 socklen_t addrlen;
2516 *addr = NULL;
2518 if (!getsockaddrlen(s, &addrlen))
2519 return -1;
2521 if (!IS_SELECTABLE(s)) {
2522 select_error();
2523 return -1;
2526 Py_BEGIN_ALLOW_THREADS
2527 memset(&addrbuf, 0, addrlen);
2528 timeout = internal_select(s, 0);
2529 if (!timeout) {
2530 #ifndef MS_WINDOWS
2531 #if defined(PYOS_OS2) && !defined(PYCC_GCC)
2532 n = recvfrom(s->sock_fd, cbuf, len, flags,
2533 SAS2SA(&addrbuf), &addrlen);
2534 #else
2535 n = recvfrom(s->sock_fd, cbuf, len, flags,
2536 (void *) &addrbuf, &addrlen);
2537 #endif
2538 #else
2539 n = recvfrom(s->sock_fd, cbuf, len, flags,
2540 SAS2SA(&addrbuf), &addrlen);
2541 #endif
2543 Py_END_ALLOW_THREADS
2545 if (timeout == 1) {
2546 PyErr_SetString(socket_timeout, "timed out");
2547 return -1;
2549 if (n < 0) {
2550 s->errorhandler();
2551 return -1;
2554 if (!(*addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
2555 addrlen, s->sock_proto)))
2556 return -1;
2558 return n;
2561 /* s.recvfrom(nbytes [,flags]) method */
2563 static PyObject *
2564 sock_recvfrom(PySocketSockObject *s, PyObject *args)
2566 PyObject *buf = NULL;
2567 PyObject *addr = NULL;
2568 PyObject *ret = NULL;
2569 int recvlen, flags = 0;
2570 ssize_t outlen;
2572 if (!PyArg_ParseTuple(args, "i|i:recvfrom", &recvlen, &flags))
2573 return NULL;
2575 if (recvlen < 0) {
2576 PyErr_SetString(PyExc_ValueError,
2577 "negative buffersize in recvfrom");
2578 return NULL;
2581 buf = PyString_FromStringAndSize((char *) 0, recvlen);
2582 if (buf == NULL)
2583 return NULL;
2585 outlen = sock_recvfrom_guts(s, PyString_AS_STRING(buf),
2586 recvlen, flags, &addr);
2587 if (outlen < 0) {
2588 goto finally;
2591 if (outlen != recvlen) {
2592 /* We did not read as many bytes as we anticipated, resize the
2593 string if possible and be succesful. */
2594 if (_PyString_Resize(&buf, outlen) < 0)
2595 /* Oopsy, not so succesful after all. */
2596 goto finally;
2599 ret = PyTuple_Pack(2, buf, addr);
2601 finally:
2602 Py_XDECREF(buf);
2603 Py_XDECREF(addr);
2604 return ret;
2607 PyDoc_STRVAR(recvfrom_doc,
2608 "recvfrom(buffersize[, flags]) -> (data, address info)\n\
2610 Like recv(buffersize, flags) but also return the sender's address info.");
2613 /* s.recvfrom_into(buffer[, nbytes [,flags]]) method */
2615 static PyObject *
2616 sock_recvfrom_into(PySocketSockObject *s, PyObject *args, PyObject* kwds)
2618 static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
2620 int recvlen = 0, flags = 0;
2621 ssize_t readlen;
2622 char *buf;
2623 int buflen;
2625 PyObject *addr = NULL;
2627 if (!PyArg_ParseTupleAndKeywords(args, kwds, "w#|ii:recvfrom_into",
2628 kwlist, &buf, &buflen,
2629 &recvlen, &flags))
2630 return NULL;
2631 assert(buf != 0 && buflen > 0);
2633 if (recvlen < 0) {
2634 PyErr_SetString(PyExc_ValueError,
2635 "negative buffersize in recvfrom_into");
2636 return NULL;
2638 if (recvlen == 0) {
2639 /* If nbytes was not specified, use the buffer's length */
2640 recvlen = buflen;
2643 readlen = sock_recvfrom_guts(s, buf, recvlen, flags, &addr);
2644 if (readlen < 0) {
2645 /* Return an error */
2646 Py_XDECREF(addr);
2647 return NULL;
2650 /* Return the number of bytes read and the address. Note that we do
2651 not do anything special here in the case that readlen < recvlen. */
2652 return Py_BuildValue("lN", readlen, addr);
2655 PyDoc_STRVAR(recvfrom_into_doc,
2656 "recvfrom_into(buffer[, nbytes[, flags]]) -> (nbytes, address info)\n\
2658 Like recv_into(buffer[, nbytes[, flags]]) but also return the sender's address info.");
2661 /* s.send(data [,flags]) method */
2663 static PyObject *
2664 sock_send(PySocketSockObject *s, PyObject *args)
2666 char *buf;
2667 int len, n = -1, flags = 0, timeout;
2668 Py_buffer pbuf;
2670 if (!PyArg_ParseTuple(args, "s*|i:send", &pbuf, &flags))
2671 return NULL;
2673 if (!IS_SELECTABLE(s)) {
2674 PyBuffer_Release(&pbuf);
2675 return select_error();
2677 buf = pbuf.buf;
2678 len = pbuf.len;
2680 Py_BEGIN_ALLOW_THREADS
2681 timeout = internal_select(s, 1);
2682 if (!timeout)
2683 #ifdef __VMS
2684 n = sendsegmented(s->sock_fd, buf, len, flags);
2685 #else
2686 n = send(s->sock_fd, buf, len, flags);
2687 #endif
2688 Py_END_ALLOW_THREADS
2690 PyBuffer_Release(&pbuf);
2692 if (timeout == 1) {
2693 PyErr_SetString(socket_timeout, "timed out");
2694 return NULL;
2696 if (n < 0)
2697 return s->errorhandler();
2698 return PyInt_FromLong((long)n);
2701 PyDoc_STRVAR(send_doc,
2702 "send(data[, flags]) -> count\n\
2704 Send a data string to the socket. For the optional flags\n\
2705 argument, see the Unix manual. Return the number of bytes\n\
2706 sent; this may be less than len(data) if the network is busy.");
2709 /* s.sendall(data [,flags]) method */
2711 static PyObject *
2712 sock_sendall(PySocketSockObject *s, PyObject *args)
2714 char *buf;
2715 int len, n = -1, flags = 0, timeout;
2716 Py_buffer pbuf;
2718 if (!PyArg_ParseTuple(args, "s*|i:sendall", &pbuf, &flags))
2719 return NULL;
2720 buf = pbuf.buf;
2721 len = pbuf.len;
2723 if (!IS_SELECTABLE(s)) {
2724 PyBuffer_Release(&pbuf);
2725 return select_error();
2728 Py_BEGIN_ALLOW_THREADS
2729 do {
2730 timeout = internal_select(s, 1);
2731 n = -1;
2732 if (timeout)
2733 break;
2734 #ifdef __VMS
2735 n = sendsegmented(s->sock_fd, buf, len, flags);
2736 #else
2737 n = send(s->sock_fd, buf, len, flags);
2738 #endif
2739 if (n < 0) {
2740 #ifdef EINTR
2741 /* We must handle EINTR here as there is no way for
2742 * the caller to know how much was sent otherwise. */
2743 if (errno == EINTR) {
2744 /* Run signal handlers. If an exception was
2745 * raised, abort and leave this socket in
2746 * an unknown state. */
2747 if (PyErr_CheckSignals())
2748 return NULL;
2749 continue;
2751 #endif
2752 break;
2754 buf += n;
2755 len -= n;
2756 } while (len > 0);
2757 Py_END_ALLOW_THREADS
2758 PyBuffer_Release(&pbuf);
2760 if (timeout == 1) {
2761 PyErr_SetString(socket_timeout, "timed out");
2762 return NULL;
2764 if (n < 0)
2765 return s->errorhandler();
2767 Py_INCREF(Py_None);
2768 return Py_None;
2771 PyDoc_STRVAR(sendall_doc,
2772 "sendall(data[, flags])\n\
2774 Send a data string to the socket. For the optional flags\n\
2775 argument, see the Unix manual. This calls send() repeatedly\n\
2776 until all data is sent. If an error occurs, it's impossible\n\
2777 to tell how much data has been sent.");
2780 /* s.sendto(data, [flags,] sockaddr) method */
2782 static PyObject *
2783 sock_sendto(PySocketSockObject *s, PyObject *args)
2785 Py_buffer pbuf;
2786 PyObject *addro;
2787 char *buf;
2788 Py_ssize_t len;
2789 sock_addr_t addrbuf;
2790 int addrlen, n = -1, flags, timeout;
2792 flags = 0;
2793 if (!PyArg_ParseTuple(args, "s*O:sendto", &pbuf, &addro)) {
2794 PyErr_Clear();
2795 if (!PyArg_ParseTuple(args, "s*iO:sendto",
2796 &pbuf, &flags, &addro))
2797 return NULL;
2799 buf = pbuf.buf;
2800 len = pbuf.len;
2802 if (!IS_SELECTABLE(s)) {
2803 PyBuffer_Release(&pbuf);
2804 return select_error();
2807 if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen)) {
2808 PyBuffer_Release(&pbuf);
2809 return NULL;
2812 Py_BEGIN_ALLOW_THREADS
2813 timeout = internal_select(s, 1);
2814 if (!timeout)
2815 n = sendto(s->sock_fd, buf, len, flags, SAS2SA(&addrbuf), addrlen);
2816 Py_END_ALLOW_THREADS
2818 PyBuffer_Release(&pbuf);
2819 if (timeout == 1) {
2820 PyErr_SetString(socket_timeout, "timed out");
2821 return NULL;
2823 if (n < 0)
2824 return s->errorhandler();
2825 return PyInt_FromLong((long)n);
2828 PyDoc_STRVAR(sendto_doc,
2829 "sendto(data[, flags], address) -> count\n\
2831 Like send(data, flags) but allows specifying the destination address.\n\
2832 For IP sockets, the address is a pair (hostaddr, port).");
2835 /* s.shutdown(how) method */
2837 static PyObject *
2838 sock_shutdown(PySocketSockObject *s, PyObject *arg)
2840 int how;
2841 int res;
2843 how = PyInt_AsLong(arg);
2844 if (how == -1 && PyErr_Occurred())
2845 return NULL;
2846 Py_BEGIN_ALLOW_THREADS
2847 res = shutdown(s->sock_fd, how);
2848 Py_END_ALLOW_THREADS
2849 if (res < 0)
2850 return s->errorhandler();
2851 Py_INCREF(Py_None);
2852 return Py_None;
2855 PyDoc_STRVAR(shutdown_doc,
2856 "shutdown(flag)\n\
2858 Shut down the reading side of the socket (flag == SHUT_RD), the writing side\n\
2859 of the socket (flag == SHUT_WR), or both ends (flag == SHUT_RDWR).");
2861 #if defined(MS_WINDOWS) && defined(SIO_RCVALL)
2862 static PyObject*
2863 sock_ioctl(PySocketSockObject *s, PyObject *arg)
2865 unsigned long cmd = SIO_RCVALL;
2866 unsigned int option = RCVALL_ON;
2867 DWORD recv;
2869 if (!PyArg_ParseTuple(arg, "kI:ioctl", &cmd, &option))
2870 return NULL;
2872 if (WSAIoctl(s->sock_fd, cmd, &option, sizeof(option),
2873 NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
2874 return set_error();
2876 return PyLong_FromUnsignedLong(recv);
2878 PyDoc_STRVAR(sock_ioctl_doc,
2879 "ioctl(cmd, option) -> long\n\
2881 Control the socket with WSAIoctl syscall. Currently only socket.SIO_RCVALL\n\
2882 is supported as control. Options must be one of the socket.RCVALL_*\n\
2883 constants.");
2885 #endif
2887 /* List of methods for socket objects */
2889 static PyMethodDef sock_methods[] = {
2890 {"accept", (PyCFunction)sock_accept, METH_NOARGS,
2891 accept_doc},
2892 {"bind", (PyCFunction)sock_bind, METH_O,
2893 bind_doc},
2894 {"close", (PyCFunction)sock_close, METH_NOARGS,
2895 close_doc},
2896 {"connect", (PyCFunction)sock_connect, METH_O,
2897 connect_doc},
2898 {"connect_ex", (PyCFunction)sock_connect_ex, METH_O,
2899 connect_ex_doc},
2900 #ifndef NO_DUP
2901 {"dup", (PyCFunction)sock_dup, METH_NOARGS,
2902 dup_doc},
2903 #endif
2904 {"fileno", (PyCFunction)sock_fileno, METH_NOARGS,
2905 fileno_doc},
2906 #ifdef HAVE_GETPEERNAME
2907 {"getpeername", (PyCFunction)sock_getpeername,
2908 METH_NOARGS, getpeername_doc},
2909 #endif
2910 {"getsockname", (PyCFunction)sock_getsockname,
2911 METH_NOARGS, getsockname_doc},
2912 {"getsockopt", (PyCFunction)sock_getsockopt, METH_VARARGS,
2913 getsockopt_doc},
2914 #if defined(MS_WINDOWS) && defined(SIO_RCVALL)
2915 {"ioctl", (PyCFunction)sock_ioctl, METH_VARARGS,
2916 sock_ioctl_doc},
2917 #endif
2918 {"listen", (PyCFunction)sock_listen, METH_O,
2919 listen_doc},
2920 #ifndef NO_DUP
2921 {"makefile", (PyCFunction)sock_makefile, METH_VARARGS,
2922 makefile_doc},
2923 #endif
2924 {"recv", (PyCFunction)sock_recv, METH_VARARGS,
2925 recv_doc},
2926 {"recv_into", (PyCFunction)sock_recv_into, METH_VARARGS | METH_KEYWORDS,
2927 recv_into_doc},
2928 {"recvfrom", (PyCFunction)sock_recvfrom, METH_VARARGS,
2929 recvfrom_doc},
2930 {"recvfrom_into", (PyCFunction)sock_recvfrom_into, METH_VARARGS | METH_KEYWORDS,
2931 recvfrom_into_doc},
2932 {"send", (PyCFunction)sock_send, METH_VARARGS,
2933 send_doc},
2934 {"sendall", (PyCFunction)sock_sendall, METH_VARARGS,
2935 sendall_doc},
2936 {"sendto", (PyCFunction)sock_sendto, METH_VARARGS,
2937 sendto_doc},
2938 {"setblocking", (PyCFunction)sock_setblocking, METH_O,
2939 setblocking_doc},
2940 {"settimeout", (PyCFunction)sock_settimeout, METH_O,
2941 settimeout_doc},
2942 {"gettimeout", (PyCFunction)sock_gettimeout, METH_NOARGS,
2943 gettimeout_doc},
2944 {"setsockopt", (PyCFunction)sock_setsockopt, METH_VARARGS,
2945 setsockopt_doc},
2946 {"shutdown", (PyCFunction)sock_shutdown, METH_O,
2947 shutdown_doc},
2948 #ifdef RISCOS
2949 {"sleeptaskw", (PyCFunction)sock_sleeptaskw, METH_O,
2950 sleeptaskw_doc},
2951 #endif
2952 {NULL, NULL} /* sentinel */
2955 /* SockObject members */
2956 static PyMemberDef sock_memberlist[] = {
2957 {"family", T_INT, offsetof(PySocketSockObject, sock_family), READONLY, "the socket family"},
2958 {"type", T_INT, offsetof(PySocketSockObject, sock_type), READONLY, "the socket type"},
2959 {"proto", T_INT, offsetof(PySocketSockObject, sock_proto), READONLY, "the socket protocol"},
2960 {"timeout", T_DOUBLE, offsetof(PySocketSockObject, sock_timeout), READONLY, "the socket timeout"},
2961 {0},
2964 /* Deallocate a socket object in response to the last Py_DECREF().
2965 First close the file description. */
2967 static void
2968 sock_dealloc(PySocketSockObject *s)
2970 if (s->sock_fd != -1)
2971 (void) SOCKETCLOSE(s->sock_fd);
2972 Py_TYPE(s)->tp_free((PyObject *)s);
2976 static PyObject *
2977 sock_repr(PySocketSockObject *s)
2979 char buf[512];
2980 #if SIZEOF_SOCKET_T > SIZEOF_LONG
2981 if (s->sock_fd > LONG_MAX) {
2982 /* this can occur on Win64, and actually there is a special
2983 ugly printf formatter for decimal pointer length integer
2984 printing, only bother if necessary*/
2985 PyErr_SetString(PyExc_OverflowError,
2986 "no printf formatter to display "
2987 "the socket descriptor in decimal");
2988 return NULL;
2990 #endif
2991 PyOS_snprintf(
2992 buf, sizeof(buf),
2993 "<socket object, fd=%ld, family=%d, type=%d, protocol=%d>",
2994 (long)s->sock_fd, s->sock_family,
2995 s->sock_type,
2996 s->sock_proto);
2997 return PyString_FromString(buf);
3001 /* Create a new, uninitialized socket object. */
3003 static PyObject *
3004 sock_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
3006 PyObject *new;
3008 new = type->tp_alloc(type, 0);
3009 if (new != NULL) {
3010 ((PySocketSockObject *)new)->sock_fd = -1;
3011 ((PySocketSockObject *)new)->sock_timeout = -1.0;
3012 ((PySocketSockObject *)new)->errorhandler = &set_error;
3014 return new;
3018 /* Initialize a new socket object. */
3020 /*ARGSUSED*/
3021 static int
3022 sock_initobj(PyObject *self, PyObject *args, PyObject *kwds)
3024 PySocketSockObject *s = (PySocketSockObject *)self;
3025 SOCKET_T fd;
3026 int family = AF_INET, type = SOCK_STREAM, proto = 0;
3027 static char *keywords[] = {"family", "type", "proto", 0};
3029 if (!PyArg_ParseTupleAndKeywords(args, kwds,
3030 "|iii:socket", keywords,
3031 &family, &type, &proto))
3032 return -1;
3034 Py_BEGIN_ALLOW_THREADS
3035 fd = socket(family, type, proto);
3036 Py_END_ALLOW_THREADS
3038 #ifdef MS_WINDOWS
3039 if (fd == INVALID_SOCKET)
3040 #else
3041 if (fd < 0)
3042 #endif
3044 set_error();
3045 return -1;
3047 init_sockobject(s, fd, family, type, proto);
3049 return 0;
3054 /* Type object for socket objects. */
3056 static PyTypeObject sock_type = {
3057 PyVarObject_HEAD_INIT(0, 0) /* Must fill in type value later */
3058 "_socket.socket", /* tp_name */
3059 sizeof(PySocketSockObject), /* tp_basicsize */
3060 0, /* tp_itemsize */
3061 (destructor)sock_dealloc, /* tp_dealloc */
3062 0, /* tp_print */
3063 0, /* tp_getattr */
3064 0, /* tp_setattr */
3065 0, /* tp_compare */
3066 (reprfunc)sock_repr, /* tp_repr */
3067 0, /* tp_as_number */
3068 0, /* tp_as_sequence */
3069 0, /* tp_as_mapping */
3070 0, /* tp_hash */
3071 0, /* tp_call */
3072 0, /* tp_str */
3073 PyObject_GenericGetAttr, /* tp_getattro */
3074 0, /* tp_setattro */
3075 0, /* tp_as_buffer */
3076 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
3077 sock_doc, /* tp_doc */
3078 0, /* tp_traverse */
3079 0, /* tp_clear */
3080 0, /* tp_richcompare */
3081 0, /* tp_weaklistoffset */
3082 0, /* tp_iter */
3083 0, /* tp_iternext */
3084 sock_methods, /* tp_methods */
3085 sock_memberlist, /* tp_members */
3086 0, /* tp_getset */
3087 0, /* tp_base */
3088 0, /* tp_dict */
3089 0, /* tp_descr_get */
3090 0, /* tp_descr_set */
3091 0, /* tp_dictoffset */
3092 sock_initobj, /* tp_init */
3093 PyType_GenericAlloc, /* tp_alloc */
3094 sock_new, /* tp_new */
3095 PyObject_Del, /* tp_free */
3099 /* Python interface to gethostname(). */
3101 /*ARGSUSED*/
3102 static PyObject *
3103 socket_gethostname(PyObject *self, PyObject *unused)
3105 char buf[1024];
3106 int res;
3107 Py_BEGIN_ALLOW_THREADS
3108 res = gethostname(buf, (int) sizeof buf - 1);
3109 Py_END_ALLOW_THREADS
3110 if (res < 0)
3111 return set_error();
3112 buf[sizeof buf - 1] = '\0';
3113 return PyString_FromString(buf);
3116 PyDoc_STRVAR(gethostname_doc,
3117 "gethostname() -> string\n\
3119 Return the current host name.");
3122 /* Python interface to gethostbyname(name). */
3124 /*ARGSUSED*/
3125 static PyObject *
3126 socket_gethostbyname(PyObject *self, PyObject *args)
3128 char *name;
3129 sock_addr_t addrbuf;
3131 if (!PyArg_ParseTuple(args, "s:gethostbyname", &name))
3132 return NULL;
3133 if (setipaddr(name, SAS2SA(&addrbuf), sizeof(addrbuf), AF_INET) < 0)
3134 return NULL;
3135 return makeipaddr(SAS2SA(&addrbuf), sizeof(struct sockaddr_in));
3138 PyDoc_STRVAR(gethostbyname_doc,
3139 "gethostbyname(host) -> address\n\
3141 Return the IP address (a string of the form '255.255.255.255') for a host.");
3144 /* Convenience function common to gethostbyname_ex and gethostbyaddr */
3146 static PyObject *
3147 gethost_common(struct hostent *h, struct sockaddr *addr, int alen, int af)
3149 char **pch;
3150 PyObject *rtn_tuple = (PyObject *)NULL;
3151 PyObject *name_list = (PyObject *)NULL;
3152 PyObject *addr_list = (PyObject *)NULL;
3153 PyObject *tmp;
3155 if (h == NULL) {
3156 /* Let's get real error message to return */
3157 #ifndef RISCOS
3158 set_herror(h_errno);
3159 #else
3160 PyErr_SetString(socket_error, "host not found");
3161 #endif
3162 return NULL;
3165 if (h->h_addrtype != af) {
3166 /* Let's get real error message to return */
3167 PyErr_SetString(socket_error,
3168 (char *)strerror(EAFNOSUPPORT));
3170 return NULL;
3173 switch (af) {
3175 case AF_INET:
3176 if (alen < sizeof(struct sockaddr_in))
3177 return NULL;
3178 break;
3180 #ifdef ENABLE_IPV6
3181 case AF_INET6:
3182 if (alen < sizeof(struct sockaddr_in6))
3183 return NULL;
3184 break;
3185 #endif
3189 if ((name_list = PyList_New(0)) == NULL)
3190 goto err;
3192 if ((addr_list = PyList_New(0)) == NULL)
3193 goto err;
3195 /* SF #1511317: h_aliases can be NULL */
3196 if (h->h_aliases) {
3197 for (pch = h->h_aliases; *pch != NULL; pch++) {
3198 int status;
3199 tmp = PyString_FromString(*pch);
3200 if (tmp == NULL)
3201 goto err;
3203 status = PyList_Append(name_list, tmp);
3204 Py_DECREF(tmp);
3206 if (status)
3207 goto err;
3211 for (pch = h->h_addr_list; *pch != NULL; pch++) {
3212 int status;
3214 switch (af) {
3216 case AF_INET:
3218 struct sockaddr_in sin;
3219 memset(&sin, 0, sizeof(sin));
3220 sin.sin_family = af;
3221 #ifdef HAVE_SOCKADDR_SA_LEN
3222 sin.sin_len = sizeof(sin);
3223 #endif
3224 memcpy(&sin.sin_addr, *pch, sizeof(sin.sin_addr));
3225 tmp = makeipaddr((struct sockaddr *)&sin, sizeof(sin));
3227 if (pch == h->h_addr_list && alen >= sizeof(sin))
3228 memcpy((char *) addr, &sin, sizeof(sin));
3229 break;
3232 #ifdef ENABLE_IPV6
3233 case AF_INET6:
3235 struct sockaddr_in6 sin6;
3236 memset(&sin6, 0, sizeof(sin6));
3237 sin6.sin6_family = af;
3238 #ifdef HAVE_SOCKADDR_SA_LEN
3239 sin6.sin6_len = sizeof(sin6);
3240 #endif
3241 memcpy(&sin6.sin6_addr, *pch, sizeof(sin6.sin6_addr));
3242 tmp = makeipaddr((struct sockaddr *)&sin6,
3243 sizeof(sin6));
3245 if (pch == h->h_addr_list && alen >= sizeof(sin6))
3246 memcpy((char *) addr, &sin6, sizeof(sin6));
3247 break;
3249 #endif
3251 default: /* can't happen */
3252 PyErr_SetString(socket_error,
3253 "unsupported address family");
3254 return NULL;
3257 if (tmp == NULL)
3258 goto err;
3260 status = PyList_Append(addr_list, tmp);
3261 Py_DECREF(tmp);
3263 if (status)
3264 goto err;
3267 rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list);
3269 err:
3270 Py_XDECREF(name_list);
3271 Py_XDECREF(addr_list);
3272 return rtn_tuple;
3276 /* Python interface to gethostbyname_ex(name). */
3278 /*ARGSUSED*/
3279 static PyObject *
3280 socket_gethostbyname_ex(PyObject *self, PyObject *args)
3282 char *name;
3283 struct hostent *h;
3284 #ifdef ENABLE_IPV6
3285 struct sockaddr_storage addr;
3286 #else
3287 struct sockaddr_in addr;
3288 #endif
3289 struct sockaddr *sa;
3290 PyObject *ret;
3291 #ifdef HAVE_GETHOSTBYNAME_R
3292 struct hostent hp_allocated;
3293 #ifdef HAVE_GETHOSTBYNAME_R_3_ARG
3294 struct hostent_data data;
3295 #else
3296 char buf[16384];
3297 int buf_len = (sizeof buf) - 1;
3298 int errnop;
3299 #endif
3300 #if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
3301 int result;
3302 #endif
3303 #endif /* HAVE_GETHOSTBYNAME_R */
3305 if (!PyArg_ParseTuple(args, "s:gethostbyname_ex", &name))
3306 return NULL;
3307 if (setipaddr(name, (struct sockaddr *)&addr, sizeof(addr), AF_INET) < 0)
3308 return NULL;
3309 Py_BEGIN_ALLOW_THREADS
3310 #ifdef HAVE_GETHOSTBYNAME_R
3311 #if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
3312 result = gethostbyname_r(name, &hp_allocated, buf, buf_len,
3313 &h, &errnop);
3314 #elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
3315 h = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop);
3316 #else /* HAVE_GETHOSTBYNAME_R_3_ARG */
3317 memset((void *) &data, '\0', sizeof(data));
3318 result = gethostbyname_r(name, &hp_allocated, &data);
3319 h = (result != 0) ? NULL : &hp_allocated;
3320 #endif
3321 #else /* not HAVE_GETHOSTBYNAME_R */
3322 #ifdef USE_GETHOSTBYNAME_LOCK
3323 PyThread_acquire_lock(netdb_lock, 1);
3324 #endif
3325 h = gethostbyname(name);
3326 #endif /* HAVE_GETHOSTBYNAME_R */
3327 Py_END_ALLOW_THREADS
3328 /* Some C libraries would require addr.__ss_family instead of
3329 addr.ss_family.
3330 Therefore, we cast the sockaddr_storage into sockaddr to
3331 access sa_family. */
3332 sa = (struct sockaddr*)&addr;
3333 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr),
3334 sa->sa_family);
3335 #ifdef USE_GETHOSTBYNAME_LOCK
3336 PyThread_release_lock(netdb_lock);
3337 #endif
3338 return ret;
3341 PyDoc_STRVAR(ghbn_ex_doc,
3342 "gethostbyname_ex(host) -> (name, aliaslist, addresslist)\n\
3344 Return the true host name, a list of aliases, and a list of IP addresses,\n\
3345 for a host. The host argument is a string giving a host name or IP number.");
3348 /* Python interface to gethostbyaddr(IP). */
3350 /*ARGSUSED*/
3351 static PyObject *
3352 socket_gethostbyaddr(PyObject *self, PyObject *args)
3354 #ifdef ENABLE_IPV6
3355 struct sockaddr_storage addr;
3356 #else
3357 struct sockaddr_in addr;
3358 #endif
3359 struct sockaddr *sa = (struct sockaddr *)&addr;
3360 char *ip_num;
3361 struct hostent *h;
3362 PyObject *ret;
3363 #ifdef HAVE_GETHOSTBYNAME_R
3364 struct hostent hp_allocated;
3365 #ifdef HAVE_GETHOSTBYNAME_R_3_ARG
3366 struct hostent_data data;
3367 #else
3368 /* glibcs up to 2.10 assume that the buf argument to
3369 gethostbyaddr_r is 8-byte aligned, which at least llvm-gcc
3370 does not ensure. The attribute below instructs the compiler
3371 to maintain this alignment. */
3372 char buf[16384] Py_ALIGNED(8);
3373 int buf_len = (sizeof buf) - 1;
3374 int errnop;
3375 #endif
3376 #if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
3377 int result;
3378 #endif
3379 #endif /* HAVE_GETHOSTBYNAME_R */
3380 char *ap;
3381 int al;
3382 int af;
3384 if (!PyArg_ParseTuple(args, "s:gethostbyaddr", &ip_num))
3385 return NULL;
3386 af = AF_UNSPEC;
3387 if (setipaddr(ip_num, sa, sizeof(addr), af) < 0)
3388 return NULL;
3389 af = sa->sa_family;
3390 ap = NULL;
3391 al = 0;
3392 switch (af) {
3393 case AF_INET:
3394 ap = (char *)&((struct sockaddr_in *)sa)->sin_addr;
3395 al = sizeof(((struct sockaddr_in *)sa)->sin_addr);
3396 break;
3397 #ifdef ENABLE_IPV6
3398 case AF_INET6:
3399 ap = (char *)&((struct sockaddr_in6 *)sa)->sin6_addr;
3400 al = sizeof(((struct sockaddr_in6 *)sa)->sin6_addr);
3401 break;
3402 #endif
3403 default:
3404 PyErr_SetString(socket_error, "unsupported address family");
3405 return NULL;
3407 Py_BEGIN_ALLOW_THREADS
3408 #ifdef HAVE_GETHOSTBYNAME_R
3409 #if defined(HAVE_GETHOSTBYNAME_R_6_ARG)
3410 result = gethostbyaddr_r(ap, al, af,
3411 &hp_allocated, buf, buf_len,
3412 &h, &errnop);
3413 #elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
3414 h = gethostbyaddr_r(ap, al, af,
3415 &hp_allocated, buf, buf_len, &errnop);
3416 #else /* HAVE_GETHOSTBYNAME_R_3_ARG */
3417 memset((void *) &data, '\0', sizeof(data));
3418 result = gethostbyaddr_r(ap, al, af, &hp_allocated, &data);
3419 h = (result != 0) ? NULL : &hp_allocated;
3420 #endif
3421 #else /* not HAVE_GETHOSTBYNAME_R */
3422 #ifdef USE_GETHOSTBYNAME_LOCK
3423 PyThread_acquire_lock(netdb_lock, 1);
3424 #endif
3425 h = gethostbyaddr(ap, al, af);
3426 #endif /* HAVE_GETHOSTBYNAME_R */
3427 Py_END_ALLOW_THREADS
3428 ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr), af);
3429 #ifdef USE_GETHOSTBYNAME_LOCK
3430 PyThread_release_lock(netdb_lock);
3431 #endif
3432 return ret;
3435 PyDoc_STRVAR(gethostbyaddr_doc,
3436 "gethostbyaddr(host) -> (name, aliaslist, addresslist)\n\
3438 Return the true host name, a list of aliases, and a list of IP addresses,\n\
3439 for a host. The host argument is a string giving a host name or IP number.");
3442 /* Python interface to getservbyname(name).
3443 This only returns the port number, since the other info is already
3444 known or not useful (like the list of aliases). */
3446 /*ARGSUSED*/
3447 static PyObject *
3448 socket_getservbyname(PyObject *self, PyObject *args)
3450 char *name, *proto=NULL;
3451 struct servent *sp;
3452 if (!PyArg_ParseTuple(args, "s|s:getservbyname", &name, &proto))
3453 return NULL;
3454 Py_BEGIN_ALLOW_THREADS
3455 sp = getservbyname(name, proto);
3456 Py_END_ALLOW_THREADS
3457 if (sp == NULL) {
3458 PyErr_SetString(socket_error, "service/proto not found");
3459 return NULL;
3461 return PyInt_FromLong((long) ntohs(sp->s_port));
3464 PyDoc_STRVAR(getservbyname_doc,
3465 "getservbyname(servicename[, protocolname]) -> integer\n\
3467 Return a port number from a service name and protocol name.\n\
3468 The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3469 otherwise any protocol will match.");
3472 /* Python interface to getservbyport(port).
3473 This only returns the service name, since the other info is already
3474 known or not useful (like the list of aliases). */
3476 /*ARGSUSED*/
3477 static PyObject *
3478 socket_getservbyport(PyObject *self, PyObject *args)
3480 int port;
3481 char *proto=NULL;
3482 struct servent *sp;
3483 if (!PyArg_ParseTuple(args, "i|s:getservbyport", &port, &proto))
3484 return NULL;
3485 if (port < 0 || port > 0xffff) {
3486 PyErr_SetString(
3487 PyExc_OverflowError,
3488 "getservbyport: port must be 0-65535.");
3489 return NULL;
3491 Py_BEGIN_ALLOW_THREADS
3492 sp = getservbyport(htons((short)port), proto);
3493 Py_END_ALLOW_THREADS
3494 if (sp == NULL) {
3495 PyErr_SetString(socket_error, "port/proto not found");
3496 return NULL;
3498 return PyString_FromString(sp->s_name);
3501 PyDoc_STRVAR(getservbyport_doc,
3502 "getservbyport(port[, protocolname]) -> string\n\
3504 Return the service name from a port number and protocol name.\n\
3505 The optional protocol name, if given, should be 'tcp' or 'udp',\n\
3506 otherwise any protocol will match.");
3508 /* Python interface to getprotobyname(name).
3509 This only returns the protocol number, since the other info is
3510 already known or not useful (like the list of aliases). */
3512 /*ARGSUSED*/
3513 static PyObject *
3514 socket_getprotobyname(PyObject *self, PyObject *args)
3516 char *name;
3517 struct protoent *sp;
3518 #ifdef __BEOS__
3519 /* Not available in BeOS yet. - [cjh] */
3520 PyErr_SetString(socket_error, "getprotobyname not supported");
3521 return NULL;
3522 #else
3523 if (!PyArg_ParseTuple(args, "s:getprotobyname", &name))
3524 return NULL;
3525 Py_BEGIN_ALLOW_THREADS
3526 sp = getprotobyname(name);
3527 Py_END_ALLOW_THREADS
3528 if (sp == NULL) {
3529 PyErr_SetString(socket_error, "protocol not found");
3530 return NULL;
3532 return PyInt_FromLong((long) sp->p_proto);
3533 #endif
3536 PyDoc_STRVAR(getprotobyname_doc,
3537 "getprotobyname(name) -> integer\n\
3539 Return the protocol number for the named protocol. (Rarely used.)");
3542 #ifdef HAVE_SOCKETPAIR
3543 /* Create a pair of sockets using the socketpair() function.
3544 Arguments as for socket() except the default family is AF_UNIX if
3545 defined on the platform; otherwise, the default is AF_INET. */
3547 /*ARGSUSED*/
3548 static PyObject *
3549 socket_socketpair(PyObject *self, PyObject *args)
3551 PySocketSockObject *s0 = NULL, *s1 = NULL;
3552 SOCKET_T sv[2];
3553 int family, type = SOCK_STREAM, proto = 0;
3554 PyObject *res = NULL;
3556 #if defined(AF_UNIX)
3557 family = AF_UNIX;
3558 #else
3559 family = AF_INET;
3560 #endif
3561 if (!PyArg_ParseTuple(args, "|iii:socketpair",
3562 &family, &type, &proto))
3563 return NULL;
3564 /* Create a pair of socket fds */
3565 if (socketpair(family, type, proto, sv) < 0)
3566 return set_error();
3567 s0 = new_sockobject(sv[0], family, type, proto);
3568 if (s0 == NULL)
3569 goto finally;
3570 s1 = new_sockobject(sv[1], family, type, proto);
3571 if (s1 == NULL)
3572 goto finally;
3573 res = PyTuple_Pack(2, s0, s1);
3575 finally:
3576 if (res == NULL) {
3577 if (s0 == NULL)
3578 SOCKETCLOSE(sv[0]);
3579 if (s1 == NULL)
3580 SOCKETCLOSE(sv[1]);
3582 Py_XDECREF(s0);
3583 Py_XDECREF(s1);
3584 return res;
3587 PyDoc_STRVAR(socketpair_doc,
3588 "socketpair([family[, type[, proto]]]) -> (socket object, socket object)\n\
3590 Create a pair of socket objects from the sockets returned by the platform\n\
3591 socketpair() function.\n\
3592 The arguments are the same as for socket() except the default family is\n\
3593 AF_UNIX if defined on the platform; otherwise, the default is AF_INET.");
3595 #endif /* HAVE_SOCKETPAIR */
3598 #ifndef NO_DUP
3599 /* Create a socket object from a numeric file description.
3600 Useful e.g. if stdin is a socket.
3601 Additional arguments as for socket(). */
3603 /*ARGSUSED*/
3604 static PyObject *
3605 socket_fromfd(PyObject *self, PyObject *args)
3607 PySocketSockObject *s;
3608 SOCKET_T fd;
3609 int family, type, proto = 0;
3610 if (!PyArg_ParseTuple(args, "iii|i:fromfd",
3611 &fd, &family, &type, &proto))
3612 return NULL;
3613 /* Dup the fd so it and the socket can be closed independently */
3614 fd = dup(fd);
3615 if (fd < 0)
3616 return set_error();
3617 s = new_sockobject(fd, family, type, proto);
3618 return (PyObject *) s;
3621 PyDoc_STRVAR(fromfd_doc,
3622 "fromfd(fd, family, type[, proto]) -> socket object\n\
3624 Create a socket object from a duplicate of the given\n\
3625 file descriptor.\n\
3626 The remaining arguments are the same as for socket().");
3628 #endif /* NO_DUP */
3631 static PyObject *
3632 socket_ntohs(PyObject *self, PyObject *args)
3634 int x1, x2;
3636 if (!PyArg_ParseTuple(args, "i:ntohs", &x1)) {
3637 return NULL;
3639 if (x1 < 0) {
3640 PyErr_SetString(PyExc_OverflowError,
3641 "can't convert negative number to unsigned long");
3642 return NULL;
3644 x2 = (unsigned int)ntohs((unsigned short)x1);
3645 return PyInt_FromLong(x2);
3648 PyDoc_STRVAR(ntohs_doc,
3649 "ntohs(integer) -> integer\n\
3651 Convert a 16-bit integer from network to host byte order.");
3654 static PyObject *
3655 socket_ntohl(PyObject *self, PyObject *arg)
3657 unsigned long x;
3659 if (PyInt_Check(arg)) {
3660 x = PyInt_AS_LONG(arg);
3661 if (x == (unsigned long) -1 && PyErr_Occurred())
3662 return NULL;
3663 if ((long)x < 0) {
3664 PyErr_SetString(PyExc_OverflowError,
3665 "can't convert negative number to unsigned long");
3666 return NULL;
3669 else if (PyLong_Check(arg)) {
3670 x = PyLong_AsUnsignedLong(arg);
3671 if (x == (unsigned long) -1 && PyErr_Occurred())
3672 return NULL;
3673 #if SIZEOF_LONG > 4
3675 unsigned long y;
3676 /* only want the trailing 32 bits */
3677 y = x & 0xFFFFFFFFUL;
3678 if (y ^ x)
3679 return PyErr_Format(PyExc_OverflowError,
3680 "long int larger than 32 bits");
3681 x = y;
3683 #endif
3685 else
3686 return PyErr_Format(PyExc_TypeError,
3687 "expected int/long, %s found",
3688 Py_TYPE(arg)->tp_name);
3689 if (x == (unsigned long) -1 && PyErr_Occurred())
3690 return NULL;
3691 return PyLong_FromUnsignedLong(ntohl(x));
3694 PyDoc_STRVAR(ntohl_doc,
3695 "ntohl(integer) -> integer\n\
3697 Convert a 32-bit integer from network to host byte order.");
3700 static PyObject *
3701 socket_htons(PyObject *self, PyObject *args)
3703 int x1, x2;
3705 if (!PyArg_ParseTuple(args, "i:htons", &x1)) {
3706 return NULL;
3708 if (x1 < 0) {
3709 PyErr_SetString(PyExc_OverflowError,
3710 "can't convert negative number to unsigned long");
3711 return NULL;
3713 x2 = (unsigned int)htons((unsigned short)x1);
3714 return PyInt_FromLong(x2);
3717 PyDoc_STRVAR(htons_doc,
3718 "htons(integer) -> integer\n\
3720 Convert a 16-bit integer from host to network byte order.");
3723 static PyObject *
3724 socket_htonl(PyObject *self, PyObject *arg)
3726 unsigned long x;
3728 if (PyInt_Check(arg)) {
3729 x = PyInt_AS_LONG(arg);
3730 if (x == (unsigned long) -1 && PyErr_Occurred())
3731 return NULL;
3732 if ((long)x < 0) {
3733 PyErr_SetString(PyExc_OverflowError,
3734 "can't convert negative number to unsigned long");
3735 return NULL;
3738 else if (PyLong_Check(arg)) {
3739 x = PyLong_AsUnsignedLong(arg);
3740 if (x == (unsigned long) -1 && PyErr_Occurred())
3741 return NULL;
3742 #if SIZEOF_LONG > 4
3744 unsigned long y;
3745 /* only want the trailing 32 bits */
3746 y = x & 0xFFFFFFFFUL;
3747 if (y ^ x)
3748 return PyErr_Format(PyExc_OverflowError,
3749 "long int larger than 32 bits");
3750 x = y;
3752 #endif
3754 else
3755 return PyErr_Format(PyExc_TypeError,
3756 "expected int/long, %s found",
3757 Py_TYPE(arg)->tp_name);
3758 return PyLong_FromUnsignedLong(htonl((unsigned long)x));
3761 PyDoc_STRVAR(htonl_doc,
3762 "htonl(integer) -> integer\n\
3764 Convert a 32-bit integer from host to network byte order.");
3766 /* socket.inet_aton() and socket.inet_ntoa() functions. */
3768 PyDoc_STRVAR(inet_aton_doc,
3769 "inet_aton(string) -> packed 32-bit IP representation\n\
3771 Convert an IP address in string format (123.45.67.89) to the 32-bit packed\n\
3772 binary format used in low-level network functions.");
3774 static PyObject*
3775 socket_inet_aton(PyObject *self, PyObject *args)
3777 #ifndef INADDR_NONE
3778 #define INADDR_NONE (-1)
3779 #endif
3780 #ifdef HAVE_INET_ATON
3781 struct in_addr buf;
3782 #endif
3784 #if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
3785 #if (SIZEOF_INT != 4)
3786 #error "Not sure if in_addr_t exists and int is not 32-bits."
3787 #endif
3788 /* Have to use inet_addr() instead */
3789 unsigned int packed_addr;
3790 #endif
3791 char *ip_addr;
3793 if (!PyArg_ParseTuple(args, "s:inet_aton", &ip_addr))
3794 return NULL;
3797 #ifdef HAVE_INET_ATON
3799 #ifdef USE_INET_ATON_WEAKLINK
3800 if (inet_aton != NULL) {
3801 #endif
3802 if (inet_aton(ip_addr, &buf))
3803 return PyString_FromStringAndSize((char *)(&buf),
3804 sizeof(buf));
3806 PyErr_SetString(socket_error,
3807 "illegal IP address string passed to inet_aton");
3808 return NULL;
3810 #ifdef USE_INET_ATON_WEAKLINK
3811 } else {
3812 #endif
3814 #endif
3816 #if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
3818 /* special-case this address as inet_addr might return INADDR_NONE
3819 * for this */
3820 if (strcmp(ip_addr, "255.255.255.255") == 0) {
3821 packed_addr = 0xFFFFFFFF;
3822 } else {
3824 packed_addr = inet_addr(ip_addr);
3826 if (packed_addr == INADDR_NONE) { /* invalid address */
3827 PyErr_SetString(socket_error,
3828 "illegal IP address string passed to inet_aton");
3829 return NULL;
3832 return PyString_FromStringAndSize((char *) &packed_addr,
3833 sizeof(packed_addr));
3835 #ifdef USE_INET_ATON_WEAKLINK
3837 #endif
3839 #endif
3842 PyDoc_STRVAR(inet_ntoa_doc,
3843 "inet_ntoa(packed_ip) -> ip_address_string\n\
3845 Convert an IP address from 32-bit packed binary format to string format");
3847 static PyObject*
3848 socket_inet_ntoa(PyObject *self, PyObject *args)
3850 char *packed_str;
3851 int addr_len;
3852 struct in_addr packed_addr;
3854 if (!PyArg_ParseTuple(args, "s#:inet_ntoa", &packed_str, &addr_len)) {
3855 return NULL;
3858 if (addr_len != sizeof(packed_addr)) {
3859 PyErr_SetString(socket_error,
3860 "packed IP wrong length for inet_ntoa");
3861 return NULL;
3864 memcpy(&packed_addr, packed_str, addr_len);
3866 return PyString_FromString(inet_ntoa(packed_addr));
3869 #ifdef HAVE_INET_PTON
3871 PyDoc_STRVAR(inet_pton_doc,
3872 "inet_pton(af, ip) -> packed IP address string\n\
3874 Convert an IP address from string format to a packed string suitable\n\
3875 for use with low-level network functions.");
3877 static PyObject *
3878 socket_inet_pton(PyObject *self, PyObject *args)
3880 int af;
3881 char* ip;
3882 int retval;
3883 #ifdef ENABLE_IPV6
3884 char packed[MAX(sizeof(struct in_addr), sizeof(struct in6_addr))];
3885 #else
3886 char packed[sizeof(struct in_addr)];
3887 #endif
3888 if (!PyArg_ParseTuple(args, "is:inet_pton", &af, &ip)) {
3889 return NULL;
3892 #if !defined(ENABLE_IPV6) && defined(AF_INET6)
3893 if(af == AF_INET6) {
3894 PyErr_SetString(socket_error,
3895 "can't use AF_INET6, IPv6 is disabled");
3896 return NULL;
3898 #endif
3900 retval = inet_pton(af, ip, packed);
3901 if (retval < 0) {
3902 PyErr_SetFromErrno(socket_error);
3903 return NULL;
3904 } else if (retval == 0) {
3905 PyErr_SetString(socket_error,
3906 "illegal IP address string passed to inet_pton");
3907 return NULL;
3908 } else if (af == AF_INET) {
3909 return PyString_FromStringAndSize(packed,
3910 sizeof(struct in_addr));
3911 #ifdef ENABLE_IPV6
3912 } else if (af == AF_INET6) {
3913 return PyString_FromStringAndSize(packed,
3914 sizeof(struct in6_addr));
3915 #endif
3916 } else {
3917 PyErr_SetString(socket_error, "unknown address family");
3918 return NULL;
3922 PyDoc_STRVAR(inet_ntop_doc,
3923 "inet_ntop(af, packed_ip) -> string formatted IP address\n\
3925 Convert a packed IP address of the given family to string format.");
3927 static PyObject *
3928 socket_inet_ntop(PyObject *self, PyObject *args)
3930 int af;
3931 char* packed;
3932 int len;
3933 const char* retval;
3934 #ifdef ENABLE_IPV6
3935 char ip[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN) + 1];
3936 #else
3937 char ip[INET_ADDRSTRLEN + 1];
3938 #endif
3940 /* Guarantee NUL-termination for PyString_FromString() below */
3941 memset((void *) &ip[0], '\0', sizeof(ip));
3943 if (!PyArg_ParseTuple(args, "is#:inet_ntop", &af, &packed, &len)) {
3944 return NULL;
3947 if (af == AF_INET) {
3948 if (len != sizeof(struct in_addr)) {
3949 PyErr_SetString(PyExc_ValueError,
3950 "invalid length of packed IP address string");
3951 return NULL;
3953 #ifdef ENABLE_IPV6
3954 } else if (af == AF_INET6) {
3955 if (len != sizeof(struct in6_addr)) {
3956 PyErr_SetString(PyExc_ValueError,
3957 "invalid length of packed IP address string");
3958 return NULL;
3960 #endif
3961 } else {
3962 PyErr_Format(PyExc_ValueError,
3963 "unknown address family %d", af);
3964 return NULL;
3967 retval = inet_ntop(af, packed, ip, sizeof(ip));
3968 if (!retval) {
3969 PyErr_SetFromErrno(socket_error);
3970 return NULL;
3971 } else {
3972 return PyString_FromString(retval);
3975 /* NOTREACHED */
3976 PyErr_SetString(PyExc_RuntimeError, "invalid handling of inet_ntop");
3977 return NULL;
3980 #endif /* HAVE_INET_PTON */
3982 /* Python interface to getaddrinfo(host, port). */
3984 /*ARGSUSED*/
3985 static PyObject *
3986 socket_getaddrinfo(PyObject *self, PyObject *args)
3988 struct addrinfo hints, *res;
3989 struct addrinfo *res0 = NULL;
3990 PyObject *hobj = NULL;
3991 PyObject *pobj = (PyObject *)NULL;
3992 char pbuf[30];
3993 char *hptr, *pptr;
3994 int family, socktype, protocol, flags;
3995 int error;
3996 PyObject *all = (PyObject *)NULL;
3997 PyObject *single = (PyObject *)NULL;
3998 PyObject *idna = NULL;
4000 family = socktype = protocol = flags = 0;
4001 family = AF_UNSPEC;
4002 if (!PyArg_ParseTuple(args, "OO|iiii:getaddrinfo",
4003 &hobj, &pobj, &family, &socktype,
4004 &protocol, &flags)) {
4005 return NULL;
4007 if (hobj == Py_None) {
4008 hptr = NULL;
4009 } else if (PyUnicode_Check(hobj)) {
4010 idna = PyObject_CallMethod(hobj, "encode", "s", "idna");
4011 if (!idna)
4012 return NULL;
4013 hptr = PyString_AsString(idna);
4014 } else if (PyString_Check(hobj)) {
4015 hptr = PyString_AsString(hobj);
4016 } else {
4017 PyErr_SetString(PyExc_TypeError,
4018 "getaddrinfo() argument 1 must be string or None");
4019 return NULL;
4021 if (PyInt_Check(pobj)) {
4022 PyOS_snprintf(pbuf, sizeof(pbuf), "%ld", PyInt_AsLong(pobj));
4023 pptr = pbuf;
4024 } else if (PyString_Check(pobj)) {
4025 pptr = PyString_AsString(pobj);
4026 } else if (pobj == Py_None) {
4027 pptr = (char *)NULL;
4028 } else {
4029 PyErr_SetString(socket_error, "Int or String expected");
4030 goto err;
4032 memset(&hints, 0, sizeof(hints));
4033 hints.ai_family = family;
4034 hints.ai_socktype = socktype;
4035 hints.ai_protocol = protocol;
4036 hints.ai_flags = flags;
4037 Py_BEGIN_ALLOW_THREADS
4038 ACQUIRE_GETADDRINFO_LOCK
4039 error = getaddrinfo(hptr, pptr, &hints, &res0);
4040 Py_END_ALLOW_THREADS
4041 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
4042 if (error) {
4043 set_gaierror(error);
4044 goto err;
4047 if ((all = PyList_New(0)) == NULL)
4048 goto err;
4049 for (res = res0; res; res = res->ai_next) {
4050 PyObject *addr =
4051 makesockaddr(-1, res->ai_addr, res->ai_addrlen, protocol);
4052 if (addr == NULL)
4053 goto err;
4054 single = Py_BuildValue("iiisO", res->ai_family,
4055 res->ai_socktype, res->ai_protocol,
4056 res->ai_canonname ? res->ai_canonname : "",
4057 addr);
4058 Py_DECREF(addr);
4059 if (single == NULL)
4060 goto err;
4062 if (PyList_Append(all, single))
4063 goto err;
4064 Py_XDECREF(single);
4066 Py_XDECREF(idna);
4067 if (res0)
4068 freeaddrinfo(res0);
4069 return all;
4070 err:
4071 Py_XDECREF(single);
4072 Py_XDECREF(all);
4073 Py_XDECREF(idna);
4074 if (res0)
4075 freeaddrinfo(res0);
4076 return (PyObject *)NULL;
4079 PyDoc_STRVAR(getaddrinfo_doc,
4080 "getaddrinfo(host, port [, family, socktype, proto, flags])\n\
4081 -> list of (family, socktype, proto, canonname, sockaddr)\n\
4083 Resolve host and port into addrinfo struct.");
4085 /* Python interface to getnameinfo(sa, flags). */
4087 /*ARGSUSED*/
4088 static PyObject *
4089 socket_getnameinfo(PyObject *self, PyObject *args)
4091 PyObject *sa = (PyObject *)NULL;
4092 int flags;
4093 char *hostp;
4094 int port, flowinfo, scope_id;
4095 char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
4096 struct addrinfo hints, *res = NULL;
4097 int error;
4098 PyObject *ret = (PyObject *)NULL;
4100 flags = flowinfo = scope_id = 0;
4101 if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags))
4102 return NULL;
4103 if (!PyArg_ParseTuple(sa, "si|ii",
4104 &hostp, &port, &flowinfo, &scope_id))
4105 return NULL;
4106 PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port);
4107 memset(&hints, 0, sizeof(hints));
4108 hints.ai_family = AF_UNSPEC;
4109 hints.ai_socktype = SOCK_DGRAM; /* make numeric port happy */
4110 Py_BEGIN_ALLOW_THREADS
4111 ACQUIRE_GETADDRINFO_LOCK
4112 error = getaddrinfo(hostp, pbuf, &hints, &res);
4113 Py_END_ALLOW_THREADS
4114 RELEASE_GETADDRINFO_LOCK /* see comment in setipaddr() */
4115 if (error) {
4116 set_gaierror(error);
4117 goto fail;
4119 if (res->ai_next) {
4120 PyErr_SetString(socket_error,
4121 "sockaddr resolved to multiple addresses");
4122 goto fail;
4124 switch (res->ai_family) {
4125 case AF_INET:
4127 char *t1;
4128 int t2;
4129 if (PyArg_ParseTuple(sa, "si", &t1, &t2) == 0) {
4130 PyErr_SetString(socket_error,
4131 "IPv4 sockaddr must be 2 tuple");
4132 goto fail;
4134 break;
4136 #ifdef ENABLE_IPV6
4137 case AF_INET6:
4139 struct sockaddr_in6 *sin6;
4140 sin6 = (struct sockaddr_in6 *)res->ai_addr;
4141 sin6->sin6_flowinfo = flowinfo;
4142 sin6->sin6_scope_id = scope_id;
4143 break;
4145 #endif
4147 error = getnameinfo(res->ai_addr, res->ai_addrlen,
4148 hbuf, sizeof(hbuf), pbuf, sizeof(pbuf), flags);
4149 if (error) {
4150 set_gaierror(error);
4151 goto fail;
4153 ret = Py_BuildValue("ss", hbuf, pbuf);
4155 fail:
4156 if (res)
4157 freeaddrinfo(res);
4158 return ret;
4161 PyDoc_STRVAR(getnameinfo_doc,
4162 "getnameinfo(sockaddr, flags) --> (host, port)\n\
4164 Get host and port for a sockaddr.");
4167 /* Python API to getting and setting the default timeout value. */
4169 static PyObject *
4170 socket_getdefaulttimeout(PyObject *self)
4172 if (defaulttimeout < 0.0) {
4173 Py_INCREF(Py_None);
4174 return Py_None;
4176 else
4177 return PyFloat_FromDouble(defaulttimeout);
4180 PyDoc_STRVAR(getdefaulttimeout_doc,
4181 "getdefaulttimeout() -> timeout\n\
4183 Returns the default timeout in floating seconds for new socket objects.\n\
4184 A value of None indicates that new socket objects have no timeout.\n\
4185 When the socket module is first imported, the default is None.");
4187 static PyObject *
4188 socket_setdefaulttimeout(PyObject *self, PyObject *arg)
4190 double timeout;
4192 if (arg == Py_None)
4193 timeout = -1.0;
4194 else {
4195 timeout = PyFloat_AsDouble(arg);
4196 if (timeout < 0.0) {
4197 if (!PyErr_Occurred())
4198 PyErr_SetString(PyExc_ValueError,
4199 "Timeout value out of range");
4200 return NULL;
4204 defaulttimeout = timeout;
4206 Py_INCREF(Py_None);
4207 return Py_None;
4210 PyDoc_STRVAR(setdefaulttimeout_doc,
4211 "setdefaulttimeout(timeout)\n\
4213 Set the default timeout in floating seconds for new socket objects.\n\
4214 A value of None indicates that new socket objects have no timeout.\n\
4215 When the socket module is first imported, the default is None.");
4218 /* List of functions exported by this module. */
4220 static PyMethodDef socket_methods[] = {
4221 {"gethostbyname", socket_gethostbyname,
4222 METH_VARARGS, gethostbyname_doc},
4223 {"gethostbyname_ex", socket_gethostbyname_ex,
4224 METH_VARARGS, ghbn_ex_doc},
4225 {"gethostbyaddr", socket_gethostbyaddr,
4226 METH_VARARGS, gethostbyaddr_doc},
4227 {"gethostname", socket_gethostname,
4228 METH_NOARGS, gethostname_doc},
4229 {"getservbyname", socket_getservbyname,
4230 METH_VARARGS, getservbyname_doc},
4231 {"getservbyport", socket_getservbyport,
4232 METH_VARARGS, getservbyport_doc},
4233 {"getprotobyname", socket_getprotobyname,
4234 METH_VARARGS, getprotobyname_doc},
4235 #ifndef NO_DUP
4236 {"fromfd", socket_fromfd,
4237 METH_VARARGS, fromfd_doc},
4238 #endif
4239 #ifdef HAVE_SOCKETPAIR
4240 {"socketpair", socket_socketpair,
4241 METH_VARARGS, socketpair_doc},
4242 #endif
4243 {"ntohs", socket_ntohs,
4244 METH_VARARGS, ntohs_doc},
4245 {"ntohl", socket_ntohl,
4246 METH_O, ntohl_doc},
4247 {"htons", socket_htons,
4248 METH_VARARGS, htons_doc},
4249 {"htonl", socket_htonl,
4250 METH_O, htonl_doc},
4251 {"inet_aton", socket_inet_aton,
4252 METH_VARARGS, inet_aton_doc},
4253 {"inet_ntoa", socket_inet_ntoa,
4254 METH_VARARGS, inet_ntoa_doc},
4255 #ifdef HAVE_INET_PTON
4256 {"inet_pton", socket_inet_pton,
4257 METH_VARARGS, inet_pton_doc},
4258 {"inet_ntop", socket_inet_ntop,
4259 METH_VARARGS, inet_ntop_doc},
4260 #endif
4261 {"getaddrinfo", socket_getaddrinfo,
4262 METH_VARARGS, getaddrinfo_doc},
4263 {"getnameinfo", socket_getnameinfo,
4264 METH_VARARGS, getnameinfo_doc},
4265 {"getdefaulttimeout", (PyCFunction)socket_getdefaulttimeout,
4266 METH_NOARGS, getdefaulttimeout_doc},
4267 {"setdefaulttimeout", socket_setdefaulttimeout,
4268 METH_O, setdefaulttimeout_doc},
4269 {NULL, NULL} /* Sentinel */
4273 #ifdef RISCOS
4274 #define OS_INIT_DEFINED
4276 static int
4277 os_init(void)
4279 _kernel_swi_regs r;
4281 r.r[0] = 0;
4282 _kernel_swi(0x43380, &r, &r);
4283 taskwindow = r.r[0];
4285 return 1;
4288 #endif /* RISCOS */
4291 #ifdef MS_WINDOWS
4292 #define OS_INIT_DEFINED
4294 /* Additional initialization and cleanup for Windows */
4296 static void
4297 os_cleanup(void)
4299 WSACleanup();
4302 static int
4303 os_init(void)
4305 WSADATA WSAData;
4306 int ret;
4307 char buf[100];
4308 ret = WSAStartup(0x0101, &WSAData);
4309 switch (ret) {
4310 case 0: /* No error */
4311 Py_AtExit(os_cleanup);
4312 return 1; /* Success */
4313 case WSASYSNOTREADY:
4314 PyErr_SetString(PyExc_ImportError,
4315 "WSAStartup failed: network not ready");
4316 break;
4317 case WSAVERNOTSUPPORTED:
4318 case WSAEINVAL:
4319 PyErr_SetString(
4320 PyExc_ImportError,
4321 "WSAStartup failed: requested version not supported");
4322 break;
4323 default:
4324 PyOS_snprintf(buf, sizeof(buf),
4325 "WSAStartup failed: error code %d", ret);
4326 PyErr_SetString(PyExc_ImportError, buf);
4327 break;
4329 return 0; /* Failure */
4332 #endif /* MS_WINDOWS */
4335 #ifdef PYOS_OS2
4336 #define OS_INIT_DEFINED
4338 /* Additional initialization for OS/2 */
4340 static int
4341 os_init(void)
4343 #ifndef PYCC_GCC
4344 char reason[64];
4345 int rc = sock_init();
4347 if (rc == 0) {
4348 return 1; /* Success */
4351 PyOS_snprintf(reason, sizeof(reason),
4352 "OS/2 TCP/IP Error# %d", sock_errno());
4353 PyErr_SetString(PyExc_ImportError, reason);
4355 return 0; /* Failure */
4356 #else
4357 /* No need to initialise sockets with GCC/EMX */
4358 return 1; /* Success */
4359 #endif
4362 #endif /* PYOS_OS2 */
4365 #ifndef OS_INIT_DEFINED
4366 static int
4367 os_init(void)
4369 return 1; /* Success */
4371 #endif
4374 /* C API table - always add new things to the end for binary
4375 compatibility. */
4376 static
4377 PySocketModule_APIObject PySocketModuleAPI =
4379 &sock_type,
4380 NULL
4384 /* Initialize the _socket module.
4386 This module is actually called "_socket", and there's a wrapper
4387 "socket.py" which implements some additional functionality. On some
4388 platforms (e.g. Windows and OS/2), socket.py also implements a
4389 wrapper for the socket type that provides missing functionality such
4390 as makefile(), dup() and fromfd(). The import of "_socket" may fail
4391 with an ImportError exception if os-specific initialization fails.
4392 On Windows, this does WINSOCK initialization. When WINSOCK is
4393 initialized succesfully, a call to WSACleanup() is scheduled to be
4394 made at exit time.
4397 PyDoc_STRVAR(socket_doc,
4398 "Implementation module for socket operations.\n\
4400 See the socket module for documentation.");
4402 PyMODINIT_FUNC
4403 init_socket(void)
4405 PyObject *m, *has_ipv6;
4407 if (!os_init())
4408 return;
4410 Py_TYPE(&sock_type) = &PyType_Type;
4411 m = Py_InitModule3(PySocket_MODULE_NAME,
4412 socket_methods,
4413 socket_doc);
4414 if (m == NULL)
4415 return;
4417 socket_error = PyErr_NewException("socket.error",
4418 PyExc_IOError, NULL);
4419 if (socket_error == NULL)
4420 return;
4421 PySocketModuleAPI.error = socket_error;
4422 Py_INCREF(socket_error);
4423 PyModule_AddObject(m, "error", socket_error);
4424 socket_herror = PyErr_NewException("socket.herror",
4425 socket_error, NULL);
4426 if (socket_herror == NULL)
4427 return;
4428 Py_INCREF(socket_herror);
4429 PyModule_AddObject(m, "herror", socket_herror);
4430 socket_gaierror = PyErr_NewException("socket.gaierror", socket_error,
4431 NULL);
4432 if (socket_gaierror == NULL)
4433 return;
4434 Py_INCREF(socket_gaierror);
4435 PyModule_AddObject(m, "gaierror", socket_gaierror);
4436 socket_timeout = PyErr_NewException("socket.timeout",
4437 socket_error, NULL);
4438 if (socket_timeout == NULL)
4439 return;
4440 Py_INCREF(socket_timeout);
4441 PyModule_AddObject(m, "timeout", socket_timeout);
4442 Py_INCREF((PyObject *)&sock_type);
4443 if (PyModule_AddObject(m, "SocketType",
4444 (PyObject *)&sock_type) != 0)
4445 return;
4446 Py_INCREF((PyObject *)&sock_type);
4447 if (PyModule_AddObject(m, "socket",
4448 (PyObject *)&sock_type) != 0)
4449 return;
4451 #ifdef ENABLE_IPV6
4452 has_ipv6 = Py_True;
4453 #else
4454 has_ipv6 = Py_False;
4455 #endif
4456 Py_INCREF(has_ipv6);
4457 PyModule_AddObject(m, "has_ipv6", has_ipv6);
4459 /* Export C API */
4460 if (PyModule_AddObject(m, PySocket_CAPI_NAME,
4461 PyCObject_FromVoidPtr((void *)&PySocketModuleAPI, NULL)
4462 ) != 0)
4463 return;
4465 /* Address families (we only support AF_INET and AF_UNIX) */
4466 #ifdef AF_UNSPEC
4467 PyModule_AddIntConstant(m, "AF_UNSPEC", AF_UNSPEC);
4468 #endif
4469 PyModule_AddIntConstant(m, "AF_INET", AF_INET);
4470 #ifdef AF_INET6
4471 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6);
4472 #endif /* AF_INET6 */
4473 #if defined(AF_UNIX)
4474 PyModule_AddIntConstant(m, "AF_UNIX", AF_UNIX);
4475 #endif /* AF_UNIX */
4476 #ifdef AF_AX25
4477 /* Amateur Radio AX.25 */
4478 PyModule_AddIntConstant(m, "AF_AX25", AF_AX25);
4479 #endif
4480 #ifdef AF_IPX
4481 PyModule_AddIntConstant(m, "AF_IPX", AF_IPX); /* Novell IPX */
4482 #endif
4483 #ifdef AF_APPLETALK
4484 /* Appletalk DDP */
4485 PyModule_AddIntConstant(m, "AF_APPLETALK", AF_APPLETALK);
4486 #endif
4487 #ifdef AF_NETROM
4488 /* Amateur radio NetROM */
4489 PyModule_AddIntConstant(m, "AF_NETROM", AF_NETROM);
4490 #endif
4491 #ifdef AF_BRIDGE
4492 /* Multiprotocol bridge */
4493 PyModule_AddIntConstant(m, "AF_BRIDGE", AF_BRIDGE);
4494 #endif
4495 #ifdef AF_ATMPVC
4496 /* ATM PVCs */
4497 PyModule_AddIntConstant(m, "AF_ATMPVC", AF_ATMPVC);
4498 #endif
4499 #ifdef AF_AAL5
4500 /* Reserved for Werner's ATM */
4501 PyModule_AddIntConstant(m, "AF_AAL5", AF_AAL5);
4502 #endif
4503 #ifdef AF_X25
4504 /* Reserved for X.25 project */
4505 PyModule_AddIntConstant(m, "AF_X25", AF_X25);
4506 #endif
4507 #ifdef AF_INET6
4508 PyModule_AddIntConstant(m, "AF_INET6", AF_INET6); /* IP version 6 */
4509 #endif
4510 #ifdef AF_ROSE
4511 /* Amateur Radio X.25 PLP */
4512 PyModule_AddIntConstant(m, "AF_ROSE", AF_ROSE);
4513 #endif
4514 #ifdef AF_DECnet
4515 /* Reserved for DECnet project */
4516 PyModule_AddIntConstant(m, "AF_DECnet", AF_DECnet);
4517 #endif
4518 #ifdef AF_NETBEUI
4519 /* Reserved for 802.2LLC project */
4520 PyModule_AddIntConstant(m, "AF_NETBEUI", AF_NETBEUI);
4521 #endif
4522 #ifdef AF_SECURITY
4523 /* Security callback pseudo AF */
4524 PyModule_AddIntConstant(m, "AF_SECURITY", AF_SECURITY);
4525 #endif
4526 #ifdef AF_KEY
4527 /* PF_KEY key management API */
4528 PyModule_AddIntConstant(m, "AF_KEY", AF_KEY);
4529 #endif
4530 #ifdef AF_NETLINK
4531 /* */
4532 PyModule_AddIntConstant(m, "AF_NETLINK", AF_NETLINK);
4533 PyModule_AddIntConstant(m, "NETLINK_ROUTE", NETLINK_ROUTE);
4534 #ifdef NETLINK_SKIP
4535 PyModule_AddIntConstant(m, "NETLINK_SKIP", NETLINK_SKIP);
4536 #endif
4537 #ifdef NETLINK_W1
4538 PyModule_AddIntConstant(m, "NETLINK_W1", NETLINK_W1);
4539 #endif
4540 PyModule_AddIntConstant(m, "NETLINK_USERSOCK", NETLINK_USERSOCK);
4541 PyModule_AddIntConstant(m, "NETLINK_FIREWALL", NETLINK_FIREWALL);
4542 #ifdef NETLINK_TCPDIAG
4543 PyModule_AddIntConstant(m, "NETLINK_TCPDIAG", NETLINK_TCPDIAG);
4544 #endif
4545 #ifdef NETLINK_NFLOG
4546 PyModule_AddIntConstant(m, "NETLINK_NFLOG", NETLINK_NFLOG);
4547 #endif
4548 #ifdef NETLINK_XFRM
4549 PyModule_AddIntConstant(m, "NETLINK_XFRM", NETLINK_XFRM);
4550 #endif
4551 #ifdef NETLINK_ARPD
4552 PyModule_AddIntConstant(m, "NETLINK_ARPD", NETLINK_ARPD);
4553 #endif
4554 #ifdef NETLINK_ROUTE6
4555 PyModule_AddIntConstant(m, "NETLINK_ROUTE6", NETLINK_ROUTE6);
4556 #endif
4557 PyModule_AddIntConstant(m, "NETLINK_IP6_FW", NETLINK_IP6_FW);
4558 #ifdef NETLINK_DNRTMSG
4559 PyModule_AddIntConstant(m, "NETLINK_DNRTMSG", NETLINK_DNRTMSG);
4560 #endif
4561 #ifdef NETLINK_TAPBASE
4562 PyModule_AddIntConstant(m, "NETLINK_TAPBASE", NETLINK_TAPBASE);
4563 #endif
4564 #endif /* AF_NETLINK */
4565 #ifdef AF_ROUTE
4566 /* Alias to emulate 4.4BSD */
4567 PyModule_AddIntConstant(m, "AF_ROUTE", AF_ROUTE);
4568 #endif
4569 #ifdef AF_ASH
4570 /* Ash */
4571 PyModule_AddIntConstant(m, "AF_ASH", AF_ASH);
4572 #endif
4573 #ifdef AF_ECONET
4574 /* Acorn Econet */
4575 PyModule_AddIntConstant(m, "AF_ECONET", AF_ECONET);
4576 #endif
4577 #ifdef AF_ATMSVC
4578 /* ATM SVCs */
4579 PyModule_AddIntConstant(m, "AF_ATMSVC", AF_ATMSVC);
4580 #endif
4581 #ifdef AF_SNA
4582 /* Linux SNA Project (nutters!) */
4583 PyModule_AddIntConstant(m, "AF_SNA", AF_SNA);
4584 #endif
4585 #ifdef AF_IRDA
4586 /* IRDA sockets */
4587 PyModule_AddIntConstant(m, "AF_IRDA", AF_IRDA);
4588 #endif
4589 #ifdef AF_PPPOX
4590 /* PPPoX sockets */
4591 PyModule_AddIntConstant(m, "AF_PPPOX", AF_PPPOX);
4592 #endif
4593 #ifdef AF_WANPIPE
4594 /* Wanpipe API Sockets */
4595 PyModule_AddIntConstant(m, "AF_WANPIPE", AF_WANPIPE);
4596 #endif
4597 #ifdef AF_LLC
4598 /* Linux LLC */
4599 PyModule_AddIntConstant(m, "AF_LLC", AF_LLC);
4600 #endif
4602 #ifdef USE_BLUETOOTH
4603 PyModule_AddIntConstant(m, "AF_BLUETOOTH", AF_BLUETOOTH);
4604 PyModule_AddIntConstant(m, "BTPROTO_L2CAP", BTPROTO_L2CAP);
4605 PyModule_AddIntConstant(m, "BTPROTO_HCI", BTPROTO_HCI);
4606 PyModule_AddIntConstant(m, "SOL_HCI", SOL_HCI);
4607 PyModule_AddIntConstant(m, "HCI_FILTER", HCI_FILTER);
4608 #if !defined(__FreeBSD__)
4609 PyModule_AddIntConstant(m, "HCI_TIME_STAMP", HCI_TIME_STAMP);
4610 PyModule_AddIntConstant(m, "HCI_DATA_DIR", HCI_DATA_DIR);
4611 PyModule_AddIntConstant(m, "BTPROTO_SCO", BTPROTO_SCO);
4612 #endif
4613 PyModule_AddIntConstant(m, "BTPROTO_RFCOMM", BTPROTO_RFCOMM);
4614 PyModule_AddStringConstant(m, "BDADDR_ANY", "00:00:00:00:00:00");
4615 PyModule_AddStringConstant(m, "BDADDR_LOCAL", "00:00:00:FF:FF:FF");
4616 #endif
4618 #ifdef HAVE_NETPACKET_PACKET_H
4619 PyModule_AddIntConstant(m, "AF_PACKET", AF_PACKET);
4620 PyModule_AddIntConstant(m, "PF_PACKET", PF_PACKET);
4621 PyModule_AddIntConstant(m, "PACKET_HOST", PACKET_HOST);
4622 PyModule_AddIntConstant(m, "PACKET_BROADCAST", PACKET_BROADCAST);
4623 PyModule_AddIntConstant(m, "PACKET_MULTICAST", PACKET_MULTICAST);
4624 PyModule_AddIntConstant(m, "PACKET_OTHERHOST", PACKET_OTHERHOST);
4625 PyModule_AddIntConstant(m, "PACKET_OUTGOING", PACKET_OUTGOING);
4626 PyModule_AddIntConstant(m, "PACKET_LOOPBACK", PACKET_LOOPBACK);
4627 PyModule_AddIntConstant(m, "PACKET_FASTROUTE", PACKET_FASTROUTE);
4628 #endif
4630 #ifdef HAVE_LINUX_TIPC_H
4631 PyModule_AddIntConstant(m, "AF_TIPC", AF_TIPC);
4633 /* for addresses */
4634 PyModule_AddIntConstant(m, "TIPC_ADDR_NAMESEQ", TIPC_ADDR_NAMESEQ);
4635 PyModule_AddIntConstant(m, "TIPC_ADDR_NAME", TIPC_ADDR_NAME);
4636 PyModule_AddIntConstant(m, "TIPC_ADDR_ID", TIPC_ADDR_ID);
4638 PyModule_AddIntConstant(m, "TIPC_ZONE_SCOPE", TIPC_ZONE_SCOPE);
4639 PyModule_AddIntConstant(m, "TIPC_CLUSTER_SCOPE", TIPC_CLUSTER_SCOPE);
4640 PyModule_AddIntConstant(m, "TIPC_NODE_SCOPE", TIPC_NODE_SCOPE);
4642 /* for setsockopt() */
4643 PyModule_AddIntConstant(m, "SOL_TIPC", SOL_TIPC);
4644 PyModule_AddIntConstant(m, "TIPC_IMPORTANCE", TIPC_IMPORTANCE);
4645 PyModule_AddIntConstant(m, "TIPC_SRC_DROPPABLE", TIPC_SRC_DROPPABLE);
4646 PyModule_AddIntConstant(m, "TIPC_DEST_DROPPABLE",
4647 TIPC_DEST_DROPPABLE);
4648 PyModule_AddIntConstant(m, "TIPC_CONN_TIMEOUT", TIPC_CONN_TIMEOUT);
4650 PyModule_AddIntConstant(m, "TIPC_LOW_IMPORTANCE",
4651 TIPC_LOW_IMPORTANCE);
4652 PyModule_AddIntConstant(m, "TIPC_MEDIUM_IMPORTANCE",
4653 TIPC_MEDIUM_IMPORTANCE);
4654 PyModule_AddIntConstant(m, "TIPC_HIGH_IMPORTANCE",
4655 TIPC_HIGH_IMPORTANCE);
4656 PyModule_AddIntConstant(m, "TIPC_CRITICAL_IMPORTANCE",
4657 TIPC_CRITICAL_IMPORTANCE);
4659 /* for subscriptions */
4660 PyModule_AddIntConstant(m, "TIPC_SUB_PORTS", TIPC_SUB_PORTS);
4661 PyModule_AddIntConstant(m, "TIPC_SUB_SERVICE", TIPC_SUB_SERVICE);
4662 #ifdef TIPC_SUB_CANCEL
4663 /* doesn't seem to be available everywhere */
4664 PyModule_AddIntConstant(m, "TIPC_SUB_CANCEL", TIPC_SUB_CANCEL);
4665 #endif
4666 PyModule_AddIntConstant(m, "TIPC_WAIT_FOREVER", TIPC_WAIT_FOREVER);
4667 PyModule_AddIntConstant(m, "TIPC_PUBLISHED", TIPC_PUBLISHED);
4668 PyModule_AddIntConstant(m, "TIPC_WITHDRAWN", TIPC_WITHDRAWN);
4669 PyModule_AddIntConstant(m, "TIPC_SUBSCR_TIMEOUT", TIPC_SUBSCR_TIMEOUT);
4670 PyModule_AddIntConstant(m, "TIPC_CFG_SRV", TIPC_CFG_SRV);
4671 PyModule_AddIntConstant(m, "TIPC_TOP_SRV", TIPC_TOP_SRV);
4672 #endif
4674 /* Socket types */
4675 PyModule_AddIntConstant(m, "SOCK_STREAM", SOCK_STREAM);
4676 PyModule_AddIntConstant(m, "SOCK_DGRAM", SOCK_DGRAM);
4677 #ifndef __BEOS__
4678 /* We have incomplete socket support. */
4679 PyModule_AddIntConstant(m, "SOCK_RAW", SOCK_RAW);
4680 PyModule_AddIntConstant(m, "SOCK_SEQPACKET", SOCK_SEQPACKET);
4681 #if defined(SOCK_RDM)
4682 PyModule_AddIntConstant(m, "SOCK_RDM", SOCK_RDM);
4683 #endif
4684 #endif
4686 #ifdef SO_DEBUG
4687 PyModule_AddIntConstant(m, "SO_DEBUG", SO_DEBUG);
4688 #endif
4689 #ifdef SO_ACCEPTCONN
4690 PyModule_AddIntConstant(m, "SO_ACCEPTCONN", SO_ACCEPTCONN);
4691 #endif
4692 #ifdef SO_REUSEADDR
4693 PyModule_AddIntConstant(m, "SO_REUSEADDR", SO_REUSEADDR);
4694 #endif
4695 #ifdef SO_EXCLUSIVEADDRUSE
4696 PyModule_AddIntConstant(m, "SO_EXCLUSIVEADDRUSE", SO_EXCLUSIVEADDRUSE);
4697 #endif
4699 #ifdef SO_KEEPALIVE
4700 PyModule_AddIntConstant(m, "SO_KEEPALIVE", SO_KEEPALIVE);
4701 #endif
4702 #ifdef SO_DONTROUTE
4703 PyModule_AddIntConstant(m, "SO_DONTROUTE", SO_DONTROUTE);
4704 #endif
4705 #ifdef SO_BROADCAST
4706 PyModule_AddIntConstant(m, "SO_BROADCAST", SO_BROADCAST);
4707 #endif
4708 #ifdef SO_USELOOPBACK
4709 PyModule_AddIntConstant(m, "SO_USELOOPBACK", SO_USELOOPBACK);
4710 #endif
4711 #ifdef SO_LINGER
4712 PyModule_AddIntConstant(m, "SO_LINGER", SO_LINGER);
4713 #endif
4714 #ifdef SO_OOBINLINE
4715 PyModule_AddIntConstant(m, "SO_OOBINLINE", SO_OOBINLINE);
4716 #endif
4717 #ifdef SO_REUSEPORT
4718 PyModule_AddIntConstant(m, "SO_REUSEPORT", SO_REUSEPORT);
4719 #endif
4720 #ifdef SO_SNDBUF
4721 PyModule_AddIntConstant(m, "SO_SNDBUF", SO_SNDBUF);
4722 #endif
4723 #ifdef SO_RCVBUF
4724 PyModule_AddIntConstant(m, "SO_RCVBUF", SO_RCVBUF);
4725 #endif
4726 #ifdef SO_SNDLOWAT
4727 PyModule_AddIntConstant(m, "SO_SNDLOWAT", SO_SNDLOWAT);
4728 #endif
4729 #ifdef SO_RCVLOWAT
4730 PyModule_AddIntConstant(m, "SO_RCVLOWAT", SO_RCVLOWAT);
4731 #endif
4732 #ifdef SO_SNDTIMEO
4733 PyModule_AddIntConstant(m, "SO_SNDTIMEO", SO_SNDTIMEO);
4734 #endif
4735 #ifdef SO_RCVTIMEO
4736 PyModule_AddIntConstant(m, "SO_RCVTIMEO", SO_RCVTIMEO);
4737 #endif
4738 #ifdef SO_ERROR
4739 PyModule_AddIntConstant(m, "SO_ERROR", SO_ERROR);
4740 #endif
4741 #ifdef SO_TYPE
4742 PyModule_AddIntConstant(m, "SO_TYPE", SO_TYPE);
4743 #endif
4745 /* Maximum number of connections for "listen" */
4746 #ifdef SOMAXCONN
4747 PyModule_AddIntConstant(m, "SOMAXCONN", SOMAXCONN);
4748 #else
4749 PyModule_AddIntConstant(m, "SOMAXCONN", 5); /* Common value */
4750 #endif
4752 /* Flags for send, recv */
4753 #ifdef MSG_OOB
4754 PyModule_AddIntConstant(m, "MSG_OOB", MSG_OOB);
4755 #endif
4756 #ifdef MSG_PEEK
4757 PyModule_AddIntConstant(m, "MSG_PEEK", MSG_PEEK);
4758 #endif
4759 #ifdef MSG_DONTROUTE
4760 PyModule_AddIntConstant(m, "MSG_DONTROUTE", MSG_DONTROUTE);
4761 #endif
4762 #ifdef MSG_DONTWAIT
4763 PyModule_AddIntConstant(m, "MSG_DONTWAIT", MSG_DONTWAIT);
4764 #endif
4765 #ifdef MSG_EOR
4766 PyModule_AddIntConstant(m, "MSG_EOR", MSG_EOR);
4767 #endif
4768 #ifdef MSG_TRUNC
4769 PyModule_AddIntConstant(m, "MSG_TRUNC", MSG_TRUNC);
4770 #endif
4771 #ifdef MSG_CTRUNC
4772 PyModule_AddIntConstant(m, "MSG_CTRUNC", MSG_CTRUNC);
4773 #endif
4774 #ifdef MSG_WAITALL
4775 PyModule_AddIntConstant(m, "MSG_WAITALL", MSG_WAITALL);
4776 #endif
4777 #ifdef MSG_BTAG
4778 PyModule_AddIntConstant(m, "MSG_BTAG", MSG_BTAG);
4779 #endif
4780 #ifdef MSG_ETAG
4781 PyModule_AddIntConstant(m, "MSG_ETAG", MSG_ETAG);
4782 #endif
4784 /* Protocol level and numbers, usable for [gs]etsockopt */
4785 #ifdef SOL_SOCKET
4786 PyModule_AddIntConstant(m, "SOL_SOCKET", SOL_SOCKET);
4787 #endif
4788 #ifdef SOL_IP
4789 PyModule_AddIntConstant(m, "SOL_IP", SOL_IP);
4790 #else
4791 PyModule_AddIntConstant(m, "SOL_IP", 0);
4792 #endif
4793 #ifdef SOL_IPX
4794 PyModule_AddIntConstant(m, "SOL_IPX", SOL_IPX);
4795 #endif
4796 #ifdef SOL_AX25
4797 PyModule_AddIntConstant(m, "SOL_AX25", SOL_AX25);
4798 #endif
4799 #ifdef SOL_ATALK
4800 PyModule_AddIntConstant(m, "SOL_ATALK", SOL_ATALK);
4801 #endif
4802 #ifdef SOL_NETROM
4803 PyModule_AddIntConstant(m, "SOL_NETROM", SOL_NETROM);
4804 #endif
4805 #ifdef SOL_ROSE
4806 PyModule_AddIntConstant(m, "SOL_ROSE", SOL_ROSE);
4807 #endif
4808 #ifdef SOL_TCP
4809 PyModule_AddIntConstant(m, "SOL_TCP", SOL_TCP);
4810 #else
4811 PyModule_AddIntConstant(m, "SOL_TCP", 6);
4812 #endif
4813 #ifdef SOL_UDP
4814 PyModule_AddIntConstant(m, "SOL_UDP", SOL_UDP);
4815 #else
4816 PyModule_AddIntConstant(m, "SOL_UDP", 17);
4817 #endif
4818 #ifdef IPPROTO_IP
4819 PyModule_AddIntConstant(m, "IPPROTO_IP", IPPROTO_IP);
4820 #else
4821 PyModule_AddIntConstant(m, "IPPROTO_IP", 0);
4822 #endif
4823 #ifdef IPPROTO_HOPOPTS
4824 PyModule_AddIntConstant(m, "IPPROTO_HOPOPTS", IPPROTO_HOPOPTS);
4825 #endif
4826 #ifdef IPPROTO_ICMP
4827 PyModule_AddIntConstant(m, "IPPROTO_ICMP", IPPROTO_ICMP);
4828 #else
4829 PyModule_AddIntConstant(m, "IPPROTO_ICMP", 1);
4830 #endif
4831 #ifdef IPPROTO_IGMP
4832 PyModule_AddIntConstant(m, "IPPROTO_IGMP", IPPROTO_IGMP);
4833 #endif
4834 #ifdef IPPROTO_GGP
4835 PyModule_AddIntConstant(m, "IPPROTO_GGP", IPPROTO_GGP);
4836 #endif
4837 #ifdef IPPROTO_IPV4
4838 PyModule_AddIntConstant(m, "IPPROTO_IPV4", IPPROTO_IPV4);
4839 #endif
4840 #ifdef IPPROTO_IPV6
4841 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
4842 #endif
4843 #ifdef IPPROTO_IPIP
4844 PyModule_AddIntConstant(m, "IPPROTO_IPIP", IPPROTO_IPIP);
4845 #endif
4846 #ifdef IPPROTO_TCP
4847 PyModule_AddIntConstant(m, "IPPROTO_TCP", IPPROTO_TCP);
4848 #else
4849 PyModule_AddIntConstant(m, "IPPROTO_TCP", 6);
4850 #endif
4851 #ifdef IPPROTO_EGP
4852 PyModule_AddIntConstant(m, "IPPROTO_EGP", IPPROTO_EGP);
4853 #endif
4854 #ifdef IPPROTO_PUP
4855 PyModule_AddIntConstant(m, "IPPROTO_PUP", IPPROTO_PUP);
4856 #endif
4857 #ifdef IPPROTO_UDP
4858 PyModule_AddIntConstant(m, "IPPROTO_UDP", IPPROTO_UDP);
4859 #else
4860 PyModule_AddIntConstant(m, "IPPROTO_UDP", 17);
4861 #endif
4862 #ifdef IPPROTO_IDP
4863 PyModule_AddIntConstant(m, "IPPROTO_IDP", IPPROTO_IDP);
4864 #endif
4865 #ifdef IPPROTO_HELLO
4866 PyModule_AddIntConstant(m, "IPPROTO_HELLO", IPPROTO_HELLO);
4867 #endif
4868 #ifdef IPPROTO_ND
4869 PyModule_AddIntConstant(m, "IPPROTO_ND", IPPROTO_ND);
4870 #endif
4871 #ifdef IPPROTO_TP
4872 PyModule_AddIntConstant(m, "IPPROTO_TP", IPPROTO_TP);
4873 #endif
4874 #ifdef IPPROTO_IPV6
4875 PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
4876 #endif
4877 #ifdef IPPROTO_ROUTING
4878 PyModule_AddIntConstant(m, "IPPROTO_ROUTING", IPPROTO_ROUTING);
4879 #endif
4880 #ifdef IPPROTO_FRAGMENT
4881 PyModule_AddIntConstant(m, "IPPROTO_FRAGMENT", IPPROTO_FRAGMENT);
4882 #endif
4883 #ifdef IPPROTO_RSVP
4884 PyModule_AddIntConstant(m, "IPPROTO_RSVP", IPPROTO_RSVP);
4885 #endif
4886 #ifdef IPPROTO_GRE
4887 PyModule_AddIntConstant(m, "IPPROTO_GRE", IPPROTO_GRE);
4888 #endif
4889 #ifdef IPPROTO_ESP
4890 PyModule_AddIntConstant(m, "IPPROTO_ESP", IPPROTO_ESP);
4891 #endif
4892 #ifdef IPPROTO_AH
4893 PyModule_AddIntConstant(m, "IPPROTO_AH", IPPROTO_AH);
4894 #endif
4895 #ifdef IPPROTO_MOBILE
4896 PyModule_AddIntConstant(m, "IPPROTO_MOBILE", IPPROTO_MOBILE);
4897 #endif
4898 #ifdef IPPROTO_ICMPV6
4899 PyModule_AddIntConstant(m, "IPPROTO_ICMPV6", IPPROTO_ICMPV6);
4900 #endif
4901 #ifdef IPPROTO_NONE
4902 PyModule_AddIntConstant(m, "IPPROTO_NONE", IPPROTO_NONE);
4903 #endif
4904 #ifdef IPPROTO_DSTOPTS
4905 PyModule_AddIntConstant(m, "IPPROTO_DSTOPTS", IPPROTO_DSTOPTS);
4906 #endif
4907 #ifdef IPPROTO_XTP
4908 PyModule_AddIntConstant(m, "IPPROTO_XTP", IPPROTO_XTP);
4909 #endif
4910 #ifdef IPPROTO_EON
4911 PyModule_AddIntConstant(m, "IPPROTO_EON", IPPROTO_EON);
4912 #endif
4913 #ifdef IPPROTO_PIM
4914 PyModule_AddIntConstant(m, "IPPROTO_PIM", IPPROTO_PIM);
4915 #endif
4916 #ifdef IPPROTO_IPCOMP
4917 PyModule_AddIntConstant(m, "IPPROTO_IPCOMP", IPPROTO_IPCOMP);
4918 #endif
4919 #ifdef IPPROTO_VRRP
4920 PyModule_AddIntConstant(m, "IPPROTO_VRRP", IPPROTO_VRRP);
4921 #endif
4922 #ifdef IPPROTO_BIP
4923 PyModule_AddIntConstant(m, "IPPROTO_BIP", IPPROTO_BIP);
4924 #endif
4925 /**/
4926 #ifdef IPPROTO_RAW
4927 PyModule_AddIntConstant(m, "IPPROTO_RAW", IPPROTO_RAW);
4928 #else
4929 PyModule_AddIntConstant(m, "IPPROTO_RAW", 255);
4930 #endif
4931 #ifdef IPPROTO_MAX
4932 PyModule_AddIntConstant(m, "IPPROTO_MAX", IPPROTO_MAX);
4933 #endif
4935 /* Some port configuration */
4936 #ifdef IPPORT_RESERVED
4937 PyModule_AddIntConstant(m, "IPPORT_RESERVED", IPPORT_RESERVED);
4938 #else
4939 PyModule_AddIntConstant(m, "IPPORT_RESERVED", 1024);
4940 #endif
4941 #ifdef IPPORT_USERRESERVED
4942 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", IPPORT_USERRESERVED);
4943 #else
4944 PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", 5000);
4945 #endif
4947 /* Some reserved IP v.4 addresses */
4948 #ifdef INADDR_ANY
4949 PyModule_AddIntConstant(m, "INADDR_ANY", INADDR_ANY);
4950 #else
4951 PyModule_AddIntConstant(m, "INADDR_ANY", 0x00000000);
4952 #endif
4953 #ifdef INADDR_BROADCAST
4954 PyModule_AddIntConstant(m, "INADDR_BROADCAST", INADDR_BROADCAST);
4955 #else
4956 PyModule_AddIntConstant(m, "INADDR_BROADCAST", 0xffffffff);
4957 #endif
4958 #ifdef INADDR_LOOPBACK
4959 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", INADDR_LOOPBACK);
4960 #else
4961 PyModule_AddIntConstant(m, "INADDR_LOOPBACK", 0x7F000001);
4962 #endif
4963 #ifdef INADDR_UNSPEC_GROUP
4964 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP);
4965 #else
4966 PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", 0xe0000000);
4967 #endif
4968 #ifdef INADDR_ALLHOSTS_GROUP
4969 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP",
4970 INADDR_ALLHOSTS_GROUP);
4971 #else
4972 PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP", 0xe0000001);
4973 #endif
4974 #ifdef INADDR_MAX_LOCAL_GROUP
4975 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP",
4976 INADDR_MAX_LOCAL_GROUP);
4977 #else
4978 PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff);
4979 #endif
4980 #ifdef INADDR_NONE
4981 PyModule_AddIntConstant(m, "INADDR_NONE", INADDR_NONE);
4982 #else
4983 PyModule_AddIntConstant(m, "INADDR_NONE", 0xffffffff);
4984 #endif
4986 /* IPv4 [gs]etsockopt options */
4987 #ifdef IP_OPTIONS
4988 PyModule_AddIntConstant(m, "IP_OPTIONS", IP_OPTIONS);
4989 #endif
4990 #ifdef IP_HDRINCL
4991 PyModule_AddIntConstant(m, "IP_HDRINCL", IP_HDRINCL);
4992 #endif
4993 #ifdef IP_TOS
4994 PyModule_AddIntConstant(m, "IP_TOS", IP_TOS);
4995 #endif
4996 #ifdef IP_TTL
4997 PyModule_AddIntConstant(m, "IP_TTL", IP_TTL);
4998 #endif
4999 #ifdef IP_RECVOPTS
5000 PyModule_AddIntConstant(m, "IP_RECVOPTS", IP_RECVOPTS);
5001 #endif
5002 #ifdef IP_RECVRETOPTS
5003 PyModule_AddIntConstant(m, "IP_RECVRETOPTS", IP_RECVRETOPTS);
5004 #endif
5005 #ifdef IP_RECVDSTADDR
5006 PyModule_AddIntConstant(m, "IP_RECVDSTADDR", IP_RECVDSTADDR);
5007 #endif
5008 #ifdef IP_RETOPTS
5009 PyModule_AddIntConstant(m, "IP_RETOPTS", IP_RETOPTS);
5010 #endif
5011 #ifdef IP_MULTICAST_IF
5012 PyModule_AddIntConstant(m, "IP_MULTICAST_IF", IP_MULTICAST_IF);
5013 #endif
5014 #ifdef IP_MULTICAST_TTL
5015 PyModule_AddIntConstant(m, "IP_MULTICAST_TTL", IP_MULTICAST_TTL);
5016 #endif
5017 #ifdef IP_MULTICAST_LOOP
5018 PyModule_AddIntConstant(m, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP);
5019 #endif
5020 #ifdef IP_ADD_MEMBERSHIP
5021 PyModule_AddIntConstant(m, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP);
5022 #endif
5023 #ifdef IP_DROP_MEMBERSHIP
5024 PyModule_AddIntConstant(m, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP);
5025 #endif
5026 #ifdef IP_DEFAULT_MULTICAST_TTL
5027 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_TTL",
5028 IP_DEFAULT_MULTICAST_TTL);
5029 #endif
5030 #ifdef IP_DEFAULT_MULTICAST_LOOP
5031 PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_LOOP",
5032 IP_DEFAULT_MULTICAST_LOOP);
5033 #endif
5034 #ifdef IP_MAX_MEMBERSHIPS
5035 PyModule_AddIntConstant(m, "IP_MAX_MEMBERSHIPS", IP_MAX_MEMBERSHIPS);
5036 #endif
5038 /* IPv6 [gs]etsockopt options, defined in RFC2553 */
5039 #ifdef IPV6_JOIN_GROUP
5040 PyModule_AddIntConstant(m, "IPV6_JOIN_GROUP", IPV6_JOIN_GROUP);
5041 #endif
5042 #ifdef IPV6_LEAVE_GROUP
5043 PyModule_AddIntConstant(m, "IPV6_LEAVE_GROUP", IPV6_LEAVE_GROUP);
5044 #endif
5045 #ifdef IPV6_MULTICAST_HOPS
5046 PyModule_AddIntConstant(m, "IPV6_MULTICAST_HOPS", IPV6_MULTICAST_HOPS);
5047 #endif
5048 #ifdef IPV6_MULTICAST_IF
5049 PyModule_AddIntConstant(m, "IPV6_MULTICAST_IF", IPV6_MULTICAST_IF);
5050 #endif
5051 #ifdef IPV6_MULTICAST_LOOP
5052 PyModule_AddIntConstant(m, "IPV6_MULTICAST_LOOP", IPV6_MULTICAST_LOOP);
5053 #endif
5054 #ifdef IPV6_UNICAST_HOPS
5055 PyModule_AddIntConstant(m, "IPV6_UNICAST_HOPS", IPV6_UNICAST_HOPS);
5056 #endif
5057 /* Additional IPV6 socket options, defined in RFC 3493 */
5058 #ifdef IPV6_V6ONLY
5059 PyModule_AddIntConstant(m, "IPV6_V6ONLY", IPV6_V6ONLY);
5060 #endif
5061 /* Advanced IPV6 socket options, from RFC 3542 */
5062 #ifdef IPV6_CHECKSUM
5063 PyModule_AddIntConstant(m, "IPV6_CHECKSUM", IPV6_CHECKSUM);
5064 #endif
5065 #ifdef IPV6_DONTFRAG
5066 PyModule_AddIntConstant(m, "IPV6_DONTFRAG", IPV6_DONTFRAG);
5067 #endif
5068 #ifdef IPV6_DSTOPTS
5069 PyModule_AddIntConstant(m, "IPV6_DSTOPTS", IPV6_DSTOPTS);
5070 #endif
5071 #ifdef IPV6_HOPLIMIT
5072 PyModule_AddIntConstant(m, "IPV6_HOPLIMIT", IPV6_HOPLIMIT);
5073 #endif
5074 #ifdef IPV6_HOPOPTS
5075 PyModule_AddIntConstant(m, "IPV6_HOPOPTS", IPV6_HOPOPTS);
5076 #endif
5077 #ifdef IPV6_NEXTHOP
5078 PyModule_AddIntConstant(m, "IPV6_NEXTHOP", IPV6_NEXTHOP);
5079 #endif
5080 #ifdef IPV6_PATHMTU
5081 PyModule_AddIntConstant(m, "IPV6_PATHMTU", IPV6_PATHMTU);
5082 #endif
5083 #ifdef IPV6_PKTINFO
5084 PyModule_AddIntConstant(m, "IPV6_PKTINFO", IPV6_PKTINFO);
5085 #endif
5086 #ifdef IPV6_RECVDSTOPTS
5087 PyModule_AddIntConstant(m, "IPV6_RECVDSTOPTS", IPV6_RECVDSTOPTS);
5088 #endif
5089 #ifdef IPV6_RECVHOPLIMIT
5090 PyModule_AddIntConstant(m, "IPV6_RECVHOPLIMIT", IPV6_RECVHOPLIMIT);
5091 #endif
5092 #ifdef IPV6_RECVHOPOPTS
5093 PyModule_AddIntConstant(m, "IPV6_RECVHOPOPTS", IPV6_RECVHOPOPTS);
5094 #endif
5095 #ifdef IPV6_RECVPKTINFO
5096 PyModule_AddIntConstant(m, "IPV6_RECVPKTINFO", IPV6_RECVPKTINFO);
5097 #endif
5098 #ifdef IPV6_RECVRTHDR
5099 PyModule_AddIntConstant(m, "IPV6_RECVRTHDR", IPV6_RECVRTHDR);
5100 #endif
5101 #ifdef IPV6_RECVTCLASS
5102 PyModule_AddIntConstant(m, "IPV6_RECVTCLASS", IPV6_RECVTCLASS);
5103 #endif
5104 #ifdef IPV6_RTHDR
5105 PyModule_AddIntConstant(m, "IPV6_RTHDR", IPV6_RTHDR);
5106 #endif
5107 #ifdef IPV6_RTHDRDSTOPTS
5108 PyModule_AddIntConstant(m, "IPV6_RTHDRDSTOPTS", IPV6_RTHDRDSTOPTS);
5109 #endif
5110 #ifdef IPV6_RTHDR_TYPE_0
5111 PyModule_AddIntConstant(m, "IPV6_RTHDR_TYPE_0", IPV6_RTHDR_TYPE_0);
5112 #endif
5113 #ifdef IPV6_RECVPATHMTU
5114 PyModule_AddIntConstant(m, "IPV6_RECVPATHMTU", IPV6_RECVPATHMTU);
5115 #endif
5116 #ifdef IPV6_TCLASS
5117 PyModule_AddIntConstant(m, "IPV6_TCLASS", IPV6_TCLASS);
5118 #endif
5119 #ifdef IPV6_USE_MIN_MTU
5120 PyModule_AddIntConstant(m, "IPV6_USE_MIN_MTU", IPV6_USE_MIN_MTU);
5121 #endif
5123 /* TCP options */
5124 #ifdef TCP_NODELAY
5125 PyModule_AddIntConstant(m, "TCP_NODELAY", TCP_NODELAY);
5126 #endif
5127 #ifdef TCP_MAXSEG
5128 PyModule_AddIntConstant(m, "TCP_MAXSEG", TCP_MAXSEG);
5129 #endif
5130 #ifdef TCP_CORK
5131 PyModule_AddIntConstant(m, "TCP_CORK", TCP_CORK);
5132 #endif
5133 #ifdef TCP_KEEPIDLE
5134 PyModule_AddIntConstant(m, "TCP_KEEPIDLE", TCP_KEEPIDLE);
5135 #endif
5136 #ifdef TCP_KEEPINTVL
5137 PyModule_AddIntConstant(m, "TCP_KEEPINTVL", TCP_KEEPINTVL);
5138 #endif
5139 #ifdef TCP_KEEPCNT
5140 PyModule_AddIntConstant(m, "TCP_KEEPCNT", TCP_KEEPCNT);
5141 #endif
5142 #ifdef TCP_SYNCNT
5143 PyModule_AddIntConstant(m, "TCP_SYNCNT", TCP_SYNCNT);
5144 #endif
5145 #ifdef TCP_LINGER2
5146 PyModule_AddIntConstant(m, "TCP_LINGER2", TCP_LINGER2);
5147 #endif
5148 #ifdef TCP_DEFER_ACCEPT
5149 PyModule_AddIntConstant(m, "TCP_DEFER_ACCEPT", TCP_DEFER_ACCEPT);
5150 #endif
5151 #ifdef TCP_WINDOW_CLAMP
5152 PyModule_AddIntConstant(m, "TCP_WINDOW_CLAMP", TCP_WINDOW_CLAMP);
5153 #endif
5154 #ifdef TCP_INFO
5155 PyModule_AddIntConstant(m, "TCP_INFO", TCP_INFO);
5156 #endif
5157 #ifdef TCP_QUICKACK
5158 PyModule_AddIntConstant(m, "TCP_QUICKACK", TCP_QUICKACK);
5159 #endif
5162 /* IPX options */
5163 #ifdef IPX_TYPE
5164 PyModule_AddIntConstant(m, "IPX_TYPE", IPX_TYPE);
5165 #endif
5167 /* get{addr,name}info parameters */
5168 #ifdef EAI_ADDRFAMILY
5169 PyModule_AddIntConstant(m, "EAI_ADDRFAMILY", EAI_ADDRFAMILY);
5170 #endif
5171 #ifdef EAI_AGAIN
5172 PyModule_AddIntConstant(m, "EAI_AGAIN", EAI_AGAIN);
5173 #endif
5174 #ifdef EAI_BADFLAGS
5175 PyModule_AddIntConstant(m, "EAI_BADFLAGS", EAI_BADFLAGS);
5176 #endif
5177 #ifdef EAI_FAIL
5178 PyModule_AddIntConstant(m, "EAI_FAIL", EAI_FAIL);
5179 #endif
5180 #ifdef EAI_FAMILY
5181 PyModule_AddIntConstant(m, "EAI_FAMILY", EAI_FAMILY);
5182 #endif
5183 #ifdef EAI_MEMORY
5184 PyModule_AddIntConstant(m, "EAI_MEMORY", EAI_MEMORY);
5185 #endif
5186 #ifdef EAI_NODATA
5187 PyModule_AddIntConstant(m, "EAI_NODATA", EAI_NODATA);
5188 #endif
5189 #ifdef EAI_NONAME
5190 PyModule_AddIntConstant(m, "EAI_NONAME", EAI_NONAME);
5191 #endif
5192 #ifdef EAI_OVERFLOW
5193 PyModule_AddIntConstant(m, "EAI_OVERFLOW", EAI_OVERFLOW);
5194 #endif
5195 #ifdef EAI_SERVICE
5196 PyModule_AddIntConstant(m, "EAI_SERVICE", EAI_SERVICE);
5197 #endif
5198 #ifdef EAI_SOCKTYPE
5199 PyModule_AddIntConstant(m, "EAI_SOCKTYPE", EAI_SOCKTYPE);
5200 #endif
5201 #ifdef EAI_SYSTEM
5202 PyModule_AddIntConstant(m, "EAI_SYSTEM", EAI_SYSTEM);
5203 #endif
5204 #ifdef EAI_BADHINTS
5205 PyModule_AddIntConstant(m, "EAI_BADHINTS", EAI_BADHINTS);
5206 #endif
5207 #ifdef EAI_PROTOCOL
5208 PyModule_AddIntConstant(m, "EAI_PROTOCOL", EAI_PROTOCOL);
5209 #endif
5210 #ifdef EAI_MAX
5211 PyModule_AddIntConstant(m, "EAI_MAX", EAI_MAX);
5212 #endif
5213 #ifdef AI_PASSIVE
5214 PyModule_AddIntConstant(m, "AI_PASSIVE", AI_PASSIVE);
5215 #endif
5216 #ifdef AI_CANONNAME
5217 PyModule_AddIntConstant(m, "AI_CANONNAME", AI_CANONNAME);
5218 #endif
5219 #ifdef AI_NUMERICHOST
5220 PyModule_AddIntConstant(m, "AI_NUMERICHOST", AI_NUMERICHOST);
5221 #endif
5222 #ifdef AI_NUMERICSERV
5223 PyModule_AddIntConstant(m, "AI_NUMERICSERV", AI_NUMERICSERV);
5224 #endif
5225 #ifdef AI_MASK
5226 PyModule_AddIntConstant(m, "AI_MASK", AI_MASK);
5227 #endif
5228 #ifdef AI_ALL
5229 PyModule_AddIntConstant(m, "AI_ALL", AI_ALL);
5230 #endif
5231 #ifdef AI_V4MAPPED_CFG
5232 PyModule_AddIntConstant(m, "AI_V4MAPPED_CFG", AI_V4MAPPED_CFG);
5233 #endif
5234 #ifdef AI_ADDRCONFIG
5235 PyModule_AddIntConstant(m, "AI_ADDRCONFIG", AI_ADDRCONFIG);
5236 #endif
5237 #ifdef AI_V4MAPPED
5238 PyModule_AddIntConstant(m, "AI_V4MAPPED", AI_V4MAPPED);
5239 #endif
5240 #ifdef AI_DEFAULT
5241 PyModule_AddIntConstant(m, "AI_DEFAULT", AI_DEFAULT);
5242 #endif
5243 #ifdef NI_MAXHOST
5244 PyModule_AddIntConstant(m, "NI_MAXHOST", NI_MAXHOST);
5245 #endif
5246 #ifdef NI_MAXSERV
5247 PyModule_AddIntConstant(m, "NI_MAXSERV", NI_MAXSERV);
5248 #endif
5249 #ifdef NI_NOFQDN
5250 PyModule_AddIntConstant(m, "NI_NOFQDN", NI_NOFQDN);
5251 #endif
5252 #ifdef NI_NUMERICHOST
5253 PyModule_AddIntConstant(m, "NI_NUMERICHOST", NI_NUMERICHOST);
5254 #endif
5255 #ifdef NI_NAMEREQD
5256 PyModule_AddIntConstant(m, "NI_NAMEREQD", NI_NAMEREQD);
5257 #endif
5258 #ifdef NI_NUMERICSERV
5259 PyModule_AddIntConstant(m, "NI_NUMERICSERV", NI_NUMERICSERV);
5260 #endif
5261 #ifdef NI_DGRAM
5262 PyModule_AddIntConstant(m, "NI_DGRAM", NI_DGRAM);
5263 #endif
5265 /* shutdown() parameters */
5266 #ifdef SHUT_RD
5267 PyModule_AddIntConstant(m, "SHUT_RD", SHUT_RD);
5268 #elif defined(SD_RECEIVE)
5269 PyModule_AddIntConstant(m, "SHUT_RD", SD_RECEIVE);
5270 #else
5271 PyModule_AddIntConstant(m, "SHUT_RD", 0);
5272 #endif
5273 #ifdef SHUT_WR
5274 PyModule_AddIntConstant(m, "SHUT_WR", SHUT_WR);
5275 #elif defined(SD_SEND)
5276 PyModule_AddIntConstant(m, "SHUT_WR", SD_SEND);
5277 #else
5278 PyModule_AddIntConstant(m, "SHUT_WR", 1);
5279 #endif
5280 #ifdef SHUT_RDWR
5281 PyModule_AddIntConstant(m, "SHUT_RDWR", SHUT_RDWR);
5282 #elif defined(SD_BOTH)
5283 PyModule_AddIntConstant(m, "SHUT_RDWR", SD_BOTH);
5284 #else
5285 PyModule_AddIntConstant(m, "SHUT_RDWR", 2);
5286 #endif
5288 #ifdef SIO_RCVALL
5290 PyObject *tmp;
5291 tmp = PyLong_FromUnsignedLong(SIO_RCVALL);
5292 if (tmp == NULL)
5293 return;
5294 PyModule_AddObject(m, "SIO_RCVALL", tmp);
5296 PyModule_AddIntConstant(m, "RCVALL_OFF", RCVALL_OFF);
5297 PyModule_AddIntConstant(m, "RCVALL_ON", RCVALL_ON);
5298 PyModule_AddIntConstant(m, "RCVALL_SOCKETLEVELONLY", RCVALL_SOCKETLEVELONLY);
5299 #ifdef RCVALL_IPLEVEL
5300 PyModule_AddIntConstant(m, "RCVALL_IPLEVEL", RCVALL_IPLEVEL);
5301 #endif
5302 #ifdef RCVALL_MAX
5303 PyModule_AddIntConstant(m, "RCVALL_MAX", RCVALL_MAX);
5304 #endif
5305 #endif /* _MSTCPIP_ */
5307 /* Initialize gethostbyname lock */
5308 #if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
5309 netdb_lock = PyThread_allocate_lock();
5310 #endif
5314 #ifndef HAVE_INET_PTON
5315 #if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
5317 /* Simplistic emulation code for inet_pton that only works for IPv4 */
5318 /* These are not exposed because they do not set errno properly */
5321 inet_pton(int af, const char *src, void *dst)
5323 if (af == AF_INET) {
5324 #if (SIZEOF_INT != 4)
5325 #error "Not sure if in_addr_t exists and int is not 32-bits."
5326 #endif
5327 unsigned int packed_addr;
5328 packed_addr = inet_addr(src);
5329 if (packed_addr == INADDR_NONE)
5330 return 0;
5331 memcpy(dst, &packed_addr, 4);
5332 return 1;
5334 /* Should set errno to EAFNOSUPPORT */
5335 return -1;
5338 const char *
5339 inet_ntop(int af, const void *src, char *dst, socklen_t size)
5341 if (af == AF_INET) {
5342 struct in_addr packed_addr;
5343 if (size < 16)
5344 /* Should set errno to ENOSPC. */
5345 return NULL;
5346 memcpy(&packed_addr, src, sizeof(packed_addr));
5347 return strncpy(dst, inet_ntoa(packed_addr), size);
5349 /* Should set errno to EAFNOSUPPORT */
5350 return NULL;
5353 #endif
5354 #endif