diff options
Diffstat (limited to 'src/python')
| -rwxr-xr-x | src/python/botan2.py | 44 |
1 files changed, 43 insertions, 1 deletions
diff --git a/src/python/botan2.py b/src/python/botan2.py index 989a07160..715a60627 100755 --- a/src/python/botan2.py +++ b/src/python/botan2.py @@ -19,7 +19,7 @@ Botan 2.8 """ from ctypes import CDLL, POINTER, byref, create_string_buffer, \ - c_void_p, c_size_t, c_uint8, c_uint32, c_int, c_char, c_char_p + c_void_p, c_size_t, c_uint8, c_uint32, c_uint64, c_int, c_char, c_char_p from sys import version_info from time import strptime, mktime @@ -167,6 +167,9 @@ botan.botan_cipher_init.errcheck = errcheck_for('botan_cipher_init') botan.botan_cipher_destroy.argtypes = [c_void_p] botan.botan_cipher_destroy.errcheck = errcheck_for('botan_cipher_destroy') +botan.botan_cipher_reset.argtypes = [c_void_p] +botan.botan_cipher_reset.errcheck = errcheck_for('botan_cipher_reset') + botan.botan_cipher_name.argtypes = [c_void_p, POINTER(c_char), POINTER(c_size_t)] botan.botan_cipher_name.errcheck = errcheck_for('botan_cipher_name') @@ -498,6 +501,21 @@ botan.botan_fpe_decrypt.argtypes = [c_void_p, c_void_p, POINTER(c_char), c_size_ botan.botan_fpe_decrypt.errcheck = errcheck_for('botan_fpe_decrypt') # +# HOTP +# +botan.botan_hotp_init.argtype = [c_void_p, POINTER(c_char), c_size_t, c_char_p, c_size_t] +botan.botan_hotp_init.errcheck = errcheck_for('botan_hotp_init') + +botan.botan_hotp_destroy.argtype = [c_void_p] +botan.botan_hotp_destroy.errcheck = errcheck_for('botan_hotp_destroy') + +botan.botan_hotp_generate.argtype = [c_void_p, POINTER(c_uint32), c_uint64] +botan.botan_hotp_generate.errcheck = errcheck_for('botan_hotp_generate') + +botan.botan_hotp_check.argtype = [c_void_p, POINTER(c_uint64), c_uint32, c_uint64, c_size_t] +botan.botan_hotp_check.errcheck = errcheck_for('botan_hotp_check') + +# # Internal utilities # def _call_fn_returning_vec(guess, fn): @@ -733,6 +751,9 @@ class SymmetricCipher(object): rc = botan.botan_cipher_valid_nonce_length(self.__obj, nonce_len) return True if rc == 1 else False + def reset(self): + botan.botan_cipher_reset(self.__obj) + def clear(self): botan.botan_cipher_clear(self.__obj) @@ -1312,6 +1333,27 @@ class FormatPreservingEncryptionFE1(object): botan.botan_fpe_decrypt(self.__obj, r.handle_(), _ctype_bits(tweak), len(tweak)) return r +class HOTP(object): + def __init__(self, key, digest="SHA-1", digits=6): + self.__obj = c_void_p(0) + botan.botan_hotp_init(byref(self.__obj), key, len(key), _ctype_str(digest), digits) + + def __del__(self): + botan.botan_hotp_destroy(self.__obj) + + def generate(self, counter): + code = c_uint32(0) + botan.botan_hotp_generate(self.__obj, byref(code), counter) + return code.value + + def check(self, code, counter, resync_range=0): + next = c_uint64(0) + rc = botan.botan_hotp_check(self.__obj, byref(next), code, counter, resync_range) + if rc == 0: + return (True,next.value) + else: + return (False,counter) + # Typedefs for compat with older versions cipher = SymmetricCipher # pylint: disable=invalid-name rng = RandomNumberGenerator # pylint: disable=invalid-name |