Alex Gaynor | c3fac9d | 2016-01-16 12:15:09 -0500 | [diff] [blame] | 1 | from threading import RLock as _RLock |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 2 | |
Alex Gaynor | c3fac9d | 2016-01-16 12:15:09 -0500 | [diff] [blame] | 3 | from OpenSSL import SSL as _ssl |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 4 | |
Hynek Schlawack | ba1d66c | 2015-09-05 18:17:42 +0200 | [diff] [blame] | 5 | |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 6 | class Connection: |
| 7 | def __init__(self, *args): |
Jean-Paul Calderone | 65627a2 | 2014-03-25 20:58:30 -0400 | [diff] [blame] | 8 | self._ssl_conn = _ssl.Connection(*args) |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 9 | 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 Schlawack | ba1d66c | 2015-09-05 18:17:42 +0200 | [diff] [blame] | 16 | '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 Calderone | e628465 | 2010-08-11 22:40:57 -0400 | [diff] [blame] | 19 | exec("""def %s(self, *args): |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 20 | self._lock.acquire() |
| 21 | try: |
Jean-Paul Calderone | e628465 | 2010-08-11 22:40:57 -0400 | [diff] [blame] | 22 | return self._ssl_conn.%s(*args) |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 23 | finally: |
Jean-Paul Calderone | e628465 | 2010-08-11 22:40:57 -0400 | [diff] [blame] | 24 | self._lock.release()\n""" % (f, f)) |