| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1 | /*********************************************************** | 
| Guido van Rossum | 524b588 | 1995-01-04 19:10:35 +0000 | [diff] [blame] | 2 | Copyright 1991-1995 by Stichting Mathematisch Centrum, Amsterdam, | 
 | 3 | The Netherlands. | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 4 |  | 
 | 5 |                         All Rights Reserved | 
 | 6 |  | 
 | 7 | Permission to use, copy, modify, and distribute this software and its  | 
 | 8 | documentation for any purpose and without fee is hereby granted,  | 
 | 9 | provided that the above copyright notice appear in all copies and that | 
 | 10 | both that copyright notice and this permission notice appear in  | 
 | 11 | supporting documentation, and that the names of Stichting Mathematisch | 
 | 12 | Centrum or CWI not be used in advertising or publicity pertaining to | 
 | 13 | distribution of the software without specific, written prior permission. | 
 | 14 |  | 
 | 15 | STICHTING MATHEMATISCH CENTRUM DISCLAIMS ALL WARRANTIES WITH REGARD TO | 
 | 16 | THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND | 
 | 17 | FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH CENTRUM BE LIABLE | 
 | 18 | FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES | 
 | 19 | WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN | 
 | 20 | ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT | 
 | 21 | OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. | 
 | 22 |  | 
 | 23 | ******************************************************************/ | 
 | 24 |  | 
 | 25 | /* Socket module */ | 
 | 26 |  | 
 | 27 | /* | 
 | 28 | This module provides an interface to Berkeley socket IPC. | 
 | 29 |  | 
 | 30 | Limitations: | 
 | 31 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 32 | - only AF_INET and AF_UNIX address families are supported | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 33 | - no asynchronous I/O (but you can use select() on sockets) | 
| Guido van Rossum | 8119447 | 1991-07-27 21:42:02 +0000 | [diff] [blame] | 34 | - no read/write operations (use send/recv or makefile instead) | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 35 | - setsockopt() and getsockopt() only support integer options | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 36 |  | 
 | 37 | Interface: | 
 | 38 |  | 
| Guido van Rossum | 8119447 | 1991-07-27 21:42:02 +0000 | [diff] [blame] | 39 | - socket.gethostname() --> host name (string) | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 40 | - socket.gethostbyname(hostname) --> host IP address (string: 'dd.dd.dd.dd') | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 41 | - socket.gethostbyaddr(IP address) --> (hostname, [alias, ...], [IP addr, ...]) | 
| Guido van Rossum | 4dd2a7e | 1991-07-01 18:51:33 +0000 | [diff] [blame] | 42 | - socket.getservbyname(servername, protocolname) --> port number | 
 | 43 | - socket.socket(family, type [, proto]) --> new socket object | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 44 | - family and type constants from <socket.h> are accessed as socket.AF_INET etc. | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 45 | - errors are reported as the exception socket.error | 
 | 46 | - an Internet socket address is a pair (hostname, port) | 
 | 47 |   where hostname can be anything recognized by gethostbyname() | 
 | 48 |   (including the dd.dd.dd.dd notation) and port is in host byte order | 
 | 49 | - where a hostname is returned, the dd.dd.dd.dd notation is used | 
 | 50 | - a UNIX domain socket is a string specifying the pathname | 
 | 51 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 52 | Socket methods: | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 53 |  | 
| Guido van Rossum | 8119447 | 1991-07-27 21:42:02 +0000 | [diff] [blame] | 54 | - s.accept() --> new socket object, sockaddr | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 55 | - s.setsockopt(level, optname, flag) --> Py_None | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 56 | - s.getsockopt(level, optname) --> flag | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 57 | - s.bind(sockaddr) --> Py_None | 
 | 58 | - s.connect(sockaddr) --> Py_None | 
| Guido van Rossum | c89705d | 1992-11-26 08:54:07 +0000 | [diff] [blame] | 59 | - s.getsockname() --> sockaddr | 
 | 60 | - s.getpeername() --> sockaddr | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 61 | - s.listen(n) --> Py_None | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 62 | - s.makefile(mode) --> file object | 
| Guido van Rossum | eb6b33a | 1993-05-25 09:38:27 +0000 | [diff] [blame] | 63 | - s.recv(nbytes [,flags]) --> string | 
| Guido van Rossum | 18c9a4f | 1993-05-25 12:16:29 +0000 | [diff] [blame] | 64 | - s.recvfrom(nbytes [,flags]) --> string, sockaddr | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 65 | - s.send(string [,flags]) --> nbytes | 
 | 66 | - s.sendto(string, [flags,] sockaddr) --> nbytes | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 67 | - s.setblocking(1 | 0) --> Py_None | 
 | 68 | - s.shutdown(how) --> Py_None | 
 | 69 | - s.close() --> Py_None | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 70 | - repr(s) --> "<socket object, fd=%d, family=%d, type=%d, protocol=%d>" | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 71 |  | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 72 | */ | 
 | 73 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 74 | #include "Python.h" | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 75 |  | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 76 | #include <sys/types.h> | 
 | 77 | #include "mytime.h" | 
| Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 78 |  | 
| Guido van Rossum | 8119447 | 1991-07-27 21:42:02 +0000 | [diff] [blame] | 79 | #include <signal.h> | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 80 | #ifndef NT | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 81 | #include <netdb.h> | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 82 | #include <sys/socket.h> | 
 | 83 | #include <netinet/in.h> | 
| Guido van Rossum | e4485b0 | 1994-09-07 14:32:49 +0000 | [diff] [blame] | 84 | #include <fcntl.h> | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 85 | #else | 
 | 86 | #include <winsock.h> | 
 | 87 | #endif | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 88 | #ifdef HAVE_SYS_UN_H | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 89 | #include <sys/un.h> | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 90 | #else | 
 | 91 | #undef AF_UNIX | 
