Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 1 | from OpenSSL import SSL |
| 2 | _ssl = SSL |
| 3 | del SSL |
| 4 | |
| 5 | import threading |
| 6 | _RLock = threading.RLock |
| 7 | del threading |
| 8 | |
| 9 | class Connection: |
| 10 | def __init__(self, *args): |
Jean-Paul Calderone | 65627a2 | 2014-03-25 20:58:30 -0400 | [diff] [blame] | 11 | self._ssl_conn = _ssl.Connection(*args) |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 12 | self._lock = _RLock() |
| 13 | |
| 14 | for f in ('get_context', 'pending', 'send', 'write', 'recv', 'read', |
| 15 | 'renegotiate', 'bind', 'listen', 'connect', 'accept', |
| 16 | 'setblocking', 'fileno', 'shutdown', 'close', 'get_cipher_list', |
| 17 | 'getpeername', 'getsockname', 'getsockopt', 'setsockopt', |
| 18 | 'makefile', 'get_app_data', 'set_app_data', 'state_string', |
Jean-Paul Calderone | 95b92c7 | 2011-05-17 15:45:21 -0400 | [diff] [blame] | 19 | 'sock_shutdown', 'get_peer_certificate', 'get_peer_cert_chain', 'want_read', |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 20 | 'want_write', 'set_connect_state', 'set_accept_state', |
| 21 | 'connect_ex', 'sendall'): |
Jean-Paul Calderone | e628465 | 2010-08-11 22:40:57 -0400 | [diff] [blame] | 22 | exec("""def %s(self, *args): |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 23 | self._lock.acquire() |
| 24 | try: |
Jean-Paul Calderone | e628465 | 2010-08-11 22:40:57 -0400 | [diff] [blame] | 25 | return self._ssl_conn.%s(*args) |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 26 | finally: |
Jean-Paul Calderone | e628465 | 2010-08-11 22:40:57 -0400 | [diff] [blame] | 27 | self._lock.release()\n""" % (f, f)) |
Jean-Paul Calderone | 897bc25 | 2008-02-18 20:50:23 -0500 | [diff] [blame] | 28 | |