Andrew K. Choi
91c7e04474
All checks were successful
continuous-integration/drone/push Build is passing
76 lines
2.7 KiB
Python
76 lines
2.7 KiB
Python
# pylint: skip-file
|
|
# vendored from https://github.com/mhils/backports.socketpair
|
|
from __future__ import absolute_import
|
|
|
|
import sys
|
|
import socket
|
|
import errno
|
|
|
|
_LOCALHOST = '127.0.0.1'
|
|
_LOCALHOST_V6 = '::1'
|
|
|
|
if not hasattr(socket, "socketpair"):
|
|
# Origin: https://gist.github.com/4325783, by Geert Jansen. Public domain.
|
|
def socketpair(family=socket.AF_INET, type=socket.SOCK_STREAM, proto=0):
|
|
if family == socket.AF_INET:
|
|
host = _LOCALHOST
|
|
elif family == socket.AF_INET6:
|
|
host = _LOCALHOST_V6
|
|
else:
|
|
raise ValueError("Only AF_INET and AF_INET6 socket address families "
|
|
"are supported")
|
|
if type != socket.SOCK_STREAM:
|
|
raise ValueError("Only SOCK_STREAM socket type is supported")
|
|
if proto != 0:
|
|
raise ValueError("Only protocol zero is supported")
|
|
|
|
# We create a connected TCP socket. Note the trick with
|
|
# setblocking(False) that prevents us from having to create a thread.
|
|
lsock = socket.socket(family, type, proto)
|
|
try:
|
|
lsock.bind((host, 0))
|
|
lsock.listen(min(socket.SOMAXCONN, 128))
|
|
# On IPv6, ignore flow_info and scope_id
|
|
addr, port = lsock.getsockname()[:2]
|
|
csock = socket.socket(family, type, proto)
|
|
try:
|
|
csock.setblocking(False)
|
|
if sys.version_info >= (3, 0):
|
|
try:
|
|
csock.connect((addr, port))
|
|
except (BlockingIOError, InterruptedError):
|
|
pass
|
|
else:
|
|
try:
|
|
csock.connect((addr, port))
|
|
except socket.error as e:
|
|
if e.errno != errno.WSAEWOULDBLOCK:
|
|
raise
|
|
csock.setblocking(True)
|
|
ssock, _ = lsock.accept()
|
|
except Exception:
|
|
csock.close()
|
|
raise
|
|
finally:
|
|
lsock.close()
|
|
|
|
# Authenticating avoids using a connection from something else
|
|
# able to connect to {host}:{port} instead of us.
|
|
# We expect only AF_INET and AF_INET6 families.
|
|
try:
|
|
if (
|
|
ssock.getsockname() != csock.getpeername()
|
|
or csock.getsockname() != ssock.getpeername()
|
|
):
|
|
raise ConnectionError("Unexpected peer connection")
|
|
except:
|
|
# getsockname() and getpeername() can fail
|
|
# if either socket isn't connected.
|
|
ssock.close()
|
|
csock.close()
|
|
raise
|
|
|
|
return (ssock, csock)
|
|
|
|
socket.socketpair = socketpair
|