| Guido van Rossum | 9575a44 | 1993-04-07 14:06:14 +0000 | [diff] [blame] | 92 | #endif | 
 | 93 |  | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 94 | #ifndef O_NDELAY | 
 | 95 | #define O_NDELAY O_NONBLOCK	/* For QNX only? */ | 
 | 96 | #endif | 
 | 97 |  | 
 | 98 |  | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 99 | /* Here we have some hacks to choose between K&R or ANSI style function | 
 | 100 |    definitions.  For NT to build this as an extension module (ie, DLL) | 
 | 101 |    it must be compiled by the C++ compiler, as it takes the address of | 
 | 102 |    a static data item exported from the main Python DLL. | 
 | 103 | */ | 
 | 104 | #ifdef NT | 
 | 105 | /* seem to be a few differences in the API */ | 
 | 106 | #define close closesocket | 
 | 107 | #define NO_DUP	/* I wont trust passing a socket to NT's RTL!!  */ | 
 | 108 | #define FORCE_ANSI_FUNC_DEFS | 
 | 109 | #endif | 
 | 110 |  | 
 | 111 | #ifdef FORCE_ANSI_FUNC_DEFS | 
 | 112 | #define BUILD_FUNC_DEF_1( fnname, arg1type, arg1name )	\ | 
 | 113 | fnname( arg1type arg1name ) | 
 | 114 |  | 
 | 115 | #define BUILD_FUNC_DEF_2( fnname, arg1type, arg1name, arg2type, arg2name ) \ | 
 | 116 | fnname( arg1type arg1name, arg2type arg2name ) | 
 | 117 |  | 
 | 118 | #define BUILD_FUNC_DEF_3( fnname, arg1type, arg1name, arg2type, arg2name , arg3type, arg3name )	\ | 
 | 119 | fnname( arg1type arg1name, arg2type arg2name, arg3type arg3name ) | 
 | 120 |  | 
 | 121 | #define BUILD_FUNC_DEF_4( fnname, arg1type, arg1name, arg2type, arg2name , arg3type, arg3name, arg4type, arg4name )	\ | 
 | 122 | fnname( arg1type arg1name, arg2type arg2name, arg3type arg3name, arg4type arg4name ) | 
 | 123 |  | 
 | 124 | #else /* !FORCE_ANSI_FN_DEFS */ | 
 | 125 | #define BUILD_FUNC_DEF_1( fnname, arg1type, arg1name )	\ | 
 | 126 | fnname( arg1name )	\ | 
 | 127 | 	arg1type arg1name; | 
 | 128 |  | 
 | 129 | #define BUILD_FUNC_DEF_2( fnname, arg1type, arg1name, arg2type, arg2name ) \ | 
 | 130 | fnname( arg1name, arg2name )	\ | 
 | 131 | 	arg1type arg1name;	\ | 
 | 132 | 	arg2type arg2name; | 
 | 133 |  | 
 | 134 | #define BUILD_FUNC_DEF_3( fnname, arg1type, arg1name, arg2type, arg2name, arg3type, arg3name ) \ | 
 | 135 | fnname( arg1name, arg2name, arg3name )	\ | 
 | 136 | 	arg1type arg1name;	\ | 
 | 137 | 	arg2type arg2name;	\ | 
 | 138 | 	arg3type arg3name; | 
 | 139 |  | 
 | 140 | #define BUILD_FUNC_DEF_4( fnname, arg1type, arg1name, arg2type, arg2name, arg3type, arg3name, arg4type, arg4name ) \ | 
 | 141 | fnname( arg1name, arg2name, arg3name, arg4name )	\ | 
 | 142 | 	arg1type arg1name;	\ | 
 | 143 | 	arg2type arg2name;	\ | 
 | 144 | 	arg3type arg3name;	\ | 
 | 145 | 	arg4type arg4name; | 
 | 146 |  | 
 | 147 | #endif /* !FORCE_ANSI_FN_DEFS */ | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 148 |  | 
 | 149 | /* Global variable holding the exception type for errors detected | 
 | 150 |    by this module (but not argument type or memory errors, etc.). */ | 
 | 151 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 152 | static PyObject *PySocket_Error; | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 153 |  | 
 | 154 |  | 
 | 155 | /* Convenience function to raise an error according to errno | 
 | 156 |    and return a NULL pointer from a function. */ | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 157 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 158 | static PyObject * | 
 | 159 | PySocket_Err() | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 160 | { | 
| Guido van Rossum | 14ed0b2 | 1994-09-29 09:50:09 +0000 | [diff] [blame] | 161 | #ifdef NT | 
 | 162 | 	if (WSAGetLastError()) { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 163 | 		PyObject *v; | 
 | 164 | 		v = Py_BuildValue("(is)", WSAGetLastError(), "winsock error"); | 
| Guido van Rossum | 14ed0b2 | 1994-09-29 09:50:09 +0000 | [diff] [blame] | 165 | 		if (v != NULL) { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 166 | 			PyErr_SetObject(PySocket_Error, v); | 
 | 167 | 			Py_DECREF(v); | 
| Guido van Rossum | 14ed0b2 | 1994-09-29 09:50:09 +0000 | [diff] [blame] | 168 | 		} | 
 | 169 | 		return NULL; | 
 | 170 | 	} | 
 | 171 | 	else | 
 | 172 | #endif | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 173 | 	return PyErr_SetFromErrno(PySocket_Error); | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 174 | } | 
 | 175 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 176 |  | 
 | 177 | /* The object holding a socket.  It holds some extra information, | 
 | 178 |    like the address family, which is used to decode socket address | 
 | 179 |    arguments properly. */ | 
 | 180 |  | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 181 | typedef struct { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 182 | 	PyObject_HEAD | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 183 | 	int sock_fd;		/* Socket file descriptor */ | 
 | 184 | 	int sock_family;	/* Address family, e.g., AF_INET */ | 
 | 185 | 	int sock_type;		/* Socket type, e.g., SOCK_STREAM */ | 
 | 186 | 	int sock_proto;		/* Protocol type, usually 0 */ | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 187 | 	union sock_addr { | 
 | 188 | 		struct sockaddr_in in; | 
 | 189 | #ifdef AF_UNIX | 
 | 190 | 		struct sockaddr_un un; | 
 | 191 | #endif | 
 | 192 | 	} sock_addr; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 193 | } PySocketSockObject; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 194 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 195 |  | 
 | 196 | /* A forward reference to the Socktype type object. | 
 | 197 |    The Socktype variable contains pointers to various functions, | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 198 |    some of which call newsockobject(), which uses Socktype, so | 
| Guido van Rossum | 54ba21b | 1991-09-10 14:57:12 +0000 | [diff] [blame] | 199 |    there has to be a circular reference. */ | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 200 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 201 | staticforward PyTypeObject PySocketSock_Type; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 202 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 203 |  | 
 | 204 | /* Create a new socket object. | 
 | 205 |    This just creates the object and initializes it. | 
 | 206 |    If the creation fails, return NULL and set an exception (implicit | 
 | 207 |    in NEWOBJ()). */ | 
 | 208 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 209 | static PySocketSockObject * | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 210 | BUILD_FUNC_DEF_4(PySocketSock_New,int,fd, int,family, int,type, int,proto) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 211 | { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 212 | 	PySocketSockObject *s; | 
 | 213 | 	s = PyObject_NEW(PySocketSockObject, &PySocketSock_Type); | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 214 | 	if (s != NULL) { | 
 | 215 | 		s->sock_fd = fd; | 
 | 216 | 		s->sock_family = family; | 
 | 217 | 		s->sock_type = type; | 
 | 218 | 		s->sock_proto = proto; | 
 | 219 | 	} | 
 | 220 | 	return s; | 
 | 221 | } | 
 | 222 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 223 |  | 
 | 224 | /* Convert a string specifying a host name or one of a few symbolic | 
 | 225 |    names to a numeric IP address.  This usually calls gethostbyname() | 
 | 226 |    to do the work; the names "" and "<broadcast>" are special. | 
 | 227 |    Return the length (should always be 4 bytes), or negative if | 
 | 228 |    an error occurred; then an exception is raised. */ | 
 | 229 |  | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 230 | static int | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 231 | BUILD_FUNC_DEF_2(setipaddr, char*,name, struct sockaddr_in *,addr_ret) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 232 | { | 
 | 233 | 	struct hostent *hp; | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 234 | 	int d1, d2, d3, d4; | 
 | 235 | 	char ch; | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 236 | #ifdef HAVE_GETHOSTBYNAME_R | 
 | 237 | 	struct hostent hp_allocated; | 
 | 238 | 	char buf[1001]; | 
 | 239 | 	int buf_len = (sizeof buf) - 1; | 
 | 240 | 	int errnop; | 
 | 241 | #endif /* HAVE_GETHOSTBYNAME_R */ | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 242 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 243 | 	if (name[0] == '\0') { | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 244 | 		addr_ret->sin_addr.s_addr = INADDR_ANY; | 
 | 245 | 		return 4; | 
 | 246 | 	} | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 247 | 	if (name[0] == '<' && strcmp(name, "<broadcast>") == 0) { | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 248 | 		addr_ret->sin_addr.s_addr = INADDR_BROADCAST; | 
 | 249 | 		return 4; | 
 | 250 | 	} | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 251 | 	if (sscanf(name, "%d.%d.%d.%d%c", &d1, &d2, &d3, &d4, &ch) == 4 && | 
 | 252 | 	    0 <= d1 && d1 <= 255 && 0 <= d2 && d2 <= 255 && | 
 | 253 | 	    0 <= d3 && d3 <= 255 && 0 <= d4 && d4 <= 255) { | 
 | 254 | 		addr_ret->sin_addr.s_addr = htonl( | 
 | 255 | 			((long) d1 << 24) | ((long) d2 << 16) | | 
 | 256 | 			((long) d3 << 8) | ((long) d4 << 0)); | 
 | 257 | 		return 4; | 
 | 258 | 	} | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 259 | #ifdef HAVE_GETHOSTBYNAME_R | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 260 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 261 | 	hp = gethostbyname_r(name, &hp_allocated, buf, buf_len, &errnop); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 262 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 263 | #else /* not HAVE_GETHOSTBYNAME_R */ | 
 | 264 | 	hp = gethostbyname(name); | 
 | 265 | #endif /* HAVE_GETHOSTBYNAME_R */ | 
 | 266 |  | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 267 | 	if (hp == NULL) { | 
| Guido van Rossum | ca9b323 | 1995-01-10 16:30:54 +0000 | [diff] [blame] | 268 | #ifdef HAVE_HSTRERROR | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 269 | 	        /* Let's get real error message to return */ | 
 | 270 | 	        extern int h_errno; | 
 | 271 | 		PyErr_SetString(PySocket_Error, (char *)hstrerror(h_errno)); | 
 | 272 | #else | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 273 | 		PyErr_SetString(PySocket_Error, "host not found"); | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 274 | #endif | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 275 | 		return -1; | 
 | 276 | 	} | 
 | 277 | 	memcpy((char *) &addr_ret->sin_addr, hp->h_addr, hp->h_length); | 
 | 278 | 	return hp->h_length; | 
 | 279 | } | 
 | 280 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 281 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 282 | /* Create a string object representing an IP address. | 
 | 283 |    This is always a string of the form 'dd.dd.dd.dd' (with variable | 
 | 284 |    size numbers). */ | 
 | 285 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 286 | static PyObject * | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 287 | BUILD_FUNC_DEF_1(makeipaddr, struct sockaddr_in *,addr) | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 288 | { | 
 | 289 | 	long x = ntohl(addr->sin_addr.s_addr); | 
 | 290 | 	char buf[100]; | 
 | 291 | 	sprintf(buf, "%d.%d.%d.%d", | 
 | 292 | 		(int) (x>>24) & 0xff, (int) (x>>16) & 0xff, | 
 | 293 | 		(int) (x>> 8) & 0xff, (int) (x>> 0) & 0xff); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 294 | 	return PyString_FromString(buf); | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 295 | } | 
 | 296 |  | 
 | 297 |  | 
 | 298 | /* Create an object representing the given socket address, | 
 | 299 |    suitable for passing it back to bind(), connect() etc. | 
 | 300 |    The family field of the sockaddr structure is inspected | 
 | 301 |    to determine what kind of address it really is. */ | 
 | 302 |  | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 303 | /*ARGSUSED*/ | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 304 | static PyObject * | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 305 | BUILD_FUNC_DEF_2(makesockaddr,struct sockaddr *,addr, int,addrlen) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 306 | { | 
| Guido van Rossum | 25bec8c | 1992-08-05 19:00:45 +0000 | [diff] [blame] | 307 | 	if (addrlen == 0) { | 
 | 308 | 		/* No address -- may be recvfrom() from known socket */ | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 309 | 		Py_INCREF(Py_None); | 
 | 310 | 		return Py_None; | 
| Guido van Rossum | 25bec8c | 1992-08-05 19:00:45 +0000 | [diff] [blame] | 311 | 	} | 
 | 312 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 313 | 	switch (addr->sa_family) { | 
 | 314 |  | 
 | 315 | 	case AF_INET: | 
 | 316 | 	{ | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 317 | 		struct sockaddr_in *a = (struct sockaddr_in *) addr; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 318 | 		PyObject *addr = makeipaddr(a); | 
 | 319 | 		PyObject *ret = Py_BuildValue("Oi", addr, ntohs(a->sin_port)); | 
 | 320 | 		Py_XDECREF(addr); | 
| Guido van Rossum | 6f5afc9 | 1993-02-05 09:46:15 +0000 | [diff] [blame] | 321 | 		return ret; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 322 | 	} | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 323 |  | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 324 | #ifdef AF_UNIX | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 325 | 	case AF_UNIX: | 
 | 326 | 	{ | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 327 | 		struct sockaddr_un *a = (struct sockaddr_un *) addr; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 328 | 		return PyString_FromString(a->sun_path); | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 329 | 	} | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 330 | #endif /* AF_UNIX */ | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 331 |  | 
 | 332 | 	/* More cases here... */ | 
 | 333 |  | 
 | 334 | 	default: | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 335 | 		PyErr_SetString(PySocket_Error, "return unknown socket address type"); | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 336 | 		return NULL; | 
| Guido van Rossum | 25bec8c | 1992-08-05 19:00:45 +0000 | [diff] [blame] | 337 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 338 | 	} | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 339 | } | 
 | 340 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 341 |  | 
 | 342 | /* Parse a socket address argument according to the socket object's | 
 | 343 |    address family.  Return 1 if the address was in the proper format, | 
 | 344 |    0 of not.  The address is returned through addr_ret, its length | 
 | 345 |    through len_ret. */ | 
 | 346 |  | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 347 | static int | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 348 | BUILD_FUNC_DEF_4( | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 349 | getsockaddrarg,PySocketSockObject *,s, PyObject *,args, struct sockaddr **,addr_ret, int *,len_ret) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 350 | { | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 351 | 	switch (s->sock_family) { | 
 | 352 |  | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 353 | #ifdef AF_UNIX | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 354 | 	case AF_UNIX: | 
 | 355 | 	{ | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 356 | 		struct sockaddr_un* addr; | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 357 | 		char *path; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 358 | 		int len; | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 359 | 		addr = (struct sockaddr_un* )&(s->sock_addr).un; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 360 | 		if (!PyArg_Parse(args, "s#", &path, &len)) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 361 | 			return 0; | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 362 | 		if (len > sizeof addr->sun_path) { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 363 | 			PyErr_SetString(PySocket_Error, "AF_UNIX path too long"); | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 364 | 			return 0; | 
 | 365 | 		} | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 366 | 		addr->sun_family = AF_UNIX; | 
 | 367 | 		memcpy(addr->sun_path, path, len); | 
 | 368 | 		*addr_ret = (struct sockaddr *) addr; | 
 | 369 | 		*len_ret = len + sizeof addr->sun_family; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 370 | 		return 1; | 
 | 371 | 	} | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 372 | #endif /* AF_UNIX */ | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 373 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 374 | 	case AF_INET: | 
 | 375 | 	{ | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 376 | 		struct sockaddr_in* addr; | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 377 | 		char *host; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 378 | 		int port; | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 379 |  		addr=(struct sockaddr_in*)&(s->sock_addr).in; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 380 | 		if (!PyArg_Parse(args, "(si)", &host, &port)) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 381 | 			return 0; | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 382 | 		if (setipaddr(host, addr) < 0) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 383 | 			return 0; | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 384 | 		addr->sin_family = AF_INET; | 
 | 385 | 		addr->sin_port = htons(port); | 
 | 386 | 		*addr_ret = (struct sockaddr *) addr; | 
 | 387 | 		*len_ret = sizeof *addr; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 388 | 		return 1; | 
 | 389 | 	} | 
 | 390 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 391 | 	/* More cases here... */ | 
 | 392 |  | 
 | 393 | 	default: | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 394 | 		PyErr_SetString(PySocket_Error, "getsockaddrarg: bad family"); | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 395 | 		return 0; | 
 | 396 |  | 
 | 397 | 	} | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 398 | } | 
 | 399 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 400 |  | 
