mirror of
https://github.com/python/cpython.git
synced 2024-11-25 19:03:49 +08:00
c18993f84a
to being a new-style class, to be more similar to the socket class in the _socket module; it is now the same as the _socketobject class. Added __slots__. Added docstrings, copied from the real socket class where possible. The _fileobject class is now also a new-style class with __slots__ (though without docstrings). The mode, name, softspace, bufsize and closed attributes are properly supported (closed as a property; name as a class attributes; the softspace, mode and bufsize as slots).
364 lines
11 KiB
Python
364 lines
11 KiB
Python
# Wrapper module for _socket, providing some additional facilities
|
|
# implemented in Python.
|
|
|
|
"""\
|
|
This module provides socket operations and some related functions.
|
|
On Unix, it supports IP (Internet Protocol) and Unix domain sockets.
|
|
On other systems, it only supports IP. Functions specific for a
|
|
socket are available as methods of the socket object.
|
|
|
|
Functions:
|
|
|
|
socket() -- create a new socket object
|
|
fromfd() -- create a socket object from an open file descriptor [*]
|
|
gethostname() -- return the current hostname
|
|
gethostbyname() -- map a hostname to its IP number
|
|
gethostbyaddr() -- map an IP number or hostname to DNS info
|
|
getservbyname() -- map a service name and a protocol name to a port number
|
|
getprotobyname() -- mape a protocol name (e.g. 'tcp') to a number
|
|
ntohs(), ntohl() -- convert 16, 32 bit int from network to host byte order
|
|
htons(), htonl() -- convert 16, 32 bit int from host to network byte order
|
|
inet_aton() -- convert IP addr string (123.45.67.89) to 32-bit packed format
|
|
inet_ntoa() -- convert 32-bit packed format IP to string (123.45.67.89)
|
|
ssl() -- secure socket layer support (only available if configured)
|
|
socket.getdefaulttimeout() -- get the default timeout value
|
|
socket.setdefaulttimeout() -- set the default timeout value
|
|
|
|
[*] not available on all platforms!
|
|
|
|
Special objects:
|
|
|
|
SocketType -- type object for socket objects
|
|
error -- exception raised for I/O errors
|
|
|
|
Integer constants:
|
|
|
|
AF_INET, AF_UNIX -- socket domains (first argument to socket() call)
|
|
SOCK_STREAM, SOCK_DGRAM, SOCK_RAW -- socket types (second argument)
|
|
|
|
Many other constants may be defined; these may be used in calls to
|
|
the setsockopt() and getsockopt() methods.
|
|
"""
|
|
|
|
import _socket
|
|
from _socket import *
|
|
|
|
_have_ssl = False
|
|
try:
|
|
from _ssl import *
|
|
_have_ssl = True
|
|
except ImportError:
|
|
pass
|
|
|
|
import os, sys
|
|
|
|
__all__ = ["getfqdn"]
|
|
__all__.extend(os._get_exports_list(_socket))
|
|
# XXX shouldn't there be something similar to the above for _ssl exports?
|
|
|
|
_realsocket = socket
|
|
_needwrapper = False
|
|
if (sys.platform.lower().startswith("win")
|
|
or (hasattr(os, 'uname') and os.uname()[0] == "BeOS")
|
|
or sys.platform=="riscos"):
|
|
|
|
_needwrapper = True
|
|
|
|
if _have_ssl:
|
|
_realssl = ssl
|
|
def ssl(sock, keyfile=None, certfile=None):
|
|
if hasattr(sock, "_sock"):
|
|
sock = sock._sock
|
|
return _realssl(sock, keyfile, certfile)
|
|
|
|
# WSA error codes
|
|
if sys.platform.lower().startswith("win"):
|
|
errorTab = {}
|
|
errorTab[10004] = "The operation was interrupted."
|
|
errorTab[10009] = "A bad file handle was passed."
|
|
errorTab[10013] = "Permission denied."
|
|
errorTab[10014] = "A fault occurred on the network??" # WSAEFAULT
|
|
errorTab[10022] = "An invalid operation was attempted."
|
|
errorTab[10035] = "The socket operation would block"
|
|
errorTab[10036] = "A blocking operation is already in progress."
|
|
errorTab[10048] = "The network address is in use."
|
|
errorTab[10054] = "The connection has been reset."
|
|
errorTab[10058] = "The network has been shut down."
|
|
errorTab[10060] = "The operation timed out."
|
|
errorTab[10061] = "Connection refused."
|
|
errorTab[10063] = "The name is too long."
|
|
errorTab[10064] = "The host is down."
|
|
errorTab[10065] = "The host is unreachable."
|
|
__all__.append("errorTab")
|
|
del os, sys
|
|
|
|
|
|
def getfqdn(name=''):
|
|
"""Get fully qualified domain name from name.
|
|
|
|
An empty argument is interpreted as meaning the local host.
|
|
|
|
First the hostname returned by gethostbyaddr() is checked, then
|
|
possibly existing aliases. In case no FQDN is available, hostname
|
|
is returned.
|
|
"""
|
|
name = name.strip()
|
|
if not name or name == '0.0.0.0':
|
|
name = gethostname()
|
|
try:
|
|
hostname, aliases, ipaddrs = gethostbyaddr(name)
|
|
except error:
|
|
pass
|
|
else:
|
|
aliases.insert(0, hostname)
|
|
for name in aliases:
|
|
if '.' in name:
|
|
break
|
|
else:
|
|
name = hostname
|
|
return name
|
|
|
|
|
|
#
|
|
# These classes are used by the socket() defined on Windows and BeOS
|
|
# platforms to provide a best-effort implementation of the cleanup
|
|
# semantics needed when sockets can't be dup()ed.
|
|
#
|
|
# These are not actually used on other platforms.
|
|
#
|
|
|
|
_socketmethods = (
|
|
'bind', 'connect', 'connect_ex', 'fileno', 'listen',
|
|
'getpeername', 'getsockname', 'getsockopt', 'setsockopt',
|
|
'recv', 'recvfrom', 'send', 'sendall', 'sendto', 'setblocking',
|
|
'settimeout', 'gettimeout', 'shutdown')
|
|
|
|
class _socketobject(object):
|
|
|
|
__doc__ = _realsocket.__doc__
|
|
|
|
__slots__ = ["_sock"]
|
|
|
|
class _closedsocket(object):
|
|
__slots__ = []
|
|
def __getattr__(self, name):
|
|
raise error(9, 'Bad file descriptor')
|
|
|
|
def __init__(self, family=AF_INET, type=SOCK_STREAM, proto=0, _sock=None):
|
|
if _sock is None:
|
|
_sock = _realsocket(family, type, proto)
|
|
self._sock = _sock
|
|
|
|
def close(self):
|
|
# Avoid referencing globals here
|
|
self._sock = self.__class__._closedsocket()
|
|
close.__doc__ = _realsocket.close.__doc__
|
|
|
|
def __del__(self):
|
|
self.close()
|
|
|
|
def accept(self):
|
|
sock, addr = self._sock.accept()
|
|
return _socketobject(_sock=sock), addr
|
|
accept.__doc__ = _realsocket.accept.__doc__
|
|
|
|
def dup(self):
|
|
"""dup() -> socket object
|
|
|
|
Return a new socket object connected to the same system resource."""
|
|
return _socketobject(_sock=self._sock)
|
|
|
|
def makefile(self, mode='r', bufsize=-1):
|
|
"""makefile([mode[, bufsize]]) -> file object
|
|
|
|
Return a regular file object corresponding to the socket. The mode
|
|
and bufsize arguments are as for the built-in open() function."""
|
|
return _fileobject(self._sock, mode, bufsize)
|
|
|
|
_s = ("def %s(self, *args): return self._sock.%s(*args)\n\n"
|
|
"%s.__doc__ = _realsocket.%s.__doc__\n")
|
|
for _m in _socketmethods:
|
|
exec _s % (_m, _m, _m, _m)
|
|
|
|
if _needwrapper:
|
|
socket = _socketobject
|
|
|
|
class _fileobject(object):
|
|
"""Faux file object attached to a socket object."""
|
|
|
|
default_bufsize = 8192
|
|
name = "<socket>"
|
|
|
|
__slots__ = ["mode", "bufsize", "softspace",
|
|
# "closed" is a property, see below
|
|
"_sock", "_rbufsize", "_wbufsize", "_rbuf", "_wbuf"]
|
|
|
|
def __init__(self, sock, mode='rb', bufsize=-1):
|
|
self._sock = sock
|
|
self.mode = mode # Not actually used in this version
|
|
if bufsize < 0:
|
|
bufsize = self.default_bufsize
|
|
self.bufsize = bufsize
|
|
self.softspace = False
|
|
if bufsize == 0:
|
|
self._rbufsize = 1
|
|
elif bufsize == 1:
|
|
self._rbufsize = self.default_bufsize
|
|
else:
|
|
self._rbufsize = bufsize
|
|
self._wbufsize = bufsize
|
|
# The buffers are lists of non-empty strings
|
|
self._rbuf = []
|
|
self._wbuf = []
|
|
|
|
def _getclosed(self):
|
|
return self._sock is not None
|
|
closed = property(_getclosed, doc="True if the file is closed")
|
|
|
|
def close(self):
|
|
try:
|
|
if self._sock:
|
|
self.flush()
|
|
finally:
|
|
self._sock = None
|
|
|
|
def __del__(self):
|
|
self.close()
|
|
|
|
def flush(self):
|
|
if self._wbuf:
|
|
buffer = "".join(self._wbuf)
|
|
self._wbuf = []
|
|
self._sock.sendall(buffer)
|
|
|
|
def fileno(self):
|
|
return self._sock.fileno()
|
|
|
|
def write(self, data):
|
|
data = str(data) # XXX Should really reject non-string non-buffers
|
|
if not data:
|
|
return
|
|
self._wbuf.append(data)
|
|
if (self._wbufsize == 0 or
|
|
self._wbufsize == 1 and '\n' in data or
|
|
self._get_wbuf_len() >= self._wbufsize):
|
|
self.flush()
|
|
|
|
def writelines(self, list):
|
|
# XXX We could do better here for very long lists
|
|
# XXX Should really reject non-string non-buffers
|
|
self._wbuf.extend(filter(None, map(str, list)))
|
|
if (self._wbufsize <= 1 or
|
|
self._get_wbuf_len() >= self._wbufsize):
|
|
self.flush()
|
|
|
|
def _get_wbuf_len(self):
|
|
buf_len = 0
|
|
for x in self._wbuf:
|
|
buf_len += len(x)
|
|
return buf_len
|
|
|
|
def _get_rbuf_len(self):
|
|
buf_len = 0
|
|
for x in self._rbuf:
|
|
buf_len += len(x)
|
|
return buf_len
|
|
|
|
def read(self, size=-1):
|
|
if size < 0:
|
|
# Read until EOF
|
|
if self._rbufsize <= 1:
|
|
recv_size = self.default_bufsize
|
|
else:
|
|
recv_size = self._rbufsize
|
|
while 1:
|
|
data = self._sock.recv(recv_size)
|
|
if not data:
|
|
break
|
|
self._rbuf.append(data)
|
|
else:
|
|
buf_len = self._get_rbuf_len()
|
|
while buf_len < size:
|
|
recv_size = max(self._rbufsize, size - buf_len)
|
|
data = self._sock.recv(recv_size)
|
|
if not data:
|
|
break
|
|
buf_len += len(data)
|
|
self._rbuf.append(data)
|
|
data = "".join(self._rbuf)
|
|
self._rbuf = []
|
|
if 0 <= size < buf_len:
|
|
self._rbuf.append(data[size:])
|
|
data = data[:size]
|
|
return data
|
|
|
|
def readline(self, size=-1):
|
|
data_len = 0
|
|
for index, x in enumerate(self._rbuf):
|
|
data_len += len(x)
|
|
if '\n' in x or 0 <= size <= data_len:
|
|
index += 1
|
|
data = "".join(self._rbuf[:index])
|
|
end = data.find('\n')
|
|
if end < 0:
|
|
end = len(data)
|
|
else:
|
|
end += 1
|
|
if 0 <= size < end:
|
|
end = size
|
|
data, rest = data[:end], data[end:]
|
|
if rest:
|
|
self._rbuf[:index] = [rest]
|
|
else:
|
|
del self._rbuf[:index]
|
|
return data
|
|
recv_size = self._rbufsize
|
|
while 1:
|
|
if size >= 0:
|
|
recv_size = min(self._rbufsize, size - data_len)
|
|
x = self._sock.recv(recv_size)
|
|
if not x:
|
|
break
|
|
data_len += len(x)
|
|
self._rbuf.append(x)
|
|
if '\n' in x or 0 <= size <= data_len:
|
|
break
|
|
data = "".join(self._rbuf)
|
|
end = data.find('\n')
|
|
if end < 0:
|
|
end = len(data)
|
|
else:
|
|
end += 1
|
|
if 0 <= size < end:
|
|
end = size
|
|
data, rest = data[:end], data[end:]
|
|
if rest:
|
|
self._rbuf = [rest]
|
|
else:
|
|
self._rbuf = []
|
|
return data
|
|
|
|
def readlines(self, sizehint=0):
|
|
total = 0
|
|
list = []
|
|
while 1:
|
|
line = self.readline()
|
|
if not line:
|
|
break
|
|
list.append(line)
|
|
total += len(line)
|
|
if sizehint and total >= sizehint:
|
|
break
|
|
return list
|
|
|
|
# Iterator protocols
|
|
|
|
def __iter__(self):
|
|
return self
|
|
|
|
def next(self):
|
|
line = self.readline()
|
|
if not line:
|
|
raise StopIteration
|
|
return line
|