mirror of
				https://github.com/python/cpython.git
				synced 2025-11-04 11:49:12 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			6456 lines
		
	
	
	
		
			188 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			6456 lines
		
	
	
	
		
			188 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/* Socket module */
 | 
						|
 | 
						|
/*
 | 
						|
 | 
						|
This module provides an interface to Berkeley socket IPC.
 | 
						|
 | 
						|
Limitations:
 | 
						|
 | 
						|
- Only AF_INET, AF_INET6 and AF_UNIX address families are supported in a
 | 
						|
  portable manner, though AF_PACKET, AF_NETLINK and AF_TIPC are supported
 | 
						|
  under Linux.
 | 
						|
- No read/write operations (use sendall/recv or makefile instead).
 | 
						|
- Additional restrictions apply on some non-Unix platforms (compensated
 | 
						|
  for by socket.py).
 | 
						|
 | 
						|
Module interface:
 | 
						|
 | 
						|
- socket.error: exception raised for socket specific errors
 | 
						|
- socket.gaierror: exception raised for getaddrinfo/getnameinfo errors,
 | 
						|
    a subclass of socket.error
 | 
						|
- socket.herror: exception raised for gethostby* errors,
 | 
						|
    a subclass of socket.error
 | 
						|
- socket.gethostbyname(hostname) --> host IP address (string: 'dd.dd.dd.dd')
 | 
						|
- socket.gethostbyaddr(IP address) --> (hostname, [alias, ...], [IP addr, ...])
 | 
						|
- socket.gethostname() --> host name (string: 'spam' or 'spam.domain.com')
 | 
						|
- socket.getprotobyname(protocolname) --> protocol number
 | 
						|
- socket.getservbyname(servicename[, protocolname]) --> port number
 | 
						|
- socket.getservbyport(portnumber[, protocolname]) --> service name
 | 
						|
- socket.socket([family[, type [, proto, fileno]]]) --> new socket object
 | 
						|
    (fileno specifies a pre-existing socket file descriptor)
 | 
						|
- socket.socketpair([family[, type [, proto]]]) --> (socket, socket)
 | 
						|
- socket.ntohs(16 bit value) --> new int object
 | 
						|
- socket.ntohl(32 bit value) --> new int object
 | 
						|
- socket.htons(16 bit value) --> new int object
 | 
						|
- socket.htonl(32 bit value) --> new int object
 | 
						|
- socket.getaddrinfo(host, port [, family, socktype, proto, flags])
 | 
						|
    --> List of (family, socktype, proto, canonname, sockaddr)
 | 
						|
- socket.getnameinfo(sockaddr, flags) --> (host, port)
 | 
						|
- socket.AF_INET, socket.SOCK_STREAM, etc.: constants from <socket.h>
 | 
						|
- socket.has_ipv6: boolean value indicating if IPv6 is supported
 | 
						|
- socket.inet_aton(IP address) -> 32-bit packed IP representation
 | 
						|
- socket.inet_ntoa(packed IP) -> IP address string
 | 
						|
- socket.getdefaulttimeout() -> None | float
 | 
						|
- socket.setdefaulttimeout(None | float)
 | 
						|
- socket.if_nameindex() -> list of tuples (if_index, if_name)
 | 
						|
- socket.if_nametoindex(name) -> corresponding interface index
 | 
						|
- socket.if_indextoname(index) -> corresponding interface name
 | 
						|
- an Internet socket address is a pair (hostname, port)
 | 
						|
  where hostname can be anything recognized by gethostbyname()
 | 
						|
  (including the dd.dd.dd.dd notation) and port is in host byte order
 | 
						|
- where a hostname is returned, the dd.dd.dd.dd notation is used
 | 
						|
- a UNIX domain socket address is a string specifying the pathname
 | 
						|
- an AF_PACKET socket address is a tuple containing a string
 | 
						|
  specifying the ethernet interface and an integer specifying
 | 
						|
  the Ethernet protocol number to be received. For example:
 | 
						|
  ("eth0",0x1234).  Optional 3rd,4th,5th elements in the tuple
 | 
						|
  specify packet-type and ha-type/addr.
 | 
						|
- an AF_TIPC socket address is expressed as
 | 
						|
 (addr_type, v1, v2, v3 [, scope]); where addr_type can be one of:
 | 
						|
    TIPC_ADDR_NAMESEQ, TIPC_ADDR_NAME, and TIPC_ADDR_ID;
 | 
						|
  and scope can be one of:
 | 
						|
    TIPC_ZONE_SCOPE, TIPC_CLUSTER_SCOPE, and TIPC_NODE_SCOPE.
 | 
						|
  The meaning of v1, v2 and v3 depends on the value of addr_type:
 | 
						|
    if addr_type is TIPC_ADDR_NAME:
 | 
						|
        v1 is the server type
 | 
						|
        v2 is the port identifier
 | 
						|
        v3 is ignored
 | 
						|
    if addr_type is TIPC_ADDR_NAMESEQ:
 | 
						|
        v1 is the server type
 | 
						|
        v2 is the lower port number
 | 
						|
        v3 is the upper port number
 | 
						|
    if addr_type is TIPC_ADDR_ID:
 | 
						|
        v1 is the node
 | 
						|
        v2 is the ref
 | 
						|
        v3 is ignored
 | 
						|
 | 
						|
 | 
						|
Local naming conventions:
 | 
						|
 | 
						|
- names starting with sock_ are socket object methods
 | 
						|
- names starting with socket_ are module-level functions
 | 
						|
- names starting with PySocket are exported through socketmodule.h
 | 
						|
 | 
						|
*/
 | 
						|
 | 
						|
#ifdef __APPLE__
 | 
						|
  /*
 | 
						|
   * inet_aton is not available on OSX 10.3, yet we want to use a binary
 | 
						|
   * that was build on 10.4 or later to work on that release, weak linking
 | 
						|
   * comes to the rescue.
 | 
						|
   */
 | 
						|
# pragma weak inet_aton
 | 
						|
#endif
 | 
						|
 | 
						|
#include "Python.h"
 | 
						|
#include "structmember.h"
 | 
						|
 | 
						|
#undef MAX
 | 
						|
#define MAX(x, y) ((x) < (y) ? (y) : (x))
 | 
						|
 | 
						|
/* Socket object documentation */
 | 
						|