| Guido van Rossum | 710e1df | 1992-06-12 10:39:36 +0000 | [diff] [blame] | 401 | /* Get the address length according to the socket object's address family.  | 
 | 402 |    Return 1 if the family is known, 0 otherwise.  The length is returned | 
 | 403 |    through len_ret. */ | 
 | 404 |  | 
 | 405 | static int | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 406 | BUILD_FUNC_DEF_2(getsockaddrlen,PySocketSockObject *,s, int *,len_ret) | 
| Guido van Rossum | 710e1df | 1992-06-12 10:39:36 +0000 | [diff] [blame] | 407 | { | 
 | 408 | 	switch (s->sock_family) { | 
 | 409 |  | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 410 | #ifdef AF_UNIX | 
| Guido van Rossum | 710e1df | 1992-06-12 10:39:36 +0000 | [diff] [blame] | 411 | 	case AF_UNIX: | 
 | 412 | 	{ | 
 | 413 | 		*len_ret = sizeof (struct sockaddr_un); | 
 | 414 | 		return 1; | 
 | 415 | 	} | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 416 | #endif /* AF_UNIX */ | 
| Guido van Rossum | 710e1df | 1992-06-12 10:39:36 +0000 | [diff] [blame] | 417 |  | 
 | 418 | 	case AF_INET: | 
 | 419 | 	{ | 
 | 420 | 		*len_ret = sizeof (struct sockaddr_in); | 
 | 421 | 		return 1; | 
 | 422 | 	} | 
 | 423 |  | 
 | 424 | 	/* More cases here... */ | 
 | 425 |  | 
 | 426 | 	default: | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 427 | 		PyErr_SetString(PySocket_Error, "getsockaddrarg: bad family"); | 
| Guido van Rossum | 710e1df | 1992-06-12 10:39:36 +0000 | [diff] [blame] | 428 | 		return 0; | 
 | 429 |  | 
 | 430 | 	} | 
 | 431 | } | 
 | 432 |  | 
 | 433 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 434 | /* s.accept() method */ | 
 | 435 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 436 | static PyObject * | 
 | 437 | BUILD_FUNC_DEF_2(PySocketSock_accept,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 438 | { | 
 | 439 | 	char addrbuf[256]; | 
 | 440 | 	int addrlen, newfd; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 441 | 	PyObject *sock, *addr, *res; | 
 | 442 | 	if (!PyArg_NoArgs(args)) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 443 | 		return NULL; | 
| Guido van Rossum | 710e1df | 1992-06-12 10:39:36 +0000 | [diff] [blame] | 444 | 	if (!getsockaddrlen(s, &addrlen)) | 
 | 445 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 446 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 447 | 	newfd = accept(s->sock_fd, (struct sockaddr *) addrbuf, &addrlen); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 448 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 449 | 	if (newfd < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 450 | 		return PySocket_Err(); | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 451 | 	/* Create the new object with unspecified family, | 
 | 452 | 	   to avoid calls to bind() etc. on it. */ | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 453 | 	sock = (PyObject *) PySocketSock_New(newfd, | 
| Guido van Rossum | 6f5afc9 | 1993-02-05 09:46:15 +0000 | [diff] [blame] | 454 | 					s->sock_family, | 
 | 455 | 					s->sock_type, | 
 | 456 | 					s->sock_proto); | 
 | 457 | 	if (sock == NULL) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 458 | 		close(newfd); | 
| Guido van Rossum | 6f5afc9 | 1993-02-05 09:46:15 +0000 | [diff] [blame] | 459 | 	addr = makesockaddr((struct sockaddr *) addrbuf, addrlen); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 460 | 	res = Py_BuildValue("OO", sock, addr); | 
 | 461 | 	Py_XDECREF(sock); | 
 | 462 | 	Py_XDECREF(addr); | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 463 | 	return res; | 
 | 464 | } | 
 | 465 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 466 |  | 
| Guido van Rossum | c65a525 | 1994-08-05 13:44:50 +0000 | [diff] [blame] | 467 | #if 0 | 
| Guido van Rossum | e0e5edc | 1991-10-20 20:22:28 +0000 | [diff] [blame] | 468 | /* s.allowbroadcast() method */ | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 469 | /* XXX obsolete -- will disappear in next release */ | 
| Guido van Rossum | e0e5edc | 1991-10-20 20:22:28 +0000 | [diff] [blame] | 470 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 471 | static PyObject * | 
 | 472 | BUILD_FUNC_DEF_2(PySocketSock_allowbroadcast,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | e0e5edc | 1991-10-20 20:22:28 +0000 | [diff] [blame] | 473 | { | 
 | 474 | 	int flag; | 
 | 475 | 	int res; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 476 | 	if (!PyArg_Parse(args, "i", &flag)) | 
| Guido van Rossum | e0e5edc | 1991-10-20 20:22:28 +0000 | [diff] [blame] | 477 | 		return NULL; | 
 | 478 | 	res = setsockopt(s->sock_fd, SOL_SOCKET, SO_BROADCAST, | 
| Guido van Rossum | b376a4a | 1993-11-23 17:53:17 +0000 | [diff] [blame] | 479 | 			 (ANY *)&flag, sizeof flag); | 
| Guido van Rossum | e0e5edc | 1991-10-20 20:22:28 +0000 | [diff] [blame] | 480 | 	if (res < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 481 | 		return PySocket_Err(); | 
 | 482 | 	Py_INCREF(Py_None); | 
 | 483 | 	return Py_None; | 
| Guido van Rossum | e0e5edc | 1991-10-20 20:22:28 +0000 | [diff] [blame] | 484 | } | 
| Guido van Rossum | c65a525 | 1994-08-05 13:44:50 +0000 | [diff] [blame] | 485 | #endif | 
| Guido van Rossum | e0e5edc | 1991-10-20 20:22:28 +0000 | [diff] [blame] | 486 |  | 
 | 487 |  | 
| Guido van Rossum | e4485b0 | 1994-09-07 14:32:49 +0000 | [diff] [blame] | 488 | #ifndef NT | 
 | 489 |  | 
 | 490 | /* s.setblocking(1 | 0) method */ | 
 | 491 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 492 | static PyObject * | 
 | 493 | PySocketSock_setblocking(s, args) | 
 | 494 | 	PySocketSockObject *s; | 
 | 495 | 	PyObject *args; | 
| Guido van Rossum | e4485b0 | 1994-09-07 14:32:49 +0000 | [diff] [blame] | 496 | { | 
 | 497 | 	int block; | 
 | 498 | 	int delay_flag; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 499 | 	if (!PyArg_GetInt(args, &block)) | 
| Guido van Rossum | e4485b0 | 1994-09-07 14:32:49 +0000 | [diff] [blame] | 500 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 501 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | e4485b0 | 1994-09-07 14:32:49 +0000 | [diff] [blame] | 502 | 	delay_flag = fcntl (s->sock_fd, F_GETFL, 0); | 
 | 503 | 	if (block) | 
 | 504 | 		delay_flag &= (~O_NDELAY); | 
 | 505 | 	else | 
 | 506 | 		delay_flag |= O_NDELAY; | 
 | 507 | 	fcntl (s->sock_fd, F_SETFL, delay_flag); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 508 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | e4485b0 | 1994-09-07 14:32:49 +0000 | [diff] [blame] | 509 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 510 | 	Py_INCREF(Py_None); | 
 | 511 | 	return Py_None; | 
| Guido van Rossum | e4485b0 | 1994-09-07 14:32:49 +0000 | [diff] [blame] | 512 | } | 
 | 513 | #endif | 
 | 514 |  | 
 | 515 |  | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 516 | /* s.setsockopt() method. | 
 | 517 |    With an integer third argument, sets an integer option. | 
 | 518 |    With a string third argument, sets an option from a buffer; | 
 | 519 |    use optional built-in module 'struct' to encode the string. */ | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 520 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 521 | static PyObject * | 
 | 522 | BUILD_FUNC_DEF_2(PySocketSock_setsockopt,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 523 | { | 
 | 524 | 	int level; | 
 | 525 | 	int optname; | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 526 | 	int res; | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 527 | 	char *buf; | 
 | 528 | 	int buflen; | 
 | 529 | 	int flag; | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 530 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 531 | 	if (PyArg_Parse(args, "(iii)", &level, &optname, &flag)) { | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 532 | 		buf = (char *) &flag; | 
 | 533 | 		buflen = sizeof flag; | 
 | 534 | 	} | 
 | 535 | 	else { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 536 | 		PyErr_Clear(); | 
 | 537 | 		if (!PyArg_Parse(args, "(iis#)", &level, &optname, &buf, &buflen)) | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 538 | 			return NULL; | 
 | 539 | 	} | 
| Guido van Rossum | b376a4a | 1993-11-23 17:53:17 +0000 | [diff] [blame] | 540 | 	res = setsockopt(s->sock_fd, level, optname, (ANY *)buf, buflen); | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 541 | 	if (res < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 542 | 		return PySocket_Err(); | 
 | 543 | 	Py_INCREF(Py_None); | 
 | 544 | 	return Py_None; | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 545 | } | 
 | 546 |  | 
 | 547 |  | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 548 | /* s.getsockopt() method. | 
 | 549 |    With two arguments, retrieves an integer option. | 
 | 550 |    With a third integer argument, retrieves a string buffer of that size; | 
 | 551 |    use optional built-in module 'struct' to decode the string. */ | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 552 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 553 | static PyObject * | 
 | 554 | BUILD_FUNC_DEF_2(PySocketSock_getsockopt,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 555 | { | 
 | 556 | 	int level; | 
 | 557 | 	int optname; | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 558 | 	int res; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 559 | 	PyObject *buf; | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 560 | 	int buflen; | 
 | 561 | 	int flag; | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 562 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 563 | 	if (PyArg_Parse(args, "(ii)", &level, &optname)) { | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 564 | 		int flag = 0; | 
 | 565 | 		int flagsize = sizeof flag; | 
| Guido van Rossum | b376a4a | 1993-11-23 17:53:17 +0000 | [diff] [blame] | 566 | 		res = getsockopt(s->sock_fd, level, optname, | 
 | 567 | 				 (ANY *)&flag, &flagsize); | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 568 | 		if (res < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 569 | 			return PySocket_Err(); | 
 | 570 | 		return PyInt_FromLong(flag); | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 571 | 	} | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 572 | 	PyErr_Clear(); | 
 | 573 | 	if (!PyArg_Parse(args, "(iii)", &level, &optname, &buflen)) | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 574 | 		return NULL; | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 575 | 	if (buflen <= 0 || buflen > 1024) { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 576 | 		PyErr_SetString(PySocket_Error, "getsockopt buflen out of range"); | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 577 | 		return NULL; | 
 | 578 | 	} | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 579 | 	buf = PyString_FromStringAndSize((char *)NULL, buflen); | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 580 | 	if (buf == NULL) | 
 | 581 | 		return NULL; | 
| Guido van Rossum | b376a4a | 1993-11-23 17:53:17 +0000 | [diff] [blame] | 582 | 	res = getsockopt(s->sock_fd, level, optname, | 
| Guido van Rossum | a597dde | 1995-01-10 20:56:29 +0000 | [diff] [blame] | 583 | 			 (ANY *)PyString_AsString(buf), &buflen); | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 584 | 	if (res < 0) { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 585 | 		Py_DECREF(buf); | 
 | 586 | 		return PySocket_Err(); | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 587 | 	} | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 588 | 	_PyString_Resize(&buf, buflen); | 
| Guido van Rossum | aee0879 | 1992-09-08 09:05:33 +0000 | [diff] [blame] | 589 | 	return buf; | 
| Guido van Rossum | 0e69587d | 1992-06-05 15:11:30 +0000 | [diff] [blame] | 590 | } | 
 | 591 |  | 
 | 592 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 593 | /* s.bind(sockaddr) method */ | 
 | 594 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 595 | static PyObject * | 
 | 596 | BUILD_FUNC_DEF_2(PySocketSock_bind,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 597 | { | 
 | 598 | 	struct sockaddr *addr; | 
 | 599 | 	int addrlen; | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 600 | 	int res; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 601 | 	if (!getsockaddrarg(s, args, &addr, &addrlen)) | 
 | 602 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 603 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 604 | 	res = bind(s->sock_fd, addr, addrlen); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 605 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 606 | 	if (res < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 607 | 		return PySocket_Err(); | 
 | 608 | 	Py_INCREF(Py_None); | 
 | 609 | 	return Py_None; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 610 | } | 
 | 611 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 612 |  | 
 | 613 | /* s.close() method. | 
 | 614 |    Set the file descriptor to -1 so operations tried subsequently | 
 | 615 |    will surely fail. */ | 
 | 616 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 617 | static PyObject * | 
 | 618 | BUILD_FUNC_DEF_2(PySocketSock_close,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 619 | { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 620 | 	if (!PyArg_NoArgs(args)) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 621 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 622 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 623 | 	(void) close(s->sock_fd); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 624 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 625 | 	s->sock_fd = -1; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 626 | 	Py_INCREF(Py_None); | 
 | 627 | 	return Py_None; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 628 | } | 
 | 629 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 630 |  | 
 | 631 | /* s.connect(sockaddr) method */ | 
 | 632 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 633 | static PyObject * | 
 | 634 | BUILD_FUNC_DEF_2(PySocketSock_connect,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 635 | { | 
 | 636 | 	struct sockaddr *addr; | 
 | 637 | 	int addrlen; | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 638 | 	int res; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 639 | 	if (!getsockaddrarg(s, args, &addr, &addrlen)) | 
 | 640 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 641 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 642 | 	res = connect(s->sock_fd, addr, addrlen); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 643 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 644 | 	if (res < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 645 | 		return PySocket_Err(); | 
 | 646 | 	Py_INCREF(Py_None); | 
 | 647 | 	return Py_None; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 648 | } | 
 | 649 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 650 |  | 
| Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 651 | /* s.fileno() method */ | 
 | 652 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 653 | static PyObject * | 
 | 654 | BUILD_FUNC_DEF_2(PySocketSock_fileno,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 655 | { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 656 | 	if (!PyArg_NoArgs(args)) | 
| Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 657 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 658 | 	return PyInt_FromLong((long) s->sock_fd); | 
| Guido van Rossum | ed233a5 | 1992-06-23 09:07:03 +0000 | [diff] [blame] | 659 | } | 
 | 660 |  | 
 | 661 |  | 
| Guido van Rossum | c89705d | 1992-11-26 08:54:07 +0000 | [diff] [blame] | 662 | /* s.getsockname() method */ | 
 | 663 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 664 | static PyObject * | 
 | 665 | BUILD_FUNC_DEF_2(PySocketSock_getsockname,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | c89705d | 1992-11-26 08:54:07 +0000 | [diff] [blame] | 666 | { | 
 | 667 | 	char addrbuf[256]; | 
 | 668 | 	int addrlen, res; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 669 | 	if (!PyArg_NoArgs(args)) | 
| Guido van Rossum | c89705d | 1992-11-26 08:54:07 +0000 | [diff] [blame] | 670 | 		return NULL; | 
 | 671 | 	if (!getsockaddrlen(s, &addrlen)) | 
 | 672 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 673 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | c89705d | 1992-11-26 08:54:07 +0000 | [diff] [blame] | 674 | 	res = getsockname(s->sock_fd, (struct sockaddr *) addrbuf, &addrlen); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 675 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | c89705d | 1992-11-26 08:54:07 +0000 | [diff] [blame] | 676 | 	if (res < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 677 | 		return PySocket_Err(); | 
| Guido van Rossum | c89705d | 1992-11-26 08:54:07 +0000 | [diff] [blame] | 678 | 	return makesockaddr((struct sockaddr *) addrbuf, addrlen); | 
 | 679 | } | 
 | 680 |  | 
 | 681 |  | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 682 | #ifdef HAVE_GETPEERNAME		/* Cray APP doesn't have this :-( */ | 
| Guido van Rossum | c89705d | 1992-11-26 08:54:07 +0000 | [diff] [blame] | 683 | /* s.getpeername() method */ | 
 | 684 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 685 | static PyObject * | 
 | 686 | BUILD_FUNC_DEF_2(PySocketSock_getpeername,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | c89705d | 1992-11-26 08:54:07 +0000 | [diff] [blame] | 687 | { | 
 | 688 | 	char addrbuf[256]; | 
 | 689 | 	int addrlen, res; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 690 | 	if (!PyArg_NoArgs(args)) | 
| Guido van Rossum | c89705d | 1992-11-26 08:54:07 +0000 | [diff] [blame] | 691 | 		return NULL; | 
 | 692 | 	if (!getsockaddrlen(s, &addrlen)) | 
 | 693 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 694 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | c89705d | 1992-11-26 08:54:07 +0000 | [diff] [blame] | 695 | 	res = getpeername(s->sock_fd, (struct sockaddr *) addrbuf, &addrlen); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 696 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | c89705d | 1992-11-26 08:54:07 +0000 | [diff] [blame] | 697 | 	if (res < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 698 | 		return PySocket_Err(); | 
| Guido van Rossum | c89705d | 1992-11-26 08:54:07 +0000 | [diff] [blame] | 699 | 	return makesockaddr((struct sockaddr *) addrbuf, addrlen); | 
 | 700 | } | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 701 | #endif /* HAVE_GETPEERNAME */ | 
| Guido van Rossum | c89705d | 1992-11-26 08:54:07 +0000 | [diff] [blame] | 702 |  | 
 | 703 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 704 | /* s.listen(n) method */ | 
 | 705 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 706 | static PyObject * | 
 | 707 | BUILD_FUNC_DEF_2(PySocketSock_listen,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 708 | { | 
 | 709 | 	int backlog; | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 710 | 	int res; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 711 | 	if (!PyArg_GetInt(args, &backlog)) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 712 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 713 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 714 | 	if (backlog < 1) | 
 | 715 | 		backlog = 1; | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 716 | 	res = listen(s->sock_fd, backlog); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 717 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 718 | 	if (res < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 719 | 		return PySocket_Err(); | 
 | 720 | 	Py_INCREF(Py_None); | 
 | 721 | 	return Py_None; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 722 | } | 
 | 723 |  | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 724 | #ifndef NO_DUP | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 725 | /* s.makefile(mode) method. | 
 | 726 |    Create a new open file object referring to a dupped version of | 
 | 727 |    the socket's file descriptor.  (The dup() call is necessary so | 
 | 728 |    that the open file and socket objects may be closed independent | 
 | 729 |    of each other.) | 
 | 730 |    The mode argument specifies 'r' or 'w' passed to fdopen(). */ | 
 | 731 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 732 | static PyObject * | 
 | 733 | BUILD_FUNC_DEF_2(PySocketSock_makefile,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 734 | { | 
| Guido van Rossum | a597dde | 1995-01-10 20:56:29 +0000 | [diff] [blame] | 735 | 	extern int fclose Py_PROTO((FILE *)); | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 736 | 	char *mode; | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 737 | 	int fd; | 
 | 738 | 	FILE *fp; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 739 | 	if (!PyArg_Parse(args, "s", &mode)) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 740 | 		return NULL; | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 741 | 	if ((fd = dup(s->sock_fd)) < 0 || | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 742 | 	    (fp = fdopen(fd, mode)) == NULL) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 743 | 		return PySocket_Err(); | 
 | 744 | 	return PyFile_FromFile(fp, "<socket>", mode, fclose); | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 745 | } | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 746 | #endif /* NO_DUP */ | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 747 |  | 
| Guido van Rossum | eb6b33a | 1993-05-25 09:38:27 +0000 | [diff] [blame] | 748 | /* s.recv(nbytes [,flags]) method */ | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 749 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 750 | static PyObject * | 
 | 751 | BUILD_FUNC_DEF_2(PySocketSock_recv,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 752 | { | 
 | 753 | 	int len, n, flags; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 754 | 	PyObject *buf; | 
| Guido van Rossum | eb6b33a | 1993-05-25 09:38:27 +0000 | [diff] [blame] | 755 | 	flags = 0; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 756 | 	if (!PyArg_Parse(args, "i", &len)) { | 
 | 757 | 		PyErr_Clear(); | 
 | 758 | 		if (!PyArg_Parse(args, "(ii)", &len, &flags)) | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 759 | 			return NULL; | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 760 | 	} | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 761 | 	buf = PyString_FromStringAndSize((char *) 0, len); | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 762 | 	if (buf == NULL) | 
 | 763 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 764 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | a597dde | 1995-01-10 20:56:29 +0000 | [diff] [blame] | 765 | 	n = recv(s->sock_fd, PyString_AsString(buf), len, flags); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 766 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 767 | 	if (n < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 768 | 		return PySocket_Err(); | 
 | 769 | 	if (_PyString_Resize(&buf, n) < 0) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 770 | 		return NULL; | 
 | 771 | 	return buf; | 
 | 772 | } | 
 | 773 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 774 |  | 
| Guido van Rossum | eb6b33a | 1993-05-25 09:38:27 +0000 | [diff] [blame] | 775 | /* s.recvfrom(nbytes [,flags]) method */ | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 776 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 777 | static PyObject * | 
 | 778 | BUILD_FUNC_DEF_2(PySocketSock_recvfrom,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 779 | { | 
 | 780 | 	char addrbuf[256]; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 781 | 	PyObject *buf, *addr, *ret; | 
| Guido van Rossum | eb6b33a | 1993-05-25 09:38:27 +0000 | [diff] [blame] | 782 | 	int addrlen, len, n, flags; | 
 | 783 | 	flags = 0; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 784 | 	if (!PyArg_Parse(args, "i", &len)) { | 
 | 785 | 		PyErr_Clear(); | 
 | 786 | 		if (!PyArg_Parse(args, "(ii)", &len, &flags)) | 
| Guido van Rossum | eb6b33a | 1993-05-25 09:38:27 +0000 | [diff] [blame] | 787 | 		    return NULL; | 
 | 788 | 	} | 
| Guido van Rossum | 18c9a4f | 1993-05-25 12:16:29 +0000 | [diff] [blame] | 789 | 	if (!getsockaddrlen(s, &addrlen)) | 
 | 790 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 791 | 	buf = PyString_FromStringAndSize((char *) 0, len); | 
| Guido van Rossum | 18c9a4f | 1993-05-25 12:16:29 +0000 | [diff] [blame] | 792 | 	if (buf == NULL) | 
 | 793 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 794 | 	Py_BEGIN_ALLOW_THREADS | 
 | 795 | 	n = recvfrom(s->sock_fd, PyString_AsString(buf), len, flags, | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 796 | #ifndef NT | 
| Guido van Rossum | b376a4a | 1993-11-23 17:53:17 +0000 | [diff] [blame] | 797 | 		     (ANY *)addrbuf, &addrlen); | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 798 | #else | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 799 | 		     (struct sockaddr *)addrbuf, &addrlen); | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 800 | #endif | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 801 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 802 | 	if (n < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 803 | 		return PySocket_Err(); | 
 | 804 | 	if (_PyString_Resize(&buf, n) < 0) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 805 | 		return NULL; | 
| Guido van Rossum | 6f5afc9 | 1993-02-05 09:46:15 +0000 | [diff] [blame] | 806 | 	addr = makesockaddr((struct sockaddr *)addrbuf, addrlen); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 807 | 	ret = Py_BuildValue("OO", buf, addr); | 
 | 808 | 	Py_XDECREF(addr); | 
 | 809 | 	Py_XDECREF(buf); | 
| Guido van Rossum | 6f5afc9 | 1993-02-05 09:46:15 +0000 | [diff] [blame] | 810 | 	return ret; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 811 | } | 
 | 812 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 813 |  | 
| Guido van Rossum | eb6b33a | 1993-05-25 09:38:27 +0000 | [diff] [blame] | 814 | /* s.send(data [,flags]) method */ | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 815 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 816 | static PyObject * | 
 | 817 | BUILD_FUNC_DEF_2(PySocketSock_send,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 818 | { | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 819 | 	char *buf; | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 820 | 	int len, n, flags; | 
| Guido van Rossum | eb6b33a | 1993-05-25 09:38:27 +0000 | [diff] [blame] | 821 | 	flags = 0; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 822 | 	if (!PyArg_Parse(args, "s#", &buf, &len)) { | 
 | 823 | 		PyErr_Clear(); | 
 | 824 | 		if (!PyArg_Parse(args, "(s#i)", &buf, &len, &flags)) | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 825 | 			return NULL; | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 826 | 	} | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 827 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 828 | 	n = send(s->sock_fd, buf, len, flags); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 829 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 830 | 	if (n < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 831 | 		return PySocket_Err(); | 
 | 832 | 	return PyInt_FromLong((long)n); | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 833 | } | 
 | 834 |  | 
 | 835 |  | 
| Guido van Rossum | eb6b33a | 1993-05-25 09:38:27 +0000 | [diff] [blame] | 836 | /* s.sendto(data, [flags,] sockaddr) method */ | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 837 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 838 | static PyObject * | 
 | 839 | BUILD_FUNC_DEF_2(PySocketSock_sendto,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 840 | { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 841 | 	PyObject *addro; | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 842 | 	char *buf; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 843 | 	struct sockaddr *addr; | 
| Guido van Rossum | eb6b33a | 1993-05-25 09:38:27 +0000 | [diff] [blame] | 844 | 	int addrlen, len, n, flags; | 
 | 845 | 	flags = 0; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 846 | 	if (!PyArg_Parse(args, "(s#O)", &buf, &len, &addro)) { | 
 | 847 | 		PyErr_Clear(); | 
 | 848 | 		if (!PyArg_Parse(args, "(s#iO)", &buf, &len, &flags, &addro)) | 
| Guido van Rossum | eb6b33a | 1993-05-25 09:38:27 +0000 | [diff] [blame] | 849 | 			return NULL; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 850 | 	} | 
| Guido van Rossum | eb6b33a | 1993-05-25 09:38:27 +0000 | [diff] [blame] | 851 | 	if (!getsockaddrarg(s, addro, &addr, &addrlen)) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 852 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 853 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | eb6b33a | 1993-05-25 09:38:27 +0000 | [diff] [blame] | 854 | 	n = sendto(s->sock_fd, buf, len, flags, addr, addrlen); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 855 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 856 | 	if (n < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 857 | 		return PySocket_Err(); | 
 | 858 | 	return PyInt_FromLong((long)n); | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 859 | } | 
 | 860 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 861 |  | 
 | 862 | /* s.shutdown(how) method */ | 
 | 863 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 864 | static PyObject * | 
 | 865 | BUILD_FUNC_DEF_2(PySocketSock_shutdown,PySocketSockObject *,s, PyObject *,args) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 866 | { | 
 | 867 | 	int how; | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 868 | 	int res; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 869 | 	if (!PyArg_GetInt(args, &how)) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 870 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 871 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 872 | 	res = shutdown(s->sock_fd, how); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 873 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 874 | 	if (res < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 875 | 		return PySocket_Err(); | 
 | 876 | 	Py_INCREF(Py_None); | 
 | 877 | 	return Py_None; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 878 | } | 
 | 879 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 880 |  | 
 | 881 | /* List of methods for socket objects */ | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 882 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 883 | static PyMethodDef PySocketSock_methods[] = { | 
 | 884 | 	{"accept",		(PyCFunction)PySocketSock_accept}, | 
| Guido van Rossum | c65a525 | 1994-08-05 13:44:50 +0000 | [diff] [blame] | 885 | #if 0 | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 886 | 	{"allowbroadcast",	(PyCFunction)PySocketSock_allowbroadcast}, | 
| Guido van Rossum | c65a525 | 1994-08-05 13:44:50 +0000 | [diff] [blame] | 887 | #endif | 
| Guido van Rossum | e4485b0 | 1994-09-07 14:32:49 +0000 | [diff] [blame] | 888 | #ifndef NT | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 889 | 	{"setblocking",		(PyCFunction)PySocketSock_setblocking}, | 
| Guido van Rossum | e4485b0 | 1994-09-07 14:32:49 +0000 | [diff] [blame] | 890 | #endif | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 891 | 	{"setsockopt",		(PyCFunction)PySocketSock_setsockopt}, | 
 | 892 | 	{"getsockopt",		(PyCFunction)PySocketSock_getsockopt}, | 
 | 893 | 	{"bind",		(PyCFunction)PySocketSock_bind}, | 
 | 894 | 	{"close",		(PyCFunction)PySocketSock_close}, | 
 | 895 | 	{"connect",		(PyCFunction)PySocketSock_connect}, | 
 | 896 | 	{"fileno",		(PyCFunction)PySocketSock_fileno}, | 
 | 897 | 	{"getsockname",		(PyCFunction)PySocketSock_getsockname}, | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 898 | #ifdef HAVE_GETPEERNAME | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 899 | 	{"getpeername",		(PyCFunction)PySocketSock_getpeername}, | 
| Guido van Rossum | 9575a44 | 1993-04-07 14:06:14 +0000 | [diff] [blame] | 900 | #endif | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 901 | 	{"listen",		(PyCFunction)PySocketSock_listen}, | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 902 | #ifndef NO_DUP | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 903 | 	{"makefile",		(PyCFunction)PySocketSock_makefile}, | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 904 | #endif | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 905 | 	{"recv",		(PyCFunction)PySocketSock_recv}, | 
 | 906 | 	{"recvfrom",		(PyCFunction)PySocketSock_recvfrom}, | 
 | 907 | 	{"send",		(PyCFunction)PySocketSock_send}, | 
 | 908 | 	{"sendto",		(PyCFunction)PySocketSock_sendto}, | 
 | 909 | 	{"shutdown",		(PyCFunction)PySocketSock_shutdown}, | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 910 | 	{NULL,			NULL}		/* sentinel */ | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 911 | }; | 
 | 912 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 913 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 914 | /* Deallocate a socket object in response to the last Py_DECREF(). | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 915 |    First close the file description. */ | 
 | 916 |  | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 917 | static void | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 918 | BUILD_FUNC_DEF_1(PySocketSock_dealloc,PySocketSockObject *,s) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 919 | { | 
 | 920 | 	(void) close(s->sock_fd); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 921 | 	PyMem_DEL(s); | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 922 | } | 
 | 923 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 924 |  | 
 | 925 | /* Return a socket object's named attribute. */ | 
 | 926 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 927 | static PyObject * | 
 | 928 | BUILD_FUNC_DEF_2(PySocketSock_getattr,PySocketSockObject *,s, char *,name) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 929 | { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 930 | 	return Py_FindMethod(PySocketSock_methods, (PyObject *) s, name); | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 931 | } | 
 | 932 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 933 |  | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 934 | static PyObject * | 
 | 935 | BUILD_FUNC_DEF_1(PySocketSock_repr,PySocketSockObject *,s) | 
 | 936 | { | 
 | 937 | 	PyObject *addro; | 
 | 938 | 	struct sockaddr *addr; | 
 | 939 | 	char buf[512]; | 
| Guido van Rossum | a597dde | 1995-01-10 20:56:29 +0000 | [diff] [blame] | 940 | 	PyObject *t, *comma, *v; | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 941 | 	int i, len; | 
 | 942 | 	sprintf(buf,  | 
 | 943 | 		"<socket object, fd=%d, family=%d, type=%d, protocol=%d>",  | 
 | 944 | 		s->sock_fd, s->sock_family, s->sock_type, s->sock_proto); | 
| Guido van Rossum | a597dde | 1995-01-10 20:56:29 +0000 | [diff] [blame] | 945 | 	t = PyString_FromString(buf); | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 946 | 	return t; | 
 | 947 | } | 
 | 948 |  | 
 | 949 |  | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 950 | /* Type object for socket objects. */ | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 951 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 952 | static PyTypeObject PySocketSock_Type = { | 
 | 953 | 	PyObject_HEAD_INIT(&PyType_Type) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 954 | 	0, | 
 | 955 | 	"socket", | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 956 | 	sizeof(PySocketSockObject), | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 957 | 	0, | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 958 | 	(destructor)PySocketSock_dealloc, /*tp_dealloc*/ | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 959 | 	0,		/*tp_print*/ | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 960 | 	(getattrfunc)PySocketSock_getattr, /*tp_getattr*/ | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 961 | 	0,		/*tp_setattr*/ | 
 | 962 | 	0,		/*tp_compare*/ | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 963 | 	(reprfunc)PySocketSock_repr, /*tp_repr*/ | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 964 | 	0,		/*tp_as_number*/ | 
 | 965 | 	0,		/*tp_as_sequence*/ | 
 | 966 | 	0,		/*tp_as_mapping*/ | 
 | 967 | }; | 
 | 968 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 969 |  | 
| Guido van Rossum | 8119447 | 1991-07-27 21:42:02 +0000 | [diff] [blame] | 970 | /* Python interface to gethostname(). */ | 
 | 971 |  | 
 | 972 | /*ARGSUSED*/ | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 973 | static PyObject * | 
 | 974 | BUILD_FUNC_DEF_2(PySocket_gethostname,PyObject *,self, PyObject *,args) | 
| Guido van Rossum | 8119447 | 1991-07-27 21:42:02 +0000 | [diff] [blame] | 975 | { | 
 | 976 | 	char buf[1024]; | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 977 | 	int res; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 978 | 	if (!PyArg_NoArgs(args)) | 
| Guido van Rossum | 8119447 | 1991-07-27 21:42:02 +0000 | [diff] [blame] | 979 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 980 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 981 | 	res = gethostname(buf, (int) sizeof buf - 1); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 982 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 983 | 	if (res < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 984 | 		return PySocket_Err(); | 
| Guido van Rossum | 8119447 | 1991-07-27 21:42:02 +0000 | [diff] [blame] | 985 | 	buf[sizeof buf - 1] = '\0'; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 986 | 	return PyString_FromString(buf); | 
| Guido van Rossum | 8119447 | 1991-07-27 21:42:02 +0000 | [diff] [blame] | 987 | } | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 988 |  | 
 | 989 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 990 | /* Python interface to gethostbyname(name). */ | 
 | 991 |  | 
 | 992 | /*ARGSUSED*/ | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 993 | static PyObject * | 
 | 994 | BUILD_FUNC_DEF_2(PySocket_gethostbyname,PyObject *,self, PyObject *,args) | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 995 | { | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 996 | 	char *name; | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 997 | 	struct sockaddr_in addrbuf; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 998 | 	if (!PyArg_Parse(args, "s", &name)) | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 999 | 		return NULL; | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 1000 | 	if (setipaddr(name, &addrbuf) < 0) | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1001 | 		return NULL; | 
 | 1002 | 	return makeipaddr(&addrbuf); | 
 | 1003 | } | 
 | 1004 |  | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 1005 | /* Python interface to gethostbyaddr(IP). */ | 
 | 1006 |  | 
 | 1007 | /*ARGSUSED*/ | 
 | 1008 | static PyObject * | 
 | 1009 | BUILD_FUNC_DEF_2(PySocket_gethostbyaddr,PyObject *,self, PyObject *, args) | 
 | 1010 | { | 
 | 1011 |         struct sockaddr_in addr; | 
 | 1012 | 	char *ip_num; | 
 | 1013 | 	struct hostent *h; | 
 | 1014 | 	int d1,d2,d3,d4; | 
 | 1015 | 	char ch, **pch; | 
 | 1016 | 	PyObject *rtn_tuple = (PyObject *)NULL; | 
 | 1017 | 	PyObject *name_list = (PyObject *)NULL; | 
 | 1018 | 	PyObject *addr_list = (PyObject *)NULL; | 
 | 1019 | 	PyObject *tmp; | 
 | 1020 |  | 
 | 1021 | 	if (!PyArg_Parse(args, "s", &ip_num)) | 
 | 1022 | 		return NULL; | 
 | 1023 | 	if (setipaddr(ip_num, &addr) < 0) | 
 | 1024 | 		return NULL; | 
 | 1025 | 	h = gethostbyaddr((char *)&addr.sin_addr, | 
 | 1026 | 			  sizeof(addr.sin_addr), | 
 | 1027 | 			  AF_INET); | 
 | 1028 | 	if (h == NULL) { | 
| Guido van Rossum | ca9b323 | 1995-01-10 16:30:54 +0000 | [diff] [blame] | 1029 | #ifdef HAVE_HSTRERROR | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 1030 | 	        /* Let's get real error message to return */ | 
 | 1031 | 	        extern int h_errno; | 
 | 1032 | 		PyErr_SetString(PySocket_Error, (char *)hstrerror(h_errno)); | 
 | 1033 | #else | 
 | 1034 | 		PyErr_SetString(PySocket_Error, "host not found"); | 
 | 1035 | #endif | 
 | 1036 | 		return NULL; | 
 | 1037 | 	} | 
 | 1038 | 	if ((name_list = PyList_New(0)) == NULL) | 
 | 1039 | 		goto err; | 
 | 1040 | 	if ((addr_list = PyList_New(0)) == NULL) | 
 | 1041 | 		goto err; | 
 | 1042 | 	for (pch = h->h_aliases; *pch != NULL; pch++) { | 
 | 1043 | 		tmp = PyString_FromString(*pch); | 
 | 1044 | 		if (tmp == NULL) | 
 | 1045 | 			goto err; | 
 | 1046 | 		PyList_Append(name_list, tmp); | 
 | 1047 | 		Py_DECREF(tmp); | 
 | 1048 | 	} | 
 | 1049 | 	for (pch = h->h_addr_list; *pch != NULL; pch++) { | 
 | 1050 | 		memcpy((char *) &addr.sin_addr, *pch, h->h_length); | 
 | 1051 | 		tmp = makeipaddr(&addr); | 
 | 1052 | 		if (tmp == NULL) | 
 | 1053 | 			goto err; | 
 | 1054 | 		PyList_Append(addr_list, tmp); | 
 | 1055 | 		Py_DECREF(tmp); | 
 | 1056 | 	} | 
 | 1057 | 	rtn_tuple = Py_BuildValue("sOO", h->h_name, name_list, addr_list); | 
 | 1058 |  err: | 
 | 1059 | 	Py_XDECREF(name_list); | 
 | 1060 | 	Py_XDECREF(addr_list); | 
 | 1061 | 	return rtn_tuple; | 
 | 1062 | } | 
 | 1063 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1064 |  | 
 | 1065 | /* Python interface to getservbyname(name). | 
 | 1066 |    This only returns the port number, since the other info is already | 
 | 1067 |    known or not useful (like the list of aliases). */ | 
 | 1068 |  | 
 | 1069 | /*ARGSUSED*/ | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1070 | static PyObject * | 
 | 1071 | BUILD_FUNC_DEF_2(PySocket_getservbyname,PyObject *,self, PyObject *,args) | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1072 | { | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 1073 | 	char *name, *proto; | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1074 | 	struct servent *sp; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1075 | 	if (!PyArg_Parse(args, "(ss)", &name, &proto)) | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1076 | 		return NULL; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1077 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | ff4949e | 1992-08-05 19:58:53 +0000 | [diff] [blame] | 1078 | 	sp = getservbyname(name, proto); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1079 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1080 | 	if (sp == NULL) { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1081 | 		PyErr_SetString(PySocket_Error, "service/proto not found"); | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1082 | 		return NULL; | 
 | 1083 | 	} | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1084 | 	return PyInt_FromLong((long) ntohs(sp->s_port)); | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1085 | } | 
 | 1086 |  | 
 | 1087 |  | 
 | 1088 | /* Python interface to socket(family, type, proto). | 
 | 1089 |    The third (protocol) argument is optional. | 
 | 1090 |    Return a new socket object. */ | 
 | 1091 |  | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1092 | /*ARGSUSED*/ | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1093 | static PyObject * | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 1094 | BUILD_FUNC_DEF_2(PySocket_socket,PyObject *,self, PyObject *,args) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1095 | { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1096 | 	PySocketSockObject *s; | 
| Guido van Rossum | 2a7178e | 1992-12-08 13:38:24 +0000 | [diff] [blame] | 1097 | 	int fd, family, type, proto; | 
 | 1098 | 	proto = 0; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1099 | 	if (!PyArg_Parse(args, "(ii)", &family, &type)) { | 
 | 1100 | 		PyErr_Clear(); | 
 | 1101 | 		if (!PyArg_Parse(args, "(iii)", &family, &type, &proto)) | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1102 | 			return NULL; | 
 | 1103 | 	} | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1104 | 	Py_BEGIN_ALLOW_THREADS | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1105 | 	fd = socket(family, type, proto); | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1106 | 	Py_END_ALLOW_THREADS | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1107 | 	if (fd < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1108 | 		return PySocket_Err(); | 
 | 1109 | 	s = PySocketSock_New(fd, family, type, proto); | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1110 | 	/* If the object can't be created, don't forget to close the | 
 | 1111 | 	   file descriptor again! */ | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1112 | 	if (s == NULL) | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1113 | 		(void) close(fd); | 
| Guido van Rossum | 8119447 | 1991-07-27 21:42:02 +0000 | [diff] [blame] | 1114 | 	/* From now on, ignore SIGPIPE and let the error checking | 
 | 1115 | 	   do the work. */ | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 1116 | #ifdef SIGPIPE       | 
| Guido van Rossum | 8119447 | 1991-07-27 21:42:02 +0000 | [diff] [blame] | 1117 | 	(void) signal(SIGPIPE, SIG_IGN); | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 1118 | #endif    | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1119 | 	return (PyObject *) s; | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1120 | } | 
 | 1121 |  | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 1122 | #ifndef NO_DUP | 
| Guido van Rossum | 2a7178e | 1992-12-08 13:38:24 +0000 | [diff] [blame] | 1123 | /* Create a socket object from a numeric file description. | 
 | 1124 |    Useful e.g. if stdin is a socket. | 
 | 1125 |    Additional arguments as for socket(). */ | 
 | 1126 |  | 
 | 1127 | /*ARGSUSED*/ | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1128 | static PyObject * | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 1129 | BUILD_FUNC_DEF_2(PySocket_fromfd,PyObject *,self, PyObject *,args) | 
| Guido van Rossum | 2a7178e | 1992-12-08 13:38:24 +0000 | [diff] [blame] | 1130 | { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1131 | 	PySocketSockObject *s; | 
| Guido van Rossum | 2a7178e | 1992-12-08 13:38:24 +0000 | [diff] [blame] | 1132 | 	int fd, family, type, proto; | 
 | 1133 | 	proto = 0; | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1134 | 	if (!PyArg_Parse(args, "(iii)", &fd, &family, &type)) { | 
 | 1135 | 		PyErr_Clear(); | 
 | 1136 | 		if (!PyArg_Parse(args, "(iiii)", &fd, &family, &type, &proto)) | 
| Guido van Rossum | 2a7178e | 1992-12-08 13:38:24 +0000 | [diff] [blame] | 1137 | 			return NULL; | 
 | 1138 | 	} | 
| Guido van Rossum | 5f59d60 | 1992-12-14 16:59:51 +0000 | [diff] [blame] | 1139 | 	/* Dup the fd so it and the socket can be closed independently */ | 
 | 1140 | 	fd = dup(fd); | 
 | 1141 | 	if (fd < 0) | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1142 | 		return PySocket_Err(); | 
 | 1143 | 	s = PySocketSock_New(fd, family, type, proto); | 
| Guido van Rossum | 2a7178e | 1992-12-08 13:38:24 +0000 | [diff] [blame] | 1144 | 	/* From now on, ignore SIGPIPE and let the error checking | 
 | 1145 | 	   do the work. */ | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 1146 | #ifdef SIGPIPE       | 
| Guido van Rossum | 2a7178e | 1992-12-08 13:38:24 +0000 | [diff] [blame] | 1147 | 	(void) signal(SIGPIPE, SIG_IGN); | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 1148 | #endif    | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1149 | 	return (PyObject *) s; | 
| Guido van Rossum | 2a7178e | 1992-12-08 13:38:24 +0000 | [diff] [blame] | 1150 | } | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 1151 | #endif /* NO_DUP */ | 
| Guido van Rossum | 2a7178e | 1992-12-08 13:38:24 +0000 | [diff] [blame] | 1152 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1153 | /* List of functions exported by this module. */ | 
 | 1154 |  | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1155 | static PyMethodDef PySocket_methods[] = { | 
 | 1156 | 	{"gethostbyname",	PySocket_gethostbyname}, | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 1157 | 	{"gethostbyaddr",	PySocket_gethostbyaddr}, | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1158 | 	{"gethostname",		PySocket_gethostname}, | 
 | 1159 | 	{"getservbyname",	PySocket_getservbyname}, | 
 | 1160 | 	{"socket",		PySocket_socket}, | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 1161 | #ifndef NO_DUP | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1162 | 	{"fromfd",		PySocket_fromfd}, | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 1163 | #endif | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1164 | 	{NULL,			NULL}		 /* Sentinel */ | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1165 | }; | 
 | 1166 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1167 |  | 
 | 1168 | /* Convenience routine to export an integer value. | 
 | 1169 |    For simplicity, errors (which are unlikely anyway) are ignored. */ | 
 | 1170 |  | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1171 | static void | 
| Guido van Rossum | 3bbc62e | 1995-01-02 19:30:30 +0000 | [diff] [blame] | 1172 | BUILD_FUNC_DEF_3(insint,PyObject *,d, char *,name, int,value) | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1173 | { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1174 | 	PyObject *v = PyInt_FromLong((long) value); | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1175 | 	if (v == NULL) { | 
 | 1176 | 		/* Don't bother reporting this error */ | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1177 | 		PyErr_Clear(); | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1178 | 	} | 
 | 1179 | 	else { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1180 | 		PyDict_SetItemString(d, name, v); | 
 | 1181 | 		Py_DECREF(v); | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1182 | 	} | 
 | 1183 | } | 
 | 1184 |  | 
| Guido van Rossum | 30a685f | 1991-06-27 15:51:29 +0000 | [diff] [blame] | 1185 |  | 
 | 1186 | /* Initialize this module. | 
 | 1187 |    This is called when the first 'import socket' is done, | 
 | 1188 |    via a table in config.c, if config.c is compiled with USE_SOCKET | 
 | 1189 |    defined. */ | 
 | 1190 |  | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1191 | void | 
 | 1192 | initsocket() | 
 | 1193 | { | 
| Guido van Rossum | 73624e9 | 1994-10-10 17:59:00 +0000 | [diff] [blame] | 1194 | 	PyObject *m, *d; | 
 | 1195 | 	m = Py_InitModule("socket", PySocket_methods); | 
 | 1196 | 	d = PyModule_GetDict(m); | 
 | 1197 | 	PySocket_Error = PyString_FromString("socket.error"); | 
 | 1198 | 	if (PySocket_Error == NULL ||  | 
 | 1199 | 	    PyDict_SetItemString(d, "error", PySocket_Error) != 0) | 
 | 1200 | 		Py_FatalError("can't define socket.error"); | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1201 | 	insint(d, "AF_INET", AF_INET); | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 1202 | #ifdef AF_UNIX | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1203 | 	insint(d, "AF_UNIX", AF_UNIX); | 
| Guido van Rossum | b6775db | 1994-08-01 11:34:53 +0000 | [diff] [blame] | 1204 | #endif /* AF_UNIX */ | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1205 | 	insint(d, "SOCK_STREAM", SOCK_STREAM); | 
 | 1206 | 	insint(d, "SOCK_DGRAM", SOCK_DGRAM); | 
 | 1207 | 	insint(d, "SOCK_RAW", SOCK_RAW); | 
 | 1208 | 	insint(d, "SOCK_SEQPACKET", SOCK_SEQPACKET); | 
 | 1209 | 	insint(d, "SOCK_RDM", SOCK_RDM); | 
| Guido van Rossum | 3fdf58b | 1995-02-07 15:39:52 +0000 | [diff] [blame] | 1210 |  | 
 | 1211 | #ifdef	SO_DEBUG | 
 | 1212 | 	insint(d, "SO_DEBUG", SO_DEBUG); | 
 | 1213 | #endif | 
 | 1214 | #ifdef	SO_ACCEPTCONN | 
 | 1215 | 	insint(d, "SO_ACCEPTCONN", SO_ACCEPTCONN); | 
 | 1216 | #endif | 
 | 1217 | #ifdef	SO_REUSEADDR | 
 | 1218 | 	insint(d, "SO_REUSEADDR", SO_REUSEADDR); | 
 | 1219 | #endif | 
 | 1220 | #ifdef	SO_KEEPALIVE | 
 | 1221 | 	insint(d, "SO_KEEPALIVE", SO_KEEPALIVE); | 
 | 1222 | #endif | 
 | 1223 | #ifdef	SO_DONTROUTE | 
 | 1224 | 	insint(d, "SO_DONTROUTE", SO_DONTROUTE); | 
 | 1225 | #endif | 
 | 1226 | #ifdef	SO_BROADCAST | 
 | 1227 | 	insint(d, "SO_BROADCAST", SO_BROADCAST); | 
 | 1228 | #endif | 
 | 1229 | #ifdef	SO_USELOOPBACK | 
 | 1230 | 	insint(d, "SO_USELOOPBACK", SO_USELOOPBACK); | 
 | 1231 | #endif | 
 | 1232 | #ifdef	SO_LINGER | 
 | 1233 | 	insint(d, "SO_LINGER", SO_LINGER); | 
 | 1234 | #endif | 
 | 1235 | #ifdef	SO_OOBINLINE | 
 | 1236 | 	insint(d, "SO_OOBINLINE", SO_OOBINLINE); | 
 | 1237 | #endif | 
 | 1238 | #ifdef	SO_REUSEPORT | 
 | 1239 | 	insint(d, "SO_REUSEPORT", SO_REUSEPORT); | 
 | 1240 | #endif | 
 | 1241 |  | 
 | 1242 | #ifdef	SO_SNDBUF | 
 | 1243 | 	insint(d, "SO_SNDBUF", SO_SNDBUF); | 
 | 1244 | #endif | 
 | 1245 | #ifdef	SO_RCVBUF | 
 | 1246 | 	insint(d, "SO_RCVBUF", SO_RCVBUF); | 
 | 1247 | #endif | 
 | 1248 | #ifdef	SO_SNDLOWAT | 
 | 1249 | 	insint(d, "SO_SNDLOWAT", SO_SNDLOWAT); | 
 | 1250 | #endif | 
 | 1251 | #ifdef	SO_RCVLOWAT | 
 | 1252 | 	insint(d, "SO_RCVLOWAT", SO_RCVLOWAT); | 
 | 1253 | #endif | 
 | 1254 | #ifdef	SO_SNDTIMEO | 
 | 1255 | 	insint(d, "SO_SNDTIMEO", SO_SNDTIMEO); | 
 | 1256 | #endif | 
 | 1257 | #ifdef	SO_RCVTIMEO | 
 | 1258 | 	insint(d, "SO_RCVTIMEO", SO_RCVTIMEO); | 
 | 1259 | #endif | 
 | 1260 | #ifdef	SO_ERROR | 
 | 1261 | 	insint(d, "SO_ERROR", SO_ERROR); | 
 | 1262 | #endif | 
 | 1263 | #ifdef	SO_TYPE | 
 | 1264 | 	insint(d, "SO_TYPE", SO_TYPE); | 
 | 1265 | #endif | 
 | 1266 |  | 
 | 1267 | 	/* Maximum number of connections for "listen" */ | 
 | 1268 | #ifdef	SOMAXCONN | 
 | 1269 | 	insint(d, "SOMAXCONN", SOMAXCONN); | 
 | 1270 | #else | 
 | 1271 | 	insint(d, "SOMAXCONN", 5);	/* Common value */ | 
 | 1272 | #endif | 
 | 1273 |  | 
 | 1274 | 	/* Flags for send, recv */ | 
 | 1275 | #ifdef	MSG_OOB | 
 | 1276 | 	insint(d, "MSG_OOB", MSG_OOB); | 
 | 1277 | #endif | 
 | 1278 | #ifdef	MSG_PEEK | 
 | 1279 | 	insint(d, "MSG_PEEK", MSG_PEEK); | 
 | 1280 | #endif | 
 | 1281 | #ifdef	MSG_DONTROUTE | 
 | 1282 | 	insint(d, "MSG_DONTROUTE", MSG_DONTROUTE); | 
 | 1283 | #endif | 
 | 1284 | #ifdef	MSG_EOR | 
 | 1285 | 	insint(d, "MSG_EOR", MSG_EOR); | 
 | 1286 | #endif | 
 | 1287 | #ifdef	MSG_TRUNC | 
 | 1288 | 	insint(d, "MSG_TRUNC", MSG_TRUNC); | 
 | 1289 | #endif | 
 | 1290 | #ifdef	MSG_CTRUNC | 
 | 1291 | 	insint(d, "MSG_CTRUNC", MSG_CTRUNC); | 
 | 1292 | #endif | 
 | 1293 | #ifdef	MSG_WAITALL | 
 | 1294 | 	insint(d, "MSG_WAITALL", MSG_WAITALL); | 
 | 1295 | #endif | 
 | 1296 | #ifdef	MSG_BTAG | 
 | 1297 | 	insint(d, "MSG_BTAG", MSG_BTAG); | 
 | 1298 | #endif | 
 | 1299 | #ifdef	MSG_ETAG | 
 | 1300 | 	insint(d, "MSG_ETAG", MSG_ETAG); | 
 | 1301 | #endif | 
 | 1302 |  | 
 | 1303 | 	/* Protocol level and numbers, usable for [gs]etsockopt */ | 
 | 1304 | #ifdef	SOL_SOCKET | 
 | 1305 | 	insint(d, "SOL_SOCKET", SOL_SOCKET); | 
 | 1306 | #endif | 
 | 1307 | #ifdef	IPPROTO_IP | 
 | 1308 | 	insint(d, "IPPROTO_IP", IPPROTO_IP); | 
 | 1309 | #endif | 
 | 1310 | #ifdef	IPPROTO_ICMP | 
 | 1311 | 	insint(d, "IPPROTO_ICMP", IPPROTO_ICMP); | 
 | 1312 | #endif | 
 | 1313 | #ifdef	IPPROTO_IGMP | 
 | 1314 | 	insint(d, "IPPROTO_IGMP", IPPROTO_IGMP); | 
 | 1315 | #endif | 
 | 1316 | #ifdef	IPPROTO_GGP | 
 | 1317 | 	insint(d, "IPPROTO_GGP", IPPROTO_GGP); | 
 | 1318 | #endif | 
 | 1319 | #ifdef	IPPROTO_TCP | 
 | 1320 | 	insint(d, "IPPROTO_TCP", IPPROTO_TCP); | 
 | 1321 | #endif | 
 | 1322 | #ifdef	IPPROTO_EGP | 
 | 1323 | 	insint(d, "IPPROTO_EGP", IPPROTO_EGP); | 
 | 1324 | #endif | 
 | 1325 | #ifdef	IPPROTO_PUP | 
 | 1326 | 	insint(d, "IPPROTO_PUP", IPPROTO_PUP); | 
 | 1327 | #endif | 
 | 1328 | #ifdef	IPPROTO_UDP | 
 | 1329 | 	insint(d, "IPPROTO_UDP", IPPROTO_UDP); | 
 | 1330 | #endif | 
 | 1331 | #ifdef	IPPROTO_IDP | 
 | 1332 | 	insint(d, "IPPROTO_IDP", IPPROTO_IDP); | 
 | 1333 | #endif | 
 | 1334 | #ifdef	IPPROTO_HELLO | 
 | 1335 | 	insint(d, "IPPROTO_HELLO", IPPROTO_HELLO); | 
 | 1336 | #endif | 
 | 1337 | #ifdef	IPPROTO_ND | 
 | 1338 | 	insint(d, "IPPROTO_ND", IPPROTO_ND); | 
 | 1339 | #endif | 
 | 1340 | #ifdef	IPPROTO_TP | 
 | 1341 | 	insint(d, "IPPROTO_TP", IPPROTO_TP); | 
 | 1342 | #endif | 
 | 1343 | #ifdef	IPPROTO_XTP | 
 | 1344 | 	insint(d, "IPPROTO_XTP", IPPROTO_XTP); | 
 | 1345 | #endif | 
 | 1346 | #ifdef	IPPROTO_EON | 
 | 1347 | 	insint(d, "IPPROTO_EON", IPPROTO_EON); | 
 | 1348 | #endif | 
 | 1349 | #ifdef	IPPROTO_BIP | 
 | 1350 | 	insint(d, "IPPROTO_BIP", IPPROTO_BIP); | 
 | 1351 | #endif | 
 | 1352 | /**/ | 
 | 1353 | #ifdef	IPPROTO_RAW | 
 | 1354 | 	insint(d, "IPPROTO_RAW", IPPROTO_RAW); | 
 | 1355 | #endif | 
 | 1356 | #ifdef	IPPROTO_MAX | 
 | 1357 | 	insint(d, "IPPROTO_MAX", IPPROTO_MAX); | 
 | 1358 | #endif | 
 | 1359 |  | 
 | 1360 | 	/* Some port configuration */ | 
 | 1361 | #ifdef	IPPORT_RESERVED | 
 | 1362 | 	insint(d, "IPPORT_RESERVED", IPPORT_RESERVED); | 
 | 1363 | #else | 
 | 1364 | 	insint(d, "IPPORT_RESERVED", 1024); | 
 | 1365 | #endif | 
 | 1366 | #ifdef	IPPORT_USERRESERVED | 
 | 1367 | 	insint(d, "IPPORT_USERRESERVED", IPPORT_USERRESERVED); | 
 | 1368 | #else | 
 | 1369 | 	insint(d, "IPPORT_USERRESERVED", 5000); | 
 | 1370 | #endif | 
 | 1371 |  | 
| Guido van Rossum | 5f05eb4 | 1995-02-17 15:11:07 +0000 | [diff] [blame] | 1372 | 	/* Some reserved IP v.4 addresses */ | 
 | 1373 | #ifdef	INADDR_ANY | 
 | 1374 | 	insint(d, "INADDR_ANY", INADDR_ANY); | 
 | 1375 | #else | 
 | 1376 | 	insint(d, "INADDR_ANY", 0x00000000); | 
 | 1377 | #endif | 
 | 1378 | #ifdef	INADDR_BROADCAST | 
 | 1379 | 	insint(d, "INADDR_BROADCAST", INADDR_BROADCAST); | 
 | 1380 | #else | 
 | 1381 | 	insint(d, "INADDR_BROADCAST", 0xffffffff); | 
 | 1382 | #endif | 
 | 1383 | #ifdef	INADDR_LOOPBACK | 
 | 1384 | 	insint(d, "INADDR_LOOPBACK", INADDR_LOOPBACK); | 
 | 1385 | #else | 
 | 1386 | 	insint(d, "INADDR_LOOPBACK", 0x7F000001); | 
 | 1387 | #endif | 
 | 1388 | #ifdef	INADDR_UNSPEC_GROUP | 
 | 1389 | 	insint(d, "INADDR_UNSPEC_GROUP", INADDR_UNSPEC_GROUP); | 
 | 1390 | #else | 
 | 1391 | 	insint(d, "INADDR_UNSPEC_GROUP", 0xe0000000); | 
 | 1392 | #endif | 
 | 1393 | #ifdef	INADDR_ALLHOSTS_GROUP | 
 | 1394 | 	insint(d, "INADDR_ALLHOSTS_GROUP", INADDR_ALLHOSTS_GROUP); | 
 | 1395 | #else | 
 | 1396 | 	insint(d, "INADDR_ALLHOSTS_GROUP", 0xe0000001); | 
 | 1397 | #endif | 
 | 1398 | #ifdef	INADDR_MAX_LOCAL_GROUP | 
 | 1399 | 	insint(d, "INADDR_MAX_LOCAL_GROUP", INADDR_MAX_LOCAL_GROUP); | 
 | 1400 | #else | 
 | 1401 | 	insint(d, "INADDR_MAX_LOCAL_GROUP", 0xe00000ff); | 
 | 1402 | #endif | 
 | 1403 | #ifdef	INADDR_NONE | 
 | 1404 | 	insint(d, "INADDR_NONE", INADDR_NONE); | 
 | 1405 | #else | 
 | 1406 | 	insint(d, "INADDR_NONE", 0xffffffff); | 
 | 1407 | #endif | 
 | 1408 |  | 
| Guido van Rossum | 3fdf58b | 1995-02-07 15:39:52 +0000 | [diff] [blame] | 1409 | 	/* IP [gs]etsockopt options */ | 
 | 1410 | #ifdef	IP_OPTIONS | 
 | 1411 | 	insint(d, "IP_OPTIONS", IP_OPTIONS); | 
 | 1412 | #endif | 
 | 1413 | #ifdef	IP_HDRINCL | 
 | 1414 | 	insint(d, "IP_HDRINCL", IP_HDRINCL); | 
 | 1415 | #endif | 
 | 1416 | #ifdef	IP_TOS | 
 | 1417 | 	insint(d, "IP_TOS", IP_TOS); | 
 | 1418 | #endif | 
 | 1419 | #ifdef	IP_TTL | 
 | 1420 | 	insint(d, "IP_TTL", IP_TTL); | 
 | 1421 | #endif | 
 | 1422 | #ifdef	IP_RECVOPTS | 
 | 1423 | 	insint(d, "IP_RECVOPTS", IP_RECVOPTS); | 
 | 1424 | #endif | 
 | 1425 | #ifdef	IP_RECVRETOPTS | 
 | 1426 | 	insint(d, "IP_RECVRETOPTS", IP_RECVRETOPTS); | 
 | 1427 | #endif | 
 | 1428 | #ifdef	IP_RECVDSTADDR | 
 | 1429 | 	insint(d, "IP_RECVDSTADDR", IP_RECVDSTADDR); | 
 | 1430 | #endif | 
 | 1431 | #ifdef	IP_RETOPTS | 
 | 1432 | 	insint(d, "IP_RETOPTS", IP_RETOPTS); | 
 | 1433 | #endif | 
 | 1434 | #ifdef	IP_MULTICAST_IF | 
 | 1435 | 	insint(d, "IP_MULTICAST_IF", IP_MULTICAST_IF); | 
 | 1436 | #endif | 
 | 1437 | #ifdef	IP_MULTICAST_TTL | 
 | 1438 | 	insint(d, "IP_MULTICAST_TTL", IP_MULTICAST_TTL); | 
 | 1439 | #endif | 
 | 1440 | #ifdef	IP_MULTICAST_LOOP | 
 | 1441 | 	insint(d, "IP_MULTICAST_LOOP", IP_MULTICAST_LOOP); | 
 | 1442 | #endif | 
 | 1443 | #ifdef	IP_ADD_MEMBERSHIP | 
 | 1444 | 	insint(d, "IP_ADD_MEMBERSHIP", IP_ADD_MEMBERSHIP); | 
 | 1445 | #endif | 
 | 1446 | #ifdef	IP_DROP_MEMBERSHIP | 
 | 1447 | 	insint(d, "IP_DROP_MEMBERSHIP", IP_DROP_MEMBERSHIP); | 
 | 1448 | #endif | 
| Guido van Rossum | 6574b3e | 1991-06-25 21:36:08 +0000 | [diff] [blame] | 1449 | } | 
| Guido van Rossum | 6a3eb5f | 1994-08-18 15:42:46 +0000 | [diff] [blame] | 1450 |  | 
 | 1451 | #ifdef NT | 
 | 1452 | BOOL	WINAPI	DllMain (HANDLE hInst,  | 
 | 1453 | 			ULONG ul_reason_for_call, | 
 | 1454 | 			LPVOID lpReserved) | 
 | 1455 | { | 
 | 1456 | 	switch (ul_reason_for_call) | 
 | 1457 | 	{ | 
 | 1458 | 		case DLL_PROCESS_ATTACH: | 
 | 1459 | 			WSADATA WSAData; | 
 | 1460 | 			if (WSAStartup(MAKEWORD(2,0), &WSAData)) { | 
 | 1461 | 				OutputDebugString("Python can't initialize Windows Sockets DLL!"); | 
 | 1462 | 				return FALSE; | 
 | 1463 | 			} | 
 | 1464 | 			break; | 
 | 1465 | 		case DLL_PROCESS_DETACH: | 
 | 1466 | 			WSACleanup(); | 
 | 1467 | 			break; | 
 | 1468 |  | 
 | 1469 | 	} | 
 | 1470 | 	return TRUE; | 
 | 1471 | } | 
 | 1472 | #endif /* NT */ |