blob: 1cc0d2bb0ff9072f73e2c49294c137d344edc6eb [file] [log] [blame]
Alex Gaynorc3fac9d2016-01-16 12:15:09 -05001from threading import RLock as _RLock
Jean-Paul Calderone897bc252008-02-18 20:50:23 -05002
Alex Gaynorc3fac9d2016-01-16 12:15:09 -05003from OpenSSL import SSL as _ssl
Jean-Paul Calderone897bc252008-02-18 20:50:23 -05004
Hynek Schlawackba1d66c2015-09-05 18:17:42 +02005
Jean-Paul Calderone897bc252008-02-18 20:50:23 -05006class Connection:
7 def __init__(self, *args):
Jean-Paul Calderone65627a22014-03-25 20:58:30 -04008 self._ssl_conn = _ssl.Connection(*args)
Jean-Paul Calderone897bc252008-02-18 20:50:23 -05009 self._lock = _RLock()
10
11 for f in ('get_context', 'pending', 'send', 'write', 'recv', 'read',
12 'renegotiate', 'bind', 'listen', 'connect', 'accept',
13 'setblocking', 'fileno', 'shutdown', 'close', 'get_cipher_list',
14 'getpeername', 'getsockname', 'getsockopt', 'setsockopt',
15 'makefile', 'get_app_data', 'set_app_data', 'state_string',
Hynek Schlawackba1d66c2015-09-05 18:17:42 +020016 'sock_shutdown', 'get_peer_certificate', 'get_peer_cert_chain',
17 'want_read', 'want_write', 'set_connect_state',
18 'set_accept_state', 'connect_ex', 'sendall'):
Jean-Paul Calderonee6284652010-08-11 22:40:57 -040019 exec("""def %s(self, *args):
Jean-Paul Calderone897bc252008-02-18 20:50:23 -050020 self._lock.acquire()
21 try:
Jean-Paul Calderonee6284652010-08-11 22:40:57 -040022 return self._ssl_conn.%s(*args)
Jean-Paul Calderone897bc252008-02-18 20:50:23 -050023 finally:
Jean-Paul Calderonee6284652010-08-11 22:40:57 -040024 self._lock.release()\n""" % (f, f))