Added interface for SSL_CTX_set_info_callback()
* dtls/openssl.py: - Added methods SSL_CTX_set_info_callback(), SSL_state_string_long(), SSL_alert_type_string_long() and SSL_alert_desc_string_long() - Added constants for state and error evaluation during callback * dtls/sslconnection.py: Added _ssl_logging_cb() as default callback function - only outputs messages when logger is activeincoming
parent
d601774e24
commit
70e78b97cb
|
@ -1,3 +1,12 @@
|
||||||
|
2017-03-17 Björn Freise <mcfreis@gmx.net>
|
||||||
|
|
||||||
|
Added interface for SSL_CTX_set_info_callback()
|
||||||
|
|
||||||
|
* dtls/openssl.py:
|
||||||
|
- Added methods SSL_CTX_set_info_callback(), SSL_state_string_long(), SSL_alert_type_string_long() and SSL_alert_desc_string_long()
|
||||||
|
- Added constants for state and error evaluation during callback
|
||||||
|
* dtls/sslconnection.py: Added _ssl_logging_cb() as default callback function - only outputs messages when logger is active
|
||||||
|
|
||||||
2017-03-17 Björn Freise <mcfreis@gmx.net>
|
2017-03-17 Björn Freise <mcfreis@gmx.net>
|
||||||
|
|
||||||
SSL_write() extended to handle ctypes.Array as data
|
SSL_write() extended to handle ctypes.Array as data
|
||||||
|
|
|
@ -102,6 +102,31 @@ GEN_DIRNAME = 4
|
||||||
NID_subject_alt_name = 85
|
NID_subject_alt_name = 85
|
||||||
CRYPTO_LOCK = 1
|
CRYPTO_LOCK = 1
|
||||||
|
|
||||||
|
SSL_ST_MASK = 0x0FFF
|
||||||
|
SSL_ST_CONNECT = 0x1000
|
||||||
|
SSL_ST_ACCEPT = 0x2000
|
||||||
|
SSL_ST_INIT = (SSL_ST_CONNECT | SSL_ST_ACCEPT)
|
||||||
|
SSL_ST_BEFORE = 0x4000
|
||||||
|
|
||||||
|
SSL_ST_OK = 0x03
|
||||||
|
SSL_ST_RENEGOTIATE = (0x04 | SSL_ST_INIT)
|
||||||
|
SSL_ST_ERR = 0x05
|
||||||
|
|
||||||
|
SSL_CB_LOOP = 0x01
|
||||||
|
SSL_CB_EXIT = 0x02
|
||||||
|
SSL_CB_READ = 0x04
|
||||||
|
SSL_CB_WRITE = 0x08
|
||||||
|
|
||||||
|
SSL_CB_ALERT = 0x4000
|
||||||
|
SSL_CB_READ_ALERT = (SSL_CB_ALERT | SSL_CB_READ)
|
||||||
|
SSL_CB_WRITE_ALERT = (SSL_CB_ALERT | SSL_CB_WRITE)
|
||||||
|
SSL_CB_ACCEPT_LOOP = (SSL_ST_ACCEPT | SSL_CB_LOOP)
|
||||||
|
SSL_CB_ACCEPT_EXIT = (SSL_ST_ACCEPT | SSL_CB_EXIT)
|
||||||
|
SSL_CB_CONNECT_LOOP = (SSL_ST_CONNECT | SSL_CB_LOOP)
|
||||||
|
SSL_CB_CONNECT_EXIT = (SSL_ST_CONNECT | SSL_CB_EXIT)
|
||||||
|
SSL_CB_HANDSHAKE_START = 0x10
|
||||||
|
SSL_CB_HANDSHAKE_DONE = 0x20
|
||||||
|
|
||||||
#
|
#
|
||||||
# Integer constants - internal
|
# Integer constants - internal
|
||||||
#
|
#
|
||||||
|
@ -486,6 +511,12 @@ __all__ = [
|
||||||
"SSL_SESS_CACHE_SERVER", "SSL_SESS_CACHE_BOTH",
|
"SSL_SESS_CACHE_SERVER", "SSL_SESS_CACHE_BOTH",
|
||||||
"SSL_SESS_CACHE_NO_AUTO_CLEAR", "SSL_SESS_CACHE_NO_INTERNAL_LOOKUP",
|
"SSL_SESS_CACHE_NO_AUTO_CLEAR", "SSL_SESS_CACHE_NO_INTERNAL_LOOKUP",
|
||||||
"SSL_SESS_CACHE_NO_INTERNAL_STORE", "SSL_SESS_CACHE_NO_INTERNAL",
|
"SSL_SESS_CACHE_NO_INTERNAL_STORE", "SSL_SESS_CACHE_NO_INTERNAL",
|
||||||
|
"SSL_ST_MASK", "SSL_ST_CONNECT", "SSL_ST_ACCEPT", "SSL_ST_INIT", "SSL_ST_BEFORE", "SSL_ST_OK",
|
||||||
|
"SSL_ST_RENEGOTIATE", "SSL_ST_ERR", "SSL_CB_LOOP", "SSL_CB_EXIT", "SSL_CB_READ", "SSL_CB_WRITE",
|
||||||
|
"SSL_CB_ALERT", "SSL_CB_READ_ALERT", "SSL_CB_WRITE_ALERT",
|
||||||
|
"SSL_CB_ACCEPT_LOOP", "SSL_CB_ACCEPT_EXIT",
|
||||||
|
"SSL_CB_CONNECT_LOOP", "SSL_CB_CONNECT_EXIT",
|
||||||
|
"SSL_CB_HANDSHAKE_START", "SSL_CB_HANDSHAKE_DONE",
|
||||||
"SSL_FILE_TYPE_PEM",
|
"SSL_FILE_TYPE_PEM",
|
||||||
"GEN_DIRNAME", "NID_subject_alt_name",
|
"GEN_DIRNAME", "NID_subject_alt_name",
|
||||||
"CRYPTO_LOCK",
|
"CRYPTO_LOCK",
|
||||||
|
@ -499,7 +530,9 @@ __all__ = [
|
||||||
"BIO_set_nbio",
|
"BIO_set_nbio",
|
||||||
"SSL_CTX_set_session_cache_mode", "SSL_CTX_set_read_ahead",
|
"SSL_CTX_set_session_cache_mode", "SSL_CTX_set_read_ahead",
|
||||||
"SSL_CTX_set_options",
|
"SSL_CTX_set_options",
|
||||||
|
"SSL_CTX_set_info_callback",
|
||||||
"SSL_read", "SSL_write",
|
"SSL_read", "SSL_write",
|
||||||
|
"SSL_state_string_long", "SSL_alert_type_string_long", "SSL_alert_desc_string_long",
|
||||||
"SSL_CTX_set_cookie_cb",
|
"SSL_CTX_set_cookie_cb",
|
||||||
"OBJ_obj2txt", "decode_ASN1_STRING", "ASN1_TIME_print",
|
"OBJ_obj2txt", "decode_ASN1_STRING", "ASN1_TIME_print",
|
||||||
"X509_get_notAfter",
|
"X509_get_notAfter",
|
||||||
|
@ -536,6 +569,8 @@ map(lambda x: _make_function(*x), (
|
||||||
((None, "ret"), (SSLCTX, "ctx"), (c_void_p, "app_gen_cookie_cb")), False),
|
((None, "ret"), (SSLCTX, "ctx"), (c_void_p, "app_gen_cookie_cb")), False),
|
||||||
("SSL_CTX_set_cookie_verify_cb", libssl,
|
("SSL_CTX_set_cookie_verify_cb", libssl,
|
||||||
((None, "ret"), (SSLCTX, "ctx"), (c_void_p, "app_verify_cookie_cb")), False),
|
((None, "ret"), (SSLCTX, "ctx"), (c_void_p, "app_verify_cookie_cb")), False),
|
||||||
|
("SSL_CTX_set_info_callback", libssl,
|
||||||
|
((None, "ret"), (SSLCTX, "ctx"), (c_void_p, "app_info_cb")), False),
|
||||||
("SSL_new", libssl,
|
("SSL_new", libssl,
|
||||||
((SSL, "ret"), (SSLCTX, "ctx"))),
|
((SSL, "ret"), (SSLCTX, "ctx"))),
|
||||||
("SSL_free", libssl,
|
("SSL_free", libssl,
|
||||||
|
@ -568,6 +603,12 @@ map(lambda x: _make_function(*x), (
|
||||||
((None, "ret"), (c_ulong, "e"), (c_char_p, "buf"), (c_size_t, "len")), False),
|
((None, "ret"), (c_ulong, "e"), (c_char_p, "buf"), (c_size_t, "len")), False),
|
||||||
("SSL_get_error", libssl,
|
("SSL_get_error", libssl,
|
||||||
((c_int, "ret"), (SSL, "ssl"), (c_int, "ret")), False, None),
|
((c_int, "ret"), (SSL, "ssl"), (c_int, "ret")), False, None),
|
||||||
|
("SSL_state_string_long", libssl,
|
||||||
|
((c_char_p, "ret"), (SSL, "ssl")), False),
|
||||||
|
("SSL_alert_type_string_long", libssl,
|
||||||
|
((c_char_p, "ret"), (c_int, "value")), False),
|
||||||
|
("SSL_alert_desc_string_long", libssl,
|
||||||
|
((c_char_p, "ret"), (c_int, "value")), False),
|
||||||
("SSL_CTX_set_cipher_list", libssl,
|
("SSL_CTX_set_cipher_list", libssl,
|
||||||
((c_int, "ret"), (SSLCTX, "ctx"), (c_char_p, "str"))),
|
((c_int, "ret"), (SSLCTX, "ctx"), (c_char_p, "str"))),
|
||||||
("SSL_CTX_use_certificate_file", libssl,
|
("SSL_CTX_use_certificate_file", libssl,
|
||||||
|
@ -681,6 +722,28 @@ def SSL_CTX_set_options(ctx, options):
|
||||||
# Returns the new option bitmaks after adding the given options
|
# Returns the new option bitmaks after adding the given options
|
||||||
return _SSL_CTX_ctrl(ctx, SSL_CTRL_OPTIONS, options, None)
|
return _SSL_CTX_ctrl(ctx, SSL_CTRL_OPTIONS, options, None)
|
||||||
|
|
||||||
|
_rvoid_voidp_int_int = CFUNCTYPE(None, c_void_p, c_int, c_int)
|
||||||
|
|
||||||
|
_info_callback = dict()
|
||||||
|
|
||||||
|
def SSL_CTX_set_info_callback(ctx, app_info_cb):
|
||||||
|
"""
|
||||||
|
Set the info callback
|
||||||
|
|
||||||
|
:param callback: The Python callback to use
|
||||||
|
:return: None
|
||||||
|
"""
|
||||||
|
def py_info_callback(ssl, where, ret):
|
||||||
|
try:
|
||||||
|
app_info_cb(SSL(ssl), where, ret)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
return
|
||||||
|
|
||||||
|
global _info_callback
|
||||||
|
_info_callback[ctx] = _rvoid_voidp_int_int(py_info_callback)
|
||||||
|
_SSL_CTX_set_info_callback(ctx, _info_callback[ctx])
|
||||||
|
|
||||||
_rint_voidp_ubytep_uintp = CFUNCTYPE(c_int, c_void_p, POINTER(c_ubyte),
|
_rint_voidp_ubytep_uintp = CFUNCTYPE(c_int, c_void_p, POINTER(c_ubyte),
|
||||||
POINTER(c_uint))
|
POINTER(c_uint))
|
||||||
_rint_voidp_ubytep_uint = CFUNCTYPE(c_int, c_void_p, POINTER(c_ubyte), c_uint)
|
_rint_voidp_ubytep_uint = CFUNCTYPE(c_int, c_void_p, POINTER(c_ubyte), c_uint)
|
||||||
|
@ -778,6 +841,27 @@ def SSL_write(ssl, data):
|
||||||
str_data = str(data)
|
str_data = str(data)
|
||||||
return _SSL_write(ssl, str_data, len(str_data))
|
return _SSL_write(ssl, str_data, len(str_data))
|
||||||
|
|
||||||
|
def SSL_state_string_long(ssl):
|
||||||
|
try:
|
||||||
|
ret = _SSL_state_string_long(ssl)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
return ret
|
||||||
|
|
||||||
|
def SSL_alert_type_string_long(value):
|
||||||
|
try:
|
||||||
|
ret = _SSL_alert_type_string_long(value)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
return ret
|
||||||
|
|
||||||
|
def SSL_alert_desc_string_long(value):
|
||||||
|
try:
|
||||||
|
ret = _SSL_alert_desc_string_long(value)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
return ret
|
||||||
|
|
||||||
def OBJ_obj2txt(asn1_object, no_name):
|
def OBJ_obj2txt(asn1_object, no_name):
|
||||||
buf = create_string_buffer(X509_NAME_MAXLEN)
|
buf = create_string_buffer(X509_NAME_MAXLEN)
|
||||||
res_len = _OBJ_obj2txt(buf, sizeof(buf), asn1_object, 1 if no_name else 0)
|
res_len = _OBJ_obj2txt(buf, sizeof(buf), asn1_object, 1 if no_name else 0)
|
||||||
|
|
|
@ -84,6 +84,58 @@ DTLS_OPENSSL_VERSION_INFO = (
|
||||||
DTLS_OPENSSL_VERSION_NUMBER & 0xF) # status
|
DTLS_OPENSSL_VERSION_NUMBER & 0xF) # status
|
||||||
|
|
||||||
|
|
||||||
|
def _ssl_logging_cb(conn, where, return_code):
|
||||||
|
_state = where & ~SSL_ST_MASK
|
||||||
|
state = "SSL"
|
||||||
|
if _state & SSL_ST_INIT == SSL_ST_INIT:
|
||||||
|
if _state & SSL_ST_RENEGOTIATE == SSL_ST_RENEGOTIATE:
|
||||||
|
state += "_renew"
|
||||||
|
else:
|
||||||
|
state += "_init"
|
||||||
|
elif _state & SSL_ST_CONNECT:
|
||||||
|
state += "_connect"
|
||||||
|
elif _state & SSL_ST_ACCEPT:
|
||||||
|
state += "_accept"
|
||||||
|
elif _state == 0:
|
||||||
|
if where & SSL_CB_HANDSHAKE_START:
|
||||||
|
state += "_handshake_start"
|
||||||
|
elif where & SSL_CB_HANDSHAKE_DONE:
|
||||||
|
state += "_handshake_done"
|
||||||
|
|
||||||
|
if where & SSL_CB_LOOP:
|
||||||
|
state += '_loop'
|
||||||
|
_logger.debug("%s:%s:%d" % (state,
|
||||||
|
SSL_state_string_long(conn),
|
||||||
|
return_code))
|
||||||
|
|
||||||
|
elif where & SSL_CB_ALERT:
|
||||||
|
state += '_alert'
|
||||||
|
state += "_read" if where & SSL_CB_READ else "_write"
|
||||||
|
_logger.debug("%s:%s:%s" % (state,
|
||||||
|
SSL_alert_type_string_long(return_code),
|
||||||
|
SSL_alert_desc_string_long(return_code)))
|
||||||
|
|
||||||
|
elif where & SSL_CB_EXIT:
|
||||||
|
state += '_exit'
|
||||||
|
if return_code == 0:
|
||||||
|
_logger.debug("%s:%s:%d(failed)" % (state,
|
||||||
|
SSL_state_string_long(conn),
|
||||||
|
return_code))
|
||||||
|
elif return_code < 0:
|
||||||
|
_logger.debug("%s:%s:%d(error)" % (state,
|
||||||
|
SSL_state_string_long(conn),
|
||||||
|
return_code))
|
||||||
|
else:
|
||||||
|
_logger.debug("%s:%s:%d" % (state,
|
||||||
|
SSL_state_string_long(conn),
|
||||||
|
return_code))
|
||||||
|
|
||||||
|
else:
|
||||||
|
_logger.debug("%s:%s:%d" % (state,
|
||||||
|
SSL_state_string_long(conn),
|
||||||
|
return_code))
|
||||||
|
|
||||||
|
|
||||||
class _CTX(_Rsrc):
|
class _CTX(_Rsrc):
|
||||||
"""SSL_CTX wrapper"""
|
"""SSL_CTX wrapper"""
|
||||||
def __init__(self, value):
|
def __init__(self, value):
|
||||||
|
@ -209,6 +261,7 @@ class SSLConnection(object):
|
||||||
SSL_CTX_set_cipher_list(self._ctx.value, self._ciphers)
|
SSL_CTX_set_cipher_list(self._ctx.value, self._ciphers)
|
||||||
except openssl_error() as err:
|
except openssl_error() as err:
|
||||||
raise_ssl_error(ERR_NO_CIPHER, err)
|
raise_ssl_error(ERR_NO_CIPHER, err)
|
||||||
|
SSL_CTX_set_info_callback(self._ctx.value, _ssl_logging_cb)
|
||||||
|
|
||||||
def _copy_server(self):
|
def _copy_server(self):
|
||||||
source = self._sock
|
source = self._sock
|
||||||
|
|
Loading…
Reference in New Issue