diff options
author | Martin v. Löwis <martin@v.loewis.de> | 2006-07-01 15:33:37 +0000 |
---|---|---|
committer | Martin v. Löwis <martin@v.loewis.de> | 2006-07-01 15:33:37 +0000 |
commit | 7596e8342ea34108d773d1802d5147e9884b547f (patch) | |
tree | aa19561ab430e4a5b9edee0409ceb4943339edd6 /Lib/socket.py | |
parent | a09803329c6315792a9dee92c9d4d9bfe6eab01a (diff) | |
download | cpython-git-7596e8342ea34108d773d1802d5147e9884b547f.tar.gz |
Release all forwarded functions in .close. Fixes #1513223.
Diffstat (limited to 'Lib/socket.py')
-rw-r--r-- | Lib/socket.py | 26 |
1 files changed, 14 insertions, 12 deletions
diff --git a/Lib/socket.py b/Lib/socket.py index fa0e663a3f..4e83f099a2 100644 --- a/Lib/socket.py +++ b/Lib/socket.py @@ -130,35 +130,37 @@ _socketmethods = ( if sys.platform == "riscos": _socketmethods = _socketmethods + ('sleeptaskw',) +# All the method names that must be delegated to either the real socket +# object or the _closedsocket object. +_delegate_methods = ("recv", "recvfrom", "recv_into", "recvfrom_into", + "send", "sendto") + class _closedsocket(object): __slots__ = [] def _dummy(*args): raise error(EBADF, 'Bad file descriptor') - send = recv = sendto = recvfrom = __getattr__ = _dummy + # All _delegate_methods must also be initialized here. + send = recv = recv_into = sendto = recvfrom = recvfrom_into = _dummy + __getattr__ = _dummy class _socketobject(object): __doc__ = _realsocket.__doc__ - __slots__ = ["_sock", - "recv", "recv_into", "recvfrom_into", - "send", "sendto", "recvfrom", - "__weakref__"] + __slots__ = ["_sock", "__weakref__"] + list(_delegate_methods) 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 - self.send = self._sock.send - self.recv = self._sock.recv - self.recv_into = self._sock.recv_into - self.sendto = self._sock.sendto - self.recvfrom = self._sock.recvfrom - self.recvfrom_into = self._sock.recvfrom_into + for method in _delegate_methods: + setattr(self, method, getattr(_sock, method)) def close(self): self._sock = _closedsocket() - self.send = self.recv = self.sendto = self.recvfrom = self._sock._dummy + dummy = self._sock._dummy + for method in _delegate_methods: + setattr(self, method, dummy) close.__doc__ = _realsocket.close.__doc__ def accept(self): |