1cfdd41f7d
Given the order of imports and monkey-patching, even when using greendns the `getaddrinfo()` call inside of _socket_nodns.create_connection would use the unpatched (i.e. non-green) version of getaddrinfo, inadvertantly blocking the process in some situations
109 lines
3.8 KiB
Python
109 lines
3.8 KiB
Python
__socket = __import__('socket')
|
|
globals().update(dict([(var, getattr(__socket, var))
|
|
for var in dir(__socket)
|
|
if not var.startswith('__')]))
|
|
|
|
os = __import__('os')
|
|
import sys
|
|
import warnings
|
|
from eventlet.hubs import get_hub
|
|
from eventlet.greenio import GreenSocket as socket
|
|
from eventlet.greenio import SSL as _SSL # for exceptions
|
|
from eventlet.greenio import _GLOBAL_DEFAULT_TIMEOUT
|
|
from eventlet.greenio import _fileobject
|
|
|
|
__all__ = __socket.__all__
|
|
__patched__ = ['fromfd', 'socketpair', 'ssl', 'socket']
|
|
try:
|
|
__original_fromfd__ = __socket.fromfd
|
|
def fromfd(*args):
|
|
return socket(__original_fromfd__(*args))
|
|
except AttributeError:
|
|
pass
|
|
|
|
try:
|
|
__original_socketpair__ = __socket.socketpair
|
|
def socketpair(*args):
|
|
one, two = __original_socketpair__(*args)
|
|
return socket(one), socket(two)
|
|
except AttributeError:
|
|
pass
|
|
|
|
|
|
|
|
def _convert_to_sslerror(ex):
|
|
""" Transliterates SSL.SysCallErrors to socket.sslerrors"""
|
|
return sslerror((ex.args[0], ex.args[1]))
|
|
|
|
|
|
class GreenSSLObject(object):
|
|
""" Wrapper object around the SSLObjects returned by socket.ssl, which have a
|
|
slightly different interface from SSL.Connection objects. """
|
|
def __init__(self, green_ssl_obj):
|
|
""" Should only be called by a 'green' socket.ssl """
|
|
self.connection = green_ssl_obj
|
|
try:
|
|
# if it's already connected, do the handshake
|
|
self.connection.getpeername()
|
|
except:
|
|
pass
|
|
else:
|
|
try:
|
|
self.connection.do_handshake()
|
|
except _SSL.SysCallError, e:
|
|
raise _convert_to_sslerror(e)
|
|
|
|
def read(self, n=1024):
|
|
"""If n is provided, read n bytes from the SSL connection, otherwise read
|
|
until EOF. The return value is a string of the bytes read."""
|
|
try:
|
|
return self.connection.read(n)
|
|
except _SSL.ZeroReturnError:
|
|
return ''
|
|
except _SSL.SysCallError, e:
|
|
raise _convert_to_sslerror(e)
|
|
|
|
def write(self, s):
|
|
"""Writes the string s to the on the object's SSL connection.
|
|
The return value is the number of bytes written. """
|
|
try:
|
|
return self.connection.write(s)
|
|
except _SSL.SysCallError, e:
|
|
raise _convert_to_sslerror(e)
|
|
|
|
def server(self):
|
|
""" Returns a string describing the server's certificate. Useful for debugging
|
|
purposes; do not parse the content of this string because its format can't be
|
|
parsed unambiguously. """
|
|
return str(self.connection.get_peer_certificate().get_subject())
|
|
|
|
def issuer(self):
|
|
"""Returns a string describing the issuer of the server's certificate. Useful
|
|
for debugging purposes; do not parse the content of this string because its
|
|
format can't be parsed unambiguously."""
|
|
return str(self.connection.get_peer_certificate().get_issuer())
|
|
|
|
|
|
try:
|
|
try:
|
|
# >= Python 2.6
|
|
from eventlet.green import ssl as ssl_module
|
|
sslerror = __socket.sslerror
|
|
__socket.ssl
|
|
def ssl(sock, certificate=None, private_key=None):
|
|
warnings.warn("socket.ssl() is deprecated. Use ssl.wrap_socket() instead.",
|
|
DeprecationWarning, stacklevel=2)
|
|
return ssl_module.sslwrap_simple(sock, private_key, certificate)
|
|
except ImportError:
|
|
# <= Python 2.5 compatibility
|
|
sslerror = __socket.sslerror
|
|
__socket.ssl
|
|
def ssl(sock, certificate=None, private_key=None):
|
|
from eventlet import util
|
|
wrapped = util.wrap_ssl(sock, certificate, private_key)
|
|
return GreenSSLObject(wrapped)
|
|
except AttributeError:
|
|
# if the real socket module doesn't have the ssl method or sslerror
|
|
# exception, we can't emulate them
|
|
pass
|