From 3202d88d62f1d67c6227fb0757dde96f718183f2 Mon Sep 17 00:00:00 2001 From: Uri Blumenthal Date: Fri, 14 Aug 2015 16:02:41 -0400 Subject: Ported python demo script to python-3 --- src/python/botan.py | 190 +++++++++++++++++++++++++++++++++++----------------- 1 file changed, 129 insertions(+), 61 deletions(-) (limited to 'src') diff --git a/src/python/botan.py b/src/python/botan.py index fe8600e19..abfc07fd7 100755 --- a/src/python/botan.py +++ b/src/python/botan.py @@ -12,6 +12,8 @@ Botan is released under the Simplified BSD License (see license.txt) import sys from ctypes import * +from binascii import hexlify, unhexlify +import base64 """ Module initialization @@ -48,7 +50,11 @@ class rng(object): def __init__(self, rng_type = 'system'): botan.botan_rng_init.argtypes = [c_void_p, c_char_p] self.rng = c_void_p(0) - rc = botan.botan_rng_init(byref(self.rng), rng_type) + if sys.version_info[0] < 3: + rc = botan.botan_rng_init(byref(self.rng), rng_type) + else: + rc = botan.botan_rng_init(byref(self.rng), rng_type.encode('ascii')) + if rc != 0 or self.rng is None: raise Exception("No rng " + algo + " for you!") @@ -65,7 +71,10 @@ class rng(object): out = create_string_buffer(length) l = c_size_t(length) rc = botan.botan_rng_get(self.rng, out, l) - return str(out.raw) + if sys.version_info[0] < 3: + return str(out.raw) + else: + return out.raw """ Hash function @@ -75,7 +84,10 @@ class hash_function(object): botan.botan_hash_init.argtypes = [c_void_p, c_char_p, c_uint32] flags = c_uint32(0) # always zero in this API version self.hash = c_void_p(0) - rc = botan.botan_hash_init(byref(self.hash), algo, flags) + if sys.version_info[0] < 3: + rc = botan.botan_hash_init(byref(self.hash), algo, flags) + else: + rc = botan.botan_hash_init(byref(self.hash), algo.encode('utf-8'), flags) if rc != 0 or self.hash is None: raise Exception("No hash " + algo + " for you!") @@ -101,7 +113,10 @@ class hash_function(object): botan.botan_hash_final.argtypes = [c_void_p, POINTER(c_char)] out = create_string_buffer(self.output_length()) botan.botan_hash_final(self.hash, out) - return str(out.raw) + if sys.version_info[0] < 3: + return str(out.raw) + else: + return out.raw """ Message authentication codes @@ -141,14 +156,20 @@ class message_authentication_code(object): botan.botan_mac_final.argtypes = [c_void_p, POINTER(c_char)] out = create_string_buffer(self.output_length()) botan.botan_mac_final(self.mac, out) - return str(out.raw) + if sys.version_info[0] < 3: + return str(out.raw) + else: + return out.raw class cipher(object): def __init__(self, algo, encrypt = True): botan.botan_cipher_init.argtypes = [c_void_p,c_char_p, c_uint32] flags = 0 if encrypt else 1 self.cipher = c_void_p(0) - rc = botan.botan_cipher_init(byref(self.cipher), algo, flags) + if sys.version_info[0] < 3: + rc = botan.botan_cipher_init(byref(self.cipher), algo, flags) + else: + rc = botan.botan_cipher_init(byref(self.cipher), algo.encode('utf-8'), flags) if rc != 0 or self.cipher is None: raise Exception("No cipher " + algo + " for you!") @@ -207,6 +228,8 @@ class cipher(object): inp = txt if txt else '' inp_sz = c_size_t(len(inp)) + if sys.version_info[0] >= 3: + inp = cast(inp, c_char_p) inp_consumed = c_size_t(0) out = create_string_buffer(inp_sz.value + (self.tag_length() if final else 0)) out_sz = c_size_t(len(out)) @@ -310,8 +333,11 @@ class public_key(object): n = hash_function(hash).output_length() buf = create_string_buffer(n) buf_len = c_size_t(n) + if sys.version_info[0] > 2: + hash = hash.encode('utf-8') + botan.botan_pubkey_fingerprint(self.pubkey, hash, buf, byref(buf_len)) - return buf[0:buf_len.value].encode('hex') + return hexlify(buf[0:buf_len.value]) class private_key(object): def __init__(self, alg, param, rng): @@ -362,6 +388,8 @@ class pk_op_encrypt(object): botan.botan_pk_op_encrypt_create.argtypes = [c_void_p, c_void_p, c_char_p, c_uint32] self.op = c_void_p(0) flags = c_uint32(0) # always zero in this ABI + if sys.version_info[0] > 2: + padding = cast(padding, c_char_p) botan.botan_pk_op_encrypt_create(byref(self.op), key.pubkey, padding, flags) if not self.op: raise Exception("No pk op for you") @@ -377,14 +405,23 @@ class pk_op_encrypt(object): outbuf_sz = c_size_t(4096) #?!?! outbuf = create_string_buffer(outbuf_sz.value) - botan.botan_pk_op_encrypt(self.op, rng.rng, outbuf, byref(outbuf_sz), msg, len(msg)) + ll = len(msg) + #print("encrypt: len=%d" % ll) + if sys.version_info[0] > 2: + msg = cast(msg, c_char_p) + ll = c_size_t(ll) + botan.botan_pk_op_encrypt(self.op, rng.rng, outbuf, byref(outbuf_sz), msg, ll) + #print("encrypt: outbuf_sz.value=%d" % outbuf_sz.value) return outbuf.raw[0:outbuf_sz.value] + class pk_op_decrypt(object): def __init__(self, key, padding): botan.botan_pk_op_decrypt_create.argtypes = [c_void_p, c_void_p, c_char_p, c_uint32] self.op = c_void_p(0) flags = c_uint32(0) # always zero in this ABI + if sys.version_info[0] > 2: + padding = cast(padding, c_char_p) botan.botan_pk_op_decrypt_create(byref(self.op), key.privkey, padding, flags) if not self.op: raise Exception("No pk op for you") @@ -400,7 +437,11 @@ class pk_op_decrypt(object): outbuf_sz = c_size_t(4096) #?!?! outbuf = create_string_buffer(outbuf_sz.value) - botan.botan_pk_op_decrypt(self.op, outbuf, byref(outbuf_sz), msg, len(msg)) + ll = len(msg) + if sys.version_info[0] > 2: + msg = cast(msg, c_char_p) + ll = c_size_t(ll) + botan.botan_pk_op_decrypt(self.op, outbuf, byref(outbuf_sz), msg, ll) return outbuf.raw[0:outbuf_sz.value] class pk_op_sign(object): @@ -408,6 +449,8 @@ class pk_op_sign(object): botan.botan_pk_op_sign_create.argtypes = [c_void_p, c_void_p, c_char_p, c_uint32] self.op = c_void_p(0) flags = c_uint32(0) # always zero in this ABI + if sys.version_info[0] > 2: + padding = cast(padding, c_char_p) botan.botan_pk_op_sign_create(byref(self.op), key.privkey, padding, flags) if not self.op: raise Exception("No pk op for you") @@ -495,46 +538,65 @@ def test(): r = rng("user") - print("\n%s" % version_string()) + print("\n%s" % version_string().decode('utf-8')) print("v%d.%d.%d\n" % (version_major(), version_minor(), version_patch())) print("KDF2(SHA-1) %s" % - kdf('KDF2(SHA-1)', '701F3480DFE95F57941F804B1B2413EF'.decode('hex'), 7, - '55A4E9DD5F4CA2EF82'.decode('hex')).encode('hex')) + base64.b16encode(kdf('KDF2(SHA-1)'.encode('ascii'), unhexlify('701F3480DFE95F57941F804B1B2413EF'), 7, + unhexlify('55A4E9DD5F4CA2EF82')) + ).decode('ascii') + ) print("PBKDF2(SHA-1) %s" % - pbkdf('PBKDF2(SHA-1)', '', 32, 10000, - '0001020304050607'.decode('hex'))[2].encode('hex').upper()) + hexlify(pbkdf('PBKDF2(SHA-1)'.encode('ascii'), ''.encode('ascii'), 32, 10000, + unhexlify('0001020304050607')) + [2] + ).upper().decode('ascii')) print("good output %s\n" % '59B2B1143B4CB1059EC58D9722FB1C72471E0D85C6F7543BA5228526375B0127') - (salt,iterations,psk) = pbkdf_timed('PBKDF2(SHA-256)', 'xyz', 32, 200) - print("PBKDF2(SHA-256) x=timed, y=iterated; salt = %s #iterations = %d\n" % - (salt.encode('hex'), iterations)) - print('x %s' % psk.encode('hex')) + + + (salt,iterations,psk) = pbkdf_timed('PBKDF2(SHA-256)'.encode('ascii'), + 'xyz'.encode('utf-8'), 32, 200) + + if sys.version_info[0] < 3: + print("PBKDF2(SHA-256) x=timed, y=iterated; salt = %s (len=%d) #iterations = %d\n" % + (hexlify(salt), len(salt), iterations) ) + else: + print("PBKDF2(SHA-256) x=timed, y=iterated; salt = %s (len=%d) #iterations = %d\n" % + (base64.b16encode(salt).decode('ascii'), len(salt), iterations) ) + + print('x %s' % hexlify(psk).decode('utf-8')) print('y %s\n' % - (pbkdf('PBKDF2(SHA-256)', 'xyz', 32, iterations, salt)[2].encode('hex'))) + (hexlify(pbkdf('PBKDF2(SHA-256)'.encode('utf-8'), + 'xyz'.encode('ascii'), 32, iterations, salt)[2]).decode('utf-8'))) - hmac = message_authentication_code('HMAC(SHA-256)') - hmac.set_key('0102030405060708090A0B0C0D0E0F101112131415161718191A1B1C1D1E1F20'.decode('hex')) - hmac.update('616263'.decode('hex')) + hmac = message_authentication_code('HMAC(SHA-256)'.encode('ascii')) + hmac.set_key(unhexlify('0102030405060708090A0B0C0D0E0F101112131415161718191A1B1C1D1E1F20')) + hmac.update(unhexlify('616263')) hmac_output = hmac.final() - - if hmac_output != 'A21B1F5D4CF4F73A4DD939750F7A066A7F98CC131CB16A6692759021CFAB8181'.decode('hex'): - print('Bad HMAC\n') + + if hmac_output != unhexlify('A21B1F5D4CF4F73A4DD939750F7A066A7F98CC131CB16A6692759021CFAB8181'): + print("Bad HMAC:\t%s" % hexlify(bytes(hmac_output, 'utf-8')).decode('utf-8')) + print("vs good: \tA21B1F5D4CF4F73A4DD939750F7A066A7F98CC131CB16A6692759021CFAB8181"); else: - print("HMAC output (good): %s\n" % hmac_output.encode('hex')) + print("HMAC output (good): %s\n" % hexlify(hmac_output).decode('utf-8')) print("rng output:\n\t%s\n\t%s\n\t%s\n" % - (r.get(42).encode('hex'), r.get(13).encode('hex'), r.get(9).encode('hex'))) + (hexlify(r.get(42)).decode('utf-8'), + hexlify(r.get(13)).decode('utf-8'), + hexlify(r.get(9)).decode('utf-8') + ) + ) h = hash_function('MD5') assert h.output_length() == 16 - h.update('h') - h.update('i') - print("md5 hash: %s\n" % (h.final().encode('hex'))) + h.update('h'.encode('utf-8')) + h.update('i'.encode('utf-8')) + print("md5 hash: %s\n" % (hexlify(h.final())).decode('utf-8')) gcm = cipher('AES-128/GCM') @@ -549,13 +611,13 @@ def test(): gcm.start(iv) assert len(gcm.update('')) == 0 ct = gcm.finish(pt) - print("GCM ct %s" % ct.encode('hex')) + print("GCM ct %s" % hexlify(ct).decode('utf-8')) gcm_dec.set_key(key) gcm_dec.start(iv) dec = gcm_dec.finish(ct) - print("GCM pt %s %d" % (pt.encode('hex'), len(pt))) - print("GCM de %s %d\n" % (dec.encode('hex'), len(dec))) + print("GCM pt %s %d" % (hexlify(pt).decode('utf-8'), len(pt))) + print("GCM de %s %d\n" % (hexlify(dec).decode('utf-8'), len(dec))) ocb = cipher('AES-128/OCB') print("AES-128/OCB: default nonce=%d update_size=%d" % @@ -569,54 +631,58 @@ def test(): ocb.start(iv) assert len(ocb.update('')) == 0 ct = ocb.finish(pt) - print("OCB ct %s" % ct.encode('hex')) + print("OCB ct %s" % hexlify(ct).decode('utf-8')) ocb_dec.set_key(key) ocb_dec.start(iv) dec = ocb_dec.finish(ct) - print("OCB pt %s %d" % (pt.encode('hex'), len(pt))) - print("OCB de %s %d\n" % (dec.encode('hex'), len(dec))) + print("OCB pt %s %d" % (hexlify(pt).decode('utf-8'), len(pt))) + print("OCB de %s %d\n" % (hexlify(dec).decode('utf-8'), len(dec))) rsapriv = private_key('rsa', 1536, r) - dec = pk_op_decrypt(rsapriv, "EME1(SHA-256)") - rsapub = rsapriv.get_public_key() + print("rsapub %s/SHA-1 fingerprint: %s (estimated strength %s)" % - (rsapub.algo_name(), rsapub.fingerprint("SHA-1"), rsapub.estimated_strength())) + (rsapub.algo_name().decode('utf-8'), rsapub.fingerprint("SHA-1").decode('utf-8'), + rsapub.estimated_strength() + ) + ) - enc = pk_op_encrypt(rsapub, "EME1(SHA-256)") + dec = pk_op_decrypt(rsapriv, "EME1(SHA-256)".encode('utf-8')) + enc = pk_op_encrypt(rsapub, "EME1(SHA-256)".encode('utf-8')) - ctext = enc.encrypt('foof', r) + ctext = enc.encrypt('foof'.encode('utf-8'), r) print("ptext \'%s\'" % 'foof') - print("ctext \'%s\'" % ctext.encode('hex')) - print("decrypt \'%s\'\n" % dec.decrypt(ctext)) + print("ctext \'%s\'" % hexlify(ctext).decode('utf-8')) + print("decrypt \'%s\'\n" % dec.decrypt(ctext).decode('utf-8')) - signer = pk_op_sign(rsapriv, 'EMSA4(SHA-384)') + signer = pk_op_sign(rsapriv, 'EMSA4(SHA-384)'.encode('utf-8')) - signer.update('messa') - signer.update('ge') + signer.update('messa'.encode('utf-8')) + signer.update('ge'.encode('utf-8')) sig = signer.finish(r) r.reseed(200) - print("EMSA4(SHA-384) signature: %s" % sig.encode('hex')) + print("EMSA4(SHA-384) signature: %s" % hexlify(sig).decode('utf-8')) - verify = pk_op_verify(rsapub, 'EMSA4(SHA-384)') + + verify = pk_op_verify(rsapub, 'EMSA4(SHA-384)'.encode('utf-8')) - verify.update('mess') - verify.update('age') + verify.update('mess'.encode('utf-8')) + verify.update('age'.encode('utf-8')) print("good sig accepted? %s" % verify.check_signature(sig)) - verify.update('mess of things') - verify.update('age') + verify.update('mess of things'.encode('utf-8')) + verify.update('age'.encode('utf-8')) print("bad sig accepted? %s" % verify.check_signature(sig)) - verify.update('message') + verify.update('message'.encode('utf-8')) print("good sig accepted? %s\n" % verify.check_signature(sig)) - dh_grp = 'secp256r1' - #dh_grp = 'curve25519' - dh_kdf = 'KDF2(SHA-384)' + dh_grp = 'secp256r1'.encode('utf-8') + #dh_grp = 'curve25519'.encode('utf-8') + dh_kdf = 'KDF2(SHA-384)'.encode('utf-8') a_dh_priv = private_key('ecdh', dh_grp, r) a_dh_pub = a_dh_priv.get_public_key() @@ -626,13 +692,15 @@ def test(): a_dh = pk_op_key_agreement(a_dh_priv, dh_kdf) b_dh = pk_op_key_agreement(b_dh_priv, dh_kdf) - print("dh pubs\n\t%s\n\t%s\n" % - (a_dh.public_value().encode('hex'), b_dh.public_value().encode('hex'))) + print("ecdh pubs:\n %s\n %s\n" % + (hexlify(a_dh.public_value()).decode('utf-8'), + hexlify(b_dh.public_value()).decode('utf-8'))) - a_key = a_dh.agree(b_dh.public_value(), 20, 'salt') - b_key = b_dh.agree(a_dh.public_value(), 20, 'salt') + a_key = a_dh.agree(b_dh.public_value(), 20, 'salt'.encode('utf-8')) + b_key = b_dh.agree(a_dh.public_value(), 20, 'salt'.encode('utf-8')) - print("dh shared\n\t%s\n\t%s\n" % (a_key.encode('hex'), b_key.encode('hex'))) + print("ecdh shared:\n %s\n %s\n" % + (hexlify(a_key).decode('utf-8'), hexlify(b_key).decode('utf-8'))) #f = open('key.ber','wb') -- cgit v1.2.3