PyDoc_STRVAR(sock_doc,
 | 
						|
"socket([family[, type[, proto]]]) -> socket object\n\
 | 
						|
\n\
 | 
						|
Open a socket of the given type.  The family argument specifies the\n\
 | 
						|
address family; it defaults to AF_INET.  The type argument specifies\n\
 | 
						|
whether this is a stream (SOCK_STREAM, this is the default)\n\
 | 
						|
or datagram (SOCK_DGRAM) socket.  The protocol argument defaults to 0,\n\
 | 
						|
specifying the default protocol.  Keyword arguments are accepted.\n\
 | 
						|
\n\
 | 
						|
A socket object represents one endpoint of a network connection.\n\
 | 
						|
\n\
 | 
						|
Methods of socket objects (keyword arguments not allowed):\n\
 | 
						|
\n\
 | 
						|
_accept() -- accept connection, returning new socket fd and client address\n\
 | 
						|
bind(addr) -- bind the socket to a local address\n\
 | 
						|
close() -- close the socket\n\
 | 
						|
connect(addr) -- connect the socket to a remote address\n\
 | 
						|
connect_ex(addr) -- connect, return an error code instead of an exception\n\
 | 
						|
_dup() -- return a new socket fd duplicated from fileno()\n\
 | 
						|
fileno() -- return underlying file descriptor\n\
 | 
						|
getpeername() -- return remote address [*]\n\
 | 
						|
getsockname() -- return local address\n\
 | 
						|
getsockopt(level, optname[, buflen]) -- get socket options\n\
 | 
						|
gettimeout() -- return timeout or None\n\
 | 
						|
listen(n) -- start listening for incoming connections\n\
 | 
						|
recv(buflen[, flags]) -- receive data\n\
 | 
						|
recv_into(buffer[, nbytes[, flags]]) -- receive data (into a buffer)\n\
 | 
						|
recvfrom(buflen[, flags]) -- receive data and sender\'s address\n\
 | 
						|
recvfrom_into(buffer[, nbytes, [, flags])\n\
 | 
						|
  -- receive data and sender\'s address (into a buffer)\n\
 | 
						|
sendall(data[, flags]) -- send all data\n\
 | 
						|
send(data[, flags]) -- send data, may not send all of it\n\
 | 
						|
sendto(data[, flags], addr) -- send data to a given address\n\
 | 
						|
setblocking(0 | 1) -- set or clear the blocking I/O flag\n\
 | 
						|
setsockopt(level, optname, value) -- set socket options\n\
 | 
						|
settimeout(None | float) -- set or clear the timeout\n\
 | 
						|
shutdown(how) -- shut down traffic in one or both directions\n\
 | 
						|
if_nameindex() -- return all network interface indices and names\n\
 | 
						|
if_nametoindex(name) -- return the corresponding interface index\n\
 | 
						|
if_indextoname(index) -- return the corresponding interface name\n\
 | 
						|
\n\
 | 
						|
 [*] not available on all platforms!");
 | 
						|
 | 
						|
/* XXX This is a terrible mess of platform-dependent preprocessor hacks.
 | 
						|
   I hope some day someone can clean this up please... */
 | 
						|
 | 
						|
/* Hacks for gethostbyname_r().  On some non-Linux platforms, the configure
 | 
						|
   script doesn't get this right, so we hardcode some platform checks below.
 | 
						|
   On the other hand, not all Linux versions agree, so there the settings
 | 
						|
   computed by the configure script are needed! */
 | 
						|
 | 
						|
#ifndef linux
 | 
						|
# undef HAVE_GETHOSTBYNAME_R_3_ARG
 | 
						|
# undef HAVE_GETHOSTBYNAME_R_5_ARG
 | 
						|
# undef HAVE_GETHOSTBYNAME_R_6_ARG
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(__OpenBSD__)
 | 
						|
# include <sys/uio.h>
 | 
						|
#endif
 | 
						|
 | 
						|
#ifndef WITH_THREAD
 | 
						|
# undef HAVE_GETHOSTBYNAME_R
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_GETHOSTBYNAME_R
 | 
						|
# if defined(_AIX)
 | 
						|
#  define HAVE_GETHOSTBYNAME_R_3_ARG
 | 
						|
# elif defined(__sun) || defined(__sgi)
 | 
						|
#  define HAVE_GETHOSTBYNAME_R_5_ARG
 | 
						|
# elif defined(linux)
 | 
						|
/* Rely on the configure script */
 | 
						|
# else
 | 
						|
#  undef HAVE_GETHOSTBYNAME_R
 | 
						|
# endif
 | 
						|
#endif
 | 
						|
 | 
						|
#if !defined(HAVE_GETHOSTBYNAME_R) && defined(WITH_THREAD) && \
 | 
						|
    !defined(MS_WINDOWS)
 | 
						|
# define USE_GETHOSTBYNAME_LOCK
 | 
						|
#endif
 | 
						|
 | 
						|
/* To use __FreeBSD_version */
 | 
						|
#ifdef HAVE_SYS_PARAM_H
 | 
						|
#include <sys/param.h>
 | 
						|
#endif
 | 
						|
/* On systems on which getaddrinfo() is believed to not be thread-safe,
 | 
						|
   (this includes the getaddrinfo emulation) protect access with a lock. */
 | 
						|
#if defined(WITH_THREAD) && (defined(__APPLE__) || \
 | 
						|
    (defined(__FreeBSD__) && __FreeBSD_version+0 < 503000) || \
 | 
						|
    defined(__OpenBSD__) || defined(__NetBSD__) || \
 | 
						|
    defined(__VMS) || !defined(HAVE_GETADDRINFO))
 | 
						|
#define USE_GETADDRINFO_LOCK
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef USE_GETADDRINFO_LOCK
 | 
						|
#define ACQUIRE_GETADDRINFO_LOCK PyThread_acquire_lock(netdb_lock, 1);
 | 
						|
#define RELEASE_GETADDRINFO_LOCK PyThread_release_lock(netdb_lock);
 | 
						|
#else
 | 
						|
#define ACQUIRE_GETADDRINFO_LOCK
 | 
						|
#define RELEASE_GETADDRINFO_LOCK
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
 | 
						|
# include "pythread.h"
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(PYCC_VACPP)
 | 
						|
# include <types.h>
 | 
						|
# include <io.h>
 | 
						|
# include <sys/ioctl.h>
 | 
						|
# include <utils.h>
 | 
						|
# include <ctype.h>
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(__VMS)
 | 
						|
#  include <ioctl.h>
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(PYOS_OS2)
 | 
						|
# define  INCL_DOS
 | 
						|
# define  INCL_DOSERRORS
 | 
						|
# define  INCL_NOPMAPI
 | 
						|
# include <os2.h>
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(__sgi) && _COMPILER_VERSION>700 && !_SGIAPI
 | 
						|
/* make sure that the reentrant (gethostbyaddr_r etc)
 | 
						|
   functions are declared correctly if compiling with
 | 
						|
   MIPSPro 7.x in ANSI C mode (default) */
 | 
						|
 | 
						|
/* XXX Using _SGIAPI is the wrong thing,
 | 
						|
   but I don't know what the right thing is. */
 | 
						|
#undef _SGIAPI /* to avoid warning */
 | 
						|
#define _SGIAPI 1
 | 
						|
 | 
						|
#undef _XOPEN_SOURCE
 | 
						|
#include <sys/socket.h>
 | 
						|
#include <sys/types.h>
 | 
						|
#include <netinet/in.h>
 | 
						|
#ifdef _SS_ALIGNSIZE
 | 
						|
#define HAVE_GETADDRINFO 1
 | 
						|
#define HAVE_GETNAMEINFO 1
 | 
						|
#endif
 | 
						|
 | 
						|
#define HAVE_INET_PTON
 | 
						|
#include <netdb.h>
 | 
						|
#endif
 | 
						|
 | 
						|
/* Irix 6.5 fails to define this variable at all. This is needed
 | 
						|
   for both GCC and SGI's compiler. I'd say that the SGI headers
 | 
						|
   are just busted. Same thing for Solaris. */
 | 
						|
#if (defined(__sgi) || defined(sun)) && !defined(INET_ADDRSTRLEN)
 | 
						|
#define INET_ADDRSTRLEN 16
 | 
						|
#endif
 | 
						|
 | 
						|
/* Generic includes */
 | 
						|
#ifdef HAVE_SYS_TYPES_H
 | 
						|
#include <sys/types.h>
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_SYS_SOCKET_H
 | 
						|
#include <sys/socket.h>
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_NET_IF_H
 | 
						|
#include <net/if.h>
 | 
						|
#endif
 | 
						|
 | 
						|
/* Generic socket object definitions and includes */
 | 
						|
#define PySocket_BUILDING_SOCKET
 | 
						|
#include "socketmodule.h"
 | 
						|
 | 
						|
/* Addressing includes */
 | 
						|
 | 
						|
#ifndef MS_WINDOWS
 | 
						|
 | 
						|
/* Non-MS WINDOWS includes */
 | 
						|
# include <netdb.h>
 | 
						|
# include <unistd.h>
 | 
						|
 | 
						|
/* Headers needed for inet_ntoa() and inet_addr() */
 | 
						|
# if defined(PYOS_OS2) && defined(PYCC_VACPP)
 | 
						|
#  include <netdb.h>
 | 
						|
typedef size_t socklen_t;
 | 
						|
# else
 | 
						|
#   include <arpa/inet.h>
 | 
						|
# endif
 | 
						|
 | 
						|
#  include <fcntl.h>
 | 
						|
 | 
						|
#else
 | 
						|
 | 
						|
/* MS_WINDOWS includes */
 | 
						|
# ifdef HAVE_FCNTL_H
 | 
						|
#  include <fcntl.h>
 | 
						|
# endif
 | 
						|
 | 
						|
#endif
 | 
						|
 | 
						|
#include <stddef.h>
 | 
						|
 | 
						|
#ifndef offsetof
 | 
						|
# define offsetof(type, member) ((size_t)(&((type *)0)->member))
 | 
						|
#endif
 | 
						|
 | 
						|
#ifndef O_NONBLOCK
 | 
						|
# define O_NONBLOCK O_NDELAY
 | 
						|
#endif
 | 
						|
 | 
						|
/* include Python's addrinfo.h unless it causes trouble */
 | 
						|
#if defined(__sgi) && _COMPILER_VERSION>700 && defined(_SS_ALIGNSIZE)
 | 
						|
  /* Do not include addinfo.h on some newer IRIX versions.
 | 
						|
   * _SS_ALIGNSIZE is defined in sys/socket.h by 6.5.21,
 | 
						|
   * for example, but not by 6.5.10.
 | 
						|
   */
 | 
						|
#elif defined(_MSC_VER) && _MSC_VER>1201
 | 
						|
  /* Do not include addrinfo.h for MSVC7 or greater. 'addrinfo' and
 | 
						|
   * EAI_* constants are defined in (the already included) ws2tcpip.h.
 | 
						|
   */
 | 
						|
#else
 | 
						|
#  include "addrinfo.h"
 | 
						|
#endif
 | 
						|
 | 
						|
#ifndef HAVE_INET_PTON
 | 
						|
#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
 | 
						|
int inet_pton(int af, const char *src, void *dst);
 | 
						|
const char *inet_ntop(int af, const void *src, char *dst, socklen_t size);
 | 
						|
#endif
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef __APPLE__
 | 
						|
/* On OS X, getaddrinfo returns no error indication of lookup
 | 
						|
   failure, so we must use the emulation instead of the libinfo
 | 
						|
   implementation. Unfortunately, performing an autoconf test
 | 
						|
   for this bug would require DNS access for the machine performing
 | 
						|
   the configuration, which is not acceptable. Therefore, we
 | 
						|
   determine the bug just by checking for __APPLE__. If this bug
 | 
						|
   gets ever fixed, perhaps checking for sys/version.h would be
 | 
						|
   appropriate, which is 10/0 on the system with the bug. */
 | 
						|
#ifndef HAVE_GETNAMEINFO
 | 
						|
/* This bug seems to be fixed in Jaguar. Ths easiest way I could
 | 
						|
   Find to check for Jaguar is that it has getnameinfo(), which
 | 
						|
   older releases don't have */
 | 
						|
#undef HAVE_GETADDRINFO
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_INET_ATON
 | 
						|
#define USE_INET_ATON_WEAKLINK
 | 
						|
#endif
 | 
						|
 | 
						|
#endif
 | 
						|
 | 
						|
/* I know this is a bad practice, but it is the easiest... */
 | 
						|
#if !defined(HAVE_GETADDRINFO)
 | 
						|
/* avoid clashes with the C library definition of the symbol. */
 | 
						|
#define getaddrinfo fake_getaddrinfo
 | 
						|
#define gai_strerror fake_gai_strerror
 | 
						|
#define freeaddrinfo fake_freeaddrinfo
 | 
						|
#include "getaddrinfo.c"
 | 
						|
#endif
 | 
						|
#if !defined(HAVE_GETNAMEINFO)
 | 
						|
#define getnameinfo fake_getnameinfo
 | 
						|
#include "getnameinfo.c"
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef MS_WINDOWS
 | 
						|
/* On Windows a socket is really a handle not an fd */
 | 
						|
static SOCKET
 | 
						|
dup_socket(SOCKET handle)
 | 
						|
{
 | 
						|
    WSAPROTOCOL_INFO info;
 | 
						|
 | 
						|
    if (WSADuplicateSocket(handle, GetCurrentProcessId(), &info))
 | 
						|
        return INVALID_SOCKET;
 | 
						|
 | 
						|
    return WSASocket(FROM_PROTOCOL_INFO, FROM_PROTOCOL_INFO,
 | 
						|
                     FROM_PROTOCOL_INFO, &info, 0, 0);
 | 
						|
}
 | 
						|
#define SOCKETCLOSE closesocket
 | 
						|
#else
 | 
						|
/* On Unix we can use dup to duplicate the file descriptor of a socket*/
 | 
						|
#define dup_socket(fd) dup(fd)
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef MS_WIN32
 | 
						|
#define EAFNOSUPPORT WSAEAFNOSUPPORT
 | 
						|
#define snprintf _snprintf
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(PYOS_OS2) && !defined(PYCC_GCC)
 | 
						|
#define SOCKETCLOSE soclose
 | 
						|
#define NO_DUP /* Sockets are Not Actual File Handles under OS/2 */
 | 
						|
#endif
 | 
						|
 | 
						|
#ifndef SOCKETCLOSE
 | 
						|
#define SOCKETCLOSE close
 | 
						|
#endif
 | 
						|
 | 
						|
#if (defined(HAVE_BLUETOOTH_H) || defined(HAVE_BLUETOOTH_BLUETOOTH_H)) && !defined(__NetBSD__) && !defined(__DragonFly__)
 | 
						|
#define USE_BLUETOOTH 1
 | 
						|
#if defined(__FreeBSD__)
 | 
						|
#define BTPROTO_L2CAP BLUETOOTH_PROTO_L2CAP
 | 
						|
#define BTPROTO_RFCOMM BLUETOOTH_PROTO_RFCOMM
 | 
						|
#define BTPROTO_HCI BLUETOOTH_PROTO_HCI
 | 
						|
#define SOL_HCI SOL_HCI_RAW
 | 
						|
#define HCI_FILTER SO_HCI_RAW_FILTER
 | 
						|
#define sockaddr_l2 sockaddr_l2cap
 | 
						|
#define sockaddr_rc sockaddr_rfcomm
 | 
						|
#define hci_dev hci_node
 | 
						|
#define _BT_L2_MEMB(sa, memb) ((sa)->l2cap_##memb)
 | 
						|
#define _BT_RC_MEMB(sa, memb) ((sa)->rfcomm_##memb)
 | 
						|
#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
 | 
						|
#elif defined(__NetBSD__) || defined(__DragonFly__)
 | 
						|
#define sockaddr_l2 sockaddr_bt
 | 
						|
#define sockaddr_rc sockaddr_bt
 | 
						|
#define sockaddr_hci sockaddr_bt
 | 
						|
#define sockaddr_sco sockaddr_bt
 | 
						|
#define SOL_HCI BTPROTO_HCI
 | 
						|
#define HCI_DATA_DIR SO_HCI_DIRECTION
 | 
						|
#define _BT_L2_MEMB(sa, memb) ((sa)->bt_##memb)
 | 
						|
#define _BT_RC_MEMB(sa, memb) ((sa)->bt_##memb)
 | 
						|
#define _BT_HCI_MEMB(sa, memb) ((sa)->bt_##memb)
 | 
						|
#define _BT_SCO_MEMB(sa, memb) ((sa)->bt_##memb)
 | 
						|
#else
 | 
						|
#define _BT_L2_MEMB(sa, memb) ((sa)->l2_##memb)
 | 
						|
#define _BT_RC_MEMB(sa, memb) ((sa)->rc_##memb)
 | 
						|
#define _BT_HCI_MEMB(sa, memb) ((sa)->hci_##memb)
 | 
						|
#define _BT_SCO_MEMB(sa, memb) ((sa)->sco_##memb)
 | 
						|
#endif
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef __VMS
 | 
						|
/* TCP/IP Services for VMS uses a maximum send/recv buffer length */
 | 
						|
#define SEGMENT_SIZE (32 * 1024 -1)
 | 
						|
#endif
 | 
						|
 | 
						|
#define SAS2SA(x)       ((struct sockaddr *)(x))
 | 
						|
 | 
						|
/*
 | 
						|
 * Constants for getnameinfo()
 | 
						|
 */
 | 
						|
#if !defined(NI_MAXHOST)
 | 
						|
#define NI_MAXHOST 1025
 | 
						|
#endif
 | 
						|
#if !defined(NI_MAXSERV)
 | 
						|
#define NI_MAXSERV 32
 | 
						|
#endif
 | 
						|
 | 
						|
#ifndef INVALID_SOCKET /* MS defines this */
 | 
						|
#define INVALID_SOCKET (-1)
 | 
						|
#endif
 | 
						|
 | 
						|
/* XXX There's a problem here: *static* functions are not supposed to have
 | 
						|
   a Py prefix (or use CapitalizedWords).  Later... */
 | 
						|
 | 
						|
/* Global variable holding the exception type for errors detected
 | 
						|
   by this module (but not argument type or memory errors, etc.). */
 | 
						|
static PyObject *socket_error;
 | 
						|
static PyObject *socket_herror;
 | 
						|
static PyObject *socket_gaierror;
 | 
						|
static PyObject *socket_timeout;
 | 
						|
 | 
						|
/* A forward reference to the socket type object.
 | 
						|
   The sock_type variable contains pointers to various functions,
 | 
						|
   some of which call new_sockobject(), which uses sock_type, so
 | 
						|
   there has to be a circular reference. */
 | 
						|
static PyTypeObject sock_type;
 | 
						|
 | 
						|
#if defined(HAVE_POLL_H)
 | 
						|
#include <poll.h>
 | 
						|
#elif defined(HAVE_SYS_POLL_H)
 | 
						|
#include <sys/poll.h>
 | 
						|
#endif
 | 
						|
 | 
						|
/* Largest value to try to store in a socklen_t (used when handling
 | 
						|
   ancillary data).  POSIX requires socklen_t to hold at least
 | 
						|
   (2**31)-1 and recommends against storing larger values, but
 | 
						|
   socklen_t was originally int in the BSD interface, so to be on the
 | 
						|
   safe side we use the smaller of (2**31)-1 and INT_MAX. */
 | 
						|
#if INT_MAX > 0x7fffffff
 | 
						|
#define SOCKLEN_T_LIMIT 0x7fffffff
 | 
						|
#else
 | 
						|
#define SOCKLEN_T_LIMIT INT_MAX
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_POLL
 | 
						|
/* Instead of select(), we'll use poll() since poll() works on any fd. */
 | 
						|
#define IS_SELECTABLE(s) 1
 | 
						|
/* Can we call select() with this socket without a buffer overrun? */
 | 
						|
#else
 | 
						|
/* If there's no timeout left, we don't have to call select, so it's a safe,
 | 
						|
 * little white lie. */
 | 
						|
#define IS_SELECTABLE(s) (_PyIsSelectable_fd((s)->sock_fd) || (s)->sock_timeout <= 0.0)
 | 
						|
#endif
 | 
						|
 | 
						|
static PyObject*
 | 
						|
select_error(void)
 | 
						|
{
 | 
						|
    PyErr_SetString(socket_error, "unable to select on socket");
 | 
						|
    return NULL;
 | 
						|
}
 | 
						|
 | 
						|
#ifdef MS_WINDOWS
 | 
						|
#ifndef WSAEAGAIN
 | 
						|
#define WSAEAGAIN WSAEWOULDBLOCK
 | 
						|
#endif
 | 
						|
#define CHECK_ERRNO(expected) \
 | 
						|
    (WSAGetLastError() == WSA ## expected)
 | 
						|
#else
 | 
						|
#define CHECK_ERRNO(expected) \
 | 
						|
    (errno == expected)
 | 
						|
#endif
 | 
						|
 | 
						|
/* Convenience function to raise an error according to errno
 | 
						|
   and return a NULL pointer from a function. */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
set_error(void)
 | 
						|
{
 | 
						|
#ifdef MS_WINDOWS
 | 
						|
    int err_no = WSAGetLastError();
 | 
						|
    /* PyErr_SetExcFromWindowsErr() invokes FormatMessage() which
 | 
						|
       recognizes the error codes used by both GetLastError() and
 | 
						|
       WSAGetLastError */
 | 
						|
    if (err_no)
 | 
						|
        return PyErr_SetExcFromWindowsErr(socket_error, err_no);
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(PYOS_OS2) && !defined(PYCC_GCC)
 | 
						|
    if (sock_errno() != NO_ERROR) {
 | 
						|
        APIRET rc;
 | 
						|
        ULONG  msglen;
 | 
						|
        char outbuf[100];
 | 
						|
        int myerrorcode = sock_errno();
 | 
						|
 | 
						|
        /* Retrieve socket-related error message from MPTN.MSG file */
 | 
						|
        rc = DosGetMessage(NULL, 0, outbuf, sizeof(outbuf),
 | 
						|
                           myerrorcode - SOCBASEERR + 26,
 | 
						|
                           "mptn.msg",
 | 
						|
                           &msglen);
 | 
						|
        if (rc == NO_ERROR) {
 | 
						|
            PyObject *v;
 | 
						|
 | 
						|
            /* OS/2 doesn't guarantee a terminator */
 | 
						|
            outbuf[msglen] = '\0';
 | 
						|
            if (strlen(outbuf) > 0) {
 | 
						|
                /* If non-empty msg, trim CRLF */
 | 
						|
                char *lastc = &outbuf[ strlen(outbuf)-1 ];
 | 
						|
                while (lastc > outbuf &&
 | 
						|
                       isspace(Py_CHARMASK(*lastc))) {
 | 
						|
                    /* Trim trailing whitespace (CRLF) */
 | 
						|
                    *lastc-- = '\0';
 | 
						|
                }
 | 
						|
            }
 | 
						|
            v = Py_BuildValue("(is)", myerrorcode, outbuf);
 | 
						|
            if (v != NULL) {
 | 
						|
                PyErr_SetObject(socket_error, v);
 | 
						|
                Py_DECREF(v);
 | 
						|
            }
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
    return PyErr_SetFromErrno(socket_error);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
static PyObject *
 | 
						|
set_herror(int h_error)
 | 
						|
{
 | 
						|
    PyObject *v;
 | 
						|
 | 
						|
#ifdef HAVE_HSTRERROR
 | 
						|
    v = Py_BuildValue("(is)", h_error, (char *)hstrerror(h_error));
 | 
						|
#else
 | 
						|
    v = Py_BuildValue("(is)", h_error, "host not found");
 | 
						|
#endif
 | 
						|
    if (v != NULL) {
 | 
						|
        PyErr_SetObject(socket_herror, v);
 | 
						|
        Py_DECREF(v);
 | 
						|
    }
 | 
						|
 | 
						|
    return NULL;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
static PyObject *
 | 
						|
set_gaierror(int error)
 | 
						|
{
 | 
						|
    PyObject *v;
 | 
						|
 | 
						|
#ifdef EAI_SYSTEM
 | 
						|
    /* EAI_SYSTEM is not available on Windows XP. */
 | 
						|
    if (error == EAI_SYSTEM)
 | 
						|
        return set_error();
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_GAI_STRERROR
 | 
						|
    v = Py_BuildValue("(is)", error, gai_strerror(error));
 | 
						|
#else
 | 
						|
    v = Py_BuildValue("(is)", error, "getaddrinfo failed");
 | 
						|
#endif
 | 
						|
    if (v != NULL) {
 | 
						|
        PyErr_SetObject(socket_gaierror, v);
 | 
						|
        Py_DECREF(v);
 | 
						|
    }
 | 
						|
 | 
						|
    return NULL;
 | 
						|
}
 | 
						|
 | 
						|
#ifdef __VMS
 | 
						|
/* Function to send in segments */
 | 
						|
static int
 | 
						|
sendsegmented(int sock_fd, char *buf, int len, int flags)
 | 
						|
{
 | 
						|
    int n = 0;
 | 
						|
    int remaining = len;
 | 
						|
 | 
						|
    while (remaining > 0) {
 | 
						|
        unsigned int segment;
 | 
						|
 | 
						|
        segment = (remaining >= SEGMENT_SIZE ? SEGMENT_SIZE : remaining);
 | 
						|
        n = send(sock_fd, buf, segment, flags);
 | 
						|
        if (n < 0) {
 | 
						|
            return n;
 | 
						|
        }
 | 
						|
        remaining -= segment;
 | 
						|
        buf += segment;
 | 
						|
    } /* end while */
 | 
						|
 | 
						|
    return len;
 | 
						|
}
 | 
						|
#endif
 | 
						|
 | 
						|
/* Function to perform the setting of socket blocking mode
 | 
						|
   internally. block = (1 | 0). */
 | 
						|
static int
 | 
						|
internal_setblocking(PySocketSockObject *s, int block)
 | 
						|
{
 | 
						|
#ifndef MS_WINDOWS
 | 
						|
    int delay_flag;
 | 
						|
#endif
 | 
						|
#ifdef SOCK_NONBLOCK
 | 
						|
    if (block)
 | 
						|
        s->sock_type &= (~SOCK_NONBLOCK);
 | 
						|
    else
 | 
						|
        s->sock_type |= SOCK_NONBLOCK;
 | 
						|
#endif
 | 
						|
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
#ifndef MS_WINDOWS
 | 
						|
#if defined(PYOS_OS2) && !defined(PYCC_GCC)
 | 
						|
    block = !block;
 | 
						|
    ioctl(s->sock_fd, FIONBIO, (caddr_t)&block, sizeof(block));
 | 
						|
#elif defined(__VMS)
 | 
						|
    block = !block;
 | 
						|
    ioctl(s->sock_fd, FIONBIO, (unsigned int *)&block);
 | 
						|
#else  /* !PYOS_OS2 && !__VMS */
 | 
						|
    delay_flag = fcntl(s->sock_fd, F_GETFL, 0);
 | 
						|
    if (block)
 | 
						|
        delay_flag &= (~O_NONBLOCK);
 | 
						|
    else
 | 
						|
        delay_flag |= O_NONBLOCK;
 | 
						|
    fcntl(s->sock_fd, F_SETFL, delay_flag);
 | 
						|
#endif /* !PYOS_OS2 */
 | 
						|
#else /* MS_WINDOWS */
 | 
						|
    block = !block;
 | 
						|
    ioctlsocket(s->sock_fd, FIONBIO, (u_long*)&block);
 | 
						|
#endif /* MS_WINDOWS */
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
 | 
						|
    /* Since these don't return anything */
 | 
						|
    return 1;
 | 
						|
}
 | 
						|
 | 
						|
/* Do a select()/poll() on the socket, if necessary (sock_timeout > 0).
 | 
						|
   The argument writing indicates the direction.
 | 
						|
   This does not raise an exception; we'll let our caller do that
 | 
						|
   after they've reacquired the interpreter lock.
 | 
						|
   Returns 1 on timeout, -1 on error, 0 otherwise. */
 | 
						|
static int
 | 
						|
internal_select_ex(PySocketSockObject *s, int writing, double interval)
 | 
						|
{
 | 
						|
    int n;
 | 
						|
 | 
						|
    /* Nothing to do unless we're in timeout mode (not non-blocking) */
 | 
						|
    if (s->sock_timeout <= 0.0)
 | 
						|
        return 0;
 | 
						|
 | 
						|
    /* Guard against closed socket */
 | 
						|
    if (s->sock_fd < 0)
 | 
						|
        return 0;
 | 
						|
 | 
						|
    /* Handling this condition here simplifies the select loops */
 | 
						|
    if (interval < 0.0)
 | 
						|
        return 1;
 | 
						|
 | 
						|
    /* Prefer poll, if available, since you can poll() any fd
 | 
						|
     * which can't be done with select(). */
 | 
						|
#ifdef HAVE_POLL
 | 
						|
    {
 | 
						|
        struct pollfd pollfd;
 | 
						|
        int timeout;
 | 
						|
 | 
						|
        pollfd.fd = s->sock_fd;
 | 
						|
        pollfd.events = writing ? POLLOUT : POLLIN;
 | 
						|
 | 
						|
        /* s->sock_timeout is in seconds, timeout in ms */
 | 
						|
        timeout = (int)(interval * 1000 + 0.5);
 | 
						|
        n = poll(&pollfd, 1, timeout);
 | 
						|
    }
 | 
						|
#else
 | 
						|
    {
 | 
						|
        /* Construct the arguments to select */
 | 
						|
        fd_set fds;
 | 
						|
        struct timeval tv;
 | 
						|
        tv.tv_sec = (int)interval;
 | 
						|
        tv.tv_usec = (int)((interval - tv.tv_sec) * 1e6);
 | 
						|
        FD_ZERO(&fds);
 | 
						|
        FD_SET(s->sock_fd, &fds);
 | 
						|
 | 
						|
        /* See if the socket is ready */
 | 
						|
        if (writing)
 | 
						|
            n = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
 | 
						|
                       NULL, &fds, NULL, &tv);
 | 
						|
        else
 | 
						|
            n = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
 | 
						|
                       &fds, NULL, NULL, &tv);
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
    if (n < 0)
 | 
						|
        return -1;
 | 
						|
    if (n == 0)
 | 
						|
        return 1;
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
internal_select(PySocketSockObject *s, int writing)
 | 
						|
{
 | 
						|
    return internal_select_ex(s, writing, s->sock_timeout);
 | 
						|
}
 | 
						|
 | 
						|
/*
 | 
						|
   Two macros for automatic retry of select() in case of false positives
 | 
						|
   (for example, select() could indicate a socket is ready for reading
 | 
						|
    but the data then discarded by the OS because of a wrong checksum).
 | 
						|
   Here is an example of use:
 | 
						|
 | 
						|
    BEGIN_SELECT_LOOP(s)
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    timeout = internal_select_ex(s, 0, interval);
 | 
						|
    if (!timeout)
 | 
						|
        outlen = recv(s->sock_fd, cbuf, len, flags);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    if (timeout == 1) {
 | 
						|
        PyErr_SetString(socket_timeout, "timed out");
 | 
						|
        return -1;
 | 
						|
    }
 | 
						|
    END_SELECT_LOOP(s)
 | 
						|
*/
 | 
						|
 | 
						|
#define BEGIN_SELECT_LOOP(s) \
 | 
						|
    { \
 | 
						|
        _PyTime_timeval now, deadline = {0, 0}; \
 | 
						|
        double interval = s->sock_timeout; \
 | 
						|
        int has_timeout = s->sock_timeout > 0.0; \
 | 
						|
        if (has_timeout) { \
 | 
						|
            _PyTime_gettimeofday(&now); \
 | 
						|
            deadline = now; \
 | 
						|
            _PyTime_ADD_SECONDS(deadline, s->sock_timeout); \
 | 
						|
        } \
 | 
						|
        while (1) { \
 | 
						|
            errno = 0; \
 | 
						|
 | 
						|
#define END_SELECT_LOOP(s) \
 | 
						|
            if (!has_timeout || \
 | 
						|
                (!CHECK_ERRNO(EWOULDBLOCK) && !CHECK_ERRNO(EAGAIN))) \
 | 
						|
                break; \
 | 
						|
            _PyTime_gettimeofday(&now); \
 | 
						|
            interval = _PyTime_INTERVAL(now, deadline); \
 | 
						|
        } \
 | 
						|
    } \
 | 
						|
 | 
						|
/* Initialize a new socket object. */
 | 
						|
 | 
						|
static double defaulttimeout = -1.0; /* Default timeout for new sockets */
 | 
						|
 | 
						|
static void
 | 
						|
init_sockobject(PySocketSockObject *s,
 | 
						|
                SOCKET_T fd, int family, int type, int proto)
 | 
						|
{
 | 
						|
    s->sock_fd = fd;
 | 
						|
    s->sock_family = family;
 | 
						|
    s->sock_type = type;
 | 
						|
    s->sock_proto = proto;
 | 
						|
 | 
						|
    s->errorhandler = &set_error;
 | 
						|
#ifdef SOCK_NONBLOCK
 | 
						|
    if (type & SOCK_NONBLOCK)
 | 
						|
        s->sock_timeout = 0.0;
 | 
						|
    else
 | 
						|
#endif
 | 
						|
    {
 | 
						|
        s->sock_timeout = defaulttimeout;
 | 
						|
        if (defaulttimeout >= 0.0)
 | 
						|
            internal_setblocking(s, 0);
 | 
						|
    }
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* Create a new socket object.
 | 
						|
   This just creates the object and initializes it.
 | 
						|
   If the creation fails, return NULL and set an exception (implicit
 | 
						|
   in NEWOBJ()). */
 | 
						|
 | 
						|
static PySocketSockObject *
 | 
						|
new_sockobject(SOCKET_T fd, int family, int type, int proto)
 | 
						|
{
 | 
						|
    PySocketSockObject *s;
 | 
						|
    s = (PySocketSockObject *)
 | 
						|
        PyType_GenericNew(&sock_type, NULL, NULL);
 | 
						|
    if (s != NULL)
 | 
						|
        init_sockobject(s, fd, family, type, proto);
 | 
						|
    return s;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* Lock to allow python interpreter to continue, but only allow one
 | 
						|
   thread to be in gethostbyname or getaddrinfo */
 | 
						|
#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
 | 
						|
PyThread_type_lock netdb_lock;
 | 
						|
#endif
 | 
						|
 | 
						|
 | 
						|
/* Convert a string specifying a host name or one of a few symbolic
 | 
						|
   names to a numeric IP address.  This usually calls gethostbyname()
 | 
						|
   to do the work; the names "" and "<broadcast>" are special.
 | 
						|
   Return the length (IPv4 should be 4 bytes), or negative if
 | 
						|
   an error occurred; then an exception is raised. */
 | 
						|
 | 
						|
static int
 | 
						|
setipaddr(char *name, struct sockaddr *addr_ret, size_t addr_ret_size, int af)
 | 
						|
{
 | 
						|
    struct addrinfo hints, *res;
 | 
						|
    int error;
 | 
						|
    int d1, d2, d3, d4;
 | 
						|
    char ch;
 | 
						|
 | 
						|
    memset((void *) addr_ret, '\0', sizeof(*addr_ret));
 | 
						|
    if (name[0] == '\0') {
 | 
						|
        int siz;
 | 
						|
        memset(&hints, 0, sizeof(hints));
 | 
						|
        hints.ai_family = af;
 | 
						|
        hints.ai_socktype = SOCK_DGRAM;         /*dummy*/
 | 
						|
        hints.ai_flags = AI_PASSIVE;
 | 
						|
        Py_BEGIN_ALLOW_THREADS
 | 
						|
        ACQUIRE_GETADDRINFO_LOCK
 | 
						|
        error = getaddrinfo(NULL, "0", &hints, &res);
 | 
						|
        Py_END_ALLOW_THREADS
 | 
						|
        /* We assume that those thread-unsafe getaddrinfo() versions
 | 
						|
           *are* safe regarding their return value, ie. that a
 | 
						|
           subsequent call to getaddrinfo() does not destroy the
 | 
						|
           outcome of the first call. */
 | 
						|
        RELEASE_GETADDRINFO_LOCK
 | 
						|
        if (error) {
 | 
						|
            set_gaierror(error);
 | 
						|
            return -1;
 | 
						|
        }
 | 
						|
        switch (res->ai_family) {
 | 
						|
        case AF_INET:
 | 
						|
            siz = 4;
 | 
						|
            break;
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
        case AF_INET6:
 | 
						|
            siz = 16;
 | 
						|
            break;
 | 
						|
#endif
 | 
						|
        default:
 | 
						|
            freeaddrinfo(res);
 | 
						|
            PyErr_SetString(socket_error,
 | 
						|
                "unsupported address family");
 | 
						|
            return -1;
 | 
						|
        }
 | 
						|
        if (res->ai_next) {
 | 
						|
            freeaddrinfo(res);
 | 
						|
            PyErr_SetString(socket_error,
 | 
						|
                "wildcard resolved to multiple address");
 | 
						|
            return -1;
 | 
						|
        }
 | 
						|
        if (res->ai_addrlen < addr_ret_size)
 | 
						|
            addr_ret_size = res->ai_addrlen;
 | 
						|
        memcpy(addr_ret, res->ai_addr, addr_ret_size);
 | 
						|
        freeaddrinfo(res);
 | 
						|
        return siz;
 | 
						|
    }
 | 
						|
    if (name[0] == '<' && strcmp(name, "<broadcast>") == 0) {
 | 
						|
        struct sockaddr_in *sin;
 | 
						|
        if (af != AF_INET && af != AF_UNSPEC) {
 | 
						|
            PyErr_SetString(socket_error,
 | 
						|
                "address family mismatched");
 | 
						|
            return -1;
 | 
						|
        }
 | 
						|
        sin = (struct sockaddr_in *)addr_ret;
 | 
						|
        memset((void *) sin, '\0', sizeof(*sin));
 | 
						|
        sin->sin_family = AF_INET;
 | 
						|
#ifdef HAVE_SOCKADDR_SA_LEN
 | 
						|
        sin->sin_len = sizeof(*sin);
 | 
						|
#endif
 | 
						|
        sin->sin_addr.s_addr = INADDR_BROADCAST;
 | 
						|
        return sizeof(sin->sin_addr);
 | 
						|
    }
 | 
						|
    if (sscanf(name, "%d.%d.%d.%d%c", &d1, &d2, &d3, &d4, &ch) == 4 &&
 | 
						|
        0 <= d1 && d1 <= 255 && 0 <= d2 && d2 <= 255 &&
 | 
						|
        0 <= d3 && d3 <= 255 && 0 <= d4 && d4 <= 255) {
 | 
						|
        struct sockaddr_in *sin;
 | 
						|
        sin = (struct sockaddr_in *)addr_ret;
 | 
						|
        sin->sin_addr.s_addr = htonl(
 | 
						|
            ((long) d1 << 24) | ((long) d2 << 16) |
 | 
						|
            ((long) d3 << 8) | ((long) d4 << 0));
 | 
						|
        sin->sin_family = AF_INET;
 | 
						|
#ifdef HAVE_SOCKADDR_SA_LEN
 | 
						|
        sin->sin_len = sizeof(*sin);
 | 
						|
#endif
 | 
						|
        return 4;
 | 
						|
    }
 | 
						|
    memset(&hints, 0, sizeof(hints));
 | 
						|
    hints.ai_family = af;
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    ACQUIRE_GETADDRINFO_LOCK
 | 
						|
    error = getaddrinfo(name, NULL, &hints, &res);
 | 
						|
#if defined(__digital__) && defined(__unix__)
 | 
						|
    if (error == EAI_NONAME && af == AF_UNSPEC) {
 | 
						|
        /* On Tru64 V5.1, numeric-to-addr conversion fails
 | 
						|
           if no address family is given. Assume IPv4 for now.*/
 | 
						|
        hints.ai_family = AF_INET;
 | 
						|
        error = getaddrinfo(name, NULL, &hints, &res);
 | 
						|
    }
 | 
						|
#endif
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    RELEASE_GETADDRINFO_LOCK  /* see comment in setipaddr() */
 | 
						|
    if (error) {
 | 
						|
        set_gaierror(error);
 | 
						|
        return -1;
 | 
						|
    }
 | 
						|
    if (res->ai_addrlen < addr_ret_size)
 | 
						|
        addr_ret_size = res->ai_addrlen;
 | 
						|
    memcpy((char *) addr_ret, res->ai_addr, addr_ret_size);
 | 
						|
    freeaddrinfo(res);
 | 
						|
    switch (addr_ret->sa_family) {
 | 
						|
    case AF_INET:
 | 
						|
        return 4;
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
    case AF_INET6:
 | 
						|
        return 16;
 | 
						|
#endif
 | 
						|
    default:
 | 
						|
        PyErr_SetString(socket_error, "unknown address family");
 | 
						|
        return -1;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* Create a string object representing an IP address.
 | 
						|
   This is always a string of the form 'dd.dd.dd.dd' (with variable
 | 
						|
   size numbers). */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
makeipaddr(struct sockaddr *addr, int addrlen)
 | 
						|
{
 | 
						|
    char buf[NI_MAXHOST];
 | 
						|
    int error;
 | 
						|
 | 
						|
    error = getnameinfo(addr, addrlen, buf, sizeof(buf), NULL, 0,
 | 
						|
        NI_NUMERICHOST);
 | 
						|
    if (error) {
 | 
						|
        set_gaierror(error);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    return PyUnicode_FromString(buf);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
#ifdef USE_BLUETOOTH
 | 
						|
/* Convert a string representation of a Bluetooth address into a numeric
 | 
						|
   address.  Returns the length (6), or raises an exception and returns -1 if
 | 
						|
   an error occurred. */
 | 
						|
 | 
						|
static int
 | 
						|
setbdaddr(char *name, bdaddr_t *bdaddr)
 | 
						|
{
 | 
						|
    unsigned int b0, b1, b2, b3, b4, b5;
 | 
						|
    char ch;
 | 
						|
    int n;
 | 
						|
 | 
						|
    n = sscanf(name, "%X:%X:%X:%X:%X:%X%c",
 | 
						|
               &b5, &b4, &b3, &b2, &b1, &b0, &ch);
 | 
						|
    if (n == 6 && (b0 | b1 | b2 | b3 | b4 | b5) < 256) {
 | 
						|
        bdaddr->b[0] = b0;
 | 
						|
        bdaddr->b[1] = b1;
 | 
						|
        bdaddr->b[2] = b2;
 | 
						|
        bdaddr->b[3] = b3;
 | 
						|
        bdaddr->b[4] = b4;
 | 
						|
        bdaddr->b[5] = b5;
 | 
						|
        return 6;
 | 
						|
    } else {
 | 
						|
        PyErr_SetString(socket_error, "bad bluetooth address");
 | 
						|
        return -1;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
/* Create a string representation of the Bluetooth address.  This is always a
 | 
						|
   string of the form 'XX:XX:XX:XX:XX:XX' where XX is a two digit hexadecimal
 | 
						|
   value (zero padded if necessary). */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
makebdaddr(bdaddr_t *bdaddr)
 | 
						|
{
 | 
						|
    char buf[(6 * 2) + 5 + 1];
 | 
						|
 | 
						|
    sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
 | 
						|
        bdaddr->b[5], bdaddr->b[4], bdaddr->b[3],
 | 
						|
        bdaddr->b[2], bdaddr->b[1], bdaddr->b[0]);
 | 
						|
    return PyUnicode_FromString(buf);
 | 
						|
}
 | 
						|
#endif
 | 
						|
 | 
						|
 | 
						|
/* Create an object representing the given socket address,
 | 
						|
   suitable for passing it back to bind(), connect() etc.
 | 
						|
   The family field of the sockaddr structure is inspected
 | 
						|
   to determine what kind of address it really is. */
 | 
						|
 | 
						|
/*ARGSUSED*/
 | 
						|
static PyObject *
 | 
						|
makesockaddr(SOCKET_T sockfd, struct sockaddr *addr, size_t addrlen, int proto)
 | 
						|
{
 | 
						|
    if (addrlen == 0) {
 | 
						|
        /* No address -- may be recvfrom() from known socket */
 | 
						|
        Py_INCREF(Py_None);
 | 
						|
        return Py_None;
 | 
						|
    }
 | 
						|
 | 
						|
    switch (addr->sa_family) {
 | 
						|
 | 
						|
    case AF_INET:
 | 
						|
    {
 | 
						|
        struct sockaddr_in *a;
 | 
						|
        PyObject *addrobj = makeipaddr(addr, sizeof(*a));
 | 
						|
        PyObject *ret = NULL;
 | 
						|
        if (addrobj) {
 | 
						|
            a = (struct sockaddr_in *)addr;
 | 
						|
            ret = Py_BuildValue("Oi", addrobj, ntohs(a->sin_port));
 | 
						|
            Py_DECREF(addrobj);
 | 
						|
        }
 | 
						|
        return ret;
 | 
						|
    }
 | 
						|
 | 
						|
#if defined(AF_UNIX)
 | 
						|
    case AF_UNIX:
 | 
						|
    {
 | 
						|
        struct sockaddr_un *a = (struct sockaddr_un *) addr;
 | 
						|
#ifdef linux
 | 
						|
        if (a->sun_path[0] == 0) {  /* Linux abstract namespace */
 | 
						|
            addrlen -= offsetof(struct sockaddr_un, sun_path);
 | 
						|
            return PyBytes_FromStringAndSize(a->sun_path, addrlen);
 | 
						|
        }
 | 
						|
        else
 | 
						|
#endif /* linux */
 | 
						|
        {
 | 
						|
            /* regular NULL-terminated string */
 | 
						|
            return PyUnicode_FromString(a->sun_path);
 | 
						|
        }
 | 
						|
    }
 | 
						|
#endif /* AF_UNIX */
 | 
						|
 | 
						|
#if defined(AF_NETLINK)
 | 
						|
       case AF_NETLINK:
 | 
						|
       {
 | 
						|
           struct sockaddr_nl *a = (struct sockaddr_nl *) addr;
 | 
						|
           return Py_BuildValue("II", a->nl_pid, a->nl_groups);
 | 
						|
       }
 | 
						|
#endif /* AF_NETLINK */
 | 
						|
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
    case AF_INET6:
 | 
						|
    {
 | 
						|
        struct sockaddr_in6 *a;
 | 
						|
        PyObject *addrobj = makeipaddr(addr, sizeof(*a));
 | 
						|
        PyObject *ret = NULL;
 | 
						|
        if (addrobj) {
 | 
						|
            a = (struct sockaddr_in6 *)addr;
 | 
						|
            ret = Py_BuildValue("Oiii",
 | 
						|
                                addrobj,
 | 
						|
                                ntohs(a->sin6_port),
 | 
						|
                                a->sin6_flowinfo,
 | 
						|
                                a->sin6_scope_id);
 | 
						|
            Py_DECREF(addrobj);
 | 
						|
        }
 | 
						|
        return ret;
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef USE_BLUETOOTH
 | 
						|
    case AF_BLUETOOTH:
 | 
						|
        switch (proto) {
 | 
						|
 | 
						|
        case BTPROTO_L2CAP:
 | 
						|
        {
 | 
						|
            struct sockaddr_l2 *a = (struct sockaddr_l2 *) addr;
 | 
						|
            PyObject *addrobj = makebdaddr(&_BT_L2_MEMB(a, bdaddr));
 | 
						|
            PyObject *ret = NULL;
 | 
						|
            if (addrobj) {
 | 
						|
                ret = Py_BuildValue("Oi",
 | 
						|
                                    addrobj,
 | 
						|
                                    _BT_L2_MEMB(a, psm));
 | 
						|
                Py_DECREF(addrobj);
 | 
						|
            }
 | 
						|
            return ret;
 | 
						|
        }
 | 
						|
 | 
						|
        case BTPROTO_RFCOMM:
 | 
						|
        {
 | 
						|
            struct sockaddr_rc *a = (struct sockaddr_rc *) addr;
 | 
						|
            PyObject *addrobj = makebdaddr(&_BT_RC_MEMB(a, bdaddr));
 | 
						|
            PyObject *ret = NULL;
 | 
						|
            if (addrobj) {
 | 
						|
                ret = Py_BuildValue("Oi",
 | 
						|
                                    addrobj,
 | 
						|
                                    _BT_RC_MEMB(a, channel));
 | 
						|
                Py_DECREF(addrobj);
 | 
						|
            }
 | 
						|
            return ret;
 | 
						|
        }
 | 
						|
 | 
						|
        case BTPROTO_HCI:
 | 
						|
        {
 | 
						|
            struct sockaddr_hci *a = (struct sockaddr_hci *) addr;
 | 
						|
#if defined(__NetBSD__) || defined(__DragonFly__)
 | 
						|
            return makebdaddr(&_BT_HCI_MEMB(a, bdaddr));
 | 
						|
#else
 | 
						|
            PyObject *ret = NULL;
 | 
						|
            ret = Py_BuildValue("i", _BT_HCI_MEMB(a, dev));
 | 
						|
            return ret;
 | 
						|
#endif
 | 
						|
        }
 | 
						|
 | 
						|
#if !defined(__FreeBSD__)
 | 
						|
        case BTPROTO_SCO:
 | 
						|
        {
 | 
						|
            struct sockaddr_sco *a = (struct sockaddr_sco *) addr;
 | 
						|
            return makebdaddr(&_BT_SCO_MEMB(a, bdaddr));
 | 
						|
        }
 | 
						|
#endif
 | 
						|
 | 
						|
        default:
 | 
						|
            PyErr_SetString(PyExc_ValueError,
 | 
						|
                            "Unknown Bluetooth protocol");
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(HAVE_NETPACKET_PACKET_H) && defined(SIOCGIFNAME)
 | 
						|
    case AF_PACKET:
 | 
						|
    {
 | 
						|
        struct sockaddr_ll *a = (struct sockaddr_ll *)addr;
 | 
						|
        char *ifname = "";
 | 
						|
        struct ifreq ifr;
 | 
						|
        /* need to look up interface name give index */
 | 
						|
        if (a->sll_ifindex) {
 | 
						|
            ifr.ifr_ifindex = a->sll_ifindex;
 | 
						|
            if (ioctl(sockfd, SIOCGIFNAME, &ifr) == 0)
 | 
						|
                ifname = ifr.ifr_name;
 | 
						|
        }
 | 
						|
        return Py_BuildValue("shbhy#",
 | 
						|
                             ifname,
 | 
						|
                             ntohs(a->sll_protocol),
 | 
						|
                             a->sll_pkttype,
 | 
						|
                             a->sll_hatype,
 | 
						|
                             a->sll_addr,
 | 
						|
                             a->sll_halen);
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_LINUX_TIPC_H
 | 
						|
    case AF_TIPC:
 | 
						|
    {
 | 
						|
        struct sockaddr_tipc *a = (struct sockaddr_tipc *) addr;
 | 
						|
        if (a->addrtype == TIPC_ADDR_NAMESEQ) {
 | 
						|
            return Py_BuildValue("IIIII",
 | 
						|
                            a->addrtype,
 | 
						|
                            a->addr.nameseq.type,
 | 
						|
                            a->addr.nameseq.lower,
 | 
						|
                            a->addr.nameseq.upper,
 | 
						|
                            a->scope);
 | 
						|
        } else if (a->addrtype == TIPC_ADDR_NAME) {
 | 
						|
            return Py_BuildValue("IIIII",
 | 
						|
                            a->addrtype,
 | 
						|
                            a->addr.name.name.type,
 | 
						|
                            a->addr.name.name.instance,
 | 
						|
                            a->addr.name.name.instance,
 | 
						|
                            a->scope);
 | 
						|
        } else if (a->addrtype == TIPC_ADDR_ID) {
 | 
						|
            return Py_BuildValue("IIIII",
 | 
						|
                            a->addrtype,
 | 
						|
                            a->addr.id.node,
 | 
						|
                            a->addr.id.ref,
 | 
						|
                            0,
 | 
						|
                            a->scope);
 | 
						|
        } else {
 | 
						|
            PyErr_SetString(PyExc_ValueError,
 | 
						|
                            "Invalid address type");
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_LINUX_CAN_H
 | 
						|
    case AF_CAN:
 | 
						|
    {
 | 
						|
        struct sockaddr_can *a = (struct sockaddr_can *)addr;
 | 
						|
        char *ifname = "";
 | 
						|
        struct ifreq ifr;
 | 
						|
        /* need to look up interface name given index */
 | 
						|
        if (a->can_ifindex) {
 | 
						|
            ifr.ifr_ifindex = a->can_ifindex;
 | 
						|
            if (ioctl(sockfd, SIOCGIFNAME, &ifr) == 0)
 | 
						|
                ifname = ifr.ifr_name;
 | 
						|
        }
 | 
						|
 | 
						|
        return Py_BuildValue("O&h", PyUnicode_DecodeFSDefault,
 | 
						|
                                    ifname,
 | 
						|
                                    a->can_family);
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
    /* More cases here... */
 | 
						|
 | 
						|
    default:
 | 
						|
        /* If we don't know the address family, don't raise an
 | 
						|
           exception -- return it as an (int, bytes) tuple. */
 | 
						|
        return Py_BuildValue("iy#",
 | 
						|
                             addr->sa_family,
 | 
						|
                             addr->sa_data,
 | 
						|
                             sizeof(addr->sa_data));
 | 
						|
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* Parse a socket address argument according to the socket object's
 | 
						|
   address family.  Return 1 if the address was in the proper format,
 | 
						|
   0 of not.  The address is returned through addr_ret, its length
 | 
						|
   through len_ret. */
 | 
						|
 | 
						|
static int
 | 
						|
getsockaddrarg(PySocketSockObject *s, PyObject *args,
 | 
						|
               struct sockaddr *addr_ret, int *len_ret)
 | 
						|
{
 | 
						|
    switch (s->sock_family) {
 | 
						|
 | 
						|
#if defined(AF_UNIX)
 | 
						|
    case AF_UNIX:
 | 
						|
    {
 | 
						|
        struct sockaddr_un* addr;
 | 
						|
        char *path;
 | 
						|
        int len;
 | 
						|
        if (!PyArg_Parse(args, "s#", &path, &len))
 | 
						|
            return 0;
 | 
						|
 | 
						|
        addr = (struct sockaddr_un*)addr_ret;
 | 
						|
#ifdef linux
 | 
						|
        if (len > 0 && path[0] == 0) {
 | 
						|
            /* Linux abstract namespace extension */
 | 
						|
            if (len > sizeof addr->sun_path) {
 | 
						|
                PyErr_SetString(socket_error,
 | 
						|
                                "AF_UNIX path too long");
 | 
						|
                return 0;
 | 
						|
            }
 | 
						|
        }
 | 
						|
        else
 | 
						|
#endif /* linux */
 | 
						|
        {
 | 
						|
            /* regular NULL-terminated string */
 | 
						|
            if (len >= sizeof addr->sun_path) {
 | 
						|
                PyErr_SetString(socket_error,
 | 
						|
                                "AF_UNIX path too long");
 | 
						|
                return 0;
 | 
						|
            }
 | 
						|
            addr->sun_path[len] = 0;
 | 
						|
        }
 | 
						|
        addr->sun_family = s->sock_family;
 | 
						|
        memcpy(addr->sun_path, path, len);
 | 
						|
#if defined(PYOS_OS2)
 | 
						|
        *len_ret = sizeof(*addr);
 | 
						|
#else
 | 
						|
        *len_ret = len + offsetof(struct sockaddr_un, sun_path);
 | 
						|
#endif
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
#endif /* AF_UNIX */
 | 
						|
 | 
						|
#if defined(AF_NETLINK)
 | 
						|
    case AF_NETLINK:
 | 
						|
    {
 | 
						|
        struct sockaddr_nl* addr;
 | 
						|
        int pid, groups;
 | 
						|
        addr = (struct sockaddr_nl *)addr_ret;
 | 
						|
        if (!PyTuple_Check(args)) {
 | 
						|
            PyErr_Format(
 | 
						|
                PyExc_TypeError,
 | 
						|
                "getsockaddrarg: "
 | 
						|
                "AF_NETLINK address must be tuple, not %.500s",
 | 
						|
                Py_TYPE(args)->tp_name);
 | 
						|
            return 0;
 | 
						|
        }
 | 
						|
        if (!PyArg_ParseTuple(args, "II:getsockaddrarg", &pid, &groups))
 | 
						|
            return 0;
 | 
						|
        addr->nl_family = AF_NETLINK;
 | 
						|
        addr->nl_pid = pid;
 | 
						|
        addr->nl_groups = groups;
 | 
						|
        *len_ret = sizeof(*addr);
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
    case AF_INET:
 | 
						|
    {
 | 
						|
        struct sockaddr_in* addr;
 | 
						|
        char *host;
 | 
						|
        int port, result;
 | 
						|
        if (!PyTuple_Check(args)) {
 | 
						|
            PyErr_Format(
 | 
						|
                PyExc_TypeError,
 | 
						|
                "getsockaddrarg: "
 | 
						|
                "AF_INET address must be tuple, not %.500s",
 | 
						|
                Py_TYPE(args)->tp_name);
 | 
						|
            return 0;
 | 
						|
        }
 | 
						|
        if (!PyArg_ParseTuple(args, "eti:getsockaddrarg",
 | 
						|
                              "idna", &host, &port))
 | 
						|
            return 0;
 | 
						|
        addr=(struct sockaddr_in*)addr_ret;
 | 
						|
        result = setipaddr(host, (struct sockaddr *)addr,
 | 
						|
                           sizeof(*addr),  AF_INET);
 | 
						|
        PyMem_Free(host);
 | 
						|
        if (result < 0)
 | 
						|
            return 0;
 | 
						|
        if (port < 0 || port > 0xffff) {
 | 
						|
            PyErr_SetString(
 | 
						|
                PyExc_OverflowError,
 | 
						|
                "getsockaddrarg: port must be 0-65535.");
 | 
						|
            return 0;
 | 
						|
        }
 | 
						|
        addr->sin_family = AF_INET;
 | 
						|
        addr->sin_port = htons((short)port);
 | 
						|
        *len_ret = sizeof *addr;
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
    case AF_INET6:
 | 
						|
    {
 | 
						|
        struct sockaddr_in6* addr;
 | 
						|
        char *host;
 | 
						|
        int port, flowinfo, scope_id, result;
 | 
						|
        flowinfo = scope_id = 0;
 | 
						|
        if (!PyTuple_Check(args)) {
 | 
						|
            PyErr_Format(
 | 
						|
                PyExc_TypeError,
 | 
						|
                "getsockaddrarg: "
 | 
						|
                "AF_INET6 address must be tuple, not %.500s",
 | 
						|
                Py_TYPE(args)->tp_name);
 | 
						|
            return 0;
 | 
						|
        }
 | 
						|
        if (!PyArg_ParseTuple(args, "eti|ii",
 | 
						|
                              "idna", &host, &port, &flowinfo,
 | 
						|
                              &scope_id)) {
 | 
						|
            return 0;
 | 
						|
        }
 | 
						|
        addr = (struct sockaddr_in6*)addr_ret;
 | 
						|
        result = setipaddr(host, (struct sockaddr *)addr,
 | 
						|
                           sizeof(*addr), AF_INET6);
 | 
						|
        PyMem_Free(host);
 | 
						|
        if (result < 0)
 | 
						|
            return 0;
 | 
						|
        if (port < 0 || port > 0xffff) {
 | 
						|
            PyErr_SetString(
 | 
						|
                PyExc_OverflowError,
 | 
						|
                "getsockaddrarg: port must be 0-65535.");
 | 
						|
            return 0;
 | 
						|
        }
 | 
						|
        addr->sin6_family = s->sock_family;
 | 
						|
        addr->sin6_port = htons((short)port);
 | 
						|
        addr->sin6_flowinfo = flowinfo;
 | 
						|
        addr->sin6_scope_id = scope_id;
 | 
						|
        *len_ret = sizeof *addr;
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef USE_BLUETOOTH
 | 
						|
    case AF_BLUETOOTH:
 | 
						|
    {
 | 
						|
        switch (s->sock_proto) {
 | 
						|
        case BTPROTO_L2CAP:
 | 
						|
        {
 | 
						|
            struct sockaddr_l2 *addr;
 | 
						|
            char *straddr;
 | 
						|
 | 
						|
            addr = (struct sockaddr_l2 *)addr_ret;
 | 
						|
            memset(addr, 0, sizeof(struct sockaddr_l2));
 | 
						|
            _BT_L2_MEMB(addr, family) = AF_BLUETOOTH;
 | 
						|
            if (!PyArg_ParseTuple(args, "si", &straddr,
 | 
						|
                                  &_BT_L2_MEMB(addr, psm))) {
 | 
						|
                PyErr_SetString(socket_error, "getsockaddrarg: "
 | 
						|
                                "wrong format");
 | 
						|
                return 0;
 | 
						|
            }
 | 
						|
            if (setbdaddr(straddr, &_BT_L2_MEMB(addr, bdaddr)) < 0)
 | 
						|
                return 0;
 | 
						|
 | 
						|
            *len_ret = sizeof *addr;
 | 
						|
            return 1;
 | 
						|
        }
 | 
						|
        case BTPROTO_RFCOMM:
 | 
						|
        {
 | 
						|
            struct sockaddr_rc *addr;
 | 
						|
            char *straddr;
 | 
						|
 | 
						|
            addr = (struct sockaddr_rc *)addr_ret;
 | 
						|
            _BT_RC_MEMB(addr, family) = AF_BLUETOOTH;
 | 
						|
            if (!PyArg_ParseTuple(args, "si", &straddr,
 | 
						|
                                  &_BT_RC_MEMB(addr, channel))) {
 | 
						|
                PyErr_SetString(socket_error, "getsockaddrarg: "
 | 
						|
                                "wrong format");
 | 
						|
                return 0;
 | 
						|
            }
 | 
						|
            if (setbdaddr(straddr, &_BT_RC_MEMB(addr, bdaddr)) < 0)
 | 
						|
                return 0;
 | 
						|
 | 
						|
            *len_ret = sizeof *addr;
 | 
						|
            return 1;
 | 
						|
        }
 | 
						|
        case BTPROTO_HCI:
 | 
						|
        {
 | 
						|
            struct sockaddr_hci *addr = (struct sockaddr_hci *)addr_ret;
 | 
						|
#if defined(__NetBSD__) || defined(__DragonFly__)
 | 
						|
                        char *straddr = PyBytes_AS_STRING(args);
 | 
						|
 | 
						|
                        _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
 | 
						|
            if (straddr == NULL) {
 | 
						|
                PyErr_SetString(socket_error, "getsockaddrarg: "
 | 
						|
                    "wrong format");
 | 
						|
                return 0;
 | 
						|
            }
 | 
						|
            if (setbdaddr(straddr, &_BT_HCI_MEMB(addr, bdaddr)) < 0)
 | 
						|
                return 0;
 | 
						|
#else
 | 
						|
            _BT_HCI_MEMB(addr, family) = AF_BLUETOOTH;
 | 
						|
            if (!PyArg_ParseTuple(args, "i", &_BT_HCI_MEMB(addr, dev))) {
 | 
						|
                PyErr_SetString(socket_error, "getsockaddrarg: "
 | 
						|
                                "wrong format");
 | 
						|
                return 0;
 | 
						|
            }
 | 
						|
#endif
 | 
						|
            *len_ret = sizeof *addr;
 | 
						|
            return 1;
 | 
						|
        }
 | 
						|
#if !defined(__FreeBSD__)
 | 
						|
        case BTPROTO_SCO:
 | 
						|
        {
 | 
						|
            struct sockaddr_sco *addr;
 | 
						|
            char *straddr;
 | 
						|
 | 
						|
            addr = (struct sockaddr_sco *)addr_ret;
 | 
						|
            _BT_SCO_MEMB(addr, family) = AF_BLUETOOTH;
 | 
						|
            if (!PyBytes_Check(args)) {
 | 
						|
                PyErr_SetString(socket_error, "getsockaddrarg: "
 | 
						|
                                "wrong format");
 | 
						|
                return 0;
 | 
						|
            }
 | 
						|
            straddr = PyBytes_AS_STRING(args);
 | 
						|
            if (setbdaddr(straddr, &_BT_SCO_MEMB(addr, bdaddr)) < 0)
 | 
						|
                return 0;
 | 
						|
 | 
						|
            *len_ret = sizeof *addr;
 | 
						|
            return 1;
 | 
						|
        }
 | 
						|
#endif
 | 
						|
        default:
 | 
						|
            PyErr_SetString(socket_error, "getsockaddrarg: unknown Bluetooth protocol");
 | 
						|
            return 0;
 | 
						|
        }
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(HAVE_NETPACKET_PACKET_H) && defined(SIOCGIFINDEX)
 | 
						|
    case AF_PACKET:
 | 
						|
    {
 | 
						|
        struct sockaddr_ll* addr;
 | 
						|
        struct ifreq ifr;
 | 
						|
        char *interfaceName;
 | 
						|
        int protoNumber;
 | 
						|
        int hatype = 0;
 | 
						|
        int pkttype = 0;
 | 
						|
        char *haddr = NULL;
 | 
						|
        unsigned int halen = 0;
 | 
						|
 | 
						|
        if (!PyTuple_Check(args)) {
 | 
						|
            PyErr_Format(
 | 
						|
                PyExc_TypeError,
 | 
						|
                "getsockaddrarg: "
 | 
						|
                "AF_PACKET address must be tuple, not %.500s",
 | 
						|
                Py_TYPE(args)->tp_name);
 | 
						|
            return 0;
 | 
						|
        }
 | 
						|
        if (!PyArg_ParseTuple(args, "si|iiy#", &interfaceName,
 | 
						|
                              &protoNumber, &pkttype, &hatype,
 | 
						|
                              &haddr, &halen))
 | 
						|
            return 0;
 | 
						|
        strncpy(ifr.ifr_name, interfaceName, sizeof(ifr.ifr_name));
 | 
						|
        ifr.ifr_name[(sizeof(ifr.ifr_name))-1] = '\0';
 | 
						|
        if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
 | 
						|
            s->errorhandler();
 | 
						|
            return 0;
 | 
						|
        }
 | 
						|
        if (halen > 8) {
 | 
						|
          PyErr_SetString(PyExc_ValueError,
 | 
						|
                          "Hardware address must be 8 bytes or less");
 | 
						|
          return 0;
 | 
						|
        }
 | 
						|
        if (protoNumber < 0 || protoNumber > 0xffff) {
 | 
						|
            PyErr_SetString(
 | 
						|
                PyExc_OverflowError,
 | 
						|
                "getsockaddrarg: protoNumber must be 0-65535.");
 | 
						|
            return 0;
 | 
						|
        }
 | 
						|
        addr = (struct sockaddr_ll*)addr_ret;
 | 
						|
        addr->sll_family = AF_PACKET;
 | 
						|
        addr->sll_protocol = htons((short)protoNumber);
 | 
						|
        addr->sll_ifindex = ifr.ifr_ifindex;
 | 
						|
        addr->sll_pkttype = pkttype;
 | 
						|
        addr->sll_hatype = hatype;
 | 
						|
        if (halen != 0) {
 | 
						|
          memcpy(&addr->sll_addr, haddr, halen);
 | 
						|
        }
 | 
						|
        addr->sll_halen = halen;
 | 
						|
        *len_ret = sizeof *addr;
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_LINUX_TIPC_H
 | 
						|
    case AF_TIPC:
 | 
						|
    {
 | 
						|
        unsigned int atype, v1, v2, v3;
 | 
						|
        unsigned int scope = TIPC_CLUSTER_SCOPE;
 | 
						|
        struct sockaddr_tipc *addr;
 | 
						|
 | 
						|
        if (!PyTuple_Check(args)) {
 | 
						|
            PyErr_Format(
 | 
						|
                PyExc_TypeError,
 | 
						|
                "getsockaddrarg: "
 | 
						|
                "AF_TIPC address must be tuple, not %.500s",
 | 
						|
                Py_TYPE(args)->tp_name);
 | 
						|
            return 0;
 | 
						|
        }
 | 
						|
 | 
						|
        if (!PyArg_ParseTuple(args,
 | 
						|
                                "IIII|I;Invalid TIPC address format",
 | 
						|
                                &atype, &v1, &v2, &v3, &scope))
 | 
						|
            return 0;
 | 
						|
 | 
						|
        addr = (struct sockaddr_tipc *) addr_ret;
 | 
						|
        memset(addr, 0, sizeof(struct sockaddr_tipc));
 | 
						|
 | 
						|
        addr->family = AF_TIPC;
 | 
						|
        addr->scope = scope;
 | 
						|
        addr->addrtype = atype;
 | 
						|
 | 
						|
        if (atype == TIPC_ADDR_NAMESEQ) {
 | 
						|
            addr->addr.nameseq.type = v1;
 | 
						|
            addr->addr.nameseq.lower = v2;
 | 
						|
            addr->addr.nameseq.upper = v3;
 | 
						|
        } else if (atype == TIPC_ADDR_NAME) {
 | 
						|
            addr->addr.name.name.type = v1;
 | 
						|
            addr->addr.name.name.instance = v2;
 | 
						|
        } else if (atype == TIPC_ADDR_ID) {
 | 
						|
            addr->addr.id.node = v1;
 | 
						|
            addr->addr.id.ref = v2;
 | 
						|
        } else {
 | 
						|
            /* Shouldn't happen */
 | 
						|
            PyErr_SetString(PyExc_TypeError, "Invalid address type");
 | 
						|
            return 0;
 | 
						|
        }
 | 
						|
 | 
						|
        *len_ret = sizeof(*addr);
 | 
						|
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_LINUX_CAN_H
 | 
						|
    case AF_CAN:
 | 
						|
        switch (s->sock_proto) {
 | 
						|
        case CAN_RAW:
 | 
						|
        {
 | 
						|
            struct sockaddr_can *addr;
 | 
						|
            PyObject *interfaceName;
 | 
						|
            struct ifreq ifr;
 | 
						|
            addr = (struct sockaddr_can *)addr_ret;
 | 
						|
            Py_ssize_t len;
 | 
						|
 | 
						|
            if (!PyArg_ParseTuple(args, "O&", PyUnicode_FSConverter,
 | 
						|
                                              &interfaceName))
 | 
						|
                return 0;
 | 
						|
 | 
						|
            len = PyBytes_GET_SIZE(interfaceName);
 | 
						|
 | 
						|
            if (len == 0) {
 | 
						|
                ifr.ifr_ifindex = 0;
 | 
						|
            } else if (len < sizeof(ifr.ifr_name)) {
 | 
						|
                strcpy(ifr.ifr_name, PyBytes_AS_STRING(interfaceName));
 | 
						|
                if (ioctl(s->sock_fd, SIOCGIFINDEX, &ifr) < 0) {
 | 
						|
                    s->errorhandler();
 | 
						|
                    Py_DECREF(interfaceName);
 | 
						|
                    return 0;
 | 
						|
                }
 | 
						|
            } else {
 | 
						|
                PyErr_SetString(socket_error,
 | 
						|
                                "AF_CAN interface name too long");
 | 
						|
                Py_DECREF(interfaceName);
 | 
						|
                return 0;
 | 
						|
            }
 | 
						|
 | 
						|
            addr->can_family = AF_CAN;
 | 
						|
            addr->can_ifindex = ifr.ifr_ifindex;
 | 
						|
 | 
						|
            *len_ret = sizeof(*addr);
 | 
						|
            Py_DECREF(interfaceName);
 | 
						|
            return 1;
 | 
						|
        }
 | 
						|
        default:
 | 
						|
            PyErr_SetString(socket_error,
 | 
						|
                            "getsockaddrarg: unsupported CAN protocol");
 | 
						|
            return 0;
 | 
						|
        }
 | 
						|
#endif
 | 
						|
 | 
						|
    /* More cases here... */
 | 
						|
 | 
						|
    default:
 | 
						|
        PyErr_SetString(socket_error, "getsockaddrarg: bad family");
 | 
						|
        return 0;
 | 
						|
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* Get the address length according to the socket object's address family.
 | 
						|
   Return 1 if the family is known, 0 otherwise.  The length is returned
 | 
						|
   through len_ret. */
 | 
						|
 | 
						|
static int
 | 
						|
getsockaddrlen(PySocketSockObject *s, socklen_t *len_ret)
 | 
						|
{
 | 
						|
    switch (s->sock_family) {
 | 
						|
 | 
						|
#if defined(AF_UNIX)
 | 
						|
    case AF_UNIX:
 | 
						|
    {
 | 
						|
        *len_ret = sizeof (struct sockaddr_un);
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
#endif /* AF_UNIX */
 | 
						|
#if defined(AF_NETLINK)
 | 
						|
       case AF_NETLINK:
 | 
						|
       {
 | 
						|
           *len_ret = sizeof (struct sockaddr_nl);
 | 
						|
           return 1;
 | 
						|
       }
 | 
						|
#endif
 | 
						|
 | 
						|
    case AF_INET:
 | 
						|
    {
 | 
						|
        *len_ret = sizeof (struct sockaddr_in);
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
    case AF_INET6:
 | 
						|
    {
 | 
						|
        *len_ret = sizeof (struct sockaddr_in6);
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef USE_BLUETOOTH
 | 
						|
    case AF_BLUETOOTH:
 | 
						|
    {
 | 
						|
        switch(s->sock_proto)
 | 
						|
        {
 | 
						|
 | 
						|
        case BTPROTO_L2CAP:
 | 
						|
            *len_ret = sizeof (struct sockaddr_l2);
 | 
						|
            return 1;
 | 
						|
        case BTPROTO_RFCOMM:
 | 
						|
            *len_ret = sizeof (struct sockaddr_rc);
 | 
						|
            return 1;
 | 
						|
        case BTPROTO_HCI:
 | 
						|
            *len_ret = sizeof (struct sockaddr_hci);
 | 
						|
            return 1;
 | 
						|
#if !defined(__FreeBSD__)
 | 
						|
        case BTPROTO_SCO:
 | 
						|
            *len_ret = sizeof (struct sockaddr_sco);
 | 
						|
            return 1;
 | 
						|
#endif
 | 
						|
        default:
 | 
						|
            PyErr_SetString(socket_error, "getsockaddrlen: "
 | 
						|
                            "unknown BT protocol");
 | 
						|
            return 0;
 | 
						|
 | 
						|
        }
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_NETPACKET_PACKET_H
 | 
						|
    case AF_PACKET:
 | 
						|
    {
 | 
						|
        *len_ret = sizeof (struct sockaddr_ll);
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_LINUX_TIPC_H
 | 
						|
    case AF_TIPC:
 | 
						|
    {
 | 
						|
        *len_ret = sizeof (struct sockaddr_tipc);
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_LINUX_CAN_H
 | 
						|
    case AF_CAN:
 | 
						|
    {
 | 
						|
        *len_ret = sizeof (struct sockaddr_can);
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
    /* More cases here... */
 | 
						|
 | 
						|
    default:
 | 
						|
        PyErr_SetString(socket_error, "getsockaddrlen: bad family");
 | 
						|
        return 0;
 | 
						|
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* Support functions for the sendmsg() and recvmsg[_into]() methods.
 | 
						|
   Currently, these methods are only compiled if the RFC 2292/3542
 | 
						|
   CMSG_LEN() macro is available.  Older systems seem to have used
 | 
						|
   sizeof(struct cmsghdr) + (length) where CMSG_LEN() is used now, so
 | 
						|
   it may be possible to define CMSG_LEN() that way if it's not
 | 
						|
   provided.  Some architectures might need extra padding after the
 | 
						|
   cmsghdr, however, and CMSG_LEN() would have to take account of
 | 
						|
   this. */
 | 
						|
#ifdef CMSG_LEN
 | 
						|
/* If length is in range, set *result to CMSG_LEN(length) and return
 | 
						|
   true; otherwise, return false. */
 | 
						|
static int
 | 
						|
get_CMSG_LEN(size_t length, size_t *result)
 | 
						|
{
 | 
						|
    size_t tmp;
 | 
						|
 | 
						|
    if (length > (SOCKLEN_T_LIMIT - CMSG_LEN(0)))
 | 
						|
        return 0;
 | 
						|
    tmp = CMSG_LEN(length);
 | 
						|
    if (tmp > SOCKLEN_T_LIMIT || tmp < length)
 | 
						|
        return 0;
 | 
						|
    *result = tmp;
 | 
						|
    return 1;
 | 
						|
}
 | 
						|
 | 
						|
#ifdef CMSG_SPACE
 | 
						|
/* If length is in range, set *result to CMSG_SPACE(length) and return
 | 
						|
   true; otherwise, return false. */
 | 
						|
static int
 | 
						|
get_CMSG_SPACE(size_t length, size_t *result)
 | 
						|
{
 | 
						|
    size_t tmp;
 | 
						|
 | 
						|
    /* Use CMSG_SPACE(1) here in order to take account of the padding
 | 
						|
       necessary before *and* after the data. */
 | 
						|
    if (length > (SOCKLEN_T_LIMIT - CMSG_SPACE(1)))
 | 
						|
        return 0;
 | 
						|
    tmp = CMSG_SPACE(length);
 | 
						|
    if (tmp > SOCKLEN_T_LIMIT || tmp < length)
 | 
						|
        return 0;
 | 
						|
    *result = tmp;
 | 
						|
    return 1;
 | 
						|
}
 | 
						|
#endif
 | 
						|
 | 
						|
/* Return true iff msg->msg_controllen is valid, cmsgh is a valid
 | 
						|
   pointer in msg->msg_control with at least "space" bytes after it,
 | 
						|
   and its cmsg_len member inside the buffer. */
 | 
						|
static int
 | 
						|
cmsg_min_space(struct msghdr *msg, struct cmsghdr *cmsgh, size_t space)
 | 
						|
{
 | 
						|
    size_t cmsg_offset;
 | 
						|
    static const size_t cmsg_len_end = (offsetof(struct cmsghdr, cmsg_len) +
 | 
						|
                                        sizeof(cmsgh->cmsg_len));
 | 
						|
 | 
						|
    /* Note that POSIX allows msg_controllen to be of signed type. */
 | 
						|
    if (cmsgh == NULL || msg->msg_control == NULL || msg->msg_controllen < 0)
 | 
						|
        return 0;
 | 
						|
    if (space < cmsg_len_end)
 | 
						|
        space = cmsg_len_end;
 | 
						|
    cmsg_offset = (char *)cmsgh - (char *)msg->msg_control;
 | 
						|
    return (cmsg_offset <= (size_t)-1 - space &&
 | 
						|
            cmsg_offset + space <= msg->msg_controllen);
 | 
						|
}
 | 
						|
 | 
						|
/* If pointer CMSG_DATA(cmsgh) is in buffer msg->msg_control, set
 | 
						|
   *space to number of bytes following it in the buffer and return
 | 
						|
   true; otherwise, return false.  Assumes cmsgh, msg->msg_control and
 | 
						|
   msg->msg_controllen are valid. */
 | 
						|
static int
 | 
						|
get_cmsg_data_space(struct msghdr *msg, struct cmsghdr *cmsgh, size_t *space)
 | 
						|
{
 | 
						|
    size_t data_offset;
 | 
						|
    char *data_ptr;
 | 
						|
 | 
						|
    if ((data_ptr = (char *)CMSG_DATA(cmsgh)) == NULL)
 | 
						|
        return 0;
 | 
						|
    data_offset = data_ptr - (char *)msg->msg_control;
 | 
						|
    if (data_offset > msg->msg_controllen)
 | 
						|
        return 0;
 | 
						|
    *space = msg->msg_controllen - data_offset;
 | 
						|
    return 1;
 | 
						|
}
 | 
						|
 | 
						|
/* If cmsgh is invalid or not contained in the buffer pointed to by
 | 
						|
   msg->msg_control, return -1.  If cmsgh is valid and its associated
 | 
						|
   data is entirely contained in the buffer, set *data_len to the
 | 
						|
   length of the associated data and return 0.  If only part of the
 | 
						|
   associated data is contained in the buffer but cmsgh is otherwise
 | 
						|
   valid, set *data_len to the length contained in the buffer and
 | 
						|
   return 1. */
 | 
						|
static int
 | 
						|
get_cmsg_data_len(struct msghdr *msg, struct cmsghdr *cmsgh, size_t *data_len)
 | 
						|
{
 | 
						|
    size_t space, cmsg_data_len;
 | 
						|
 | 
						|
    if (!cmsg_min_space(msg, cmsgh, CMSG_LEN(0)) ||
 | 
						|
        cmsgh->cmsg_len < CMSG_LEN(0))
 | 
						|
        return -1;
 | 
						|
    cmsg_data_len = cmsgh->cmsg_len - CMSG_LEN(0);
 | 
						|
    if (!get_cmsg_data_space(msg, cmsgh, &space))
 | 
						|
        return -1;
 | 
						|
    if (space >= cmsg_data_len) {
 | 
						|
        *data_len = cmsg_data_len;
 | 
						|
        return 0;
 | 
						|
    }
 | 
						|
    *data_len = space;
 | 
						|
    return 1;
 | 
						|
}
 | 
						|
#endif    /* CMSG_LEN */
 | 
						|
 | 
						|
 | 
						|
/* s._accept() -> (fd, address) */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_accept(PySocketSockObject *s)
 | 
						|
{
 | 
						|
    sock_addr_t addrbuf;
 | 
						|
    SOCKET_T newfd = INVALID_SOCKET;
 | 
						|
    socklen_t addrlen;
 | 
						|
    PyObject *sock = NULL;
 | 
						|
    PyObject *addr = NULL;
 | 
						|
    PyObject *res = NULL;
 | 
						|
    int timeout;
 | 
						|
    if (!getsockaddrlen(s, &addrlen))
 | 
						|
        return NULL;
 | 
						|
    memset(&addrbuf, 0, addrlen);
 | 
						|
 | 
						|
    if (!IS_SELECTABLE(s))
 | 
						|
        return select_error();
 | 
						|
 | 
						|
    BEGIN_SELECT_LOOP(s)
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    timeout = internal_select_ex(s, 0, interval);
 | 
						|
    if (!timeout) {
 | 
						|
        newfd = accept(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
 | 
						|
    }
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
 | 
						|
    if (timeout == 1) {
 | 
						|
        PyErr_SetString(socket_timeout, "timed out");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    END_SELECT_LOOP(s)
 | 
						|
 | 
						|
    if (newfd == INVALID_SOCKET)
 | 
						|
        return s->errorhandler();
 | 
						|
 | 
						|
    sock = PyLong_FromSocket_t(newfd);
 | 
						|
    if (sock == NULL) {
 | 
						|
        SOCKETCLOSE(newfd);
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
 | 
						|
    addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
 | 
						|
                        addrlen, s->sock_proto);
 | 
						|
    if (addr == NULL)
 | 
						|
        goto finally;
 | 
						|
 | 
						|
    res = PyTuple_Pack(2, sock, addr);
 | 
						|
 | 
						|
finally:
 | 
						|
    Py_XDECREF(sock);
 | 
						|
    Py_XDECREF(addr);
 | 
						|
    return res;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(accept_doc,
 | 
						|
"_accept() -> (integer, address info)\n\
 | 
						|
\n\
 | 
						|
Wait for an incoming connection.  Return a new socket file descriptor\n\
 | 
						|
representing the connection, and the address of the client.\n\
 | 
						|
For IP sockets, the address info is a pair (hostaddr, port).");
 | 
						|
 | 
						|
/* s.setblocking(flag) method.  Argument:
 | 
						|
   False -- non-blocking mode; same as settimeout(0)
 | 
						|
   True -- blocking mode; same as settimeout(None)
 | 
						|
*/
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_setblocking(PySocketSockObject *s, PyObject *arg)
 | 
						|
{
 | 
						|
    int block;
 | 
						|
 | 
						|
    block = PyLong_AsLong(arg);
 | 
						|
    if (block == -1 && PyErr_Occurred())
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    s->sock_timeout = block ? -1.0 : 0.0;
 | 
						|
    internal_setblocking(s, block);
 | 
						|
 | 
						|
    Py_INCREF(Py_None);
 | 
						|
    return Py_None;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(setblocking_doc,
 | 
						|
"setblocking(flag)\n\
 | 
						|
\n\
 | 
						|
Set the socket to blocking (flag is true) or non-blocking (false).\n\
 | 
						|
setblocking(True) is equivalent to settimeout(None);\n\
 | 
						|
setblocking(False) is equivalent to settimeout(0.0).");
 | 
						|
 | 
						|
/* s.settimeout(timeout) method.  Argument:
 | 
						|
   None -- no timeout, blocking mode; same as setblocking(True)
 | 
						|
   0.0  -- non-blocking mode; same as setblocking(False)
 | 
						|
   > 0  -- timeout mode; operations time out after timeout seconds
 | 
						|
   < 0  -- illegal; raises an exception
 | 
						|
*/
 | 
						|
static PyObject *
 | 
						|
sock_settimeout(PySocketSockObject *s, PyObject *arg)
 | 
						|
{
 | 
						|
    double timeout;
 | 
						|
 | 
						|
    if (arg == Py_None)
 | 
						|
        timeout = -1.0;
 | 
						|
    else {
 | 
						|
        timeout = PyFloat_AsDouble(arg);
 | 
						|
        if (timeout < 0.0) {
 | 
						|
            if (!PyErr_Occurred())
 | 
						|
                PyErr_SetString(PyExc_ValueError,
 | 
						|
                                "Timeout value out of range");
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    s->sock_timeout = timeout;
 | 
						|
    internal_setblocking(s, timeout < 0.0);
 | 
						|
 | 
						|
    Py_INCREF(Py_None);
 | 
						|
    return Py_None;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(settimeout_doc,
 | 
						|
"settimeout(timeout)\n\
 | 
						|
\n\
 | 
						|
Set a timeout on socket operations.  'timeout' can be a float,\n\
 | 
						|
giving in seconds, or None.  Setting a timeout of None disables\n\
 | 
						|
the timeout feature and is equivalent to setblocking(1).\n\
 | 
						|
Setting a timeout of zero is the same as setblocking(0).");
 | 
						|
 | 
						|
/* s.gettimeout() method.
 | 
						|
   Returns the timeout associated with a socket. */
 | 
						|
static PyObject *
 | 
						|
sock_gettimeout(PySocketSockObject *s)
 | 
						|
{
 | 
						|
    if (s->sock_timeout < 0.0) {
 | 
						|
        Py_INCREF(Py_None);
 | 
						|
        return Py_None;
 | 
						|
    }
 | 
						|
    else
 | 
						|
        return PyFloat_FromDouble(s->sock_timeout);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(gettimeout_doc,
 | 
						|
"gettimeout() -> timeout\n\
 | 
						|
\n\
 | 
						|
Returns the timeout in seconds (float) associated with socket \n\
 | 
						|
operations. A timeout of None indicates that timeouts on socket \n\
 | 
						|
operations are disabled.");
 | 
						|
 | 
						|
/* s.setsockopt() method.
 | 
						|
   With an integer third argument, sets an integer option.
 | 
						|
   With a string third argument, sets an option from a buffer;
 | 
						|
   use optional built-in module 'struct' to encode the string. */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_setsockopt(PySocketSockObject *s, PyObject *args)
 | 
						|
{
 | 
						|
    int level;
 | 
						|
    int optname;
 | 
						|
    int res;
 | 
						|
    char *buf;
 | 
						|
    int buflen;
 | 
						|
    int flag;
 | 
						|
 | 
						|
    if (PyArg_ParseTuple(args, "iii:setsockopt",
 | 
						|
                         &level, &optname, &flag)) {
 | 
						|
        buf = (char *) &flag;
 | 
						|
        buflen = sizeof flag;
 | 
						|
    }
 | 
						|
    else {
 | 
						|
        PyErr_Clear();
 | 
						|
        if (!PyArg_ParseTuple(args, "iiy#:setsockopt",
 | 
						|
                              &level, &optname, &buf, &buflen))
 | 
						|
            return NULL;
 | 
						|
    }
 | 
						|
    res = setsockopt(s->sock_fd, level, optname, (void *)buf, buflen);
 | 
						|
    if (res < 0)
 | 
						|
        return s->errorhandler();
 | 
						|
    Py_INCREF(Py_None);
 | 
						|
    return Py_None;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(setsockopt_doc,
 | 
						|
"setsockopt(level, option, value)\n\
 | 
						|
\n\
 | 
						|
Set a socket option.  See the Unix manual for level and option.\n\
 | 
						|
The value argument can either be an integer or a string.");
 | 
						|
 | 
						|
 | 
						|
/* s.getsockopt() method.
 | 
						|
   With two arguments, retrieves an integer option.
 | 
						|
   With a third integer argument, retrieves a string buffer of that size;
 | 
						|
   use optional built-in module 'struct' to decode the string. */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_getsockopt(PySocketSockObject *s, PyObject *args)
 | 
						|
{
 | 
						|
    int level;
 | 
						|
    int optname;
 | 
						|
    int res;
 | 
						|
    PyObject *buf;
 | 
						|
    socklen_t buflen = 0;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "ii|i:getsockopt",
 | 
						|
                          &level, &optname, &buflen))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    if (buflen == 0) {
 | 
						|
        int flag = 0;
 | 
						|
        socklen_t flagsize = sizeof flag;
 | 
						|
        res = getsockopt(s->sock_fd, level, optname,
 | 
						|
                         (void *)&flag, &flagsize);
 | 
						|
        if (res < 0)
 | 
						|
            return s->errorhandler();
 | 
						|
        return PyLong_FromLong(flag);
 | 
						|
    }
 | 
						|
#ifdef __VMS
 | 
						|
    /* socklen_t is unsigned so no negative test is needed,
 | 
						|
       test buflen == 0 is previously done */
 | 
						|
    if (buflen > 1024) {
 | 
						|
#else
 | 
						|
    if (buflen <= 0 || buflen > 1024) {
 | 
						|
#endif
 | 
						|
        PyErr_SetString(socket_error,
 | 
						|
                        "getsockopt buflen out of range");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    buf = PyBytes_FromStringAndSize((char *)NULL, buflen);
 | 
						|
    if (buf == NULL)
 | 
						|
        return NULL;
 | 
						|
    res = getsockopt(s->sock_fd, level, optname,
 | 
						|
                     (void *)PyBytes_AS_STRING(buf), &buflen);
 | 
						|
    if (res < 0) {
 | 
						|
        Py_DECREF(buf);
 | 
						|
        return s->errorhandler();
 | 
						|
    }
 | 
						|
    _PyBytes_Resize(&buf, buflen);
 | 
						|
    return buf;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(getsockopt_doc,
 | 
						|
"getsockopt(level, option[, buffersize]) -> value\n\
 | 
						|
\n\
 | 
						|
Get a socket option.  See the Unix manual for level and option.\n\
 | 
						|
If a nonzero buffersize argument is given, the return value is a\n\
 | 
						|
string of that length; otherwise it is an integer.");
 | 
						|
 | 
						|
 | 
						|
/* s.bind(sockaddr) method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_bind(PySocketSockObject *s, PyObject *addro)
 | 
						|
{
 | 
						|
    sock_addr_t addrbuf;
 | 
						|
    int addrlen;
 | 
						|
    int res;
 | 
						|
 | 
						|
    if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
 | 
						|
        return NULL;
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    res = bind(s->sock_fd, SAS2SA(&addrbuf), addrlen);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    if (res < 0)
 | 
						|
        return s->errorhandler();
 | 
						|
    Py_INCREF(Py_None);
 | 
						|
    return Py_None;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(bind_doc,
 | 
						|
"bind(address)\n\
 | 
						|
\n\
 | 
						|
Bind the socket to a local address.  For IP sockets, the address is a\n\
 | 
						|
pair (host, port); the host must refer to the local host. For raw packet\n\
 | 
						|
sockets the address is a tuple (ifname, proto [,pkttype [,hatype]])");
 | 
						|
 | 
						|
 | 
						|
/* s.close() method.
 | 
						|
   Set the file descriptor to -1 so operations tried subsequently
 | 
						|
   will surely fail. */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_close(PySocketSockObject *s)
 | 
						|
{
 | 
						|
    SOCKET_T fd;
 | 
						|
 | 
						|
    if ((fd = s->sock_fd) != -1) {
 | 
						|
        s->sock_fd = -1;
 | 
						|
        Py_BEGIN_ALLOW_THREADS
 | 
						|
        (void) SOCKETCLOSE(fd);
 | 
						|
        Py_END_ALLOW_THREADS
 | 
						|
    }
 | 
						|
    Py_INCREF(Py_None);
 | 
						|
    return Py_None;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(close_doc,
 | 
						|
"close()\n\
 | 
						|
\n\
 | 
						|
Close the socket.  It cannot be used after this call.");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_detach(PySocketSockObject *s)
 | 
						|
{
 | 
						|
    SOCKET_T fd = s->sock_fd;
 | 
						|
    s->sock_fd = -1;
 | 
						|
    return PyLong_FromSocket_t(fd);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(detach_doc,
 | 
						|
"detach()\n\
 | 
						|
\n\
 | 
						|
Close the socket object without closing the underlying file descriptor.\
 | 
						|
The object cannot be used after this call, but the file descriptor\
 | 
						|
can be reused for other purposes.  The file descriptor is returned.");
 | 
						|
 | 
						|
static int
 | 
						|
internal_connect(PySocketSockObject *s, struct sockaddr *addr, int addrlen,
 | 
						|
                 int *timeoutp)
 | 
						|
{
 | 
						|
    int res, timeout;
 | 
						|
 | 
						|
    timeout = 0;
 | 
						|
    res = connect(s->sock_fd, addr, addrlen);
 | 
						|
 | 
						|
#ifdef MS_WINDOWS
 | 
						|
 | 
						|
    if (s->sock_timeout > 0.0) {
 | 
						|
        if (res < 0 && WSAGetLastError() == WSAEWOULDBLOCK &&
 | 
						|
            IS_SELECTABLE(s)) {
 | 
						|
            /* This is a mess.  Best solution: trust select */
 | 
						|
            fd_set fds;
 | 
						|
            fd_set fds_exc;
 | 
						|
            struct timeval tv;
 | 
						|
            tv.tv_sec = (int)s->sock_timeout;
 | 
						|
            tv.tv_usec = (int)((s->sock_timeout - tv.tv_sec) * 1e6);
 | 
						|
            FD_ZERO(&fds);
 | 
						|
            FD_SET(s->sock_fd, &fds);
 | 
						|
            FD_ZERO(&fds_exc);
 | 
						|
            FD_SET(s->sock_fd, &fds_exc);
 | 
						|
            res = select(Py_SAFE_DOWNCAST(s->sock_fd+1, SOCKET_T, int),
 | 
						|
                         NULL, &fds, &fds_exc, &tv);
 | 
						|
            if (res == 0) {
 | 
						|
                res = WSAEWOULDBLOCK;
 | 
						|
                timeout = 1;
 | 
						|
            } else if (res > 0) {
 | 
						|
                if (FD_ISSET(s->sock_fd, &fds))
 | 
						|
                    /* The socket is in the writable set - this
 | 
						|
                       means connected */
 | 
						|
                    res = 0;
 | 
						|
                else {
 | 
						|
                    /* As per MS docs, we need to call getsockopt()
 | 
						|
                       to get the underlying error */
 | 
						|
                    int res_size = sizeof res;
 | 
						|
                    /* It must be in the exception set */
 | 
						|
                    assert(FD_ISSET(s->sock_fd, &fds_exc));
 | 
						|
                    if (0 == getsockopt(s->sock_fd, SOL_SOCKET, SO_ERROR,
 | 
						|
                                        (char *)&res, &res_size))
 | 
						|
                        /* getsockopt also clears WSAGetLastError,
 | 
						|
                           so reset it back. */
 | 
						|
                        WSASetLastError(res);
 | 
						|
                    else
 | 
						|
                        res = WSAGetLastError();
 | 
						|
                }
 | 
						|
            }
 | 
						|
            /* else if (res < 0) an error occurred */
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    if (res < 0)
 | 
						|
        res = WSAGetLastError();
 | 
						|
 | 
						|
#else
 | 
						|
 | 
						|
    if (s->sock_timeout > 0.0) {
 | 
						|
        if (res < 0 && errno == EINPROGRESS && IS_SELECTABLE(s)) {
 | 
						|
            timeout = internal_select(s, 1);
 | 
						|
            if (timeout == 0) {
 | 
						|
                /* Bug #1019808: in case of an EINPROGRESS,
 | 
						|
                   use getsockopt(SO_ERROR) to get the real
 | 
						|
                   error. */
 | 
						|
                socklen_t res_size = sizeof res;
 | 
						|
                (void)getsockopt(s->sock_fd, SOL_SOCKET,
 | 
						|
                                 SO_ERROR, &res, &res_size);
 | 
						|
                if (res == EISCONN)
 | 
						|
                    res = 0;
 | 
						|
                errno = res;
 | 
						|
            }
 | 
						|
            else if (timeout == -1) {
 | 
						|
                res = errno;            /* had error */
 | 
						|
            }
 | 
						|
            else
 | 
						|
                res = EWOULDBLOCK;                      /* timed out */
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    if (res < 0)
 | 
						|
        res = errno;
 | 
						|
 | 
						|
#endif
 | 
						|
    *timeoutp = timeout;
 | 
						|
 | 
						|
    return res;
 | 
						|
}
 | 
						|
 | 
						|
/* s.connect(sockaddr) method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_connect(PySocketSockObject *s, PyObject *addro)
 | 
						|
{
 | 
						|
    sock_addr_t addrbuf;
 | 
						|
    int addrlen;
 | 
						|
    int res;
 | 
						|
    int timeout;
 | 
						|
 | 
						|
    if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
 | 
						|
    if (timeout == 1) {
 | 
						|
        PyErr_SetString(socket_timeout, "timed out");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    if (res != 0)
 | 
						|
        return s->errorhandler();
 | 
						|
    Py_INCREF(Py_None);
 | 
						|
    return Py_None;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(connect_doc,
 | 
						|
"connect(address)\n\
 | 
						|
\n\
 | 
						|
Connect the socket to a remote address.  For IP sockets, the address\n\
 | 
						|
is a pair (host, port).");
 | 
						|
 | 
						|
 | 
						|
/* s.connect_ex(sockaddr) method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_connect_ex(PySocketSockObject *s, PyObject *addro)
 | 
						|
{
 | 
						|
    sock_addr_t addrbuf;
 | 
						|
    int addrlen;
 | 
						|
    int res;
 | 
						|
    int timeout;
 | 
						|
 | 
						|
    if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    res = internal_connect(s, SAS2SA(&addrbuf), addrlen, &timeout);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
 | 
						|
    /* Signals are not errors (though they may raise exceptions).  Adapted
 | 
						|
       from PyErr_SetFromErrnoWithFilenameObject(). */
 | 
						|
#ifdef EINTR
 | 
						|
    if (res == EINTR && PyErr_CheckSignals())
 | 
						|
        return NULL;
 | 
						|
#endif
 | 
						|
 | 
						|
    return PyLong_FromLong((long) res);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(connect_ex_doc,
 | 
						|
"connect_ex(address) -> errno\n\
 | 
						|
\n\
 | 
						|
This is like connect(address), but returns an error code (the errno value)\n\
 | 
						|
instead of raising an exception when an error occurs.");
 | 
						|
 | 
						|
 | 
						|
/* s.fileno() method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_fileno(PySocketSockObject *s)
 | 
						|
{
 | 
						|
    return PyLong_FromSocket_t(s->sock_fd);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(fileno_doc,
 | 
						|
"fileno() -> integer\n\
 | 
						|
\n\
 | 
						|
Return the integer file descriptor of the socket.");
 | 
						|
 | 
						|
 | 
						|
/* s.getsockname() method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_getsockname(PySocketSockObject *s)
 | 
						|
{
 | 
						|
    sock_addr_t addrbuf;
 | 
						|
    int res;
 | 
						|
    socklen_t addrlen;
 | 
						|
 | 
						|
    if (!getsockaddrlen(s, &addrlen))
 | 
						|
        return NULL;
 | 
						|
    memset(&addrbuf, 0, addrlen);
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    res = getsockname(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    if (res < 0)
 | 
						|
        return s->errorhandler();
 | 
						|
    return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
 | 
						|
                        s->sock_proto);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(getsockname_doc,
 | 
						|
"getsockname() -> address info\n\
 | 
						|
\n\
 | 
						|
Return the address of the local endpoint.  For IP sockets, the address\n\
 | 
						|
info is a pair (hostaddr, port).");
 | 
						|
 | 
						|
 | 
						|
#ifdef HAVE_GETPEERNAME         /* Cray APP doesn't have this :-( */
 | 
						|
/* s.getpeername() method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_getpeername(PySocketSockObject *s)
 | 
						|
{
 | 
						|
    sock_addr_t addrbuf;
 | 
						|
    int res;
 | 
						|
    socklen_t addrlen;
 | 
						|
 | 
						|
    if (!getsockaddrlen(s, &addrlen))
 | 
						|
        return NULL;
 | 
						|
    memset(&addrbuf, 0, addrlen);
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    res = getpeername(s->sock_fd, SAS2SA(&addrbuf), &addrlen);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    if (res < 0)
 | 
						|
        return s->errorhandler();
 | 
						|
    return makesockaddr(s->sock_fd, SAS2SA(&addrbuf), addrlen,
 | 
						|
                        s->sock_proto);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(getpeername_doc,
 | 
						|
"getpeername() -> address info\n\
 | 
						|
\n\
 | 
						|
Return the address of the remote endpoint.  For IP sockets, the address\n\
 | 
						|
info is a pair (hostaddr, port).");
 | 
						|
 | 
						|
#endif /* HAVE_GETPEERNAME */
 | 
						|
 | 
						|
 | 
						|
/* s.listen(n) method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_listen(PySocketSockObject *s, PyObject *arg)
 | 
						|
{
 | 
						|
    int backlog;
 | 
						|
    int res;
 | 
						|
 | 
						|
    backlog = PyLong_AsLong(arg);
 | 
						|
    if (backlog == -1 && PyErr_Occurred())
 | 
						|
        return NULL;
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    /* To avoid problems on systems that don't allow a negative backlog
 | 
						|
     * (which doesn't make sense anyway) we force a minimum value of 0. */
 | 
						|
    if (backlog < 0)
 | 
						|
        backlog = 0;
 | 
						|
    res = listen(s->sock_fd, backlog);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    if (res < 0)
 | 
						|
        return s->errorhandler();
 | 
						|
    Py_INCREF(Py_None);
 | 
						|
    return Py_None;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(listen_doc,
 | 
						|
"listen(backlog)\n\
 | 
						|
\n\
 | 
						|
Enable a server to accept connections.  The backlog argument must be at\n\
 | 
						|
least 0 (if it is lower, it is set to 0); it specifies the number of\n\
 | 
						|
unaccepted connections that the system will allow before refusing new\n\
 | 
						|
connections.");
 | 
						|
 | 
						|
 | 
						|
/*
 | 
						|
 * This is the guts of the recv() and recv_into() methods, which reads into a
 | 
						|
 * char buffer.  If you have any inc/dec ref to do to the objects that contain
 | 
						|
 * the buffer, do it in the caller.  This function returns the number of bytes
 | 
						|
 * successfully read.  If there was an error, it returns -1.  Note that it is
 | 
						|
 * also possible that we return a number of bytes smaller than the request
 | 
						|
 * bytes.
 | 
						|
 */
 | 
						|
 | 
						|
static Py_ssize_t
 | 
						|
sock_recv_guts(PySocketSockObject *s, char* cbuf, Py_ssize_t len, int flags)
 | 
						|
{
 | 
						|
    Py_ssize_t outlen = -1;
 | 
						|
    int timeout;
 | 
						|
#ifdef __VMS
 | 
						|
    int remaining;
 | 
						|
    char *read_buf;
 | 
						|
#endif
 | 
						|
 | 
						|
    if (!IS_SELECTABLE(s)) {
 | 
						|
        select_error();
 | 
						|
        return -1;
 | 
						|
    }
 | 
						|
    if (len == 0) {
 | 
						|
        /* If 0 bytes were requested, do nothing. */
 | 
						|
        return 0;
 | 
						|
    }
 | 
						|
 | 
						|
#ifndef __VMS
 | 
						|
    BEGIN_SELECT_LOOP(s)
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    timeout = internal_select_ex(s, 0, interval);
 | 
						|
    if (!timeout)
 | 
						|
        outlen = recv(s->sock_fd, cbuf, len, flags);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
 | 
						|
    if (timeout == 1) {
 | 
						|
        PyErr_SetString(socket_timeout, "timed out");
 | 
						|
        return -1;
 | 
						|
    }
 | 
						|
    END_SELECT_LOOP(s)
 | 
						|
    if (outlen < 0) {
 | 
						|
        /* Note: the call to errorhandler() ALWAYS indirectly returned
 | 
						|
           NULL, so ignore its return value */
 | 
						|
        s->errorhandler();
 | 
						|
        return -1;
 | 
						|
    }
 | 
						|
#else
 | 
						|
    read_buf = cbuf;
 | 
						|
    remaining = len;
 | 
						|
    while (remaining != 0) {
 | 
						|
        unsigned int segment;
 | 
						|
        int nread = -1;
 | 
						|
 | 
						|
        segment = remaining /SEGMENT_SIZE;
 | 
						|
        if (segment != 0) {
 | 
						|
            segment = SEGMENT_SIZE;
 | 
						|
        }
 | 
						|
        else {
 | 
						|
            segment = remaining;
 | 
						|
        }
 | 
						|
 | 
						|
        BEGIN_SELECT_LOOP(s)
 | 
						|
        Py_BEGIN_ALLOW_THREADS
 | 
						|
        timeout = internal_select_ex(s, 0, interval);
 | 
						|
        if (!timeout)
 | 
						|
            nread = recv(s->sock_fd, read_buf, segment, flags);
 | 
						|
        Py_END_ALLOW_THREADS
 | 
						|
        if (timeout == 1) {
 | 
						|
            PyErr_SetString(socket_timeout, "timed out");
 | 
						|
            return -1;
 | 
						|
        }
 | 
						|
        END_SELECT_LOOP(s)
 | 
						|
 | 
						|
        if (nread < 0) {
 | 
						|
            s->errorhandler();
 | 
						|
            return -1;
 | 
						|
        }
 | 
						|
        if (nread != remaining) {
 | 
						|
            read_buf += nread;
 | 
						|
            break;
 | 
						|
        }
 | 
						|
 | 
						|
        remaining -= segment;
 | 
						|
        read_buf += segment;
 | 
						|
    }
 | 
						|
    outlen = read_buf - cbuf;
 | 
						|
#endif /* !__VMS */
 | 
						|
 | 
						|
    return outlen;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* s.recv(nbytes [,flags]) method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_recv(PySocketSockObject *s, PyObject *args)
 | 
						|
{
 | 
						|
    Py_ssize_t recvlen, outlen;
 | 
						|
    int flags = 0;
 | 
						|
    PyObject *buf;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "n|i:recv", &recvlen, &flags))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    if (recvlen < 0) {
 | 
						|
        PyErr_SetString(PyExc_ValueError,
 | 
						|
                        "negative buffersize in recv");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Allocate a new string. */
 | 
						|
    buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
 | 
						|
    if (buf == NULL)
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    /* Call the guts */
 | 
						|
    outlen = sock_recv_guts(s, PyBytes_AS_STRING(buf), recvlen, flags);
 | 
						|
    if (outlen < 0) {
 | 
						|
        /* An error occurred, release the string and return an
 | 
						|
           error. */
 | 
						|
        Py_DECREF(buf);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    if (outlen != recvlen) {
 | 
						|
        /* We did not read as many bytes as we anticipated, resize the
 | 
						|
           string if possible and be successful. */
 | 
						|
        _PyBytes_Resize(&buf, outlen);
 | 
						|
    }
 | 
						|
 | 
						|
    return buf;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(recv_doc,
 | 
						|
"recv(buffersize[, flags]) -> data\n\
 | 
						|
\n\
 | 
						|
Receive up to buffersize bytes from the socket.  For the optional flags\n\
 | 
						|
argument, see the Unix manual.  When no data is available, block until\n\
 | 
						|
at least one byte is available or until the remote end is closed.  When\n\
 | 
						|
the remote end is closed and all data is read, return the empty string.");
 | 
						|
 | 
						|
 | 
						|
/* s.recv_into(buffer, [nbytes [,flags]]) method */
 | 
						|
 | 
						|
static PyObject*
 | 
						|
sock_recv_into(PySocketSockObject *s, PyObject *args, PyObject *kwds)
 | 
						|
{
 | 
						|
    static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
 | 
						|
 | 
						|
    int flags = 0;
 | 
						|
    Py_buffer pbuf;
 | 
						|
    char *buf;
 | 
						|
    Py_ssize_t buflen, readlen, recvlen = 0;
 | 
						|
 | 
						|
    /* Get the buffer's memory */
 | 
						|
    if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ni:recv_into", kwlist,
 | 
						|
                                     &pbuf, &recvlen, &flags))
 | 
						|
        return NULL;
 | 
						|
    buf = pbuf.buf;
 | 
						|
    buflen = pbuf.len;
 | 
						|
 | 
						|
    if (recvlen < 0) {
 | 
						|
        PyBuffer_Release(&pbuf);
 | 
						|
        PyErr_SetString(PyExc_ValueError,
 | 
						|
                        "negative buffersize in recv_into");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    if (recvlen == 0) {
 | 
						|
        /* If nbytes was not specified, use the buffer's length */
 | 
						|
        recvlen = buflen;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Check if the buffer is large enough */
 | 
						|
    if (buflen < recvlen) {
 | 
						|
        PyBuffer_Release(&pbuf);
 | 
						|
        PyErr_SetString(PyExc_ValueError,
 | 
						|
                        "buffer too small for requested bytes");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Call the guts */
 | 
						|
    readlen = sock_recv_guts(s, buf, recvlen, flags);
 | 
						|
    if (readlen < 0) {
 | 
						|
        /* Return an error. */
 | 
						|
        PyBuffer_Release(&pbuf);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    PyBuffer_Release(&pbuf);
 | 
						|
    /* Return the number of bytes read.  Note that we do not do anything
 | 
						|
       special here in the case that readlen < recvlen. */
 | 
						|
    return PyLong_FromSsize_t(readlen);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(recv_into_doc,
 | 
						|
"recv_into(buffer, [nbytes[, flags]]) -> nbytes_read\n\
 | 
						|
\n\
 | 
						|
A version of recv() that stores its data into a buffer rather than creating \n\
 | 
						|
a new string.  Receive up to buffersize bytes from the socket.  If buffersize \n\
 | 
						|
is not specified (or 0), receive up to the size available in the given buffer.\n\
 | 
						|
\n\
 | 
						|
See recv() for documentation about the flags.");
 | 
						|
 | 
						|
 | 
						|
/*
 | 
						|
 * This is the guts of the recvfrom() and recvfrom_into() methods, which reads
 | 
						|
 * into a char buffer.  If you have any inc/def ref to do to the objects that
 | 
						|
 * contain the buffer, do it in the caller.  This function returns the number
 | 
						|
 * of bytes successfully read.  If there was an error, it returns -1.  Note
 | 
						|
 * that it is also possible that we return a number of bytes smaller than the
 | 
						|
 * request bytes.
 | 
						|
 *
 | 
						|
 * 'addr' is a return value for the address object.  Note that you must decref
 | 
						|
 * it yourself.
 | 
						|
 */
 | 
						|
static Py_ssize_t
 | 
						|
sock_recvfrom_guts(PySocketSockObject *s, char* cbuf, Py_ssize_t len, int flags,
 | 
						|
                   PyObject** addr)
 | 
						|
{
 | 
						|
    sock_addr_t addrbuf;
 | 
						|
    int timeout;
 | 
						|
    Py_ssize_t n = -1;
 | 
						|
    socklen_t addrlen;
 | 
						|
 | 
						|
    *addr = NULL;
 | 
						|
 | 
						|
    if (!getsockaddrlen(s, &addrlen))
 | 
						|
        return -1;
 | 
						|
 | 
						|
    if (!IS_SELECTABLE(s)) {
 | 
						|
        select_error();
 | 
						|
        return -1;
 | 
						|
    }
 | 
						|
 | 
						|
    BEGIN_SELECT_LOOP(s)
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    memset(&addrbuf, 0, addrlen);
 | 
						|
    timeout = internal_select_ex(s, 0, interval);
 | 
						|
    if (!timeout) {
 | 
						|
#ifndef MS_WINDOWS
 | 
						|
#if defined(PYOS_OS2) && !defined(PYCC_GCC)
 | 
						|
        n = recvfrom(s->sock_fd, cbuf, len, flags,
 | 
						|
                     SAS2SA(&addrbuf), &addrlen);
 | 
						|
#else
 | 
						|
        n = recvfrom(s->sock_fd, cbuf, len, flags,
 | 
						|
                     (void *) &addrbuf, &addrlen);
 | 
						|
#endif
 | 
						|
#else
 | 
						|
        n = recvfrom(s->sock_fd, cbuf, len, flags,
 | 
						|
                     SAS2SA(&addrbuf), &addrlen);
 | 
						|
#endif
 | 
						|
    }
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
 | 
						|
    if (timeout == 1) {
 | 
						|
        PyErr_SetString(socket_timeout, "timed out");
 | 
						|
        return -1;
 | 
						|
    }
 | 
						|
    END_SELECT_LOOP(s)
 | 
						|
    if (n < 0) {
 | 
						|
        s->errorhandler();
 | 
						|
        return -1;
 | 
						|
    }
 | 
						|
 | 
						|
    if (!(*addr = makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
 | 
						|
                               addrlen, s->sock_proto)))
 | 
						|
        return -1;
 | 
						|
 | 
						|
    return n;
 | 
						|
}
 | 
						|
 | 
						|
/* s.recvfrom(nbytes [,flags]) method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_recvfrom(PySocketSockObject *s, PyObject *args)
 | 
						|
{
 | 
						|
    PyObject *buf = NULL;
 | 
						|
    PyObject *addr = NULL;
 | 
						|
    PyObject *ret = NULL;
 | 
						|
    int flags = 0;
 | 
						|
    Py_ssize_t recvlen, outlen;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "n|i:recvfrom", &recvlen, &flags))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    if (recvlen < 0) {
 | 
						|
        PyErr_SetString(PyExc_ValueError,
 | 
						|
                        "negative buffersize in recvfrom");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    buf = PyBytes_FromStringAndSize((char *) 0, recvlen);
 | 
						|
    if (buf == NULL)
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    outlen = sock_recvfrom_guts(s, PyBytes_AS_STRING(buf),
 | 
						|
                                recvlen, flags, &addr);
 | 
						|
    if (outlen < 0) {
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
 | 
						|
    if (outlen != recvlen) {
 | 
						|
        /* We did not read as many bytes as we anticipated, resize the
 | 
						|
           string if possible and be successful. */
 | 
						|
        if (_PyBytes_Resize(&buf, outlen) < 0)
 | 
						|
            /* Oopsy, not so successful after all. */
 | 
						|
            goto finally;
 | 
						|
    }
 | 
						|
 | 
						|
    ret = PyTuple_Pack(2, buf, addr);
 | 
						|
 | 
						|
finally:
 | 
						|
    Py_XDECREF(buf);
 | 
						|
    Py_XDECREF(addr);
 | 
						|
    return ret;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(recvfrom_doc,
 | 
						|
"recvfrom(buffersize[, flags]) -> (data, address info)\n\
 | 
						|
\n\
 | 
						|
Like recv(buffersize, flags) but also return the sender's address info.");
 | 
						|
 | 
						|
 | 
						|
/* s.recvfrom_into(buffer[, nbytes [,flags]]) method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_recvfrom_into(PySocketSockObject *s, PyObject *args, PyObject* kwds)
 | 
						|
{
 | 
						|
    static char *kwlist[] = {"buffer", "nbytes", "flags", 0};
 | 
						|
 | 
						|
    int flags = 0;
 | 
						|
    Py_buffer pbuf;
 | 
						|
    char *buf;
 | 
						|
    Py_ssize_t readlen, buflen, recvlen = 0;
 | 
						|
 | 
						|
    PyObject *addr = NULL;
 | 
						|
 | 
						|
    if (!PyArg_ParseTupleAndKeywords(args, kwds, "w*|ni:recvfrom_into",
 | 
						|
                                     kwlist, &pbuf,
 | 
						|
                                     &recvlen, &flags))
 | 
						|
        return NULL;
 | 
						|
    buf = pbuf.buf;
 | 
						|
    buflen = pbuf.len;
 | 
						|
    assert(buf != 0 && buflen > 0);
 | 
						|
 | 
						|
    if (recvlen < 0) {
 | 
						|
        PyBuffer_Release(&pbuf);
 | 
						|
        PyErr_SetString(PyExc_ValueError,
 | 
						|
                        "negative buffersize in recvfrom_into");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    if (recvlen == 0) {
 | 
						|
        /* If nbytes was not specified, use the buffer's length */
 | 
						|
        recvlen = buflen;
 | 
						|
    }
 | 
						|
 | 
						|
    readlen = sock_recvfrom_guts(s, buf, recvlen, flags, &addr);
 | 
						|
    if (readlen < 0) {
 | 
						|
        PyBuffer_Release(&pbuf);
 | 
						|
        /* Return an error */
 | 
						|
        Py_XDECREF(addr);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    PyBuffer_Release(&pbuf);
 | 
						|
    /* Return the number of bytes read and the address.  Note that we do
 | 
						|
       not do anything special here in the case that readlen < recvlen. */
 | 
						|
    return Py_BuildValue("nN", readlen, addr);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(recvfrom_into_doc,
 | 
						|
"recvfrom_into(buffer[, nbytes[, flags]]) -> (nbytes, address info)\n\
 | 
						|
\n\
 | 
						|
Like recv_into(buffer[, nbytes[, flags]]) but also return the sender's address info.");
 | 
						|
 | 
						|
 | 
						|
/* The sendmsg() and recvmsg[_into]() methods require a working
 | 
						|
   CMSG_LEN().  See the comment near get_CMSG_LEN(). */
 | 
						|
#ifdef CMSG_LEN
 | 
						|
/*
 | 
						|
 * Call recvmsg() with the supplied iovec structures, flags, and
 | 
						|
 * ancillary data buffer size (controllen).  Returns the tuple return
 | 
						|
 * value for recvmsg() or recvmsg_into(), with the first item provided
 | 
						|
 * by the supplied makeval() function.  makeval() will be called with
 | 
						|
 * the length read and makeval_data as arguments, and must return a
 | 
						|
 * new reference (which will be decrefed if there is a subsequent
 | 
						|
 * error).  On error, closes any file descriptors received via
 | 
						|
 * SCM_RIGHTS.
 | 
						|
 */
 | 
						|
static PyObject *
 | 
						|
sock_recvmsg_guts(PySocketSockObject *s, struct iovec *iov, int iovlen,
 | 
						|
                  int flags, Py_ssize_t controllen,
 | 
						|
                  PyObject *(*makeval)(ssize_t, void *), void *makeval_data)
 | 
						|
{
 | 
						|
    ssize_t bytes_received = -1;
 | 
						|
    int timeout;
 | 
						|
    sock_addr_t addrbuf;
 | 
						|
    socklen_t addrbuflen;
 | 
						|
    struct msghdr msg = {0};
 | 
						|
    PyObject *cmsg_list = NULL, *retval = NULL;
 | 
						|
    void *controlbuf = NULL;
 | 
						|
    struct cmsghdr *cmsgh;
 | 
						|
    size_t cmsgdatalen = 0;
 | 
						|
    int cmsg_status;
 | 
						|
 | 
						|
    /* XXX: POSIX says that msg_name and msg_namelen "shall be
 | 
						|
       ignored" when the socket is connected (Linux fills them in
 | 
						|
       anyway for AF_UNIX sockets at least).  Normally msg_namelen
 | 
						|
       seems to be set to 0 if there's no address, but try to
 | 
						|
       initialize msg_name to something that won't be mistaken for a
 | 
						|
       real address if that doesn't happen. */
 | 
						|
    if (!getsockaddrlen(s, &addrbuflen))
 | 
						|
        return NULL;
 | 
						|
    memset(&addrbuf, 0, addrbuflen);
 | 
						|
    SAS2SA(&addrbuf)->sa_family = AF_UNSPEC;
 | 
						|
 | 
						|
    if (controllen < 0 || controllen > SOCKLEN_T_LIMIT) {
 | 
						|
        PyErr_SetString(PyExc_ValueError,
 | 
						|
                        "invalid ancillary data buffer length");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    if (controllen > 0 && (controlbuf = PyMem_Malloc(controllen)) == NULL)
 | 
						|
        return PyErr_NoMemory();
 | 
						|
 | 
						|
    /* Make the system call. */
 | 
						|
    if (!IS_SELECTABLE(s)) {
 | 
						|
        select_error();
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
 | 
						|
    BEGIN_SELECT_LOOP(s)
 | 
						|
    Py_BEGIN_ALLOW_THREADS;
 | 
						|
    msg.msg_name = SAS2SA(&addrbuf);
 | 
						|
    msg.msg_namelen = addrbuflen;
 | 
						|
    msg.msg_iov = iov;
 | 
						|
    msg.msg_iovlen = iovlen;
 | 
						|
    msg.msg_control = controlbuf;
 | 
						|
    msg.msg_controllen = controllen;
 | 
						|
    timeout = internal_select_ex(s, 0, interval);
 | 
						|
    if (!timeout)
 | 
						|
        bytes_received = recvmsg(s->sock_fd, &msg, flags);
 | 
						|
    Py_END_ALLOW_THREADS;
 | 
						|
    if (timeout == 1) {
 | 
						|
        PyErr_SetString(socket_timeout, "timed out");
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
    END_SELECT_LOOP(s)
 | 
						|
 | 
						|
    if (bytes_received < 0) {
 | 
						|
        s->errorhandler();
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Make list of (level, type, data) tuples from control messages. */
 | 
						|
    if ((cmsg_list = PyList_New(0)) == NULL)
 | 
						|
        goto err_closefds;
 | 
						|
    /* Check for empty ancillary data as old CMSG_FIRSTHDR()
 | 
						|
       implementations didn't do so. */
 | 
						|
    for (cmsgh = ((msg.msg_controllen > 0) ? CMSG_FIRSTHDR(&msg) : NULL);
 | 
						|
         cmsgh != NULL; cmsgh = CMSG_NXTHDR(&msg, cmsgh)) {
 | 
						|
        PyObject *bytes, *tuple;
 | 
						|
        int tmp;
 | 
						|
 | 
						|
        cmsg_status = get_cmsg_data_len(&msg, cmsgh, &cmsgdatalen);
 | 
						|
        if (cmsg_status != 0) {
 | 
						|
            if (PyErr_WarnEx(PyExc_RuntimeWarning,
 | 
						|
                             "received malformed or improperly-truncated "
 | 
						|
                             "ancillary data", 1) == -1)
 | 
						|
                goto err_closefds;
 | 
						|
        }
 | 
						|
        if (cmsg_status < 0)
 | 
						|
            break;
 | 
						|
        if (cmsgdatalen > PY_SSIZE_T_MAX) {
 | 
						|
            PyErr_SetString(socket_error, "control message too long");
 | 
						|
            goto err_closefds;
 | 
						|
        }
 | 
						|
 | 
						|
        bytes = PyBytes_FromStringAndSize((char *)CMSG_DATA(cmsgh),
 | 
						|
                                          cmsgdatalen);
 | 
						|
        tuple = Py_BuildValue("iiN", (int)cmsgh->cmsg_level,
 | 
						|
                              (int)cmsgh->cmsg_type, bytes);
 | 
						|
        if (tuple == NULL)
 | 
						|
            goto err_closefds;
 | 
						|
        tmp = PyList_Append(cmsg_list, tuple);
 | 
						|
        Py_DECREF(tuple);
 | 
						|
        if (tmp != 0)
 | 
						|
            goto err_closefds;
 | 
						|
 | 
						|
        if (cmsg_status != 0)
 | 
						|
            break;
 | 
						|
    }
 | 
						|
 | 
						|
    retval = Py_BuildValue("NOiN",
 | 
						|
                           (*makeval)(bytes_received, makeval_data),
 | 
						|
                           cmsg_list,
 | 
						|
                           (int)msg.msg_flags,
 | 
						|
                           makesockaddr(s->sock_fd, SAS2SA(&addrbuf),
 | 
						|
                                        ((msg.msg_namelen > addrbuflen) ?
 | 
						|
                                         addrbuflen : msg.msg_namelen),
 | 
						|
                                        s->sock_proto));
 | 
						|
    if (retval == NULL)
 | 
						|
        goto err_closefds;
 | 
						|
 | 
						|
finally:
 | 
						|
    Py_XDECREF(cmsg_list);
 | 
						|
    PyMem_Free(controlbuf);
 | 
						|
    return retval;
 | 
						|
 | 
						|
err_closefds:
 | 
						|
#ifdef SCM_RIGHTS
 | 
						|
    /* Close all descriptors coming from SCM_RIGHTS, so they don't leak. */
 | 
						|
    for (cmsgh = ((msg.msg_controllen > 0) ? CMSG_FIRSTHDR(&msg) : NULL);
 | 
						|
         cmsgh != NULL; cmsgh = CMSG_NXTHDR(&msg, cmsgh)) {
 | 
						|
        cmsg_status = get_cmsg_data_len(&msg, cmsgh, &cmsgdatalen);
 | 
						|
        if (cmsg_status < 0)
 | 
						|
            break;
 | 
						|
        if (cmsgh->cmsg_level == SOL_SOCKET &&
 | 
						|
            cmsgh->cmsg_type == SCM_RIGHTS) {
 | 
						|
            size_t numfds;
 | 
						|
            int *fdp;
 | 
						|
 | 
						|
            numfds = cmsgdatalen / sizeof(int);
 | 
						|
            fdp = (int *)CMSG_DATA(cmsgh);
 | 
						|
            while (numfds-- > 0)
 | 
						|
                close(*fdp++);
 | 
						|
        }
 | 
						|
        if (cmsg_status != 0)
 | 
						|
            break;
 | 
						|
    }
 | 
						|
#endif /* SCM_RIGHTS */
 | 
						|
    goto finally;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
static PyObject *
 | 
						|
makeval_recvmsg(ssize_t received, void *data)
 | 
						|
{
 | 
						|
    PyObject **buf = data;
 | 
						|
 | 
						|
    if (received < PyBytes_GET_SIZE(*buf))
 | 
						|
        _PyBytes_Resize(buf, received);
 | 
						|
    Py_XINCREF(*buf);
 | 
						|
    return *buf;
 | 
						|
}
 | 
						|
 | 
						|
/* s.recvmsg(bufsize[, ancbufsize[, flags]]) method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_recvmsg(PySocketSockObject *s, PyObject *args)
 | 
						|
{
 | 
						|
    Py_ssize_t bufsize, ancbufsize = 0;
 | 
						|
    int flags = 0;
 | 
						|
    struct iovec iov;
 | 
						|
    PyObject *buf = NULL, *retval = NULL;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "n|ni:recvmsg", &bufsize, &ancbufsize, &flags))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    if (bufsize < 0) {
 | 
						|
        PyErr_SetString(PyExc_ValueError, "negative buffer size in recvmsg()");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    if ((buf = PyBytes_FromStringAndSize(NULL, bufsize)) == NULL)
 | 
						|
        return NULL;
 | 
						|
    iov.iov_base = PyBytes_AS_STRING(buf);
 | 
						|
    iov.iov_len = bufsize;
 | 
						|
 | 
						|
    /* Note that we're passing a pointer to *our pointer* to the bytes
 | 
						|
       object here (&buf); makeval_recvmsg() may incref the object, or
 | 
						|
       deallocate it and set our pointer to NULL. */
 | 
						|
    retval = sock_recvmsg_guts(s, &iov, 1, flags, ancbufsize,
 | 
						|
                               &makeval_recvmsg, &buf);
 | 
						|
    Py_XDECREF(buf);
 | 
						|
    return retval;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(recvmsg_doc,
 | 
						|
"recvmsg(bufsize[, ancbufsize[, flags]]) -> (data, ancdata, msg_flags, address)\n\
 | 
						|
\n\
 | 
						|
Receive normal data (up to bufsize bytes) and ancillary data from the\n\
 | 
						|
socket.  The ancbufsize argument sets the size in bytes of the\n\
 | 
						|
internal buffer used to receive the ancillary data; it defaults to 0,\n\
 | 
						|
meaning that no ancillary data will be received.  Appropriate buffer\n\
 | 
						|
sizes for ancillary data can be calculated using CMSG_SPACE() or\n\
 | 
						|
CMSG_LEN(), and items which do not fit into the buffer might be\n\
 | 
						|
truncated or discarded.  The flags argument defaults to 0 and has the\n\
 | 
						|
same meaning as for recv().\n\
 | 
						|
\n\
 | 
						|
The return value is a 4-tuple: (data, ancdata, msg_flags, address).\n\
 | 
						|
The data item is a bytes object holding the non-ancillary data\n\
 | 
						|
received.  The ancdata item is a list of zero or more tuples\n\
 | 
						|
(cmsg_level, cmsg_type, cmsg_data) representing the ancillary data\n\
 | 
						|
(control messages) received: cmsg_level and cmsg_type are integers\n\
 | 
						|
specifying the protocol level and protocol-specific type respectively,\n\
 | 
						|
and cmsg_data is a bytes object holding the associated data.  The\n\
 | 
						|
msg_flags item is the bitwise OR of various flags indicating\n\
 | 
						|
conditions on the received message; see your system documentation for\n\
 | 
						|
details.  If the receiving socket is unconnected, address is the\n\
 | 
						|
address of the sending socket, if available; otherwise, its value is\n\
 | 
						|
unspecified.\n\
 | 
						|
\n\
 | 
						|
If recvmsg() raises an exception after the system call returns, it\n\
 | 
						|
will first attempt to close any file descriptors received via the\n\
 | 
						|
SCM_RIGHTS mechanism.");
 | 
						|
 | 
						|
 | 
						|
static PyObject *
 | 
						|
makeval_recvmsg_into(ssize_t received, void *data)
 | 
						|
{
 | 
						|
    return PyLong_FromSsize_t(received);
 | 
						|
}
 | 
						|
 | 
						|
/* s.recvmsg_into(buffers[, ancbufsize[, flags]]) method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_recvmsg_into(PySocketSockObject *s, PyObject *args)
 | 
						|
{
 | 
						|
    Py_ssize_t ancbufsize = 0;
 | 
						|
    int flags = 0;
 | 
						|
    struct iovec *iovs = NULL;
 | 
						|
    Py_ssize_t i, nitems, nbufs = 0;
 | 
						|
    Py_buffer *bufs = NULL;
 | 
						|
    PyObject *buffers_arg, *fast, *retval = NULL;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "O|ni:recvmsg_into",
 | 
						|
                          &buffers_arg, &ancbufsize, &flags))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    if ((fast = PySequence_Fast(buffers_arg,
 | 
						|
                                "recvmsg_into() argument 1 must be an "
 | 
						|
                                "iterable")) == NULL)
 | 
						|
        return NULL;
 | 
						|
    nitems = PySequence_Fast_GET_SIZE(fast);
 | 
						|
    if (nitems > INT_MAX) {
 | 
						|
        PyErr_SetString(socket_error, "recvmsg_into() argument 1 is too long");
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Fill in an iovec for each item, and save the Py_buffer
 | 
						|
       structs to release afterwards. */
 | 
						|
    if (nitems > 0 && ((iovs = PyMem_New(struct iovec, nitems)) == NULL ||
 | 
						|
                       (bufs = PyMem_New(Py_buffer, nitems)) == NULL)) {
 | 
						|
        PyErr_NoMemory();
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
    for (; nbufs < nitems; nbufs++) {
 | 
						|
        if (!PyArg_Parse(PySequence_Fast_GET_ITEM(fast, nbufs),
 | 
						|
                         "w*;recvmsg_into() argument 1 must be an iterable "
 | 
						|
                         "of single-segment read-write buffers",
 | 
						|
                         &bufs[nbufs]))
 | 
						|
            goto finally;
 | 
						|
        iovs[nbufs].iov_base = bufs[nbufs].buf;
 | 
						|
        iovs[nbufs].iov_len = bufs[nbufs].len;
 | 
						|
    }
 | 
						|
 | 
						|
    retval = sock_recvmsg_guts(s, iovs, nitems, flags, ancbufsize,
 | 
						|
                               &makeval_recvmsg_into, NULL);
 | 
						|
finally:
 | 
						|
    for (i = 0; i < nbufs; i++)
 | 
						|
        PyBuffer_Release(&bufs[i]);
 | 
						|
    PyMem_Free(bufs);
 | 
						|
    PyMem_Free(iovs);
 | 
						|
    Py_DECREF(fast);
 | 
						|
    return retval;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(recvmsg_into_doc,
 | 
						|
"recvmsg_into(buffers[, ancbufsize[, flags]]) -> (nbytes, ancdata, msg_flags, address)\n\
 | 
						|
\n\
 | 
						|
Receive normal data and ancillary data from the socket, scattering the\n\
 | 
						|
non-ancillary data into a series of buffers.  The buffers argument\n\
 | 
						|
must be an iterable of objects that export writable buffers\n\
 | 
						|
(e.g. bytearray objects); these will be filled with successive chunks\n\
 | 
						|
of the non-ancillary data until it has all been written or there are\n\
 | 
						|
no more buffers.  The ancbufsize argument sets the size in bytes of\n\
 | 
						|
the internal buffer used to receive the ancillary data; it defaults to\n\
 | 
						|
0, meaning that no ancillary data will be received.  Appropriate\n\
 | 
						|
buffer sizes for ancillary data can be calculated using CMSG_SPACE()\n\
 | 
						|
or CMSG_LEN(), and items which do not fit into the buffer might be\n\
 | 
						|
truncated or discarded.  The flags argument defaults to 0 and has the\n\
 | 
						|
same meaning as for recv().\n\
 | 
						|
\n\
 | 
						|
The return value is a 4-tuple: (nbytes, ancdata, msg_flags, address).\n\
 | 
						|
The nbytes item is the total number of bytes of non-ancillary data\n\
 | 
						|
written into the buffers.  The ancdata item is a list of zero or more\n\
 | 
						|
tuples (cmsg_level, cmsg_type, cmsg_data) representing the ancillary\n\
 | 
						|
data (control messages) received: cmsg_level and cmsg_type are\n\
 | 
						|
integers specifying the protocol level and protocol-specific type\n\
 | 
						|
respectively, and cmsg_data is a bytes object holding the associated\n\
 | 
						|
data.  The msg_flags item is the bitwise OR of various flags\n\
 | 
						|
indicating conditions on the received message; see your system\n\
 | 
						|
documentation for details.  If the receiving socket is unconnected,\n\
 | 
						|
address is the address of the sending socket, if available; otherwise,\n\
 | 
						|
its value is unspecified.\n\
 | 
						|
\n\
 | 
						|
If recvmsg_into() raises an exception after the system call returns,\n\
 | 
						|
it will first attempt to close any file descriptors received via the\n\
 | 
						|
SCM_RIGHTS mechanism.");
 | 
						|
#endif    /* CMSG_LEN */
 | 
						|
 | 
						|
 | 
						|
/* s.send(data [,flags]) method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_send(PySocketSockObject *s, PyObject *args)
 | 
						|
{
 | 
						|
    char *buf;
 | 
						|
    Py_ssize_t len, n = -1;
 | 
						|
    int flags = 0, timeout;
 | 
						|
    Py_buffer pbuf;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "y*|i:send", &pbuf, &flags))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    if (!IS_SELECTABLE(s)) {
 | 
						|
        PyBuffer_Release(&pbuf);
 | 
						|
        return select_error();
 | 
						|
    }
 | 
						|
    buf = pbuf.buf;
 | 
						|
    len = pbuf.len;
 | 
						|
 | 
						|
    BEGIN_SELECT_LOOP(s)
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    timeout = internal_select_ex(s, 1, interval);
 | 
						|
    if (!timeout)
 | 
						|
#ifdef __VMS
 | 
						|
        n = sendsegmented(s->sock_fd, buf, len, flags);
 | 
						|
#else
 | 
						|
        n = send(s->sock_fd, buf, len, flags);
 | 
						|
#endif
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    if (timeout == 1) {
 | 
						|
        PyBuffer_Release(&pbuf);
 | 
						|
        PyErr_SetString(socket_timeout, "timed out");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    END_SELECT_LOOP(s)
 | 
						|
 | 
						|
    PyBuffer_Release(&pbuf);
 | 
						|
    if (n < 0)
 | 
						|
        return s->errorhandler();
 | 
						|
    return PyLong_FromSsize_t(n);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(send_doc,
 | 
						|
"send(data[, flags]) -> count\n\
 | 
						|
\n\
 | 
						|
Send a data string to the socket.  For the optional flags\n\
 | 
						|
argument, see the Unix manual.  Return the number of bytes\n\
 | 
						|
sent; this may be less than len(data) if the network is busy.");
 | 
						|
 | 
						|
 | 
						|
/* s.sendall(data [,flags]) method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_sendall(PySocketSockObject *s, PyObject *args)
 | 
						|
{
 | 
						|
    char *buf;
 | 
						|
    Py_ssize_t len, n = -1;
 | 
						|
    int flags = 0, timeout, saved_errno;
 | 
						|
    Py_buffer pbuf;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "y*|i:sendall", &pbuf, &flags))
 | 
						|
        return NULL;
 | 
						|
    buf = pbuf.buf;
 | 
						|
    len = pbuf.len;
 | 
						|
 | 
						|
    if (!IS_SELECTABLE(s)) {
 | 
						|
        PyBuffer_Release(&pbuf);
 | 
						|
        return select_error();
 | 
						|
    }
 | 
						|
 | 
						|
    do {
 | 
						|
        Py_BEGIN_ALLOW_THREADS
 | 
						|
        timeout = internal_select(s, 1);
 | 
						|
        n = -1;
 | 
						|
        if (!timeout) {
 | 
						|
#ifdef __VMS
 | 
						|
            n = sendsegmented(s->sock_fd, buf, len, flags);
 | 
						|
#else
 | 
						|
            n = send(s->sock_fd, buf, len, flags);
 | 
						|
#endif
 | 
						|
        }
 | 
						|
        Py_END_ALLOW_THREADS
 | 
						|
        if (timeout == 1) {
 | 
						|
            PyBuffer_Release(&pbuf);
 | 
						|
            PyErr_SetString(socket_timeout, "timed out");
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
        /* PyErr_CheckSignals() might change errno */
 | 
						|
        saved_errno = errno;
 | 
						|
        /* We must run our signal handlers before looping again.
 | 
						|
           send() can return a successful partial write when it is
 | 
						|
           interrupted, so we can't restrict ourselves to EINTR. */
 | 
						|
        if (PyErr_CheckSignals()) {
 | 
						|
            PyBuffer_Release(&pbuf);
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
        if (n < 0) {
 | 
						|
            /* If interrupted, try again */
 | 
						|
            if (saved_errno == EINTR)
 | 
						|
                continue;
 | 
						|
            else
 | 
						|
                break;
 | 
						|
        }
 | 
						|
        buf += n;
 | 
						|
        len -= n;
 | 
						|
    } while (len > 0);
 | 
						|
    PyBuffer_Release(&pbuf);
 | 
						|
 | 
						|
    if (n < 0)
 | 
						|
        return s->errorhandler();
 | 
						|
 | 
						|
    Py_INCREF(Py_None);
 | 
						|
    return Py_None;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(sendall_doc,
 | 
						|
"sendall(data[, flags])\n\
 | 
						|
\n\
 | 
						|
Send a data string to the socket.  For the optional flags\n\
 | 
						|
argument, see the Unix manual.  This calls send() repeatedly\n\
 | 
						|
until all data is sent.  If an error occurs, it's impossible\n\
 | 
						|
to tell how much data has been sent.");
 | 
						|
 | 
						|
 | 
						|
/* s.sendto(data, [flags,] sockaddr) method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_sendto(PySocketSockObject *s, PyObject *args)
 | 
						|
{
 | 
						|
    Py_buffer pbuf;
 | 
						|
    PyObject *addro;
 | 
						|
    char *buf;
 | 
						|
    Py_ssize_t len, arglen;
 | 
						|
    sock_addr_t addrbuf;
 | 
						|
    int addrlen, n = -1, flags, timeout;
 | 
						|
 | 
						|
    flags = 0;
 | 
						|
    arglen = PyTuple_Size(args);
 | 
						|
    switch (arglen) {
 | 
						|
        case 2:
 | 
						|
            PyArg_ParseTuple(args, "y*O:sendto", &pbuf, &addro);
 | 
						|
            break;
 | 
						|
        case 3:
 | 
						|
            PyArg_ParseTuple(args, "y*iO:sendto",
 | 
						|
                             &pbuf, &flags, &addro);
 | 
						|
            break;
 | 
						|
        default:
 | 
						|
            PyErr_Format(PyExc_TypeError,
 | 
						|
                         "sendto() takes 2 or 3 arguments (%d given)",
 | 
						|
                         arglen);
 | 
						|
            return NULL;
 | 
						|
    }
 | 
						|
    if (PyErr_Occurred())
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    buf = pbuf.buf;
 | 
						|
    len = pbuf.len;
 | 
						|
 | 
						|
    if (!IS_SELECTABLE(s)) {
 | 
						|
        PyBuffer_Release(&pbuf);
 | 
						|
        return select_error();
 | 
						|
    }
 | 
						|
 | 
						|
    if (!getsockaddrarg(s, addro, SAS2SA(&addrbuf), &addrlen)) {
 | 
						|
        PyBuffer_Release(&pbuf);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    BEGIN_SELECT_LOOP(s)
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    timeout = internal_select_ex(s, 1, interval);
 | 
						|
    if (!timeout)
 | 
						|
        n = sendto(s->sock_fd, buf, len, flags, SAS2SA(&addrbuf), addrlen);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
 | 
						|
    if (timeout == 1) {
 | 
						|
        PyBuffer_Release(&pbuf);
 | 
						|
        PyErr_SetString(socket_timeout, "timed out");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    END_SELECT_LOOP(s)
 | 
						|
    PyBuffer_Release(&pbuf);
 | 
						|
    if (n < 0)
 | 
						|
        return s->errorhandler();
 | 
						|
    return PyLong_FromSsize_t(n);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(sendto_doc,
 | 
						|
"sendto(data[, flags], address) -> count\n\
 | 
						|
\n\
 | 
						|
Like send(data, flags) but allows specifying the destination address.\n\
 | 
						|
For IP sockets, the address is a pair (hostaddr, port).");
 | 
						|
 | 
						|
 | 
						|
/* The sendmsg() and recvmsg[_into]() methods require a working
 | 
						|
   CMSG_LEN().  See the comment near get_CMSG_LEN(). */
 | 
						|
#ifdef CMSG_LEN
 | 
						|
/* s.sendmsg(buffers[, ancdata[, flags[, address]]]) method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_sendmsg(PySocketSockObject *s, PyObject *args)
 | 
						|
{
 | 
						|
    Py_ssize_t i, ndataparts, ndatabufs = 0, ncmsgs, ncmsgbufs = 0;
 | 
						|
    Py_buffer *databufs = NULL;
 | 
						|
    struct iovec *iovs = NULL;
 | 
						|
    sock_addr_t addrbuf;
 | 
						|
    struct msghdr msg = {0};
 | 
						|
    struct cmsginfo {
 | 
						|
        int level;
 | 
						|
        int type;
 | 
						|
        Py_buffer data;
 | 
						|
    } *cmsgs = NULL;
 | 
						|
    void *controlbuf = NULL;
 | 
						|
    size_t controllen, controllen_last;
 | 
						|
    ssize_t bytes_sent = -1;
 | 
						|
    int addrlen, timeout, flags = 0;
 | 
						|
    PyObject *data_arg, *cmsg_arg = NULL, *addr_arg = NULL, *data_fast = NULL,
 | 
						|
        *cmsg_fast = NULL, *retval = NULL;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "O|OiO:sendmsg",
 | 
						|
                          &data_arg, &cmsg_arg, &flags, &addr_arg))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    /* Parse destination address. */
 | 
						|
    if (addr_arg != NULL && addr_arg != Py_None) {
 | 
						|
        if (!getsockaddrarg(s, addr_arg, SAS2SA(&addrbuf), &addrlen))
 | 
						|
            goto finally;
 | 
						|
        msg.msg_name = &addrbuf;
 | 
						|
        msg.msg_namelen = addrlen;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Fill in an iovec for each message part, and save the Py_buffer
 | 
						|
       structs to release afterwards. */
 | 
						|
    if ((data_fast = PySequence_Fast(data_arg,
 | 
						|
                                     "sendmsg() argument 1 must be an "
 | 
						|
                                     "iterable")) == NULL)
 | 
						|
        goto finally;
 | 
						|
    ndataparts = PySequence_Fast_GET_SIZE(data_fast);
 | 
						|
    if (ndataparts > INT_MAX) {
 | 
						|
        PyErr_SetString(socket_error, "sendmsg() argument 1 is too long");
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
    msg.msg_iovlen = ndataparts;
 | 
						|
    if (ndataparts > 0 &&
 | 
						|
        ((msg.msg_iov = iovs = PyMem_New(struct iovec, ndataparts)) == NULL ||
 | 
						|
         (databufs = PyMem_New(Py_buffer, ndataparts)) == NULL)) {
 | 
						|
        PyErr_NoMemory();
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
    for (; ndatabufs < ndataparts; ndatabufs++) {
 | 
						|
        if (!PyArg_Parse(PySequence_Fast_GET_ITEM(data_fast, ndatabufs),
 | 
						|
                         "y*;sendmsg() argument 1 must be an iterable of "
 | 
						|
                         "buffer-compatible objects",
 | 
						|
                         &databufs[ndatabufs]))
 | 
						|
            goto finally;
 | 
						|
        iovs[ndatabufs].iov_base = databufs[ndatabufs].buf;
 | 
						|
        iovs[ndatabufs].iov_len = databufs[ndatabufs].len;
 | 
						|
    }
 | 
						|
 | 
						|
    if (cmsg_arg == NULL)
 | 
						|
        ncmsgs = 0;
 | 
						|
    else {
 | 
						|
        if ((cmsg_fast = PySequence_Fast(cmsg_arg,
 | 
						|
                                         "sendmsg() argument 2 must be an "
 | 
						|
                                         "iterable")) == NULL)
 | 
						|
            goto finally;
 | 
						|
        ncmsgs = PySequence_Fast_GET_SIZE(cmsg_fast);
 | 
						|
    }
 | 
						|
 | 
						|
#ifndef CMSG_SPACE
 | 
						|
    if (ncmsgs > 1) {
 | 
						|
        PyErr_SetString(socket_error,
 | 
						|
                        "sending multiple control messages is not supported "
 | 
						|
                        "on this system");
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
#endif
 | 
						|
    /* Save level, type and Py_buffer for each control message,
 | 
						|
       and calculate total size. */
 | 
						|
    if (ncmsgs > 0 && (cmsgs = PyMem_New(struct cmsginfo, ncmsgs)) == NULL) {
 | 
						|
        PyErr_NoMemory();
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
    controllen = controllen_last = 0;
 | 
						|
    while (ncmsgbufs < ncmsgs) {
 | 
						|
        size_t bufsize, space;
 | 
						|
 | 
						|
        if (!PyArg_Parse(PySequence_Fast_GET_ITEM(cmsg_fast, ncmsgbufs),
 | 
						|
                         "(iiy*):[sendmsg() ancillary data items]",
 | 
						|
                         &cmsgs[ncmsgbufs].level,
 | 
						|
                         &cmsgs[ncmsgbufs].type,
 | 
						|
                         &cmsgs[ncmsgbufs].data))
 | 
						|
            goto finally;
 | 
						|
        bufsize = cmsgs[ncmsgbufs++].data.len;
 | 
						|
 | 
						|
#ifdef CMSG_SPACE
 | 
						|
        if (!get_CMSG_SPACE(bufsize, &space)) {
 | 
						|
#else
 | 
						|
        if (!get_CMSG_LEN(bufsize, &space)) {
 | 
						|
#endif
 | 
						|
            PyErr_SetString(socket_error, "ancillary data item too large");
 | 
						|
            goto finally;
 | 
						|
        }
 | 
						|
        controllen += space;
 | 
						|
        if (controllen > SOCKLEN_T_LIMIT || controllen < controllen_last) {
 | 
						|
            PyErr_SetString(socket_error, "too much ancillary data");
 | 
						|
            goto finally;
 | 
						|
        }
 | 
						|
        controllen_last = controllen;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Construct ancillary data block from control message info. */
 | 
						|
    if (ncmsgbufs > 0) {
 | 
						|
        struct cmsghdr *cmsgh = NULL;
 | 
						|
 | 
						|
        if ((msg.msg_control = controlbuf =
 | 
						|
             PyMem_Malloc(controllen)) == NULL) {
 | 
						|
            PyErr_NoMemory();
 | 
						|
            goto finally;
 | 
						|
        }
 | 
						|
        msg.msg_controllen = controllen;
 | 
						|
 | 
						|
        /* Need to zero out the buffer as a workaround for glibc's
 | 
						|
           CMSG_NXTHDR() implementation.  After getting the pointer to
 | 
						|
           the next header, it checks its (uninitialized) cmsg_len
 | 
						|
           member to see if the "message" fits in the buffer, and
 | 
						|
           returns NULL if it doesn't.  Zero-filling the buffer
 | 
						|
           ensures that that doesn't happen. */
 | 
						|
        memset(controlbuf, 0, controllen);
 | 
						|
 | 
						|
        for (i = 0; i < ncmsgbufs; i++) {
 | 
						|
            size_t msg_len, data_len = cmsgs[i].data.len;
 | 
						|
            int enough_space = 0;
 | 
						|
 | 
						|
            cmsgh = (i == 0) ? CMSG_FIRSTHDR(&msg) : CMSG_NXTHDR(&msg, cmsgh);
 | 
						|
            if (cmsgh == NULL) {
 | 
						|
                PyErr_Format(PyExc_RuntimeError,
 | 
						|
                             "unexpected NULL result from %s()",
 | 
						|
                             (i == 0) ? "CMSG_FIRSTHDR" : "CMSG_NXTHDR");
 | 
						|
                goto finally;
 | 
						|
            }
 | 
						|
            if (!get_CMSG_LEN(data_len, &msg_len)) {
 | 
						|
                PyErr_SetString(PyExc_RuntimeError,
 | 
						|
                                "item size out of range for CMSG_LEN()");
 | 
						|
                goto finally;
 | 
						|
            }
 | 
						|
            if (cmsg_min_space(&msg, cmsgh, msg_len)) {
 | 
						|
                size_t space;
 | 
						|
 | 
						|
                cmsgh->cmsg_len = msg_len;
 | 
						|
                if (get_cmsg_data_space(&msg, cmsgh, &space))
 | 
						|
                    enough_space = (space >= data_len);
 | 
						|
            }
 | 
						|
            if (!enough_space) {
 | 
						|
                PyErr_SetString(PyExc_RuntimeError,
 | 
						|
                                "ancillary data does not fit in calculated "
 | 
						|
                                "space");
 | 
						|
                goto finally;
 | 
						|
            }
 | 
						|
            cmsgh->cmsg_level = cmsgs[i].level;
 | 
						|
            cmsgh->cmsg_type = cmsgs[i].type;
 | 
						|
            memcpy(CMSG_DATA(cmsgh), cmsgs[i].data.buf, data_len);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /* Make the system call. */
 | 
						|
    if (!IS_SELECTABLE(s)) {
 | 
						|
        select_error();
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
 | 
						|
    BEGIN_SELECT_LOOP(s)
 | 
						|
    Py_BEGIN_ALLOW_THREADS;
 | 
						|
    timeout = internal_select_ex(s, 1, interval);
 | 
						|
    if (!timeout)
 | 
						|
        bytes_sent = sendmsg(s->sock_fd, &msg, flags);
 | 
						|
    Py_END_ALLOW_THREADS;
 | 
						|
    if (timeout == 1) {
 | 
						|
        PyErr_SetString(socket_timeout, "timed out");
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
    END_SELECT_LOOP(s)
 | 
						|
 | 
						|
    if (bytes_sent < 0) {
 | 
						|
        s->errorhandler();
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
    retval = PyLong_FromSsize_t(bytes_sent);
 | 
						|
 | 
						|
finally:
 | 
						|
    PyMem_Free(controlbuf);
 | 
						|
    for (i = 0; i < ncmsgbufs; i++)
 | 
						|
        PyBuffer_Release(&cmsgs[i].data);
 | 
						|
    PyMem_Free(cmsgs);
 | 
						|
    Py_XDECREF(cmsg_fast);
 | 
						|
    for (i = 0; i < ndatabufs; i++)
 | 
						|
        PyBuffer_Release(&databufs[i]);
 | 
						|
    PyMem_Free(databufs);
 | 
						|
    PyMem_Free(iovs);
 | 
						|
    Py_XDECREF(data_fast);
 | 
						|
    return retval;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(sendmsg_doc,
 | 
						|
"sendmsg(buffers[, ancdata[, flags[, address]]]) -> count\n\
 | 
						|
\n\
 | 
						|
Send normal and ancillary data to the socket, gathering the\n\
 | 
						|
non-ancillary data from a series of buffers and concatenating it into\n\
 | 
						|
a single message.  The buffers argument specifies the non-ancillary\n\
 | 
						|
data as an iterable of buffer-compatible objects (e.g. bytes objects).\n\
 | 
						|
The ancdata argument specifies the ancillary data (control messages)\n\
 | 
						|
as an iterable of zero or more tuples (cmsg_level, cmsg_type,\n\
 | 
						|
cmsg_data), where cmsg_level and cmsg_type are integers specifying the\n\
 | 
						|
protocol level and protocol-specific type respectively, and cmsg_data\n\
 | 
						|
is a buffer-compatible object holding the associated data.  The flags\n\
 | 
						|
argument defaults to 0 and has the same meaning as for send().  If\n\
 | 
						|
address is supplied and not None, it sets a destination address for\n\
 | 
						|
the message.  The return value is the number of bytes of non-ancillary\n\
 | 
						|
data sent.");
 | 
						|
#endif    /* CMSG_LEN */
 | 
						|
 | 
						|
 | 
						|
/* s.shutdown(how) method */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_shutdown(PySocketSockObject *s, PyObject *arg)
 | 
						|
{
 | 
						|
    int how;
 | 
						|
    int res;
 | 
						|
 | 
						|
    how = PyLong_AsLong(arg);
 | 
						|
    if (how == -1 && PyErr_Occurred())
 | 
						|
        return NULL;
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    res = shutdown(s->sock_fd, how);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    if (res < 0)
 | 
						|
        return s->errorhandler();
 | 
						|
    Py_INCREF(Py_None);
 | 
						|
    return Py_None;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(shutdown_doc,
 | 
						|
"shutdown(flag)\n\
 | 
						|
\n\
 | 
						|
Shut down the reading side of the socket (flag == SHUT_RD), the writing side\n\
 | 
						|
of the socket (flag == SHUT_WR), or both ends (flag == SHUT_RDWR).");
 | 
						|
 | 
						|
#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
 | 
						|
static PyObject*
 | 
						|
sock_ioctl(PySocketSockObject *s, PyObject *arg)
 | 
						|
{
 | 
						|
    unsigned long cmd = SIO_RCVALL;
 | 
						|
    PyObject *argO;
 | 
						|
    DWORD recv;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(arg, "kO:ioctl", &cmd, &argO))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    switch (cmd) {
 | 
						|
    case SIO_RCVALL: {
 | 
						|
        unsigned int option = RCVALL_ON;
 | 
						|
        if (!PyArg_ParseTuple(arg, "kI:ioctl", &cmd, &option))
 | 
						|
            return NULL;
 | 
						|
        if (WSAIoctl(s->sock_fd, cmd, &option, sizeof(option),
 | 
						|
                         NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
 | 
						|
            return set_error();
 | 
						|
        }
 | 
						|
        return PyLong_FromUnsignedLong(recv); }
 | 
						|
    case SIO_KEEPALIVE_VALS: {
 | 
						|
        struct tcp_keepalive ka;
 | 
						|
        if (!PyArg_ParseTuple(arg, "k(kkk):ioctl", &cmd,
 | 
						|
                        &ka.onoff, &ka.keepalivetime, &ka.keepaliveinterval))
 | 
						|
            return NULL;
 | 
						|
        if (WSAIoctl(s->sock_fd, cmd, &ka, sizeof(ka),
 | 
						|
                         NULL, 0, &recv, NULL, NULL) == SOCKET_ERROR) {
 | 
						|
            return set_error();
 | 
						|
        }
 | 
						|
        return PyLong_FromUnsignedLong(recv); }
 | 
						|
    default:
 | 
						|
        PyErr_Format(PyExc_ValueError, "invalid ioctl command %d", cmd);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
}
 | 
						|
PyDoc_STRVAR(sock_ioctl_doc,
 | 
						|
"ioctl(cmd, option) -> long\n\
 | 
						|
\n\
 | 
						|
Control the socket with WSAIoctl syscall. Currently supported 'cmd' values are\n\
 | 
						|
SIO_RCVALL:  'option' must be one of the socket.RCVALL_* constants.\n\
 | 
						|
SIO_KEEPALIVE_VALS:  'option' is a tuple of (onoff, timeout, interval).");
 | 
						|
 | 
						|
#endif
 | 
						|
 | 
						|
/* List of methods for socket objects */
 | 
						|
 | 
						|
static PyMethodDef sock_methods[] = {
 | 
						|
    {"_accept",           (PyCFunction)sock_accept, METH_NOARGS,
 | 
						|
                      accept_doc},
 | 
						|
    {"bind",              (PyCFunction)sock_bind, METH_O,
 | 
						|
                      bind_doc},
 | 
						|
    {"close",             (PyCFunction)sock_close, METH_NOARGS,
 | 
						|
                      close_doc},
 | 
						|
    {"connect",           (PyCFunction)sock_connect, METH_O,
 | 
						|
                      connect_doc},
 | 
						|
    {"connect_ex",        (PyCFunction)sock_connect_ex, METH_O,
 | 
						|
                      connect_ex_doc},
 | 
						|
    {"detach",            (PyCFunction)sock_detach, METH_NOARGS,
 | 
						|
                      detach_doc},
 | 
						|
    {"fileno",            (PyCFunction)sock_fileno, METH_NOARGS,
 | 
						|
                      fileno_doc},
 | 
						|
#ifdef HAVE_GETPEERNAME
 | 
						|
    {"getpeername",       (PyCFunction)sock_getpeername,
 | 
						|
                      METH_NOARGS, getpeername_doc},
 | 
						|
#endif
 | 
						|
    {"getsockname",       (PyCFunction)sock_getsockname,
 | 
						|
                      METH_NOARGS, getsockname_doc},
 | 
						|
    {"getsockopt",        (PyCFunction)sock_getsockopt, METH_VARARGS,
 | 
						|
                      getsockopt_doc},
 | 
						|
#if defined(MS_WINDOWS) && defined(SIO_RCVALL)
 | 
						|
    {"ioctl",             (PyCFunction)sock_ioctl, METH_VARARGS,
 | 
						|
                      sock_ioctl_doc},
 | 
						|
#endif
 | 
						|
    {"listen",            (PyCFunction)sock_listen, METH_O,
 | 
						|
                      listen_doc},
 | 
						|
    {"recv",              (PyCFunction)sock_recv, METH_VARARGS,
 | 
						|
                      recv_doc},
 | 
						|
    {"recv_into",         (PyCFunction)sock_recv_into, METH_VARARGS | METH_KEYWORDS,
 | 
						|
                      recv_into_doc},
 | 
						|
    {"recvfrom",          (PyCFunction)sock_recvfrom, METH_VARARGS,
 | 
						|
                      recvfrom_doc},
 | 
						|
    {"recvfrom_into",  (PyCFunction)sock_recvfrom_into, METH_VARARGS | METH_KEYWORDS,
 | 
						|
                      recvfrom_into_doc},
 | 
						|
    {"send",              (PyCFunction)sock_send, METH_VARARGS,
 | 
						|
                      send_doc},
 | 
						|
    {"sendall",           (PyCFunction)sock_sendall, METH_VARARGS,
 | 
						|
                      sendall_doc},
 | 
						|
    {"sendto",            (PyCFunction)sock_sendto, METH_VARARGS,
 | 
						|
                      sendto_doc},
 | 
						|
    {"setblocking",       (PyCFunction)sock_setblocking, METH_O,
 | 
						|
                      setblocking_doc},
 | 
						|
    {"settimeout",    (PyCFunction)sock_settimeout, METH_O,
 | 
						|
                      settimeout_doc},
 | 
						|
    {"gettimeout",    (PyCFunction)sock_gettimeout, METH_NOARGS,
 | 
						|
                      gettimeout_doc},
 | 
						|
    {"setsockopt",        (PyCFunction)sock_setsockopt, METH_VARARGS,
 | 
						|
                      setsockopt_doc},
 | 
						|
    {"shutdown",          (PyCFunction)sock_shutdown, METH_O,
 | 
						|
                      shutdown_doc},
 | 
						|
#ifdef CMSG_LEN
 | 
						|
    {"recvmsg",           (PyCFunction)sock_recvmsg, METH_VARARGS,
 | 
						|
                      recvmsg_doc},
 | 
						|
    {"recvmsg_into",      (PyCFunction)sock_recvmsg_into, METH_VARARGS,
 | 
						|
                      recvmsg_into_doc,},
 | 
						|
    {"sendmsg",           (PyCFunction)sock_sendmsg, METH_VARARGS,
 | 
						|
                      sendmsg_doc},
 | 
						|
#endif
 | 
						|
    {NULL,                      NULL}           /* sentinel */
 | 
						|
};
 | 
						|
 | 
						|
/* SockObject members */
 | 
						|
static PyMemberDef sock_memberlist[] = {
 | 
						|
       {"family", T_INT, offsetof(PySocketSockObject, sock_family), READONLY, "the socket family"},
 | 
						|
       {"type", T_INT, offsetof(PySocketSockObject, sock_type), READONLY, "the socket type"},
 | 
						|
       {"proto", T_INT, offsetof(PySocketSockObject, sock_proto), READONLY, "the socket protocol"},
 | 
						|
       {"timeout", T_DOUBLE, offsetof(PySocketSockObject, sock_timeout), READONLY, "the socket timeout"},
 | 
						|
       {0},
 | 
						|
};
 | 
						|
 | 
						|
/* Deallocate a socket object in response to the last Py_DECREF().
 | 
						|
   First close the file description. */
 | 
						|
 | 
						|
static void
 | 
						|
sock_dealloc(PySocketSockObject *s)
 | 
						|
{
 | 
						|
    if (s->sock_fd != -1) {
 | 
						|
        PyObject *exc, *val, *tb;
 | 
						|
        Py_ssize_t old_refcount = Py_REFCNT(s);
 | 
						|
        ++Py_REFCNT(s);
 | 
						|
        PyErr_Fetch(&exc, &val, &tb);
 | 
						|
        if (PyErr_WarnFormat(PyExc_ResourceWarning, 1,
 | 
						|
                             "unclosed %R", s))
 | 
						|
            /* Spurious errors can appear at shutdown */
 | 
						|
            if (PyErr_ExceptionMatches(PyExc_Warning))
 | 
						|
                PyErr_WriteUnraisable((PyObject *) s);
 | 
						|
        PyErr_Restore(exc, val, tb);
 | 
						|
        (void) SOCKETCLOSE(s->sock_fd);
 | 
						|
        Py_REFCNT(s) = old_refcount;
 | 
						|
    }
 | 
						|
    Py_TYPE(s)->tp_free((PyObject *)s);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_repr(PySocketSockObject *s)
 | 
						|
{
 | 
						|
#if SIZEOF_SOCKET_T > SIZEOF_LONG
 | 
						|
    if (s->sock_fd > LONG_MAX) {
 | 
						|
        /* this can occur on Win64, and actually there is a special
 | 
						|
           ugly printf formatter for decimal pointer length integer
 | 
						|
           printing, only bother if necessary*/
 | 
						|
        PyErr_SetString(PyExc_OverflowError,
 | 
						|
                        "no printf formatter to display "
 | 
						|
                        "the socket descriptor in decimal");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
#endif
 | 
						|
    return PyUnicode_FromFormat(
 | 
						|
        "<socket object, fd=%ld, family=%d, type=%d, proto=%d>",
 | 
						|
        (long)s->sock_fd, s->sock_family,
 | 
						|
        s->sock_type,
 | 
						|
        s->sock_proto);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* Create a new, uninitialized socket object. */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
sock_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
 | 
						|
{
 | 
						|
    PyObject *new;
 | 
						|
 | 
						|
    new = type->tp_alloc(type, 0);
 | 
						|
    if (new != NULL) {
 | 
						|
        ((PySocketSockObject *)new)->sock_fd = -1;
 | 
						|
        ((PySocketSockObject *)new)->sock_timeout = -1.0;
 | 
						|
        ((PySocketSockObject *)new)->errorhandler = &set_error;
 | 
						|
    }
 | 
						|
    return new;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* Initialize a new socket object. */
 | 
						|
 | 
						|
/*ARGSUSED*/
 | 
						|
static int
 | 
						|
sock_initobj(PyObject *self, PyObject *args, PyObject *kwds)
 | 
						|
{
 | 
						|
    PySocketSockObject *s = (PySocketSockObject *)self;
 | 
						|
    PyObject *fdobj = NULL;
 | 
						|
    SOCKET_T fd = INVALID_SOCKET;
 | 
						|
    int family = AF_INET, type = SOCK_STREAM, proto = 0;
 | 
						|
    static char *keywords[] = {"family", "type", "proto", "fileno", 0};
 | 
						|
 | 
						|
    if (!PyArg_ParseTupleAndKeywords(args, kwds,
 | 
						|
                                     "|iiiO:socket", keywords,
 | 
						|
                                     &family, &type, &proto, &fdobj))
 | 
						|
        return -1;
 | 
						|
 | 
						|
    if (fdobj != NULL && fdobj != Py_None) {
 | 
						|
        fd = PyLong_AsSocket_t(fdobj);
 | 
						|
        if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
 | 
						|
            return -1;
 | 
						|
        if (fd == INVALID_SOCKET) {
 | 
						|
            PyErr_SetString(PyExc_ValueError,
 | 
						|
                            "can't use invalid socket value");
 | 
						|
            return -1;
 | 
						|
        }
 | 
						|
    }
 | 
						|
    else {
 | 
						|
        Py_BEGIN_ALLOW_THREADS
 | 
						|
        fd = socket(family, type, proto);
 | 
						|
        Py_END_ALLOW_THREADS
 | 
						|
 | 
						|
        if (fd == INVALID_SOCKET) {
 | 
						|
            set_error();
 | 
						|
            return -1;
 | 
						|
        }
 | 
						|
    }
 | 
						|
    init_sockobject(s, fd, family, type, proto);
 | 
						|
 | 
						|
    return 0;
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* Type object for socket objects. */
 | 
						|
 | 
						|
static PyTypeObject sock_type = {
 | 
						|
    PyVarObject_HEAD_INIT(0, 0)         /* Must fill in type value later */
 | 
						|
    "_socket.socket",                           /* tp_name */
 | 
						|
    sizeof(PySocketSockObject),                 /* tp_basicsize */
 | 
						|
    0,                                          /* tp_itemsize */
 | 
						|
    (destructor)sock_dealloc,                   /* tp_dealloc */
 | 
						|
    0,                                          /* tp_print */
 | 
						|
    0,                                          /* tp_getattr */
 | 
						|
    0,                                          /* tp_setattr */
 | 
						|
    0,                                          /* tp_reserved */
 | 
						|
    (reprfunc)sock_repr,                        /* tp_repr */
 | 
						|
    0,                                          /* tp_as_number */
 | 
						|
    0,                                          /* tp_as_sequence */
 | 
						|
    0,                                          /* tp_as_mapping */
 | 
						|
    0,                                          /* tp_hash */
 | 
						|
    0,                                          /* tp_call */
 | 
						|
    0,                                          /* tp_str */
 | 
						|
    PyObject_GenericGetAttr,                    /* tp_getattro */
 | 
						|
    0,                                          /* tp_setattro */
 | 
						|
    0,                                          /* tp_as_buffer */
 | 
						|
    Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
 | 
						|
    sock_doc,                                   /* tp_doc */
 | 
						|
    0,                                          /* tp_traverse */
 | 
						|
    0,                                          /* tp_clear */
 | 
						|
    0,                                          /* tp_richcompare */
 | 
						|
    0,                                          /* tp_weaklistoffset */
 | 
						|
    0,                                          /* tp_iter */
 | 
						|
    0,                                          /* tp_iternext */
 | 
						|
    sock_methods,                               /* tp_methods */
 | 
						|
    sock_memberlist,                            /* tp_members */
 | 
						|
    0,                                          /* tp_getset */
 | 
						|
    0,                                          /* tp_base */
 | 
						|
    0,                                          /* tp_dict */
 | 
						|
    0,                                          /* tp_descr_get */
 | 
						|
    0,                                          /* tp_descr_set */
 | 
						|
    0,                                          /* tp_dictoffset */
 | 
						|
    sock_initobj,                               /* tp_init */
 | 
						|
    PyType_GenericAlloc,                        /* tp_alloc */
 | 
						|
    sock_new,                                   /* tp_new */
 | 
						|
    PyObject_Del,                               /* tp_free */
 | 
						|
};
 | 
						|
 | 
						|
 | 
						|
/* Python interface to gethostname(). */
 | 
						|
 | 
						|
/*ARGSUSED*/
 | 
						|
static PyObject *
 | 
						|
socket_gethostname(PyObject *self, PyObject *unused)
 | 
						|
{
 | 
						|
#ifdef MS_WINDOWS
 | 
						|
    /* Don't use winsock's gethostname, as this returns the ANSI
 | 
						|
       version of the hostname, whereas we need a Unicode string.
 | 
						|
       Otherwise, gethostname apparently also returns the DNS name. */
 | 
						|
    wchar_t buf[MAX_COMPUTERNAME_LENGTH + 1];
 | 
						|
    DWORD size = Py_ARRAY_LENGTH(buf);
 | 
						|
    PyObject *result;
 | 
						|
    if (!GetComputerNameExW(ComputerNamePhysicalDnsHostname, buf, &size)) {
 | 
						|
        if (GetLastError() == ERROR_MORE_DATA) {
 | 
						|
            /* MSDN says this may occur "because DNS allows longer names */
 | 
						|
            if (size == 0) /* XXX: I'm not sure how to handle this */
 | 
						|
                return PyUnicode_FromUnicode(NULL, 0);
 | 
						|
            result = PyUnicode_FromUnicode(NULL, size - 1);
 | 
						|
            if (!result)
 | 
						|
                return NULL;
 | 
						|
            if (GetComputerNameExW(ComputerNamePhysicalDnsHostname,
 | 
						|
                                   PyUnicode_AS_UNICODE(result),
 | 
						|
                                   &size))
 | 
						|
                return result;
 | 
						|
            Py_DECREF(result);
 | 
						|
        }
 | 
						|
        return PyErr_SetExcFromWindowsErr(PyExc_WindowsError, GetLastError());
 | 
						|
    }
 | 
						|
    return PyUnicode_FromUnicode(buf, size);
 | 
						|
#else
 | 
						|
    char buf[1024];
 | 
						|
    int res;
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    res = gethostname(buf, (int) sizeof buf - 1);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    if (res < 0)
 | 
						|
        return set_error();
 | 
						|
    buf[sizeof buf - 1] = '\0';
 | 
						|
    return PyUnicode_FromString(buf);
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(gethostname_doc,
 | 
						|
"gethostname() -> string\n\
 | 
						|
\n\
 | 
						|
Return the current host name.");
 | 
						|
 | 
						|
#ifdef HAVE_SETHOSTNAME
 | 
						|
PyDoc_STRVAR(sethostname_doc,
 | 
						|
"sethostname(name)\n\n\
 | 
						|
Sets the hostname to name.");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_sethostname(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    PyObject *hnobj;
 | 
						|
    Py_buffer buf;
 | 
						|
    int res, flag = 0;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "S:sethostname", &hnobj)) {
 | 
						|
        PyErr_Clear();
 | 
						|
        if (!PyArg_ParseTuple(args, "O&:sethostname",
 | 
						|
                PyUnicode_FSConverter, &hnobj))
 | 
						|
            return NULL;
 | 
						|
        flag = 1;
 | 
						|
    }
 | 
						|
    res = PyObject_GetBuffer(hnobj, &buf, PyBUF_SIMPLE);
 | 
						|
    if (!res) {
 | 
						|
        res = sethostname(buf.buf, buf.len);
 | 
						|
        PyBuffer_Release(&buf);
 | 
						|
    }
 | 
						|
    if (flag)
 | 
						|
        Py_DECREF(hnobj);
 | 
						|
    if (res)
 | 
						|
        return set_error();
 | 
						|
    Py_RETURN_NONE;
 | 
						|
}
 | 
						|
#endif
 | 
						|
 | 
						|
/* Python interface to gethostbyname(name). */
 | 
						|
 | 
						|
/*ARGSUSED*/
 | 
						|
static PyObject *
 | 
						|
socket_gethostbyname(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    char *name;
 | 
						|
    sock_addr_t addrbuf;
 | 
						|
    PyObject *ret = NULL;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "et:gethostbyname", "idna", &name))
 | 
						|
        return NULL;
 | 
						|
    if (setipaddr(name, SAS2SA(&addrbuf),  sizeof(addrbuf), AF_INET) < 0)
 | 
						|
        goto finally;
 | 
						|
    ret = makeipaddr(SAS2SA(&addrbuf), sizeof(struct sockaddr_in));
 | 
						|
finally:
 | 
						|
    PyMem_Free(name);
 | 
						|
    return ret;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(gethostbyname_doc,
 | 
						|
"gethostbyname(host) -> address\n\
 | 
						|
\n\
 | 
						|
Return the IP address (a string of the form '255.255.255.255') for a host.");
 | 
						|
 | 
						|
 | 
						|
/* Convenience function common to gethostbyname_ex and gethostbyaddr */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
gethost_common(struct hostent *h, struct sockaddr *addr, int alen, int af)
 | 
						|
{
 | 
						|
    char **pch;
 | 
						|
    PyObject *rtn_tuple = (PyObject *)NULL;
 | 
						|
    PyObject *name_list = (PyObject *)NULL;
 | 
						|
    PyObject *addr_list = (PyObject *)NULL;
 | 
						|
    PyObject *tmp;
 | 
						|
 | 
						|
    if (h == NULL) {
 | 
						|
        /* Let's get real error message to return */
 | 
						|
        set_herror(h_errno);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    if (h->h_addrtype != af) {
 | 
						|
        /* Let's get real error message to return */
 | 
						|
        PyErr_SetString(socket_error,
 | 
						|
                        (char *)strerror(EAFNOSUPPORT));
 | 
						|
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    switch (af) {
 | 
						|
 | 
						|
    case AF_INET:
 | 
						|
        if (alen < sizeof(struct sockaddr_in))
 | 
						|
            return NULL;
 | 
						|
        break;
 | 
						|
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
    case AF_INET6:
 | 
						|
        if (alen < sizeof(struct sockaddr_in6))
 | 
						|
            return NULL;
 | 
						|
        break;
 | 
						|
#endif
 | 
						|
 | 
						|
    }
 | 
						|
 | 
						|
    if ((name_list = PyList_New(0)) == NULL)
 | 
						|
        goto err;
 | 
						|
 | 
						|
    if ((addr_list = PyList_New(0)) == NULL)
 | 
						|
        goto err;
 | 
						|
 | 
						|
    /* SF #1511317: h_aliases can be NULL */
 | 
						|
    if (h->h_aliases) {
 | 
						|
        for (pch = h->h_aliases; *pch != NULL; pch++) {
 | 
						|
            int status;
 | 
						|
            tmp = PyUnicode_FromString(*pch);
 | 
						|
            if (tmp == NULL)
 | 
						|
                goto err;
 | 
						|
 | 
						|
            status = PyList_Append(name_list, tmp);
 | 
						|
            Py_DECREF(tmp);
 | 
						|
 | 
						|
            if (status)
 | 
						|
                goto err;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    for (pch = h->h_addr_list; *pch != NULL; pch++) {
 | 
						|
        int status;
 | 
						|
 | 
						|
        switch (af) {
 | 
						|
 | 
						|
        case AF_INET:
 | 
						|
            {
 | 
						|
            struct sockaddr_in sin;
 | 
						|
            memset(&sin, 0, sizeof(sin));
 | 
						|
            sin.sin_family = af;
 | 
						|
#ifdef HAVE_SOCKADDR_SA_LEN
 | 
						|
            sin.sin_len = sizeof(sin);
 | 
						|
#endif
 | 
						|
            memcpy(&sin.sin_addr, *pch, sizeof(sin.sin_addr));
 | 
						|
            tmp = makeipaddr((struct sockaddr *)&sin, sizeof(sin));
 | 
						|
 | 
						|
            if (pch == h->h_addr_list && alen >= sizeof(sin))
 | 
						|
                memcpy((char *) addr, &sin, sizeof(sin));
 | 
						|
            break;
 | 
						|
            }
 | 
						|
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
        case AF_INET6:
 | 
						|
            {
 | 
						|
            struct sockaddr_in6 sin6;
 | 
						|
            memset(&sin6, 0, sizeof(sin6));
 | 
						|
            sin6.sin6_family = af;
 | 
						|
#ifdef HAVE_SOCKADDR_SA_LEN
 | 
						|
            sin6.sin6_len = sizeof(sin6);
 | 
						|
#endif
 | 
						|
            memcpy(&sin6.sin6_addr, *pch, sizeof(sin6.sin6_addr));
 | 
						|
            tmp = makeipaddr((struct sockaddr *)&sin6,
 | 
						|
                sizeof(sin6));
 | 
						|
 | 
						|
            if (pch == h->h_addr_list && alen >= sizeof(sin6))
 | 
						|
                memcpy((char *) addr, &sin6, sizeof(sin6));
 | 
						|
            break;
 | 
						|
            }
 | 
						|
#endif
 | 
						|
 | 
						|
        default:                /* can't happen */
 | 
						|
            PyErr_SetString(socket_error,
 | 
						|
                            "unsupported address family");
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
 | 
						|
        if (tmp == NULL)
 | 
						|
            goto err;
 | 
						|
 | 
						|
        status = PyList_Append(addr_list, tmp);
 | 
						|
        Py_DECREF(tmp);
 | 
						|
 | 
						|
        if (status)
 | 
						|
            goto err;
 | 
						|
    }
 | 
						|
 | 
						|
    rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list);
 | 
						|
 | 
						|
 err:
 | 
						|
    Py_XDECREF(name_list);
 | 
						|
    Py_XDECREF(addr_list);
 | 
						|
    return rtn_tuple;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* Python interface to gethostbyname_ex(name). */
 | 
						|
 | 
						|
/*ARGSUSED*/
 | 
						|
static PyObject *
 | 
						|
socket_gethostbyname_ex(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    char *name;
 | 
						|
    struct hostent *h;
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
    struct sockaddr_storage addr;
 | 
						|
#else
 | 
						|
    struct sockaddr_in addr;
 | 
						|
#endif
 | 
						|
    struct sockaddr *sa;
 | 
						|
    PyObject *ret = NULL;
 | 
						|
#ifdef HAVE_GETHOSTBYNAME_R
 | 
						|
    struct hostent hp_allocated;
 | 
						|
#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
 | 
						|
    struct hostent_data data;
 | 
						|
#else
 | 
						|
    char buf[16384];
 | 
						|
    int buf_len = (sizeof buf) - 1;
 | 
						|
    int errnop;
 | 
						|
#endif
 | 
						|
#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
 | 
						|
    int result;
 | 
						|
#endif
 | 
						|
#endif /* HAVE_GETHOSTBYNAME_R */
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "et:gethostbyname_ex", "idna", &name))
 | 
						|
        return NULL;
 | 
						|
    if (setipaddr(name, (struct sockaddr *)&addr, sizeof(addr), AF_INET) < 0)
 | 
						|
        goto finally;
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
#ifdef HAVE_GETHOSTBYNAME_R
 | 
						|
#if   defined(HAVE_GETHOSTBYNAME_R_6_ARG)
 | 
						|
    result = gethostbyname_r(name, &hp_allocated, buf, buf_len,
 | 
						|
                             &h, &errnop);
 | 
						|
#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
 | 
						|
    h = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop);
 | 
						|
#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
 | 
						|
    memset((void *) &data, '\0', sizeof(data));
 | 
						|
    result = gethostbyname_r(name, &hp_allocated, &data);
 | 
						|
    h = (result != 0) ? NULL : &hp_allocated;
 | 
						|
#endif
 | 
						|
#else /* not HAVE_GETHOSTBYNAME_R */
 | 
						|
#ifdef USE_GETHOSTBYNAME_LOCK
 | 
						|
    PyThread_acquire_lock(netdb_lock, 1);
 | 
						|
#endif
 | 
						|
    h = gethostbyname(name);
 | 
						|
#endif /* HAVE_GETHOSTBYNAME_R */
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    /* Some C libraries would require addr.__ss_family instead of
 | 
						|
       addr.ss_family.
 | 
						|
       Therefore, we cast the sockaddr_storage into sockaddr to
 | 
						|
       access sa_family. */
 | 
						|
    sa = (struct sockaddr*)&addr;
 | 
						|
    ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr),
 | 
						|
                         sa->sa_family);
 | 
						|
#ifdef USE_GETHOSTBYNAME_LOCK
 | 
						|
    PyThread_release_lock(netdb_lock);
 | 
						|
#endif
 | 
						|
finally:
 | 
						|
    PyMem_Free(name);
 | 
						|
    return ret;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(ghbn_ex_doc,
 | 
						|
"gethostbyname_ex(host) -> (name, aliaslist, addresslist)\n\
 | 
						|
\n\
 | 
						|
Return the true host name, a list of aliases, and a list of IP addresses,\n\
 | 
						|
for a host.  The host argument is a string giving a host name or IP number.");
 | 
						|
 | 
						|
 | 
						|
/* Python interface to gethostbyaddr(IP). */
 | 
						|
 | 
						|
/*ARGSUSED*/
 | 
						|
static PyObject *
 | 
						|
socket_gethostbyaddr(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
    struct sockaddr_storage addr;
 | 
						|
#else
 | 
						|
    struct sockaddr_in addr;
 | 
						|
#endif
 | 
						|
    struct sockaddr *sa = (struct sockaddr *)&addr;
 | 
						|
    char *ip_num;
 | 
						|
    struct hostent *h;
 | 
						|
    PyObject *ret = NULL;
 | 
						|
#ifdef HAVE_GETHOSTBYNAME_R
 | 
						|
    struct hostent hp_allocated;
 | 
						|
#ifdef HAVE_GETHOSTBYNAME_R_3_ARG
 | 
						|
    struct hostent_data data;
 | 
						|
#else
 | 
						|
    /* glibcs up to 2.10 assume that the buf argument to
 | 
						|
       gethostbyaddr_r is 8-byte aligned, which at least llvm-gcc
 | 
						|
       does not ensure. The attribute below instructs the compiler
 | 
						|
       to maintain this alignment. */
 | 
						|
    char buf[16384] Py_ALIGNED(8);
 | 
						|
    int buf_len = (sizeof buf) - 1;
 | 
						|
    int errnop;
 | 
						|
#endif
 | 
						|
#if defined(HAVE_GETHOSTBYNAME_R_3_ARG) || defined(HAVE_GETHOSTBYNAME_R_6_ARG)
 | 
						|
    int result;
 | 
						|
#endif
 | 
						|
#endif /* HAVE_GETHOSTBYNAME_R */
 | 
						|
    char *ap;
 | 
						|
    int al;
 | 
						|
    int af;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "et:gethostbyaddr", "idna", &ip_num))
 | 
						|
        return NULL;
 | 
						|
    af = AF_UNSPEC;
 | 
						|
    if (setipaddr(ip_num, sa, sizeof(addr), af) < 0)
 | 
						|
        goto finally;
 | 
						|
    af = sa->sa_family;
 | 
						|
    ap = NULL;
 | 
						|
    /* al = 0; */
 | 
						|
    switch (af) {
 | 
						|
    case AF_INET:
 | 
						|
        ap = (char *)&((struct sockaddr_in *)sa)->sin_addr;
 | 
						|
        al = sizeof(((struct sockaddr_in *)sa)->sin_addr);
 | 
						|
        break;
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
    case AF_INET6:
 | 
						|
        ap = (char *)&((struct sockaddr_in6 *)sa)->sin6_addr;
 | 
						|
        al = sizeof(((struct sockaddr_in6 *)sa)->sin6_addr);
 | 
						|
        break;
 | 
						|
#endif
 | 
						|
    default:
 | 
						|
        PyErr_SetString(socket_error, "unsupported address family");
 | 
						|
        goto finally;
 | 
						|
    }
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
#ifdef HAVE_GETHOSTBYNAME_R
 | 
						|
#if   defined(HAVE_GETHOSTBYNAME_R_6_ARG)
 | 
						|
    result = gethostbyaddr_r(ap, al, af,
 | 
						|
        &hp_allocated, buf, buf_len,
 | 
						|
        &h, &errnop);
 | 
						|
#elif defined(HAVE_GETHOSTBYNAME_R_5_ARG)
 | 
						|
    h = gethostbyaddr_r(ap, al, af,
 | 
						|
                        &hp_allocated, buf, buf_len, &errnop);
 | 
						|
#else /* HAVE_GETHOSTBYNAME_R_3_ARG */
 | 
						|
    memset((void *) &data, '\0', sizeof(data));
 | 
						|
    result = gethostbyaddr_r(ap, al, af, &hp_allocated, &data);
 | 
						|
    h = (result != 0) ? NULL : &hp_allocated;
 | 
						|
#endif
 | 
						|
#else /* not HAVE_GETHOSTBYNAME_R */
 | 
						|
#ifdef USE_GETHOSTBYNAME_LOCK
 | 
						|
    PyThread_acquire_lock(netdb_lock, 1);
 | 
						|
#endif
 | 
						|
    h = gethostbyaddr(ap, al, af);
 | 
						|
#endif /* HAVE_GETHOSTBYNAME_R */
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    ret = gethost_common(h, (struct sockaddr *)&addr, sizeof(addr), af);
 | 
						|
#ifdef USE_GETHOSTBYNAME_LOCK
 | 
						|
    PyThread_release_lock(netdb_lock);
 | 
						|
#endif
 | 
						|
finally:
 | 
						|
    PyMem_Free(ip_num);
 | 
						|
    return ret;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(gethostbyaddr_doc,
 | 
						|
"gethostbyaddr(host) -> (name, aliaslist, addresslist)\n\
 | 
						|
\n\
 | 
						|
Return the true host name, a list of aliases, and a list of IP addresses,\n\
 | 
						|
for a host.  The host argument is a string giving a host name or IP number.");
 | 
						|
 | 
						|
 | 
						|
/* Python interface to getservbyname(name).
 | 
						|
   This only returns the port number, since the other info is already
 | 
						|
   known or not useful (like the list of aliases). */
 | 
						|
 | 
						|
/*ARGSUSED*/
 | 
						|
static PyObject *
 | 
						|
socket_getservbyname(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    char *name, *proto=NULL;
 | 
						|
    struct servent *sp;
 | 
						|
    if (!PyArg_ParseTuple(args, "s|s:getservbyname", &name, &proto))
 | 
						|
        return NULL;
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    sp = getservbyname(name, proto);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    if (sp == NULL) {
 | 
						|
        PyErr_SetString(socket_error, "service/proto not found");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    return PyLong_FromLong((long) ntohs(sp->s_port));
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(getservbyname_doc,
 | 
						|
"getservbyname(servicename[, protocolname]) -> integer\n\
 | 
						|
\n\
 | 
						|
Return a port number from a service name and protocol name.\n\
 | 
						|
The optional protocol name, if given, should be 'tcp' or 'udp',\n\
 | 
						|
otherwise any protocol will match.");
 | 
						|
 | 
						|
 | 
						|
/* Python interface to getservbyport(port).
 | 
						|
   This only returns the service name, since the other info is already
 | 
						|
   known or not useful (like the list of aliases). */
 | 
						|
 | 
						|
/*ARGSUSED*/
 | 
						|
static PyObject *
 | 
						|
socket_getservbyport(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    int port;
 | 
						|
    char *proto=NULL;
 | 
						|
    struct servent *sp;
 | 
						|
    if (!PyArg_ParseTuple(args, "i|s:getservbyport", &port, &proto))
 | 
						|
        return NULL;
 | 
						|
    if (port < 0 || port > 0xffff) {
 | 
						|
        PyErr_SetString(
 | 
						|
            PyExc_OverflowError,
 | 
						|
            "getservbyport: port must be 0-65535.");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    sp = getservbyport(htons((short)port), proto);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    if (sp == NULL) {
 | 
						|
        PyErr_SetString(socket_error, "port/proto not found");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    return PyUnicode_FromString(sp->s_name);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(getservbyport_doc,
 | 
						|
"getservbyport(port[, protocolname]) -> string\n\
 | 
						|
\n\
 | 
						|
Return the service name from a port number and protocol name.\n\
 | 
						|
The optional protocol name, if given, should be 'tcp' or 'udp',\n\
 | 
						|
otherwise any protocol will match.");
 | 
						|
 | 
						|
/* Python interface to getprotobyname(name).
 | 
						|
   This only returns the protocol number, since the other info is
 | 
						|
   already known or not useful (like the list of aliases). */
 | 
						|
 | 
						|
/*ARGSUSED*/
 | 
						|
static PyObject *
 | 
						|
socket_getprotobyname(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    char *name;
 | 
						|
    struct protoent *sp;
 | 
						|
    if (!PyArg_ParseTuple(args, "s:getprotobyname", &name))
 | 
						|
        return NULL;
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    sp = getprotobyname(name);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    if (sp == NULL) {
 | 
						|
        PyErr_SetString(socket_error, "protocol not found");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    return PyLong_FromLong((long) sp->p_proto);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(getprotobyname_doc,
 | 
						|
"getprotobyname(name) -> integer\n\
 | 
						|
\n\
 | 
						|
Return the protocol number for the named protocol.  (Rarely used.)");
 | 
						|
 | 
						|
 | 
						|
#ifndef NO_DUP
 | 
						|
/* dup() function for socket fds */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_dup(PyObject *self, PyObject *fdobj)
 | 
						|
{
 | 
						|
    SOCKET_T fd, newfd;
 | 
						|
    PyObject *newfdobj;
 | 
						|
 | 
						|
 | 
						|
    fd = PyLong_AsSocket_t(fdobj);
 | 
						|
    if (fd == (SOCKET_T)(-1) && PyErr_Occurred())
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    newfd = dup_socket(fd);
 | 
						|
    if (newfd == INVALID_SOCKET)
 | 
						|
        return set_error();
 | 
						|
 | 
						|
    newfdobj = PyLong_FromSocket_t(newfd);
 | 
						|
    if (newfdobj == NULL)
 | 
						|
        SOCKETCLOSE(newfd);
 | 
						|
    return newfdobj;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(dup_doc,
 | 
						|
"dup(integer) -> integer\n\
 | 
						|
\n\
 | 
						|
Duplicate an integer socket file descriptor.  This is like os.dup(), but for\n\
 | 
						|
sockets; on some platforms os.dup() won't work for socket file descriptors.");
 | 
						|
#endif
 | 
						|
 | 
						|
 | 
						|
#ifdef HAVE_SOCKETPAIR
 | 
						|
/* Create a pair of sockets using the socketpair() function.
 | 
						|
   Arguments as for socket() except the default family is AF_UNIX if
 | 
						|
   defined on the platform; otherwise, the default is AF_INET. */
 | 
						|
 | 
						|
/*ARGSUSED*/
 | 
						|
static PyObject *
 | 
						|
socket_socketpair(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    PySocketSockObject *s0 = NULL, *s1 = NULL;
 | 
						|
    SOCKET_T sv[2];
 | 
						|
    int family, type = SOCK_STREAM, proto = 0;
 | 
						|
    PyObject *res = NULL;
 | 
						|
 | 
						|
#if defined(AF_UNIX)
 | 
						|
    family = AF_UNIX;
 | 
						|
#else
 | 
						|
    family = AF_INET;
 | 
						|
#endif
 | 
						|
    if (!PyArg_ParseTuple(args, "|iii:socketpair",
 | 
						|
                          &family, &type, &proto))
 | 
						|
        return NULL;
 | 
						|
    /* Create a pair of socket fds */
 | 
						|
    if (socketpair(family, type, proto, sv) < 0)
 | 
						|
        return set_error();
 | 
						|
    s0 = new_sockobject(sv[0], family, type, proto);
 | 
						|
    if (s0 == NULL)
 | 
						|
        goto finally;
 | 
						|
    s1 = new_sockobject(sv[1], family, type, proto);
 | 
						|
    if (s1 == NULL)
 | 
						|
        goto finally;
 | 
						|
    res = PyTuple_Pack(2, s0, s1);
 | 
						|
 | 
						|
finally:
 | 
						|
    if (res == NULL) {
 | 
						|
        if (s0 == NULL)
 | 
						|
            SOCKETCLOSE(sv[0]);
 | 
						|
        if (s1 == NULL)
 | 
						|
            SOCKETCLOSE(sv[1]);
 | 
						|
    }
 | 
						|
    Py_XDECREF(s0);
 | 
						|
    Py_XDECREF(s1);
 | 
						|
    return res;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(socketpair_doc,
 | 
						|
"socketpair([family[, type[, proto]]]) -> (socket object, socket object)\n\
 | 
						|
\n\
 | 
						|
Create a pair of socket objects from the sockets returned by the platform\n\
 | 
						|
socketpair() function.\n\
 | 
						|
The arguments are the same as for socket() except the default family is\n\
 | 
						|
AF_UNIX if defined on the platform; otherwise, the default is AF_INET.");
 | 
						|
 | 
						|
#endif /* HAVE_SOCKETPAIR */
 | 
						|
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_ntohs(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    int x1, x2;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "i:ntohs", &x1)) {
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    if (x1 < 0) {
 | 
						|
        PyErr_SetString(PyExc_OverflowError,
 | 
						|
            "can't convert negative number to unsigned long");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    x2 = (unsigned int)ntohs((unsigned short)x1);
 | 
						|
    return PyLong_FromLong(x2);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(ntohs_doc,
 | 
						|
"ntohs(integer) -> integer\n\
 | 
						|
\n\
 | 
						|
Convert a 16-bit integer from network to host byte order.");
 | 
						|
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_ntohl(PyObject *self, PyObject *arg)
 | 
						|
{
 | 
						|
    unsigned long x;
 | 
						|
 | 
						|
    if (PyLong_Check(arg)) {
 | 
						|
        x = PyLong_AsUnsignedLong(arg);
 | 
						|
        if (x == (unsigned long) -1 && PyErr_Occurred())
 | 
						|
            return NULL;
 | 
						|
#if SIZEOF_LONG > 4
 | 
						|
        {
 | 
						|
            unsigned long y;
 | 
						|
            /* only want the trailing 32 bits */
 | 
						|
            y = x & 0xFFFFFFFFUL;
 | 
						|
            if (y ^ x)
 | 
						|
                return PyErr_Format(PyExc_OverflowError,
 | 
						|
                            "long int larger than 32 bits");
 | 
						|
            x = y;
 | 
						|
        }
 | 
						|
#endif
 | 
						|
    }
 | 
						|
    else
 | 
						|
        return PyErr_Format(PyExc_TypeError,
 | 
						|
                            "expected int/long, %s found",
 | 
						|
                            Py_TYPE(arg)->tp_name);
 | 
						|
    if (x == (unsigned long) -1 && PyErr_Occurred())
 | 
						|
        return NULL;
 | 
						|
    return PyLong_FromUnsignedLong(ntohl(x));
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(ntohl_doc,
 | 
						|
"ntohl(integer) -> integer\n\
 | 
						|
\n\
 | 
						|
Convert a 32-bit integer from network to host byte order.");
 | 
						|
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_htons(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    int x1, x2;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "i:htons", &x1)) {
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    if (x1 < 0) {
 | 
						|
        PyErr_SetString(PyExc_OverflowError,
 | 
						|
            "can't convert negative number to unsigned long");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    x2 = (unsigned int)htons((unsigned short)x1);
 | 
						|
    return PyLong_FromLong(x2);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(htons_doc,
 | 
						|
"htons(integer) -> integer\n\
 | 
						|
\n\
 | 
						|
Convert a 16-bit integer from host to network byte order.");
 | 
						|
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_htonl(PyObject *self, PyObject *arg)
 | 
						|
{
 | 
						|
    unsigned long x;
 | 
						|
 | 
						|
    if (PyLong_Check(arg)) {
 | 
						|
        x = PyLong_AsUnsignedLong(arg);
 | 
						|
        if (x == (unsigned long) -1 && PyErr_Occurred())
 | 
						|
            return NULL;
 | 
						|
#if SIZEOF_LONG > 4
 | 
						|
        {
 | 
						|
            unsigned long y;
 | 
						|
            /* only want the trailing 32 bits */
 | 
						|
            y = x & 0xFFFFFFFFUL;
 | 
						|
            if (y ^ x)
 | 
						|
                return PyErr_Format(PyExc_OverflowError,
 | 
						|
                            "long int larger than 32 bits");
 | 
						|
            x = y;
 | 
						|
        }
 | 
						|
#endif
 | 
						|
    }
 | 
						|
    else
 | 
						|
        return PyErr_Format(PyExc_TypeError,
 | 
						|
                            "expected int/long, %s found",
 | 
						|
                            Py_TYPE(arg)->tp_name);
 | 
						|
    return PyLong_FromUnsignedLong(htonl((unsigned long)x));
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(htonl_doc,
 | 
						|
"htonl(integer) -> integer\n\
 | 
						|
\n\
 | 
						|
Convert a 32-bit integer from host to network byte order.");
 | 
						|
 | 
						|
/* socket.inet_aton() and socket.inet_ntoa() functions. */
 | 
						|
 | 
						|
PyDoc_STRVAR(inet_aton_doc,
 | 
						|
"inet_aton(string) -> bytes giving packed 32-bit IP representation\n\
 | 
						|
\n\
 | 
						|
Convert an IP address in string format (123.45.67.89) to the 32-bit packed\n\
 | 
						|
binary format used in low-level network functions.");
 | 
						|
 | 
						|
static PyObject*
 | 
						|
socket_inet_aton(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
#ifndef INADDR_NONE
 | 
						|
#define INADDR_NONE (-1)
 | 
						|
#endif
 | 
						|
#ifdef HAVE_INET_ATON
 | 
						|
    struct in_addr buf;
 | 
						|
#endif
 | 
						|
 | 
						|
#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
 | 
						|
#if (SIZEOF_INT != 4)
 | 
						|
#error "Not sure if in_addr_t exists and int is not 32-bits."
 | 
						|
#endif
 | 
						|
    /* Have to use inet_addr() instead */
 | 
						|
    unsigned int packed_addr;
 | 
						|
#endif
 | 
						|
    char *ip_addr;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "s:inet_aton", &ip_addr))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
 | 
						|
#ifdef HAVE_INET_ATON
 | 
						|
 | 
						|
#ifdef USE_INET_ATON_WEAKLINK
 | 
						|
    if (inet_aton != NULL) {
 | 
						|
#endif
 | 
						|
    if (inet_aton(ip_addr, &buf))
 | 
						|
        return PyBytes_FromStringAndSize((char *)(&buf),
 | 
						|
                                          sizeof(buf));
 | 
						|
 | 
						|
    PyErr_SetString(socket_error,
 | 
						|
                    "illegal IP address string passed to inet_aton");
 | 
						|
    return NULL;
 | 
						|
 | 
						|
#ifdef USE_INET_ATON_WEAKLINK
 | 
						|
   } else {
 | 
						|
#endif
 | 
						|
 | 
						|
#endif
 | 
						|
 | 
						|
#if !defined(HAVE_INET_ATON) || defined(USE_INET_ATON_WEAKLINK)
 | 
						|
 | 
						|
    /* special-case this address as inet_addr might return INADDR_NONE
 | 
						|
     * for this */
 | 
						|
    if (strcmp(ip_addr, "255.255.255.255") == 0) {
 | 
						|
        packed_addr = 0xFFFFFFFF;
 | 
						|
    } else {
 | 
						|
 | 
						|
        packed_addr = inet_addr(ip_addr);
 | 
						|
 | 
						|
        if (packed_addr == INADDR_NONE) {               /* invalid address */
 | 
						|
            PyErr_SetString(socket_error,
 | 
						|
                "illegal IP address string passed to inet_aton");
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
    }
 | 
						|
    return PyBytes_FromStringAndSize((char *) &packed_addr,
 | 
						|
                                      sizeof(packed_addr));
 | 
						|
 | 
						|
#ifdef USE_INET_ATON_WEAKLINK
 | 
						|
   }
 | 
						|
#endif
 | 
						|
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(inet_ntoa_doc,
 | 
						|
"inet_ntoa(packed_ip) -> ip_address_string\n\
 | 
						|
\n\
 | 
						|
Convert an IP address from 32-bit packed binary format to string format");
 | 
						|
 | 
						|
static PyObject*
 | 
						|
socket_inet_ntoa(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    char *packed_str;
 | 
						|
    int addr_len;
 | 
						|
    struct in_addr packed_addr;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "y#:inet_ntoa", &packed_str, &addr_len)) {
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    if (addr_len != sizeof(packed_addr)) {
 | 
						|
        PyErr_SetString(socket_error,
 | 
						|
            "packed IP wrong length for inet_ntoa");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    memcpy(&packed_addr, packed_str, addr_len);
 | 
						|
 | 
						|
    return PyUnicode_FromString(inet_ntoa(packed_addr));
 | 
						|
}
 | 
						|
 | 
						|
#ifdef HAVE_INET_PTON
 | 
						|
 | 
						|
PyDoc_STRVAR(inet_pton_doc,
 | 
						|
"inet_pton(af, ip) -> packed IP address string\n\
 | 
						|
\n\
 | 
						|
Convert an IP address from string format to a packed string suitable\n\
 | 
						|
for use with low-level network functions.");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_inet_pton(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    int af;
 | 
						|
    char* ip;
 | 
						|
    int retval;
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
    char packed[MAX(sizeof(struct in_addr), sizeof(struct in6_addr))];
 | 
						|
#else
 | 
						|
    char packed[sizeof(struct in_addr)];
 | 
						|
#endif
 | 
						|
    if (!PyArg_ParseTuple(args, "is:inet_pton", &af, &ip)) {
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
#if !defined(ENABLE_IPV6) && defined(AF_INET6)
 | 
						|
    if(af == AF_INET6) {
 | 
						|
        PyErr_SetString(socket_error,
 | 
						|
                        "can't use AF_INET6, IPv6 is disabled");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
    retval = inet_pton(af, ip, packed);
 | 
						|
    if (retval < 0) {
 | 
						|
        PyErr_SetFromErrno(socket_error);
 | 
						|
        return NULL;
 | 
						|
    } else if (retval == 0) {
 | 
						|
        PyErr_SetString(socket_error,
 | 
						|
            "illegal IP address string passed to inet_pton");
 | 
						|
        return NULL;
 | 
						|
    } else if (af == AF_INET) {
 | 
						|
        return PyBytes_FromStringAndSize(packed,
 | 
						|
                                          sizeof(struct in_addr));
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
    } else if (af == AF_INET6) {
 | 
						|
        return PyBytes_FromStringAndSize(packed,
 | 
						|
                                          sizeof(struct in6_addr));
 | 
						|
#endif
 | 
						|
    } else {
 | 
						|
        PyErr_SetString(socket_error, "unknown address family");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(inet_ntop_doc,
 | 
						|
"inet_ntop(af, packed_ip) -> string formatted IP address\n\
 | 
						|
\n\
 | 
						|
Convert a packed IP address of the given family to string format.");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_inet_ntop(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    int af;
 | 
						|
    char* packed;
 | 
						|
    int len;
 | 
						|
    const char* retval;
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
    char ip[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN) + 1];
 | 
						|
#else
 | 
						|
    char ip[INET_ADDRSTRLEN + 1];
 | 
						|
#endif
 | 
						|
 | 
						|
    /* Guarantee NUL-termination for PyUnicode_FromString() below */
 | 
						|
    memset((void *) &ip[0], '\0', sizeof(ip));
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "iy#:inet_ntop", &af, &packed, &len)) {
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    if (af == AF_INET) {
 | 
						|
        if (len != sizeof(struct in_addr)) {
 | 
						|
            PyErr_SetString(PyExc_ValueError,
 | 
						|
                "invalid length of packed IP address string");
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
    } else if (af == AF_INET6) {
 | 
						|
        if (len != sizeof(struct in6_addr)) {
 | 
						|
            PyErr_SetString(PyExc_ValueError,
 | 
						|
                "invalid length of packed IP address string");
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
#endif
 | 
						|
    } else {
 | 
						|
        PyErr_Format(PyExc_ValueError,
 | 
						|
            "unknown address family %d", af);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    retval = inet_ntop(af, packed, ip, sizeof(ip));
 | 
						|
    if (!retval) {
 | 
						|
        PyErr_SetFromErrno(socket_error);
 | 
						|
        return NULL;
 | 
						|
    } else {
 | 
						|
        return PyUnicode_FromString(retval);
 | 
						|
    }
 | 
						|
 | 
						|
    /* NOTREACHED */
 | 
						|
    PyErr_SetString(PyExc_RuntimeError, "invalid handling of inet_ntop");
 | 
						|
    return NULL;
 | 
						|
}
 | 
						|
 | 
						|
#endif /* HAVE_INET_PTON */
 | 
						|
 | 
						|
/* Python interface to getaddrinfo(host, port). */
 | 
						|
 | 
						|
/*ARGSUSED*/
 | 
						|
static PyObject *
 | 
						|
socket_getaddrinfo(PyObject *self, PyObject *args, PyObject* kwargs)
 | 
						|
{
 | 
						|
    static char* kwnames[] = {"host", "port", "family", "type", "proto",
 | 
						|
                              "flags", 0};
 | 
						|
    struct addrinfo hints, *res;
 | 
						|
    struct addrinfo *res0 = NULL;
 | 
						|
    PyObject *hobj = NULL;
 | 
						|
    PyObject *pobj = (PyObject *)NULL;
 | 
						|
    char pbuf[30];
 | 
						|
    char *hptr, *pptr;
 | 
						|
    int family, socktype, protocol, flags;
 | 
						|
    int error;
 | 
						|
    PyObject *all = (PyObject *)NULL;
 | 
						|
    PyObject *idna = NULL;
 | 
						|
 | 
						|
    family = socktype = protocol = flags = 0;
 | 
						|
    family = AF_UNSPEC;
 | 
						|
    if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO|iiii:getaddrinfo",
 | 
						|
                          kwnames, &hobj, &pobj, &family, &socktype,
 | 
						|
                          &protocol, &flags)) {
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    if (hobj == Py_None) {
 | 
						|
        hptr = NULL;
 | 
						|
    } else if (PyUnicode_Check(hobj)) {
 | 
						|
        idna = PyObject_CallMethod(hobj, "encode", "s", "idna");
 | 
						|
        if (!idna)
 | 
						|
            return NULL;
 | 
						|
        assert(PyBytes_Check(idna));
 | 
						|
        hptr = PyBytes_AS_STRING(idna);
 | 
						|
    } else if (PyBytes_Check(hobj)) {
 | 
						|
        hptr = PyBytes_AsString(hobj);
 | 
						|
    } else {
 | 
						|
        PyErr_SetString(PyExc_TypeError,
 | 
						|
                        "getaddrinfo() argument 1 must be string or None");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    if (PyLong_CheckExact(pobj)) {
 | 
						|
        long value = PyLong_AsLong(pobj);
 | 
						|
        if (value == -1 && PyErr_Occurred())
 | 
						|
            goto err;
 | 
						|
        PyOS_snprintf(pbuf, sizeof(pbuf), "%ld", value);
 | 
						|
        pptr = pbuf;
 | 
						|
    } else if (PyUnicode_Check(pobj)) {
 | 
						|
        pptr = _PyUnicode_AsString(pobj);
 | 
						|
        if (pptr == NULL)
 | 
						|
            goto err;
 | 
						|
    } else if (PyBytes_Check(pobj)) {
 | 
						|
        pptr = PyBytes_AS_STRING(pobj);
 | 
						|
    } else if (pobj == Py_None) {
 | 
						|
        pptr = (char *)NULL;
 | 
						|
    } else {
 | 
						|
        PyErr_SetString(socket_error, "Int or String expected");
 | 
						|
        goto err;
 | 
						|
    }
 | 
						|
    memset(&hints, 0, sizeof(hints));
 | 
						|
    hints.ai_family = family;
 | 
						|
    hints.ai_socktype = socktype;
 | 
						|
    hints.ai_protocol = protocol;
 | 
						|
    hints.ai_flags = flags;
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    ACQUIRE_GETADDRINFO_LOCK
 | 
						|
    error = getaddrinfo(hptr, pptr, &hints, &res0);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    RELEASE_GETADDRINFO_LOCK  /* see comment in setipaddr() */
 | 
						|
    if (error) {
 | 
						|
        set_gaierror(error);
 | 
						|
        goto err;
 | 
						|
    }
 | 
						|
 | 
						|
    if ((all = PyList_New(0)) == NULL)
 | 
						|
        goto err;
 | 
						|
    for (res = res0; res; res = res->ai_next) {
 | 
						|
        PyObject *single;
 | 
						|
        PyObject *addr =
 | 
						|
            makesockaddr(-1, res->ai_addr, res->ai_addrlen, protocol);
 | 
						|
        if (addr == NULL)
 | 
						|
            goto err;
 | 
						|
        single = Py_BuildValue("iiisO", res->ai_family,
 | 
						|
            res->ai_socktype, res->ai_protocol,
 | 
						|
            res->ai_canonname ? res->ai_canonname : "",
 | 
						|
            addr);
 | 
						|
        Py_DECREF(addr);
 | 
						|
        if (single == NULL)
 | 
						|
            goto err;
 | 
						|
 | 
						|
        if (PyList_Append(all, single))
 | 
						|
            goto err;
 | 
						|
        Py_XDECREF(single);
 | 
						|
    }
 | 
						|
    Py_XDECREF(idna);
 | 
						|
    if (res0)
 | 
						|
        freeaddrinfo(res0);
 | 
						|
    return all;
 | 
						|
 err:
 | 
						|
    Py_XDECREF(all);
 | 
						|
    Py_XDECREF(idna);
 | 
						|
    if (res0)
 | 
						|
        freeaddrinfo(res0);
 | 
						|
    return (PyObject *)NULL;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(getaddrinfo_doc,
 | 
						|
"getaddrinfo(host, port [, family, socktype, proto, flags])\n\
 | 
						|
    -> list of (family, socktype, proto, canonname, sockaddr)\n\
 | 
						|
\n\
 | 
						|
Resolve host and port into addrinfo struct.");
 | 
						|
 | 
						|
/* Python interface to getnameinfo(sa, flags). */
 | 
						|
 | 
						|
/*ARGSUSED*/
 | 
						|
static PyObject *
 | 
						|
socket_getnameinfo(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    PyObject *sa = (PyObject *)NULL;
 | 
						|
    int flags;
 | 
						|
    char *hostp;
 | 
						|
    int port, flowinfo, scope_id;
 | 
						|
    char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
 | 
						|
    struct addrinfo hints, *res = NULL;
 | 
						|
    int error;
 | 
						|
    PyObject *ret = (PyObject *)NULL;
 | 
						|
 | 
						|
    flags = flowinfo = scope_id = 0;
 | 
						|
    if (!PyArg_ParseTuple(args, "Oi:getnameinfo", &sa, &flags))
 | 
						|
        return NULL;
 | 
						|
    if (!PyTuple_Check(sa)) {
 | 
						|
        PyErr_SetString(PyExc_TypeError,
 | 
						|
                        "getnameinfo() argument 1 must be a tuple");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    if (!PyArg_ParseTuple(sa, "si|ii",
 | 
						|
                          &hostp, &port, &flowinfo, &scope_id))
 | 
						|
        return NULL;
 | 
						|
    PyOS_snprintf(pbuf, sizeof(pbuf), "%d", port);
 | 
						|
    memset(&hints, 0, sizeof(hints));
 | 
						|
    hints.ai_family = AF_UNSPEC;
 | 
						|
    hints.ai_socktype = SOCK_DGRAM;     /* make numeric port happy */
 | 
						|
    hints.ai_flags = AI_NUMERICHOST;    /* don't do any name resolution */
 | 
						|
    Py_BEGIN_ALLOW_THREADS
 | 
						|
    ACQUIRE_GETADDRINFO_LOCK
 | 
						|
    error = getaddrinfo(hostp, pbuf, &hints, &res);
 | 
						|
    Py_END_ALLOW_THREADS
 | 
						|
    RELEASE_GETADDRINFO_LOCK  /* see comment in setipaddr() */
 | 
						|
    if (error) {
 | 
						|
        set_gaierror(error);
 | 
						|
        goto fail;
 | 
						|
    }
 | 
						|
    if (res->ai_next) {
 | 
						|
        PyErr_SetString(socket_error,
 | 
						|
            "sockaddr resolved to multiple addresses");
 | 
						|
        goto fail;
 | 
						|
    }
 | 
						|
    switch (res->ai_family) {
 | 
						|
    case AF_INET:
 | 
						|
        {
 | 
						|
        if (PyTuple_GET_SIZE(sa) != 2) {
 | 
						|
            PyErr_SetString(socket_error,
 | 
						|
                "IPv4 sockaddr must be 2 tuple");
 | 
						|
            goto fail;
 | 
						|
        }
 | 
						|
        break;
 | 
						|
        }
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
    case AF_INET6:
 | 
						|
        {
 | 
						|
        struct sockaddr_in6 *sin6;
 | 
						|
        sin6 = (struct sockaddr_in6 *)res->ai_addr;
 | 
						|
        sin6->sin6_flowinfo = flowinfo;
 | 
						|
        sin6->sin6_scope_id = scope_id;
 | 
						|
        break;
 | 
						|
        }
 | 
						|
#endif
 | 
						|
    }
 | 
						|
    error = getnameinfo(res->ai_addr, (socklen_t) res->ai_addrlen,
 | 
						|
                    hbuf, sizeof(hbuf), pbuf, sizeof(pbuf), flags);
 | 
						|
    if (error) {
 | 
						|
        set_gaierror(error);
 | 
						|
        goto fail;
 | 
						|
    }
 | 
						|
    ret = Py_BuildValue("ss", hbuf, pbuf);
 | 
						|
 | 
						|
fail:
 | 
						|
    if (res)
 | 
						|
        freeaddrinfo(res);
 | 
						|
    return ret;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(getnameinfo_doc,
 | 
						|
"getnameinfo(sockaddr, flags) --> (host, port)\n\
 | 
						|
\n\
 | 
						|
Get host and port for a sockaddr.");
 | 
						|
 | 
						|
 | 
						|
/* Python API to getting and setting the default timeout value. */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_getdefaulttimeout(PyObject *self)
 | 
						|
{
 | 
						|
    if (defaulttimeout < 0.0) {
 | 
						|
        Py_INCREF(Py_None);
 | 
						|
        return Py_None;
 | 
						|
    }
 | 
						|
    else
 | 
						|
        return PyFloat_FromDouble(defaulttimeout);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(getdefaulttimeout_doc,
 | 
						|
"getdefaulttimeout() -> timeout\n\
 | 
						|
\n\
 | 
						|
Returns the default timeout in seconds (float) for new socket objects.\n\
 | 
						|
A value of None indicates that new socket objects have no timeout.\n\
 | 
						|
When the socket module is first imported, the default is None.");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_setdefaulttimeout(PyObject *self, PyObject *arg)
 | 
						|
{
 | 
						|
    double timeout;
 | 
						|
 | 
						|
    if (arg == Py_None)
 | 
						|
        timeout = -1.0;
 | 
						|
    else {
 | 
						|
        timeout = PyFloat_AsDouble(arg);
 | 
						|
        if (timeout < 0.0) {
 | 
						|
            if (!PyErr_Occurred())
 | 
						|
                PyErr_SetString(PyExc_ValueError,
 | 
						|
                                "Timeout value out of range");
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    defaulttimeout = timeout;
 | 
						|
 | 
						|
    Py_INCREF(Py_None);
 | 
						|
    return Py_None;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(setdefaulttimeout_doc,
 | 
						|
"setdefaulttimeout(timeout)\n\
 | 
						|
\n\
 | 
						|
Set the default timeout in seconds (float) for new socket objects.\n\
 | 
						|
A value of None indicates that new socket objects have no timeout.\n\
 | 
						|
When the socket module is first imported, the default is None.");
 | 
						|
 | 
						|
#ifdef HAVE_IF_NAMEINDEX
 | 
						|
/* Python API for getting interface indices and names */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_if_nameindex(PyObject *self, PyObject *arg)
 | 
						|
{
 | 
						|
    PyObject *list;
 | 
						|
    int i;
 | 
						|
    struct if_nameindex *ni;
 | 
						|
 | 
						|
    ni = if_nameindex();
 | 
						|
    if (ni == NULL) {
 | 
						|
        PyErr_SetFromErrno(socket_error);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    list = PyList_New(0);
 | 
						|
    if (list == NULL) {
 | 
						|
        if_freenameindex(ni);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    for (i = 0; ni[i].if_index != 0 && i < INT_MAX; i++) {
 | 
						|
        PyObject *ni_tuple = Py_BuildValue("IO&",
 | 
						|
                ni[i].if_index, PyUnicode_DecodeFSDefault, ni[i].if_name);
 | 
						|
 | 
						|
        if (ni_tuple == NULL || PyList_Append(list, ni_tuple) == -1) {
 | 
						|
            Py_XDECREF(ni_tuple);
 | 
						|
            Py_DECREF(list);
 | 
						|
            if_freenameindex(ni);
 | 
						|
            return NULL;
 | 
						|
        }
 | 
						|
        Py_DECREF(ni_tuple);
 | 
						|
    }
 | 
						|
 | 
						|
    if_freenameindex(ni);
 | 
						|
    return list;
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(if_nameindex_doc,
 | 
						|
"if_nameindex()\n\
 | 
						|
\n\
 | 
						|
Returns a list of network interface information (index, name) tuples.");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_if_nametoindex(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    PyObject *oname;
 | 
						|
    unsigned long index;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "O&:if_nametoindex",
 | 
						|
                          PyUnicode_FSConverter, &oname))
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    index = if_nametoindex(PyBytes_AS_STRING(oname));
 | 
						|
    Py_DECREF(oname);
 | 
						|
    if (index == 0) {
 | 
						|
        /* if_nametoindex() doesn't set errno */
 | 
						|
        PyErr_SetString(socket_error, "no interface with this name");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    return PyLong_FromUnsignedLong(index);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(if_nametoindex_doc,
 | 
						|
"if_nametoindex(if_name)\n\
 | 
						|
\n\
 | 
						|
Returns the interface index corresponding to the interface name if_name.");
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_if_indextoname(PyObject *self, PyObject *arg)
 | 
						|
{
 | 
						|
    unsigned long index;
 | 
						|
    char name[IF_NAMESIZE + 1];
 | 
						|
 | 
						|
    index = PyLong_AsUnsignedLong(arg);
 | 
						|
    if (index == (unsigned long) -1)
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    if (if_indextoname(index, name) == NULL) {
 | 
						|
        PyErr_SetFromErrno(socket_error);
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    return PyUnicode_DecodeFSDefault(name);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(if_indextoname_doc,
 | 
						|
"if_indextoname(if_index)\n\
 | 
						|
\n\
 | 
						|
Returns the interface name corresponding to the interface index if_index.");
 | 
						|
 | 
						|
#endif  /* HAVE_IF_NAMEINDEX */
 | 
						|
 | 
						|
 | 
						|
#ifdef CMSG_LEN
 | 
						|
/* Python interface to CMSG_LEN(length). */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_CMSG_LEN(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    Py_ssize_t length;
 | 
						|
    size_t result;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "n:CMSG_LEN", &length))
 | 
						|
        return NULL;
 | 
						|
    if (length < 0 || !get_CMSG_LEN(length, &result)) {
 | 
						|
        PyErr_Format(PyExc_OverflowError, "CMSG_LEN() argument out of range");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    return PyLong_FromSize_t(result);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(CMSG_LEN_doc,
 | 
						|
"CMSG_LEN(length) -> control message length\n\
 | 
						|
\n\
 | 
						|
Return the total length, without trailing padding, of an ancillary\n\
 | 
						|
data item with associated data of the given length.  This value can\n\
 | 
						|
often be used as the buffer size for recvmsg() to receive a single\n\
 | 
						|
item of ancillary data, but RFC 3542 requires portable applications to\n\
 | 
						|
use CMSG_SPACE() and thus include space for padding, even when the\n\
 | 
						|
item will be the last in the buffer.  Raises OverflowError if length\n\
 | 
						|
is outside the permissible range of values.");
 | 
						|
 | 
						|
 | 
						|
#ifdef CMSG_SPACE
 | 
						|
/* Python interface to CMSG_SPACE(length). */
 | 
						|
 | 
						|
static PyObject *
 | 
						|
socket_CMSG_SPACE(PyObject *self, PyObject *args)
 | 
						|
{
 | 
						|
    Py_ssize_t length;
 | 
						|
    size_t result;
 | 
						|
 | 
						|
    if (!PyArg_ParseTuple(args, "n:CMSG_SPACE", &length))
 | 
						|
        return NULL;
 | 
						|
    if (length < 0 || !get_CMSG_SPACE(length, &result)) {
 | 
						|
        PyErr_SetString(PyExc_OverflowError,
 | 
						|
                        "CMSG_SPACE() argument out of range");
 | 
						|
        return NULL;
 | 
						|
    }
 | 
						|
    return PyLong_FromSize_t(result);
 | 
						|
}
 | 
						|
 | 
						|
PyDoc_STRVAR(CMSG_SPACE_doc,
 | 
						|
"CMSG_SPACE(length) -> buffer size\n\
 | 
						|
\n\
 | 
						|
Return the buffer size needed for recvmsg() to receive an ancillary\n\
 | 
						|
data item with associated data of the given length, along with any\n\
 | 
						|
trailing padding.  The buffer space needed to receive multiple items\n\
 | 
						|
is the sum of the CMSG_SPACE() values for their associated data\n\
 | 
						|
lengths.  Raises OverflowError if length is outside the permissible\n\
 | 
						|
range of values.");
 | 
						|
#endif    /* CMSG_SPACE */
 | 
						|
#endif    /* CMSG_LEN */
 | 
						|
 | 
						|
 | 
						|
/* List of functions exported by this module. */
 | 
						|
 | 
						|
static PyMethodDef socket_methods[] = {
 | 
						|
    {"gethostbyname",           socket_gethostbyname,
 | 
						|
     METH_VARARGS, gethostbyname_doc},
 | 
						|
    {"gethostbyname_ex",        socket_gethostbyname_ex,
 | 
						|
     METH_VARARGS, ghbn_ex_doc},
 | 
						|
    {"gethostbyaddr",           socket_gethostbyaddr,
 | 
						|
     METH_VARARGS, gethostbyaddr_doc},
 | 
						|
    {"gethostname",             socket_gethostname,
 | 
						|
     METH_NOARGS,  gethostname_doc},
 | 
						|
#ifdef HAVE_SETHOSTNAME
 | 
						|
    {"sethostname",             socket_sethostname,
 | 
						|
     METH_VARARGS,  sethostname_doc},
 | 
						|
#endif
 | 
						|
    {"getservbyname",           socket_getservbyname,
 | 
						|
     METH_VARARGS, getservbyname_doc},
 | 
						|
    {"getservbyport",           socket_getservbyport,
 | 
						|
     METH_VARARGS, getservbyport_doc},
 | 
						|
    {"getprotobyname",          socket_getprotobyname,
 | 
						|
     METH_VARARGS, getprotobyname_doc},
 | 
						|
#ifndef NO_DUP
 | 
						|
    {"dup",                     socket_dup,
 | 
						|
     METH_O, dup_doc},
 | 
						|
#endif
 | 
						|
#ifdef HAVE_SOCKETPAIR
 | 
						|
    {"socketpair",              socket_socketpair,
 | 
						|
     METH_VARARGS, socketpair_doc},
 | 
						|
#endif
 | 
						|
    {"ntohs",                   socket_ntohs,
 | 
						|
     METH_VARARGS, ntohs_doc},
 | 
						|
    {"ntohl",                   socket_ntohl,
 | 
						|
     METH_O, ntohl_doc},
 | 
						|
    {"htons",                   socket_htons,
 | 
						|
     METH_VARARGS, htons_doc},
 | 
						|
    {"htonl",                   socket_htonl,
 | 
						|
     METH_O, htonl_doc},
 | 
						|
    {"inet_aton",               socket_inet_aton,
 | 
						|
     METH_VARARGS, inet_aton_doc},
 | 
						|
    {"inet_ntoa",               socket_inet_ntoa,
 | 
						|
     METH_VARARGS, inet_ntoa_doc},
 | 
						|
#ifdef HAVE_INET_PTON
 | 
						|
    {"inet_pton",               socket_inet_pton,
 | 
						|
     METH_VARARGS, inet_pton_doc},
 | 
						|
    {"inet_ntop",               socket_inet_ntop,
 | 
						|
     METH_VARARGS, inet_ntop_doc},
 | 
						|
#endif
 | 
						|
    {"getaddrinfo",             (PyCFunction)socket_getaddrinfo,
 | 
						|
     METH_VARARGS | METH_KEYWORDS, getaddrinfo_doc},
 | 
						|
    {"getnameinfo",             socket_getnameinfo,
 | 
						|
     METH_VARARGS, getnameinfo_doc},
 | 
						|
    {"getdefaulttimeout",       (PyCFunction)socket_getdefaulttimeout,
 | 
						|
     METH_NOARGS, getdefaulttimeout_doc},
 | 
						|
    {"setdefaulttimeout",       socket_setdefaulttimeout,
 | 
						|
     METH_O, setdefaulttimeout_doc},
 | 
						|
#ifdef HAVE_IF_NAMEINDEX
 | 
						|
    {"if_nameindex", socket_if_nameindex,
 | 
						|
     METH_NOARGS, if_nameindex_doc},
 | 
						|
    {"if_nametoindex", socket_if_nametoindex,
 | 
						|
     METH_VARARGS, if_nametoindex_doc},
 | 
						|
    {"if_indextoname", socket_if_indextoname,
 | 
						|
     METH_O, if_indextoname_doc},
 | 
						|
#endif
 | 
						|
#ifdef CMSG_LEN
 | 
						|
    {"CMSG_LEN",                socket_CMSG_LEN,
 | 
						|
     METH_VARARGS, CMSG_LEN_doc},
 | 
						|
#ifdef CMSG_SPACE
 | 
						|
    {"CMSG_SPACE",              socket_CMSG_SPACE,
 | 
						|
     METH_VARARGS, CMSG_SPACE_doc},
 | 
						|
#endif
 | 
						|
#endif
 | 
						|
    {NULL,                      NULL}            /* Sentinel */
 | 
						|
};
 | 
						|
 | 
						|
 | 
						|
#ifdef MS_WINDOWS
 | 
						|
#define OS_INIT_DEFINED
 | 
						|
 | 
						|
/* Additional initialization and cleanup for Windows */
 | 
						|
 | 
						|
static void
 | 
						|
os_cleanup(void)
 | 
						|
{
 | 
						|
    WSACleanup();
 | 
						|
}
 | 
						|
 | 
						|
static int
 | 
						|
os_init(void)
 | 
						|
{
 | 
						|
    WSADATA WSAData;
 | 
						|
    int ret;
 | 
						|
    ret = WSAStartup(0x0101, &WSAData);
 | 
						|
    switch (ret) {
 | 
						|
    case 0:     /* No error */
 | 
						|
        Py_AtExit(os_cleanup);
 | 
						|
        return 1; /* Success */
 | 
						|
    case WSASYSNOTREADY:
 | 
						|
        PyErr_SetString(PyExc_ImportError,
 | 
						|
                        "WSAStartup failed: network not ready");
 | 
						|
        break;
 | 
						|
    case WSAVERNOTSUPPORTED:
 | 
						|
    case WSAEINVAL:
 | 
						|
        PyErr_SetString(
 | 
						|
            PyExc_ImportError,
 | 
						|
            "WSAStartup failed: requested version not supported");
 | 
						|
        break;
 | 
						|
    default:
 | 
						|
        PyErr_Format(PyExc_ImportError, "WSAStartup failed: error code %d", ret);
 | 
						|
        break;
 | 
						|
    }
 | 
						|
    return 0; /* Failure */
 | 
						|
}
 | 
						|
 | 
						|
#endif /* MS_WINDOWS */
 | 
						|
 | 
						|
 | 
						|
#ifdef PYOS_OS2
 | 
						|
#define OS_INIT_DEFINED
 | 
						|
 | 
						|
/* Additional initialization for OS/2 */
 | 
						|
 | 
						|
static int
 | 
						|
os_init(void)
 | 
						|
{
 | 
						|
#ifndef PYCC_GCC
 | 
						|
    int rc = sock_init();
 | 
						|
 | 
						|
    if (rc == 0) {
 | 
						|
        return 1; /* Success */
 | 
						|
    }
 | 
						|
 | 
						|
    PyErr_Format(PyExc_ImportError, "OS/2 TCP/IP Error# %d", sock_errno());
 | 
						|
 | 
						|
    return 0;  /* Failure */
 | 
						|
#else
 | 
						|
    /* No need to initialize sockets with GCC/EMX */
 | 
						|
    return 1; /* Success */
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
#endif /* PYOS_OS2 */
 | 
						|
 | 
						|
 | 
						|
#ifndef OS_INIT_DEFINED
 | 
						|
static int
 | 
						|
os_init(void)
 | 
						|
{
 | 
						|
    return 1; /* Success */
 | 
						|
}
 | 
						|
#endif
 | 
						|
 | 
						|
 | 
						|
/* C API table - always add new things to the end for binary
 | 
						|
   compatibility. */
 | 
						|
static
 | 
						|
PySocketModule_APIObject PySocketModuleAPI =
 | 
						|
{
 | 
						|
    &sock_type,
 | 
						|
    NULL,
 | 
						|
    NULL
 | 
						|
};
 | 
						|
 | 
						|
 | 
						|
/* Initialize the _socket module.
 | 
						|
 | 
						|
   This module is actually called "_socket", and there's a wrapper
 | 
						|
   "socket.py" which implements some additional functionality.
 | 
						|
   The import of "_socket" may fail with an ImportError exception if
 | 
						|
   os-specific initialization fails.  On Windows, this does WINSOCK
 | 
						|
   initialization.  When WINSOCK is initialized successfully, a call to
 | 
						|
   WSACleanup() is scheduled to be made at exit time.
 | 
						|
*/
 | 
						|
 | 
						|
PyDoc_STRVAR(socket_doc,
 | 
						|
"Implementation module for socket operations.\n\
 | 
						|
\n\
 | 
						|
See the socket module for documentation.");
 | 
						|
 | 
						|
static struct PyModuleDef socketmodule = {
 | 
						|
    PyModuleDef_HEAD_INIT,
 | 
						|
    PySocket_MODULE_NAME,
 | 
						|
    socket_doc,
 | 
						|
    -1,
 | 
						|
    socket_methods,
 | 
						|
    NULL,
 | 
						|
    NULL,
 | 
						|
    NULL,
 | 
						|
    NULL
 | 
						|
};
 | 
						|
 | 
						|
PyMODINIT_FUNC
 | 
						|
PyInit__socket(void)
 | 
						|
{
 | 
						|
    PyObject *m, *has_ipv6;
 | 
						|
 | 
						|
    if (!os_init())
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    Py_TYPE(&sock_type) = &PyType_Type;
 | 
						|
    m = PyModule_Create(&socketmodule);
 | 
						|
    if (m == NULL)
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    socket_error = PyErr_NewException("socket.error",
 | 
						|
                                      PyExc_IOError, NULL);
 | 
						|
    if (socket_error == NULL)
 | 
						|
        return NULL;
 | 
						|
    PySocketModuleAPI.error = socket_error;
 | 
						|
    Py_INCREF(socket_error);
 | 
						|
    PyModule_AddObject(m, "error", socket_error);
 | 
						|
    socket_herror = PyErr_NewException("socket.herror",
 | 
						|
                                       socket_error, NULL);
 | 
						|
    if (socket_herror == NULL)
 | 
						|
        return NULL;
 | 
						|
    Py_INCREF(socket_herror);
 | 
						|
    PyModule_AddObject(m, "herror", socket_herror);
 | 
						|
    socket_gaierror = PyErr_NewException("socket.gaierror", socket_error,
 | 
						|
        NULL);
 | 
						|
    if (socket_gaierror == NULL)
 | 
						|
        return NULL;
 | 
						|
    Py_INCREF(socket_gaierror);
 | 
						|
    PyModule_AddObject(m, "gaierror", socket_gaierror);
 | 
						|
    socket_timeout = PyErr_NewException("socket.timeout",
 | 
						|
                                        socket_error, NULL);
 | 
						|
    if (socket_timeout == NULL)
 | 
						|
        return NULL;
 | 
						|
    PySocketModuleAPI.timeout_error = socket_timeout;
 | 
						|
    Py_INCREF(socket_timeout);
 | 
						|
    PyModule_AddObject(m, "timeout", socket_timeout);
 | 
						|
    Py_INCREF((PyObject *)&sock_type);
 | 
						|
    if (PyModule_AddObject(m, "SocketType",
 | 
						|
                           (PyObject *)&sock_type) != 0)
 | 
						|
        return NULL;
 | 
						|
    Py_INCREF((PyObject *)&sock_type);
 | 
						|
    if (PyModule_AddObject(m, "socket",
 | 
						|
                           (PyObject *)&sock_type) != 0)
 | 
						|
        return NULL;
 | 
						|
 | 
						|
#ifdef ENABLE_IPV6
 | 
						|
    has_ipv6 = Py_True;
 | 
						|
#else
 | 
						|
    has_ipv6 = Py_False;
 | 
						|
#endif
 | 
						|
    Py_INCREF(has_ipv6);
 | 
						|
    PyModule_AddObject(m, "has_ipv6", has_ipv6);
 | 
						|
 | 
						|
    /* Export C API */
 | 
						|
    if (PyModule_AddObject(m, PySocket_CAPI_NAME,
 | 
						|
           PyCapsule_New(&PySocketModuleAPI, PySocket_CAPSULE_NAME, NULL)
 | 
						|
                             ) != 0)
 | 
						|
        return NULL;
 | 
						|
 | 
						|
    /* Address families (we only support AF_INET and AF_UNIX) */
 | 
						|
#ifdef AF_UNSPEC
 | 
						|
    PyModule_AddIntConstant(m, "AF_UNSPEC", AF_UNSPEC);
 | 
						|
#endif
 | 
						|
    PyModule_AddIntConstant(m, "AF_INET", AF_INET);
 | 
						|
#ifdef AF_INET6
 | 
						|
    PyModule_AddIntConstant(m, "AF_INET6", AF_INET6);
 | 
						|
#endif /* AF_INET6 */
 | 
						|
#if defined(AF_UNIX)
 | 
						|
    PyModule_AddIntConstant(m, "AF_UNIX", AF_UNIX);
 | 
						|
#endif /* AF_UNIX */
 | 
						|
#ifdef AF_AX25
 | 
						|
    /* Amateur Radio AX.25 */
 | 
						|
    PyModule_AddIntConstant(m, "AF_AX25", AF_AX25);
 | 
						|
#endif
 | 
						|
#ifdef AF_IPX
 | 
						|
    PyModule_AddIntConstant(m, "AF_IPX", AF_IPX); /* Novell IPX */
 | 
						|
#endif
 | 
						|
#ifdef AF_APPLETALK
 | 
						|
    /* Appletalk DDP */
 | 
						|
    PyModule_AddIntConstant(m, "AF_APPLETALK", AF_APPLETALK);
 | 
						|
#endif
 | 
						|
#ifdef AF_NETROM
 | 
						|
    /* Amateur radio NetROM */
 | 
						|
    PyModule_AddIntConstant(m, "AF_NETROM", AF_NETROM);
 | 
						|
#endif
 | 
						|
#ifdef AF_BRIDGE
 | 
						|
    /* Multiprotocol bridge */
 | 
						|
    PyModule_AddIntConstant(m, "AF_BRIDGE", AF_BRIDGE);
 | 
						|
#endif
 | 
						|
#ifdef AF_ATMPVC
 | 
						|
    /* ATM PVCs */
 | 
						|
    PyModule_AddIntConstant(m, "AF_ATMPVC", AF_ATMPVC);
 | 
						|
#endif
 | 
						|
#ifdef AF_AAL5
 | 
						|
    /* Reserved for Werner's ATM */
 | 
						|
    PyModule_AddIntConstant(m, "AF_AAL5", AF_AAL5);
 | 
						|
#endif
 | 
						|
#ifdef AF_X25
 | 
						|
    /* Reserved for X.25 project */
 | 
						|
    PyModule_AddIntConstant(m, "AF_X25", AF_X25);
 | 
						|
#endif
 | 
						|
#ifdef AF_INET6
 | 
						|
    PyModule_AddIntConstant(m, "AF_INET6", AF_INET6); /* IP version 6 */
 | 
						|
#endif
 | 
						|
#ifdef AF_ROSE
 | 
						|
    /* Amateur Radio X.25 PLP */
 | 
						|
    PyModule_AddIntConstant(m, "AF_ROSE", AF_ROSE);
 | 
						|
#endif
 | 
						|
#ifdef AF_DECnet
 | 
						|
    /* Reserved for DECnet project */
 | 
						|
    PyModule_AddIntConstant(m, "AF_DECnet", AF_DECnet);
 | 
						|
#endif
 | 
						|
#ifdef AF_NETBEUI
 | 
						|
    /* Reserved for 802.2LLC project */
 | 
						|
    PyModule_AddIntConstant(m, "AF_NETBEUI", AF_NETBEUI);
 | 
						|
#endif
 | 
						|
#ifdef AF_SECURITY
 | 
						|
    /* Security callback pseudo AF */
 | 
						|
    PyModule_AddIntConstant(m, "AF_SECURITY", AF_SECURITY);
 | 
						|
#endif
 | 
						|
#ifdef AF_KEY
 | 
						|
    /* PF_KEY key management API */
 | 
						|
    PyModule_AddIntConstant(m, "AF_KEY", AF_KEY);
 | 
						|
#endif
 | 
						|
#ifdef AF_NETLINK
 | 
						|
    /*  */
 | 
						|
    PyModule_AddIntConstant(m, "AF_NETLINK", AF_NETLINK);
 | 
						|
    PyModule_AddIntConstant(m, "NETLINK_ROUTE", NETLINK_ROUTE);
 | 
						|
#ifdef NETLINK_SKIP
 | 
						|
    PyModule_AddIntConstant(m, "NETLINK_SKIP", NETLINK_SKIP);
 | 
						|
#endif
 | 
						|
#ifdef NETLINK_W1
 | 
						|
    PyModule_AddIntConstant(m, "NETLINK_W1", NETLINK_W1);
 | 
						|
#endif
 | 
						|
    PyModule_AddIntConstant(m, "NETLINK_USERSOCK", NETLINK_USERSOCK);
 | 
						|
    PyModule_AddIntConstant(m, "NETLINK_FIREWALL", NETLINK_FIREWALL);
 | 
						|
#ifdef NETLINK_TCPDIAG
 | 
						|
    PyModule_AddIntConstant(m, "NETLINK_TCPDIAG", NETLINK_TCPDIAG);
 | 
						|
#endif
 | 
						|
#ifdef NETLINK_NFLOG
 | 
						|
    PyModule_AddIntConstant(m, "NETLINK_NFLOG", NETLINK_NFLOG);
 | 
						|
#endif
 | 
						|
#ifdef NETLINK_XFRM
 | 
						|
    PyModule_AddIntConstant(m, "NETLINK_XFRM", NETLINK_XFRM);
 | 
						|
#endif
 | 
						|
#ifdef NETLINK_ARPD
 | 
						|
    PyModule_AddIntConstant(m, "NETLINK_ARPD", NETLINK_ARPD);
 | 
						|
#endif
 | 
						|
#ifdef NETLINK_ROUTE6
 | 
						|
    PyModule_AddIntConstant(m, "NETLINK_ROUTE6", NETLINK_ROUTE6);
 | 
						|
#endif
 | 
						|
    PyModule_AddIntConstant(m, "NETLINK_IP6_FW", NETLINK_IP6_FW);
 | 
						|
#ifdef NETLINK_DNRTMSG
 | 
						|
    PyModule_AddIntConstant(m, "NETLINK_DNRTMSG", NETLINK_DNRTMSG);
 | 
						|
#endif
 | 
						|
#ifdef NETLINK_TAPBASE
 | 
						|
    PyModule_AddIntConstant(m, "NETLINK_TAPBASE", NETLINK_TAPBASE);
 | 
						|
#endif
 | 
						|
#endif /* AF_NETLINK */
 | 
						|
#ifdef AF_ROUTE
 | 
						|
    /* Alias to emulate 4.4BSD */
 | 
						|
    PyModule_AddIntConstant(m, "AF_ROUTE", AF_ROUTE);
 | 
						|
#endif
 | 
						|
#ifdef AF_ASH
 | 
						|
    /* Ash */
 | 
						|
    PyModule_AddIntConstant(m, "AF_ASH", AF_ASH);
 | 
						|
#endif
 | 
						|
#ifdef AF_ECONET
 | 
						|
    /* Acorn Econet */
 | 
						|
    PyModule_AddIntConstant(m, "AF_ECONET", AF_ECONET);
 | 
						|
#endif
 | 
						|
#ifdef AF_ATMSVC
 | 
						|
    /* ATM SVCs */
 | 
						|
    PyModule_AddIntConstant(m, "AF_ATMSVC", AF_ATMSVC);
 | 
						|
#endif
 | 
						|
#ifdef AF_SNA
 | 
						|
    /* Linux SNA Project (nutters!) */
 | 
						|
    PyModule_AddIntConstant(m, "AF_SNA", AF_SNA);
 | 
						|
#endif
 | 
						|
#ifdef AF_IRDA
 | 
						|
    /* IRDA sockets */
 | 
						|
    PyModule_AddIntConstant(m, "AF_IRDA", AF_IRDA);
 | 
						|
#endif
 | 
						|
#ifdef AF_PPPOX
 | 
						|
    /* PPPoX sockets */
 | 
						|
    PyModule_AddIntConstant(m, "AF_PPPOX", AF_PPPOX);
 | 
						|
#endif
 | 
						|
#ifdef AF_WANPIPE
 | 
						|
    /* Wanpipe API Sockets */
 | 
						|
    PyModule_AddIntConstant(m, "AF_WANPIPE", AF_WANPIPE);
 | 
						|
#endif
 | 
						|
#ifdef AF_LLC
 | 
						|
    /* Linux LLC */
 | 
						|
    PyModule_AddIntConstant(m, "AF_LLC", AF_LLC);
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef USE_BLUETOOTH
 | 
						|
    PyModule_AddIntConstant(m, "AF_BLUETOOTH", AF_BLUETOOTH);
 | 
						|
    PyModule_AddIntConstant(m, "BTPROTO_L2CAP", BTPROTO_L2CAP);
 | 
						|
    PyModule_AddIntConstant(m, "BTPROTO_HCI", BTPROTO_HCI);
 | 
						|
    PyModule_AddIntConstant(m, "SOL_HCI", SOL_HCI);
 | 
						|
#if !defined(__NetBSD__) && !defined(__DragonFly__)
 | 
						|
    PyModule_AddIntConstant(m, "HCI_FILTER", HCI_FILTER);
 | 
						|
#endif
 | 
						|
#if !defined(__FreeBSD__)
 | 
						|
#if !defined(__NetBSD__) && !defined(__DragonFly__)
 | 
						|
    PyModule_AddIntConstant(m, "HCI_TIME_STAMP", HCI_TIME_STAMP);
 | 
						|
#endif
 | 
						|
    PyModule_AddIntConstant(m, "HCI_DATA_DIR", HCI_DATA_DIR);
 | 
						|
    PyModule_AddIntConstant(m, "BTPROTO_SCO", BTPROTO_SCO);
 | 
						|
#endif
 | 
						|
    PyModule_AddIntConstant(m, "BTPROTO_RFCOMM", BTPROTO_RFCOMM);
 | 
						|
    PyModule_AddStringConstant(m, "BDADDR_ANY", "00:00:00:00:00:00");
 | 
						|
    PyModule_AddStringConstant(m, "BDADDR_LOCAL", "00:00:00:FF:FF:FF");
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef AF_CAN
 | 
						|
    /* Controller Area Network */
 | 
						|
    PyModule_AddIntConstant(m, "AF_CAN", AF_CAN);
 | 
						|
#endif
 | 
						|
#ifdef PF_CAN
 | 
						|
    /* Controller Area Network */
 | 
						|
    PyModule_AddIntConstant(m, "PF_CAN", PF_CAN);
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef AF_PACKET
 | 
						|
    PyModule_AddIntMacro(m, AF_PACKET);
 | 
						|
#endif
 | 
						|
#ifdef PF_PACKET
 | 
						|
    PyModule_AddIntMacro(m, PF_PACKET);
 | 
						|
#endif
 | 
						|
#ifdef PACKET_HOST
 | 
						|
    PyModule_AddIntMacro(m, PACKET_HOST);
 | 
						|
#endif
 | 
						|
#ifdef PACKET_BROADCAST
 | 
						|
    PyModule_AddIntMacro(m, PACKET_BROADCAST);
 | 
						|
#endif
 | 
						|
#ifdef PACKET_MULTICAST
 | 
						|
    PyModule_AddIntMacro(m, PACKET_MULTICAST);
 | 
						|
#endif
 | 
						|
#ifdef PACKET_OTHERHOST
 | 
						|
    PyModule_AddIntMacro(m, PACKET_OTHERHOST);
 | 
						|
#endif
 | 
						|
#ifdef PACKET_OUTGOING
 | 
						|
    PyModule_AddIntMacro(m, PACKET_OUTGOING);
 | 
						|
#endif
 | 
						|
#ifdef PACKET_LOOPBACK
 | 
						|
    PyModule_AddIntMacro(m, PACKET_LOOPBACK);
 | 
						|
#endif
 | 
						|
#ifdef PACKET_FASTROUTE
 | 
						|
    PyModule_AddIntMacro(m, PACKET_FASTROUTE);
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_LINUX_TIPC_H
 | 
						|
    PyModule_AddIntConstant(m, "AF_TIPC", AF_TIPC);
 | 
						|
 | 
						|
    /* for addresses */
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_ADDR_NAMESEQ", TIPC_ADDR_NAMESEQ);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_ADDR_NAME", TIPC_ADDR_NAME);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_ADDR_ID", TIPC_ADDR_ID);
 | 
						|
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_ZONE_SCOPE", TIPC_ZONE_SCOPE);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_CLUSTER_SCOPE", TIPC_CLUSTER_SCOPE);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_NODE_SCOPE", TIPC_NODE_SCOPE);
 | 
						|
 | 
						|
    /* for setsockopt() */
 | 
						|
    PyModule_AddIntConstant(m, "SOL_TIPC", SOL_TIPC);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_IMPORTANCE", TIPC_IMPORTANCE);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_SRC_DROPPABLE", TIPC_SRC_DROPPABLE);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_DEST_DROPPABLE",
 | 
						|
                    TIPC_DEST_DROPPABLE);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_CONN_TIMEOUT", TIPC_CONN_TIMEOUT);
 | 
						|
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_LOW_IMPORTANCE",
 | 
						|
                    TIPC_LOW_IMPORTANCE);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_MEDIUM_IMPORTANCE",
 | 
						|
                    TIPC_MEDIUM_IMPORTANCE);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_HIGH_IMPORTANCE",
 | 
						|
                    TIPC_HIGH_IMPORTANCE);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_CRITICAL_IMPORTANCE",
 | 
						|
                    TIPC_CRITICAL_IMPORTANCE);
 | 
						|
 | 
						|
    /* for subscriptions */
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_SUB_PORTS", TIPC_SUB_PORTS);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_SUB_SERVICE", TIPC_SUB_SERVICE);
 | 
						|
#ifdef TIPC_SUB_CANCEL
 | 
						|
    /* doesn't seem to be available everywhere */
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_SUB_CANCEL", TIPC_SUB_CANCEL);
 | 
						|
#endif
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_WAIT_FOREVER", TIPC_WAIT_FOREVER);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_PUBLISHED", TIPC_PUBLISHED);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_WITHDRAWN", TIPC_WITHDRAWN);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_SUBSCR_TIMEOUT", TIPC_SUBSCR_TIMEOUT);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_CFG_SRV", TIPC_CFG_SRV);
 | 
						|
    PyModule_AddIntConstant(m, "TIPC_TOP_SRV", TIPC_TOP_SRV);
 | 
						|
#endif
 | 
						|
 | 
						|
    /* Socket types */
 | 
						|
    PyModule_AddIntConstant(m, "SOCK_STREAM", SOCK_STREAM);
 | 
						|
    PyModule_AddIntConstant(m, "SOCK_DGRAM", SOCK_DGRAM);
 | 
						|
/* We have incomplete socket support. */
 | 
						|
    PyModule_AddIntConstant(m, "SOCK_RAW", SOCK_RAW);
 | 
						|
    PyModule_AddIntConstant(m, "SOCK_SEQPACKET", SOCK_SEQPACKET);
 | 
						|
#if defined(SOCK_RDM)
 | 
						|
    PyModule_AddIntConstant(m, "SOCK_RDM", SOCK_RDM);
 | 
						|
#endif
 | 
						|
#ifdef SOCK_CLOEXEC
 | 
						|
    PyModule_AddIntConstant(m, "SOCK_CLOEXEC", SOCK_CLOEXEC);
 | 
						|
#endif
 | 
						|
#ifdef SOCK_NONBLOCK
 | 
						|
    PyModule_AddIntConstant(m, "SOCK_NONBLOCK", SOCK_NONBLOCK);
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef  SO_DEBUG
 | 
						|
    PyModule_AddIntConstant(m, "SO_DEBUG", SO_DEBUG);
 | 
						|
#endif
 | 
						|
#ifdef  SO_ACCEPTCONN
 | 
						|
    PyModule_AddIntConstant(m, "SO_ACCEPTCONN", SO_ACCEPTCONN);
 | 
						|
#endif
 | 
						|
#ifdef  SO_REUSEADDR
 | 
						|
    PyModule_AddIntConstant(m, "SO_REUSEADDR", SO_REUSEADDR);
 | 
						|
#endif
 | 
						|
#ifdef SO_EXCLUSIVEADDRUSE
 | 
						|
    PyModule_AddIntConstant(m, "SO_EXCLUSIVEADDRUSE", SO_EXCLUSIVEADDRUSE);
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef  SO_KEEPALIVE
 | 
						|
    PyModule_AddIntConstant(m, "SO_KEEPALIVE", SO_KEEPALIVE);
 | 
						|
#endif
 | 
						|
#ifdef  SO_DONTROUTE
 | 
						|
    PyModule_AddIntConstant(m, "SO_DONTROUTE", SO_DONTROUTE);
 | 
						|
#endif
 | 
						|
#ifdef  SO_BROADCAST
 | 
						|
    PyModule_AddIntConstant(m, "SO_BROADCAST", SO_BROADCAST);
 | 
						|
#endif
 | 
						|
#ifdef  SO_USELOOPBACK
 | 
						|
    PyModule_AddIntConstant(m, "SO_USELOOPBACK", SO_USELOOPBACK);
 | 
						|
#endif
 | 
						|
#ifdef  SO_LINGER
 | 
						|
    PyModule_AddIntConstant(m, "SO_LINGER", SO_LINGER);
 | 
						|
#endif
 | 
						|
#ifdef  SO_OOBINLINE
 | 
						|
    PyModule_AddIntConstant(m, "SO_OOBINLINE", SO_OOBINLINE);
 | 
						|
#endif
 | 
						|
#ifdef  SO_REUSEPORT
 | 
						|
    PyModule_AddIntConstant(m, "SO_REUSEPORT", SO_REUSEPORT);
 | 
						|
#endif
 | 
						|
#ifdef  SO_SNDBUF
 | 
						|
    PyModule_AddIntConstant(m, "SO_SNDBUF", SO_SNDBUF);
 | 
						|
#endif
 | 
						|
#ifdef  SO_RCVBUF
 | 
						|
    PyModule_AddIntConstant(m, "SO_RCVBUF", SO_RCVBUF);
 | 
						|
#endif
 | 
						|
#ifdef  SO_SNDLOWAT
 | 
						|
    PyModule_AddIntConstant(m, "SO_SNDLOWAT", SO_SNDLOWAT);
 | 
						|
#endif
 | 
						|
#ifdef  SO_RCVLOWAT
 | 
						|
    PyModule_AddIntConstant(m, "SO_RCVLOWAT", SO_RCVLOWAT);
 | 
						|
#endif
 | 
						|
#ifdef  SO_SNDTIMEO
 | 
						|
    PyModule_AddIntConstant(m, "SO_SNDTIMEO", SO_SNDTIMEO);
 | 
						|
#endif
 | 
						|
#ifdef  SO_RCVTIMEO
 | 
						|
    PyModule_AddIntConstant(m, "SO_RCVTIMEO", SO_RCVTIMEO);
 | 
						|
#endif
 | 
						|
#ifdef  SO_ERROR
 | 
						|
    PyModule_AddIntConstant(m, "SO_ERROR", SO_ERROR);
 | 
						|
#endif
 | 
						|
#ifdef  SO_TYPE
 | 
						|
    PyModule_AddIntConstant(m, "SO_TYPE", SO_TYPE);
 | 
						|
#endif
 | 
						|
#ifdef  SO_SETFIB
 | 
						|
    PyModule_AddIntConstant(m, "SO_SETFIB", SO_SETFIB);
 | 
						|
#endif
 | 
						|
#ifdef  SO_PASSCRED
 | 
						|
    PyModule_AddIntConstant(m, "SO_PASSCRED", SO_PASSCRED);
 | 
						|
#endif
 | 
						|
#ifdef  SO_PEERCRED
 | 
						|
    PyModule_AddIntConstant(m, "SO_PEERCRED", SO_PEERCRED);
 | 
						|
#endif
 | 
						|
#ifdef  LOCAL_PEERCRED
 | 
						|
    PyModule_AddIntConstant(m, "LOCAL_PEERCRED", LOCAL_PEERCRED);
 | 
						|
#endif
 | 
						|
 | 
						|
    /* Maximum number of connections for "listen" */
 | 
						|
#ifdef  SOMAXCONN
 | 
						|
    PyModule_AddIntConstant(m, "SOMAXCONN", SOMAXCONN);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "SOMAXCONN", 5); /* Common value */
 | 
						|
#endif
 | 
						|
 | 
						|
    /* Ancilliary message types */
 | 
						|
#ifdef  SCM_RIGHTS
 | 
						|
    PyModule_AddIntConstant(m, "SCM_RIGHTS", SCM_RIGHTS);
 | 
						|
#endif
 | 
						|
#ifdef  SCM_CREDENTIALS
 | 
						|
    PyModule_AddIntConstant(m, "SCM_CREDENTIALS", SCM_CREDENTIALS);
 | 
						|
#endif
 | 
						|
#ifdef  SCM_CREDS
 | 
						|
    PyModule_AddIntConstant(m, "SCM_CREDS", SCM_CREDS);
 | 
						|
#endif
 | 
						|
 | 
						|
    /* Flags for send, recv */
 | 
						|
#ifdef  MSG_OOB
 | 
						|
    PyModule_AddIntConstant(m, "MSG_OOB", MSG_OOB);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_PEEK
 | 
						|
    PyModule_AddIntConstant(m, "MSG_PEEK", MSG_PEEK);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_DONTROUTE
 | 
						|
    PyModule_AddIntConstant(m, "MSG_DONTROUTE", MSG_DONTROUTE);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_DONTWAIT
 | 
						|
    PyModule_AddIntConstant(m, "MSG_DONTWAIT", MSG_DONTWAIT);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_EOR
 | 
						|
    PyModule_AddIntConstant(m, "MSG_EOR", MSG_EOR);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_TRUNC
 | 
						|
    PyModule_AddIntConstant(m, "MSG_TRUNC", MSG_TRUNC);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_CTRUNC
 | 
						|
    PyModule_AddIntConstant(m, "MSG_CTRUNC", MSG_CTRUNC);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_WAITALL
 | 
						|
    PyModule_AddIntConstant(m, "MSG_WAITALL", MSG_WAITALL);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_BTAG
 | 
						|
    PyModule_AddIntConstant(m, "MSG_BTAG", MSG_BTAG);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_ETAG
 | 
						|
    PyModule_AddIntConstant(m, "MSG_ETAG", MSG_ETAG);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_NOSIGNAL
 | 
						|
    PyModule_AddIntConstant(m, "MSG_NOSIGNAL", MSG_NOSIGNAL);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_NOTIFICATION
 | 
						|
    PyModule_AddIntConstant(m, "MSG_NOTIFICATION", MSG_NOTIFICATION);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_CMSG_CLOEXEC
 | 
						|
    PyModule_AddIntConstant(m, "MSG_CMSG_CLOEXEC", MSG_CMSG_CLOEXEC);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_ERRQUEUE
 | 
						|
    PyModule_AddIntConstant(m, "MSG_ERRQUEUE", MSG_ERRQUEUE);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_CONFIRM
 | 
						|
    PyModule_AddIntConstant(m, "MSG_CONFIRM", MSG_CONFIRM);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_MORE
 | 
						|
    PyModule_AddIntConstant(m, "MSG_MORE", MSG_MORE);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_EOF
 | 
						|
    PyModule_AddIntConstant(m, "MSG_EOF", MSG_EOF);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_BCAST
 | 
						|
    PyModule_AddIntConstant(m, "MSG_BCAST", MSG_BCAST);
 | 
						|
#endif
 | 
						|
#ifdef  MSG_MCAST
 | 
						|
    PyModule_AddIntConstant(m, "MSG_MCAST", MSG_MCAST);
 | 
						|
#endif
 | 
						|
 | 
						|
    /* Protocol level and numbers, usable for [gs]etsockopt */
 | 
						|
#ifdef  SOL_SOCKET
 | 
						|
    PyModule_AddIntConstant(m, "SOL_SOCKET", SOL_SOCKET);
 | 
						|
#endif
 | 
						|
#ifdef  SOL_IP
 | 
						|
    PyModule_AddIntConstant(m, "SOL_IP", SOL_IP);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "SOL_IP", 0);
 | 
						|
#endif
 | 
						|
#ifdef  SOL_IPX
 | 
						|
    PyModule_AddIntConstant(m, "SOL_IPX", SOL_IPX);
 | 
						|
#endif
 | 
						|
#ifdef  SOL_AX25
 | 
						|
    PyModule_AddIntConstant(m, "SOL_AX25", SOL_AX25);
 | 
						|
#endif
 | 
						|
#ifdef  SOL_ATALK
 | 
						|
    PyModule_AddIntConstant(m, "SOL_ATALK", SOL_ATALK);
 | 
						|
#endif
 | 
						|
#ifdef  SOL_NETROM
 | 
						|
    PyModule_AddIntConstant(m, "SOL_NETROM", SOL_NETROM);
 | 
						|
#endif
 | 
						|
#ifdef  SOL_ROSE
 | 
						|
    PyModule_AddIntConstant(m, "SOL_ROSE", SOL_ROSE);
 | 
						|
#endif
 | 
						|
#ifdef  SOL_TCP
 | 
						|
    PyModule_AddIntConstant(m, "SOL_TCP", SOL_TCP);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "SOL_TCP", 6);
 | 
						|
#endif
 | 
						|
#ifdef  SOL_UDP
 | 
						|
    PyModule_AddIntConstant(m, "SOL_UDP", SOL_UDP);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "SOL_UDP", 17);
 | 
						|
#endif
 | 
						|
#ifdef SOL_CAN_BASE
 | 
						|
    PyModule_AddIntConstant(m, "SOL_CAN_BASE", SOL_CAN_BASE);
 | 
						|
#endif
 | 
						|
#ifdef SOL_CAN_RAW
 | 
						|
    PyModule_AddIntConstant(m, "SOL_CAN_RAW", SOL_CAN_RAW);
 | 
						|
    PyModule_AddIntConstant(m, "CAN_RAW", CAN_RAW);
 | 
						|
#endif
 | 
						|
#ifdef HAVE_LINUX_CAN_H
 | 
						|
    PyModule_AddIntConstant(m, "CAN_EFF_FLAG", CAN_EFF_FLAG);
 | 
						|
    PyModule_AddIntConstant(m, "CAN_RTR_FLAG", CAN_RTR_FLAG);
 | 
						|
    PyModule_AddIntConstant(m, "CAN_ERR_FLAG", CAN_ERR_FLAG);
 | 
						|
 | 
						|
    PyModule_AddIntConstant(m, "CAN_SFF_MASK", CAN_SFF_MASK);
 | 
						|
    PyModule_AddIntConstant(m, "CAN_EFF_MASK", CAN_EFF_MASK);
 | 
						|
    PyModule_AddIntConstant(m, "CAN_ERR_MASK", CAN_ERR_MASK);
 | 
						|
#endif
 | 
						|
#ifdef HAVE_LINUX_CAN_RAW_H
 | 
						|
    PyModule_AddIntConstant(m, "CAN_RAW_FILTER", CAN_RAW_FILTER);
 | 
						|
    PyModule_AddIntConstant(m, "CAN_RAW_ERR_FILTER", CAN_RAW_ERR_FILTER);
 | 
						|
    PyModule_AddIntConstant(m, "CAN_RAW_LOOPBACK", CAN_RAW_LOOPBACK);
 | 
						|
    PyModule_AddIntConstant(m, "CAN_RAW_RECV_OWN_MSGS", CAN_RAW_RECV_OWN_MSGS);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_IP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_IP", IPPROTO_IP);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_IP", 0);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_HOPOPTS
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_HOPOPTS", IPPROTO_HOPOPTS);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_ICMP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_ICMP", IPPROTO_ICMP);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_ICMP", 1);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_IGMP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_IGMP", IPPROTO_IGMP);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_GGP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_GGP", IPPROTO_GGP);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_IPV4
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_IPV4", IPPROTO_IPV4);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_IPV6
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_IPIP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_IPIP", IPPROTO_IPIP);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_TCP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_TCP", IPPROTO_TCP);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_TCP", 6);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_EGP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_EGP", IPPROTO_EGP);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_PUP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_PUP", IPPROTO_PUP);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_UDP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_UDP", IPPROTO_UDP);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_UDP", 17);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_IDP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_IDP", IPPROTO_IDP);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_HELLO
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_HELLO", IPPROTO_HELLO);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_ND
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_ND", IPPROTO_ND);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_TP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_TP", IPPROTO_TP);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_IPV6
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_IPV6", IPPROTO_IPV6);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_ROUTING
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_ROUTING", IPPROTO_ROUTING);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_FRAGMENT
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_FRAGMENT", IPPROTO_FRAGMENT);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_RSVP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_RSVP", IPPROTO_RSVP);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_GRE
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_GRE", IPPROTO_GRE);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_ESP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_ESP", IPPROTO_ESP);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_AH
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_AH", IPPROTO_AH);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_MOBILE
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_MOBILE", IPPROTO_MOBILE);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_ICMPV6
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_ICMPV6", IPPROTO_ICMPV6);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_NONE
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_NONE", IPPROTO_NONE);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_DSTOPTS
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_DSTOPTS", IPPROTO_DSTOPTS);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_XTP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_XTP", IPPROTO_XTP);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_EON
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_EON", IPPROTO_EON);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_PIM
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_PIM", IPPROTO_PIM);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_IPCOMP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_IPCOMP", IPPROTO_IPCOMP);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_VRRP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_VRRP", IPPROTO_VRRP);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_SCTP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_SCTP", IPPROTO_SCTP);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_BIP
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_BIP", IPPROTO_BIP);
 | 
						|
#endif
 | 
						|
/**/
 | 
						|
#ifdef  IPPROTO_RAW
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_RAW", IPPROTO_RAW);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_RAW", 255);
 | 
						|
#endif
 | 
						|
#ifdef  IPPROTO_MAX
 | 
						|
    PyModule_AddIntConstant(m, "IPPROTO_MAX", IPPROTO_MAX);
 | 
						|
#endif
 | 
						|
 | 
						|
    /* Some port configuration */
 | 
						|
#ifdef  IPPORT_RESERVED
 | 
						|
    PyModule_AddIntConstant(m, "IPPORT_RESERVED", IPPORT_RESERVED);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "IPPORT_RESERVED", 1024);
 | 
						|
#endif
 | 
						|
#ifdef  IPPORT_USERRESERVED
 | 
						|
    PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", IPPORT_USERRESERVED);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "IPPORT_USERRESERVED", 5000);
 | 
						|
#endif
 | 
						|
 | 
						|
    /* Some reserved IP v.4 addresses */
 | 
						|
#ifdef  INADDR_ANY
 | 
						|
    PyModule_AddIntConstant(m, "INADDR_ANY", INADDR_ANY);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "INADDR_ANY", 0x00000000);
 | 
						|
#endif
 | 
						|
#ifdef  INADDR_BROADCAST
 | 
						|
    PyModule_AddIntConstant(m, "INADDR_BROADCAST", INADDR_BROADCAST);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "INADDR_BROADCAST", 0xffffffff);
 | 
						|
#endif
 | 
						|
#ifdef  INADDR_LOOPBACK
 | 
						|
    PyModule_AddIntConstant(m, "INADDR_LOOPBACK", INADDR_LOOPBACK);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "INADDR_LOOPBACK", 0x7F000001);
 | 
						|
#endif
 | 
						|
#ifdef  INADDR_UNSPEC_GROUP
 | 
						|
    PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "INADDR_UNSPEC_GROUP", 0xe0000000);
 | 
						|
#endif
 | 
						|
#ifdef  INADDR_ALLHOSTS_GROUP
 | 
						|
    PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP",
 | 
						|
                            INADDR_ALLHOSTS_GROUP);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "INADDR_ALLHOSTS_GROUP", 0xe0000001);
 | 
						|
#endif
 | 
						|
#ifdef  INADDR_MAX_LOCAL_GROUP
 | 
						|
    PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP",
 | 
						|
                            INADDR_MAX_LOCAL_GROUP);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff);
 | 
						|
#endif
 | 
						|
#ifdef  INADDR_NONE
 | 
						|
    PyModule_AddIntConstant(m, "INADDR_NONE", INADDR_NONE);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "INADDR_NONE", 0xffffffff);
 | 
						|
#endif
 | 
						|
 | 
						|
    /* IPv4 [gs]etsockopt options */
 | 
						|
#ifdef  IP_OPTIONS
 | 
						|
    PyModule_AddIntConstant(m, "IP_OPTIONS", IP_OPTIONS);
 | 
						|
#endif
 | 
						|
#ifdef  IP_HDRINCL
 | 
						|
    PyModule_AddIntConstant(m, "IP_HDRINCL", IP_HDRINCL);
 | 
						|
#endif
 | 
						|
#ifdef  IP_TOS
 | 
						|
    PyModule_AddIntConstant(m, "IP_TOS", IP_TOS);
 | 
						|
#endif
 | 
						|
#ifdef  IP_TTL
 | 
						|
    PyModule_AddIntConstant(m, "IP_TTL", IP_TTL);
 | 
						|
#endif
 | 
						|
#ifdef  IP_RECVOPTS
 | 
						|
    PyModule_AddIntConstant(m, "IP_RECVOPTS", IP_RECVOPTS);
 | 
						|
#endif
 | 
						|
#ifdef  IP_RECVRETOPTS
 | 
						|
    PyModule_AddIntConstant(m, "IP_RECVRETOPTS", IP_RECVRETOPTS);
 | 
						|
#endif
 | 
						|
#ifdef  IP_RECVDSTADDR
 | 
						|
    PyModule_AddIntConstant(m, "IP_RECVDSTADDR", IP_RECVDSTADDR);
 | 
						|
#endif
 | 
						|
#ifdef  IP_RETOPTS
 | 
						|
    PyModule_AddIntConstant(m, "IP_RETOPTS", IP_RETOPTS);
 | 
						|
#endif
 | 
						|
#ifdef  IP_MULTICAST_IF
 | 
						|
    PyModule_AddIntConstant(m, "IP_MULTICAST_IF", IP_MULTICAST_IF);
 | 
						|
#endif
 | 
						|
#ifdef  IP_MULTICAST_TTL
 | 
						|
    PyModule_AddIntConstant(m, "IP_MULTICAST_TTL", IP_MULTICAST_TTL);
 | 
						|
#endif
 | 
						|
#ifdef  IP_MULTICAST_LOOP
 | 
						|
    PyModule_AddIntConstant(m, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP);
 | 
						|
#endif
 | 
						|
#ifdef  IP_ADD_MEMBERSHIP
 | 
						|
    PyModule_AddIntConstant(m, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP);
 | 
						|
#endif
 | 
						|
#ifdef  IP_DROP_MEMBERSHIP
 | 
						|
    PyModule_AddIntConstant(m, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP);
 | 
						|
#endif
 | 
						|
#ifdef  IP_DEFAULT_MULTICAST_TTL
 | 
						|
    PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_TTL",
 | 
						|
                            IP_DEFAULT_MULTICAST_TTL);
 | 
						|
#endif
 | 
						|
#ifdef  IP_DEFAULT_MULTICAST_LOOP
 | 
						|
    PyModule_AddIntConstant(m, "IP_DEFAULT_MULTICAST_LOOP",
 | 
						|
                            IP_DEFAULT_MULTICAST_LOOP);
 | 
						|
#endif
 | 
						|
#ifdef  IP_MAX_MEMBERSHIPS
 | 
						|
    PyModule_AddIntConstant(m, "IP_MAX_MEMBERSHIPS", IP_MAX_MEMBERSHIPS);
 | 
						|
#endif
 | 
						|
 | 
						|
    /* IPv6 [gs]etsockopt options, defined in RFC2553 */
 | 
						|
#ifdef  IPV6_JOIN_GROUP
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_JOIN_GROUP", IPV6_JOIN_GROUP);
 | 
						|
#endif
 | 
						|
#ifdef  IPV6_LEAVE_GROUP
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_LEAVE_GROUP", IPV6_LEAVE_GROUP);
 | 
						|
#endif
 | 
						|
#ifdef  IPV6_MULTICAST_HOPS
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_MULTICAST_HOPS", IPV6_MULTICAST_HOPS);
 | 
						|
#endif
 | 
						|
#ifdef  IPV6_MULTICAST_IF
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_MULTICAST_IF", IPV6_MULTICAST_IF);
 | 
						|
#endif
 | 
						|
#ifdef  IPV6_MULTICAST_LOOP
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_MULTICAST_LOOP", IPV6_MULTICAST_LOOP);
 | 
						|
#endif
 | 
						|
#ifdef  IPV6_UNICAST_HOPS
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_UNICAST_HOPS", IPV6_UNICAST_HOPS);
 | 
						|
#endif
 | 
						|
    /* Additional IPV6 socket options, defined in RFC 3493 */
 | 
						|
#ifdef IPV6_V6ONLY
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_V6ONLY", IPV6_V6ONLY);
 | 
						|
#endif
 | 
						|
    /* Advanced IPV6 socket options, from RFC 3542 */
 | 
						|
#ifdef IPV6_CHECKSUM
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_CHECKSUM", IPV6_CHECKSUM);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_DONTFRAG
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_DONTFRAG", IPV6_DONTFRAG);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_DSTOPTS
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_DSTOPTS", IPV6_DSTOPTS);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_HOPLIMIT
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_HOPLIMIT", IPV6_HOPLIMIT);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_HOPOPTS
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_HOPOPTS", IPV6_HOPOPTS);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_NEXTHOP
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_NEXTHOP", IPV6_NEXTHOP);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_PATHMTU
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_PATHMTU", IPV6_PATHMTU);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_PKTINFO
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_PKTINFO", IPV6_PKTINFO);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_RECVDSTOPTS
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_RECVDSTOPTS", IPV6_RECVDSTOPTS);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_RECVHOPLIMIT
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_RECVHOPLIMIT", IPV6_RECVHOPLIMIT);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_RECVHOPOPTS
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_RECVHOPOPTS", IPV6_RECVHOPOPTS);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_RECVPKTINFO
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_RECVPKTINFO", IPV6_RECVPKTINFO);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_RECVRTHDR
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_RECVRTHDR", IPV6_RECVRTHDR);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_RECVTCLASS
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_RECVTCLASS", IPV6_RECVTCLASS);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_RTHDR
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_RTHDR", IPV6_RTHDR);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_RTHDRDSTOPTS
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_RTHDRDSTOPTS", IPV6_RTHDRDSTOPTS);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_RTHDR_TYPE_0
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_RTHDR_TYPE_0", IPV6_RTHDR_TYPE_0);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_RECVPATHMTU
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_RECVPATHMTU", IPV6_RECVPATHMTU);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_TCLASS
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_TCLASS", IPV6_TCLASS);
 | 
						|
#endif
 | 
						|
#ifdef IPV6_USE_MIN_MTU
 | 
						|
    PyModule_AddIntConstant(m, "IPV6_USE_MIN_MTU", IPV6_USE_MIN_MTU);
 | 
						|
#endif
 | 
						|
 | 
						|
    /* TCP options */
 | 
						|
#ifdef  TCP_NODELAY
 | 
						|
    PyModule_AddIntConstant(m, "TCP_NODELAY", TCP_NODELAY);
 | 
						|
#endif
 | 
						|
#ifdef  TCP_MAXSEG
 | 
						|
    PyModule_AddIntConstant(m, "TCP_MAXSEG", TCP_MAXSEG);
 | 
						|
#endif
 | 
						|
#ifdef  TCP_CORK
 | 
						|
    PyModule_AddIntConstant(m, "TCP_CORK", TCP_CORK);
 | 
						|
#endif
 | 
						|
#ifdef  TCP_KEEPIDLE
 | 
						|
    PyModule_AddIntConstant(m, "TCP_KEEPIDLE", TCP_KEEPIDLE);
 | 
						|
#endif
 | 
						|
#ifdef  TCP_KEEPINTVL
 | 
						|
    PyModule_AddIntConstant(m, "TCP_KEEPINTVL", TCP_KEEPINTVL);
 | 
						|
#endif
 | 
						|
#ifdef  TCP_KEEPCNT
 | 
						|
    PyModule_AddIntConstant(m, "TCP_KEEPCNT", TCP_KEEPCNT);
 | 
						|
#endif
 | 
						|
#ifdef  TCP_SYNCNT
 | 
						|
    PyModule_AddIntConstant(m, "TCP_SYNCNT", TCP_SYNCNT);
 | 
						|
#endif
 | 
						|
#ifdef  TCP_LINGER2
 | 
						|
    PyModule_AddIntConstant(m, "TCP_LINGER2", TCP_LINGER2);
 | 
						|
#endif
 | 
						|
#ifdef  TCP_DEFER_ACCEPT
 | 
						|
    PyModule_AddIntConstant(m, "TCP_DEFER_ACCEPT", TCP_DEFER_ACCEPT);
 | 
						|
#endif
 | 
						|
#ifdef  TCP_WINDOW_CLAMP
 | 
						|
    PyModule_AddIntConstant(m, "TCP_WINDOW_CLAMP", TCP_WINDOW_CLAMP);
 | 
						|
#endif
 | 
						|
#ifdef  TCP_INFO
 | 
						|
    PyModule_AddIntConstant(m, "TCP_INFO", TCP_INFO);
 | 
						|
#endif
 | 
						|
#ifdef  TCP_QUICKACK
 | 
						|
    PyModule_AddIntConstant(m, "TCP_QUICKACK", TCP_QUICKACK);
 | 
						|
#endif
 | 
						|
 | 
						|
 | 
						|
    /* IPX options */
 | 
						|
#ifdef  IPX_TYPE
 | 
						|
    PyModule_AddIntConstant(m, "IPX_TYPE", IPX_TYPE);
 | 
						|
#endif
 | 
						|
 | 
						|
    /* get{addr,name}info parameters */
 | 
						|
#ifdef EAI_ADDRFAMILY
 | 
						|
    PyModule_AddIntConstant(m, "EAI_ADDRFAMILY", EAI_ADDRFAMILY);
 | 
						|
#endif
 | 
						|
#ifdef EAI_AGAIN
 | 
						|
    PyModule_AddIntConstant(m, "EAI_AGAIN", EAI_AGAIN);
 | 
						|
#endif
 | 
						|
#ifdef EAI_BADFLAGS
 | 
						|
    PyModule_AddIntConstant(m, "EAI_BADFLAGS", EAI_BADFLAGS);
 | 
						|
#endif
 | 
						|
#ifdef EAI_FAIL
 | 
						|
    PyModule_AddIntConstant(m, "EAI_FAIL", EAI_FAIL);
 | 
						|
#endif
 | 
						|
#ifdef EAI_FAMILY
 | 
						|
    PyModule_AddIntConstant(m, "EAI_FAMILY", EAI_FAMILY);
 | 
						|
#endif
 | 
						|
#ifdef EAI_MEMORY
 | 
						|
    PyModule_AddIntConstant(m, "EAI_MEMORY", EAI_MEMORY);
 | 
						|
#endif
 | 
						|
#ifdef EAI_NODATA
 | 
						|
    PyModule_AddIntConstant(m, "EAI_NODATA", EAI_NODATA);
 | 
						|
#endif
 | 
						|
#ifdef EAI_NONAME
 | 
						|
    PyModule_AddIntConstant(m, "EAI_NONAME", EAI_NONAME);
 | 
						|
#endif
 | 
						|
#ifdef EAI_OVERFLOW
 | 
						|
    PyModule_AddIntConstant(m, "EAI_OVERFLOW", EAI_OVERFLOW);
 | 
						|
#endif
 | 
						|
#ifdef EAI_SERVICE
 | 
						|
    PyModule_AddIntConstant(m, "EAI_SERVICE", EAI_SERVICE);
 | 
						|
#endif
 | 
						|
#ifdef EAI_SOCKTYPE
 | 
						|
    PyModule_AddIntConstant(m, "EAI_SOCKTYPE", EAI_SOCKTYPE);
 | 
						|
#endif
 | 
						|
#ifdef EAI_SYSTEM
 | 
						|
    PyModule_AddIntConstant(m, "EAI_SYSTEM", EAI_SYSTEM);
 | 
						|
#endif
 | 
						|
#ifdef EAI_BADHINTS
 | 
						|
    PyModule_AddIntConstant(m, "EAI_BADHINTS", EAI_BADHINTS);
 | 
						|
#endif
 | 
						|
#ifdef EAI_PROTOCOL
 | 
						|
    PyModule_AddIntConstant(m, "EAI_PROTOCOL", EAI_PROTOCOL);
 | 
						|
#endif
 | 
						|
#ifdef EAI_MAX
 | 
						|
    PyModule_AddIntConstant(m, "EAI_MAX", EAI_MAX);
 | 
						|
#endif
 | 
						|
#ifdef AI_PASSIVE
 | 
						|
    PyModule_AddIntConstant(m, "AI_PASSIVE", AI_PASSIVE);
 | 
						|
#endif
 | 
						|
#ifdef AI_CANONNAME
 | 
						|
    PyModule_AddIntConstant(m, "AI_CANONNAME", AI_CANONNAME);
 | 
						|
#endif
 | 
						|
#ifdef AI_NUMERICHOST
 | 
						|
    PyModule_AddIntConstant(m, "AI_NUMERICHOST", AI_NUMERICHOST);
 | 
						|
#endif
 | 
						|
#ifdef AI_NUMERICSERV
 | 
						|
    PyModule_AddIntConstant(m, "AI_NUMERICSERV", AI_NUMERICSERV);
 | 
						|
#endif
 | 
						|
#ifdef AI_MASK
 | 
						|
    PyModule_AddIntConstant(m, "AI_MASK", AI_MASK);
 | 
						|
#endif
 | 
						|
#ifdef AI_ALL
 | 
						|
    PyModule_AddIntConstant(m, "AI_ALL", AI_ALL);
 | 
						|
#endif
 | 
						|
#ifdef AI_V4MAPPED_CFG
 | 
						|
    PyModule_AddIntConstant(m, "AI_V4MAPPED_CFG", AI_V4MAPPED_CFG);
 | 
						|
#endif
 | 
						|
#ifdef AI_ADDRCONFIG
 | 
						|
    PyModule_AddIntConstant(m, "AI_ADDRCONFIG", AI_ADDRCONFIG);
 | 
						|
#endif
 | 
						|
#ifdef AI_V4MAPPED
 | 
						|
    PyModule_AddIntConstant(m, "AI_V4MAPPED", AI_V4MAPPED);
 | 
						|
#endif
 | 
						|
#ifdef AI_DEFAULT
 | 
						|
    PyModule_AddIntConstant(m, "AI_DEFAULT", AI_DEFAULT);
 | 
						|
#endif
 | 
						|
#ifdef NI_MAXHOST
 | 
						|
    PyModule_AddIntConstant(m, "NI_MAXHOST", NI_MAXHOST);
 | 
						|
#endif
 | 
						|
#ifdef NI_MAXSERV
 | 
						|
    PyModule_AddIntConstant(m, "NI_MAXSERV", NI_MAXSERV);
 | 
						|
#endif
 | 
						|
#ifdef NI_NOFQDN
 | 
						|
    PyModule_AddIntConstant(m, "NI_NOFQDN", NI_NOFQDN);
 | 
						|
#endif
 | 
						|
#ifdef NI_NUMERICHOST
 | 
						|
    PyModule_AddIntConstant(m, "NI_NUMERICHOST", NI_NUMERICHOST);
 | 
						|
#endif
 | 
						|
#ifdef NI_NAMEREQD
 | 
						|
    PyModule_AddIntConstant(m, "NI_NAMEREQD", NI_NAMEREQD);
 | 
						|
#endif
 | 
						|
#ifdef NI_NUMERICSERV
 | 
						|
    PyModule_AddIntConstant(m, "NI_NUMERICSERV", NI_NUMERICSERV);
 | 
						|
#endif
 | 
						|
#ifdef NI_DGRAM
 | 
						|
    PyModule_AddIntConstant(m, "NI_DGRAM", NI_DGRAM);
 | 
						|
#endif
 | 
						|
 | 
						|
    /* shutdown() parameters */
 | 
						|
#ifdef SHUT_RD
 | 
						|
    PyModule_AddIntConstant(m, "SHUT_RD", SHUT_RD);
 | 
						|
#elif defined(SD_RECEIVE)
 | 
						|
    PyModule_AddIntConstant(m, "SHUT_RD", SD_RECEIVE);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "SHUT_RD", 0);
 | 
						|
#endif
 | 
						|
#ifdef SHUT_WR
 | 
						|
    PyModule_AddIntConstant(m, "SHUT_WR", SHUT_WR);
 | 
						|
#elif defined(SD_SEND)
 | 
						|
    PyModule_AddIntConstant(m, "SHUT_WR", SD_SEND);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "SHUT_WR", 1);
 | 
						|
#endif
 | 
						|
#ifdef SHUT_RDWR
 | 
						|
    PyModule_AddIntConstant(m, "SHUT_RDWR", SHUT_RDWR);
 | 
						|
#elif defined(SD_BOTH)
 | 
						|
    PyModule_AddIntConstant(m, "SHUT_RDWR", SD_BOTH);
 | 
						|
#else
 | 
						|
    PyModule_AddIntConstant(m, "SHUT_RDWR", 2);
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef SIO_RCVALL
 | 
						|
    {
 | 
						|
        DWORD codes[] = {SIO_RCVALL, SIO_KEEPALIVE_VALS};
 | 
						|
        const char *names[] = {"SIO_RCVALL", "SIO_KEEPALIVE_VALS"};
 | 
						|
        int i;
 | 
						|
        for(i = 0; i<Py_ARRAY_LENGTH(codes); ++i) {
 | 
						|
            PyObject *tmp;
 | 
						|
            tmp = PyLong_FromUnsignedLong(codes[i]);
 | 
						|
            if (tmp == NULL)
 | 
						|
                return NULL;
 | 
						|
            PyModule_AddObject(m, names[i], tmp);
 | 
						|
        }
 | 
						|
    }
 | 
						|
    PyModule_AddIntConstant(m, "RCVALL_OFF", RCVALL_OFF);
 | 
						|
    PyModule_AddIntConstant(m, "RCVALL_ON", RCVALL_ON);
 | 
						|
    PyModule_AddIntConstant(m, "RCVALL_SOCKETLEVELONLY", RCVALL_SOCKETLEVELONLY);
 | 
						|
#ifdef RCVALL_IPLEVEL
 | 
						|
    PyModule_AddIntConstant(m, "RCVALL_IPLEVEL", RCVALL_IPLEVEL);
 | 
						|
#endif
 | 
						|
#ifdef RCVALL_MAX
 | 
						|
    PyModule_AddIntConstant(m, "RCVALL_MAX", RCVALL_MAX);
 | 
						|
#endif
 | 
						|
#endif /* _MSTCPIP_ */
 | 
						|
 | 
						|
    /* Initialize gethostbyname lock */
 | 
						|
#if defined(USE_GETHOSTBYNAME_LOCK) || defined(USE_GETADDRINFO_LOCK)
 | 
						|
    netdb_lock = PyThread_allocate_lock();
 | 
						|
#endif
 | 
						|
    return m;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
#ifndef HAVE_INET_PTON
 | 
						|
#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < NTDDI_LONGHORN)
 | 
						|
 | 
						|
/* Simplistic emulation code for inet_pton that only works for IPv4 */
 | 
						|
/* These are not exposed because they do not set errno properly */
 | 
						|
 | 
						|
int
 | 
						|
inet_pton(int af, const char *src, void *dst)
 | 
						|
{
 | 
						|
    if (af == AF_INET) {
 | 
						|
#if (SIZEOF_INT != 4)
 | 
						|
#error "Not sure if in_addr_t exists and int is not 32-bits."
 | 
						|
#endif
 | 
						|
        unsigned int packed_addr;
 | 
						|
        packed_addr = inet_addr(src);
 | 
						|
        if (packed_addr == INADDR_NONE)
 | 
						|
            return 0;
 | 
						|
        memcpy(dst, &packed_addr, 4);
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
    /* Should set errno to EAFNOSUPPORT */
 | 
						|
    return -1;
 | 
						|
}
 | 
						|
 | 
						|
const char *
 | 
						|
inet_ntop(int af, const void *src, char *dst, socklen_t size)
 | 
						|
{
 | 
						|
    if (af == AF_INET) {
 | 
						|
        struct in_addr packed_addr;
 | 
						|
        if (size < 16)
 | 
						|
            /* Should set errno to ENOSPC. */
 | 
						|
            return NULL;
 | 
						|
        memcpy(&packed_addr, src, sizeof(packed_addr));
 | 
						|
        return strncpy(dst, inet_ntoa(packed_addr), size);
 | 
						|
    }
 | 
						|
    /* Should set errno to EAFNOSUPPORT */
 | 
						|
    return NULL;
 | 
						|
}
 | 
						|
 | 
						|
#endif
 | 
						|
#endif
 |