arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-22 23:48:12 +00:00
Code Issues Releases Wiki Activity

embed: hdnode public derivation

Browse Source
This commit is contained in:
Tomas Susanka 2018-01-26 13:53:32 +01:00 committed by Jan Pochyla
parent 858f2583a5
commit 368ba963a6
1 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
embed/extmod/modtrezorcrypto/modtrezorcrypto-bip32.h
View File

@ -31,6 +31,7 @@ typedef struct _mp_obj_HDNode_t {
mp_obj_base_t base; mp_obj_base_t base;
uint32_t fingerprint; uint32_t fingerprint;
HDNode hdnode; HDNode hdnode;
bool private_key_set;
} mp_obj_HDNode_t; } mp_obj_HDNode_t;
STATIC const mp_obj_type_t mod_trezorcrypto_HDNode_type; STATIC const mp_obj_type_t mod_trezorcrypto_HDNode_type;
@ -110,8 +111,10 @@ STATIC mp_obj_t mod_trezorcrypto_HDNode_make_new(const mp_obj_type_t *type, size
} }
if (32 == private_key.len) { if (32 == private_key.len) {
memcpy(o->hdnode.private_key, private_key.buf, 32); memcpy(o->hdnode.private_key, private_key.buf, 32);
o->private_key_set = true;
} else { } else {
memzero(o->hdnode.private_key, 32); memzero(o->hdnode.private_key, 32);
o->private_key_set = false;
} }
if (33 == public_key.len) { if (33 == public_key.len) {
memcpy(o->hdnode.public_key, public_key.buf, 33); memcpy(o->hdnode.public_key, public_key.buf, 33);
@ -123,16 +126,27 @@ STATIC mp_obj_t mod_trezorcrypto_HDNode_make_new(const mp_obj_type_t *type, size
return MP_OBJ_FROM_PTR(o); return MP_OBJ_FROM_PTR(o);
} }
/// def derive(self, index: int) -> None: /// def derive(self, index: int, public: bool=False) -> None:
/// ''' /// '''
/// Derive a BIP0032 child node in place. /// Derive a BIP0032 child node in place.
/// ''' /// '''
STATIC mp_obj_t mod_trezorcrypto_HDNode_derive(mp_obj_t self, mp_obj_t index) { STATIC mp_obj_t mod_trezorcrypto_HDNode_derive(size_t n_args, const mp_obj_t *args) {
mp_obj_HDNode_t *o = MP_OBJ_TO_PTR(self); mp_obj_HDNode_t *o = MP_OBJ_TO_PTR(args[0]);
uint32_t i = mp_obj_get_int_truncated(index); uint32_t i = mp_obj_get_int_truncated(args[1]);
uint32_t fp = hdnode_fingerprint(&o->hdnode); uint32_t fp = hdnode_fingerprint(&o->hdnode);
bool public = n_args > 2 && args[2] == mp_const_true;
if (!hdnode_private_ckd(&o->hdnode, i)) { int res;
if (public) {
res = hdnode_public_ckd(&o->hdnode, i);
} else {
if (!o->private_key_set) {
memzero(&o->hdnode, sizeof(o->hdnode));
mp_raise_ValueError("Failed to derive, private key not set");
}
res = hdnode_private_ckd(&o->hdnode, i);
}
if (!res) {
memzero(&o->hdnode, sizeof(o->hdnode)); memzero(&o->hdnode, sizeof(o->hdnode));
mp_raise_ValueError("Failed to derive"); mp_raise_ValueError("Failed to derive");
} }
@ -140,7 +154,7 @@ STATIC mp_obj_t mod_trezorcrypto_HDNode_derive(mp_obj_t self, mp_obj_t index) {
return mp_const_none; return mp_const_none;
} }
STATIC MP_DEFINE_CONST_FUN_OBJ_2(mod_trezorcrypto_HDNode_derive_obj, mod_trezorcrypto_HDNode_derive); STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mod_trezorcrypto_HDNode_derive_obj, 2, 3, mod_trezorcrypto_HDNode_derive);
/// def derive_path(self, path: List[int]) -> None: /// def derive_path(self, path: List[int]) -> None:
/// ''' /// '''