arno
/
trezor-firmware
mirror of https://github.com/trezor/trezor-firmware.git
1
0
Fork 0
Code Issues Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '54d348228f'
${ noResults }
trezor-firmware/core/embed/extmod/modtrezorcrypto/modtrezorcrypto-monero.h

1490 lines
53 KiB
Raw Blame History

/*
* This file is part of the Trezor project, https://trezor.io/
*
* Copyright (c) SatoshiLabs
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "py/mpz.h"
#include "py/objint.h"
#include "py/objstr.h"
#include "bignum.h"
#include "memzero.h"
#include "monero/monero.h"
/// package: trezorcrypto.monero
typedef struct _mp_obj_hasher_t {
mp_obj_base_t base;
Hasher h;
} mp_obj_hasher_t;
typedef struct _mp_obj_ge25519_t {
mp_obj_base_t base;
ge25519 p;
} mp_obj_ge25519_t;
typedef struct _mp_obj_bignum256modm_t {
mp_obj_base_t base;
bignum256modm p;
} mp_obj_bignum256modm_t;
//
// Helpers
//
STATIC const mp_obj_type_t mod_trezorcrypto_monero_ge25519_type;
STATIC const mp_obj_type_t mod_trezorcrypto_monero_bignum256modm_type;
STATIC const mp_obj_type_t mod_trezorcrypto_monero_hasher_type;
#define MP_OBJ_IS_GE25519(o) \
MP_OBJ_IS_TYPE((o), &mod_trezorcrypto_monero_ge25519_type)
#define MP_OBJ_IS_SCALAR(o) \
MP_OBJ_IS_TYPE((o), &mod_trezorcrypto_monero_bignum256modm_type)
#define MP_OBJ_PTR_MPC_GE25519(o) ((const mp_obj_ge25519_t *)(o))
#define MP_OBJ_PTR_MPC_SCALAR(o) ((const mp_obj_bignum256modm_t *)(o))
#define MP_OBJ_PTR_MP_GE25519(o) ((mp_obj_ge25519_t *)(o))
#define MP_OBJ_PTR_MP_SCALAR(o) ((mp_obj_bignum256modm_t *)(o))
#define MP_OBJ_C_GE25519(o) (MP_OBJ_PTR_MPC_GE25519(o)->p)
#define MP_OBJ_GE25519(o) (MP_OBJ_PTR_MP_GE25519(o)->p)
#define MP_OBJ_C_SCALAR(o) (MP_OBJ_PTR_MPC_SCALAR(o)->p)
#define MP_OBJ_SCALAR(o) (MP_OBJ_PTR_MP_SCALAR(o)->p)
STATIC inline void assert_ge25519(const mp_obj_t o) {
if (!MP_OBJ_IS_GE25519(o)) {
mp_raise_ValueError("ge25519 expected");
}
}
STATIC inline void assert_scalar(const mp_obj_t o) {
if (!MP_OBJ_IS_SCALAR(o)) {
mp_raise_ValueError("scalar expected");
}
}
static uint64_t mp_obj_uint64_get_checked(mp_const_obj_t self_in) {
#if MICROPY_LONGINT_IMPL != MICROPY_LONGINT_IMPL_MPZ
#error "MPZ supported only"
#endif
if (MP_OBJ_IS_SMALL_INT(self_in)) {
return MP_OBJ_SMALL_INT_VALUE(self_in);
} else {
byte buff[8];
uint64_t res = 0;
mp_obj_t *o = MP_OBJ_TO_PTR(self_in);
mp_obj_int_to_bytes_impl(o, true, 8, buff);
for (int i = 0; i < 8; i++) {
res <<= i > 0 ? 8 : 0;
res |= (uint64_t)(buff[i] & 0xff);
}
return res;
}
}
static uint64_t mp_obj_get_uint64(mp_const_obj_t arg) {
if (arg == mp_const_false) {
return 0;
} else if (arg == mp_const_true) {
return 1;
} else if (MP_OBJ_IS_SMALL_INT(arg)) {
return MP_OBJ_SMALL_INT_VALUE(arg);
} else if (MP_OBJ_IS_TYPE(arg, &mp_type_int)) {
return mp_obj_uint64_get_checked(arg);
} else {
if (MICROPY_ERROR_REPORTING == MICROPY_ERROR_REPORTING_TERSE) {
mp_raise_TypeError("can't convert to int");
} else {
nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_TypeError,
"can't convert %s to int",
mp_obj_get_type_str(arg)));
}
}
}
STATIC mp_obj_t mp_obj_new_scalar() {
mp_obj_bignum256modm_t *o = m_new_obj(mp_obj_bignum256modm_t);
o->base.type = &mod_trezorcrypto_monero_bignum256modm_type;
set256_modm(o->p, 0);
return MP_OBJ_FROM_PTR(o);
}
STATIC mp_obj_t mp_obj_new_scalar_r(mp_obj_t r) {
if (r == mp_const_none) {
return mp_obj_new_scalar();
}
assert_scalar(r);
return r;
}
STATIC mp_obj_t mp_obj_new_ge25519() {
mp_obj_ge25519_t *o = m_new_obj(mp_obj_ge25519_t);
o->base.type = &mod_trezorcrypto_monero_ge25519_type;
ge25519_set_neutral(&o->p);
return MP_OBJ_FROM_PTR(o);
}
STATIC mp_obj_t mp_obj_new_ge25519_r(mp_obj_t r) {
if (r == mp_const_none) {
return mp_obj_new_ge25519();
}
assert_ge25519(r);
return r;
}
STATIC void mp_unpack_ge25519(ge25519 *r, const mp_obj_t arg, mp_int_t offset) {
mp_buffer_info_t buff;
mp_get_buffer_raise(arg, &buff, MP_BUFFER_READ);
if (buff.len < 32 + offset) {
mp_raise_ValueError("Invalid length of the EC point");
}
const int res = ge25519_unpack_vartime(r, ((uint8_t *)buff.buf) + offset);
if (res != 1) {
mp_raise_ValueError("Point decoding error");
}
}
STATIC void mp_unpack_scalar(bignum256modm r, const mp_obj_t arg,
mp_int_t offset) {
mp_buffer_info_t buff;
mp_get_buffer_raise(arg, &buff, MP_BUFFER_READ);
if (buff.len < 32 + offset) {
mp_raise_ValueError("Invalid length of secret key");
}
expand256_modm(r, ((uint8_t *)buff.buf) + offset, 32);
}
//
// Constructors
//
/// class Ge25519:
/// """
/// EC point on ED25519
/// """
///
/// def __init__(x: Optional[Union[Ge25519, bytes]] = None):
/// """
/// Constructor
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519_make_new(
const mp_obj_type_t *type, size_t n_args, size_t n_kw,
const mp_obj_t *args) {
mp_arg_check_num(n_args, n_kw, 0, 1, false);
mp_obj_ge25519_t *o = m_new_obj(mp_obj_ge25519_t);
o->base.type = type;
if (n_args == 0 || args[0] == mp_const_none) {
ge25519_set_neutral(&o->p);
} else if (n_args == 1 && MP_OBJ_IS_GE25519(args[0])) {
ge25519_copy(&o->p, &MP_OBJ_C_GE25519(args[0]));
} else if (n_args == 1 && MP_OBJ_IS_STR_OR_BYTES(args[0])) {
mp_unpack_ge25519(&o->p, args[0], 0);
} else {
mp_raise_ValueError("Invalid ge25519 constructor");
}
return MP_OBJ_FROM_PTR(o);
}
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519___del__(mp_obj_t self) {
mp_obj_ge25519_t *o = MP_OBJ_TO_PTR(self);
memzero(&(o->p), sizeof(ge25519));
return mp_const_none;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_trezorcrypto_monero_ge25519___del___obj,
mod_trezorcrypto_monero_ge25519___del__);
/// class Sc25519:
/// """
/// EC scalar on SC25519
/// """
///
/// def __init__(x: Optional[Union[Sc25519, bytes, int]] = None):
/// """
/// Constructor
/// """
///
///
STATIC mp_obj_t mod_trezorcrypto_monero_bignum256modm_make_new(
const mp_obj_type_t *type, size_t n_args, size_t n_kw,
const mp_obj_t *args) {
mp_arg_check_num(n_args, n_kw, 0, 1, false);
mp_obj_bignum256modm_t *o = m_new_obj(mp_obj_bignum256modm_t);
o->base.type = type;
if (n_args == 0 || args[0] == mp_const_none) {
set256_modm(o->p, 0);
} else if (n_args == 1 && MP_OBJ_IS_SCALAR(args[0])) {
copy256_modm(o->p, MP_OBJ_C_SCALAR(args[0]));
} else if (n_args == 1 && MP_OBJ_IS_STR_OR_BYTES(args[0])) {
mp_unpack_scalar(o->p, args[0], 0);
} else if (n_args == 1 && mp_obj_is_integer(args[0])) {
uint64_t v = mp_obj_get_uint64(args[0]);
set256_modm(o->p, v);
} else {
mp_raise_ValueError("Invalid scalar constructor");
}
return MP_OBJ_FROM_PTR(o);
}
STATIC mp_obj_t mod_trezorcrypto_monero_bignum256modm___del__(mp_obj_t self) {
mp_obj_bignum256modm_t *o = MP_OBJ_TO_PTR(self);
memzero(o->p, sizeof(bignum256modm));
return mp_const_none;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(
mod_trezorcrypto_monero_bignum256modm___del___obj,
mod_trezorcrypto_monero_bignum256modm___del__);
/// class Hasher:
/// """
/// XMR hasher
/// """
///
/// def __init__(x: Optional[bytes] = None):
/// """
/// Constructor
/// """
///
/// def update(buffer: bytes):
/// """
/// Update hasher
/// """
///
/// def digest() -> bytes:
/// """
/// Computes digest
/// """
///
/// def copy() -> Hasher:
/// """
/// Creates copy of the hasher, preserving the state
/// """
///
///
STATIC mp_obj_t mod_trezorcrypto_monero_hasher_make_new(
const mp_obj_type_t *type, size_t n_args, size_t n_kw,
const mp_obj_t *args) {
mp_arg_check_num(n_args, n_kw, 0, 1, false);
mp_obj_hasher_t *o = m_new_obj(mp_obj_hasher_t);
o->base.type = type;
xmr_hasher_init(&(o->h));
if (n_args == 1 && MP_OBJ_IS_STR_OR_BYTES(args[0])) {
mp_buffer_info_t buff;
mp_get_buffer_raise(args[0], &buff, MP_BUFFER_READ);
xmr_hasher_update(&o->h, buff.buf, buff.len);
}
return MP_OBJ_FROM_PTR(o);
}
STATIC mp_obj_t mod_trezorcrypto_monero_hasher___del__(mp_obj_t self) {
mp_obj_hasher_t *o = MP_OBJ_TO_PTR(self);
memzero(&(o->h), sizeof(Hasher));
return mp_const_none;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_trezorcrypto_monero_hasher___del___obj,
mod_trezorcrypto_monero_hasher___del__);
//
// Scalar defs
//
/// mock:global
/// def init256_modm(
/// dst: Optional[Sc25519], val: Union[int, bytes, Sc25519]
/// ) -> Sc25519:
/// """
/// Initializes Sc25519 scalar
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_init256_modm(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 2;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_scalar_r(res_arg ? args[0] : mp_const_none);
if (n_args == 0 || args[0] == mp_const_none) {
set256_modm(MP_OBJ_SCALAR(res), 0);
} else if (n_args > 0 && MP_OBJ_IS_SCALAR(args[1 + off])) {
copy256_modm(MP_OBJ_SCALAR(res), MP_OBJ_C_SCALAR(args[1 + off]));
} else if (n_args > 0 && MP_OBJ_IS_STR_OR_BYTES(args[1 + off])) {
mp_unpack_scalar(MP_OBJ_SCALAR(res), args[1 + off], 0);
} else if (n_args > 0 && mp_obj_is_integer(args[1 + off])) {
uint64_t v = mp_obj_get_uint64(args[1 + off]);
set256_modm(MP_OBJ_SCALAR(res), v);
} else {
mp_raise_ValueError("Invalid scalar def");
}
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_init256_modm_obj, 0, 2,
mod_trezorcrypto_monero_init256_modm);
/// def check256_modm(val: Sc25519):
/// """
/// Throws exception if scalar is invalid
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_check256_modm(const mp_obj_t arg) {
assert_scalar(arg);
if (check256_modm(MP_OBJ_C_SCALAR(arg)) != 1) {
mp_raise_ValueError("Ed25519 scalar invalid");
}
return mp_const_none;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_trezorcrypto_monero_check256_modm_obj,
mod_trezorcrypto_monero_check256_modm);
/// def iszero256_modm(val: Sc25519) -> bool:
/// """
/// Returns False if the scalar is zero
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_iszero256_modm(const mp_obj_t arg) {
assert_scalar(arg);
const int r = iszero256_modm(MP_OBJ_C_SCALAR(arg));
return mp_obj_new_int(r);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_trezorcrypto_monero_iszero256_modm_obj,
mod_trezorcrypto_monero_iszero256_modm);
/// def eq256_modm(a: Sc25519, b: Sc25519) -> int:
/// """
/// Compares scalars, returns 1 on the same value
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_eq256_modm(const mp_obj_t a,
const mp_obj_t b) {
assert_scalar(a);
assert_scalar(b);
int r = eq256_modm(MP_OBJ_C_SCALAR(a), MP_OBJ_C_SCALAR(b));
return MP_OBJ_NEW_SMALL_INT(r);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_2(mod_trezorcrypto_monero_eq256_modm_obj,
mod_trezorcrypto_monero_eq256_modm);
/// def get256_modm(a: Sc25519) -> int:
/// """
/// Extracts 64bit integer from the scalar. Raises exception if scalar is
/// bigger than 2^64
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_get256_modm(const mp_obj_t arg) {
assert_scalar(arg);
uint64_t v;
if (!get256_modm(&v, MP_OBJ_C_SCALAR(arg))) {
mp_raise_ValueError("Ed25519 scalar too big");
}
return mp_obj_new_int_from_ull(v);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_trezorcrypto_monero_get256_modm_obj,
mod_trezorcrypto_monero_get256_modm);
/// def add256_modm(r: Optional[Sc25519], a: Sc25519, b: Sc25519) -> Sc25519:
/// """
/// Scalar addition
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_add256_modm(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 3;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_scalar_r(res_arg ? args[0] : mp_const_none);
assert_scalar(args[1 + off]);
assert_scalar(args[2 + off]);
add256_modm(MP_OBJ_SCALAR(res), MP_OBJ_C_SCALAR(args[1 + off]),
MP_OBJ_C_SCALAR(args[2 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_add256_modm_obj, 2, 3,
mod_trezorcrypto_monero_add256_modm);
/// def sub256_modm(r: Optional[Sc25519], a: Sc25519, b: Sc25519) -> Sc25519:
/// """
/// Scalar subtraction
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_sub256_modm(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 3;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_scalar_r(res_arg ? args[0] : mp_const_none);
assert_scalar(args[1 + off]);
assert_scalar(args[2 + off]);
sub256_modm(MP_OBJ_SCALAR(res), MP_OBJ_C_SCALAR(args[1 + off]),
MP_OBJ_C_SCALAR(args[2 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_sub256_modm_obj, 2, 3,
mod_trezorcrypto_monero_sub256_modm);
/// def mul256_modm(r: Optional[Sc25519], a: Sc25519, b: Sc25519) -> Sc25519:
/// """
/// Scalar multiplication
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_mul256_modm(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 3;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_scalar_r(res_arg ? args[0] : mp_const_none);
assert_scalar(args[1 + off]);
assert_scalar(args[2 + off]);
mul256_modm(MP_OBJ_SCALAR(res), MP_OBJ_C_SCALAR(args[1 + off]),
MP_OBJ_C_SCALAR(args[2 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_mul256_modm_obj, 2, 3,
mod_trezorcrypto_monero_mul256_modm);
/// def mulsub256_modm(
/// r: Optional[Sc25519], a: Sc25519, b: Sc25519, c: Sc25519
/// ) -> Sc25519:
/// """
/// c - a*b
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_mulsub256_modm(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 4;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_scalar_r(res_arg ? args[0] : mp_const_none);
assert_scalar(args[1 + off]);
assert_scalar(args[2 + off]);
assert_scalar(args[3 + off]);
mulsub256_modm(MP_OBJ_SCALAR(res), MP_OBJ_C_SCALAR(args[1 + off]),
MP_OBJ_C_SCALAR(args[2 + off]),
MP_OBJ_C_SCALAR(args[3 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_mulsub256_modm_obj, 3, 4,
mod_trezorcrypto_monero_mulsub256_modm);
/// def muladd256_modm(
/// r: Optional[Sc25519], a: Sc25519, b: Sc25519, c: Sc25519
/// ) -> Sc25519:
/// """
/// c + a*b
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_muladd256_modm(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 4;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_scalar_r(res_arg ? args[0] : mp_const_none);
assert_scalar(args[1 + off]);
assert_scalar(args[2 + off]);
assert_scalar(args[3 + off]);
muladd256_modm(MP_OBJ_SCALAR(res), MP_OBJ_C_SCALAR(args[1 + off]),
MP_OBJ_C_SCALAR(args[2 + off]),
MP_OBJ_C_SCALAR(args[3 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_muladd256_modm_obj, 3, 4,
mod_trezorcrypto_monero_muladd256_modm);
/// def inv256_modm(r: Optional[Sc25519], a: Sc25519) -> Sc25519:
/// """
/// Scalar modular inversion
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_inv256_modm(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 2;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_scalar_r(res_arg ? args[0] : mp_const_none);
assert_scalar(args[1 + off]);
// bn_prime = curve order, little endian encoded
bignum256 bn_prime = {.val = {0x1cf5d3ed, 0x20498c69, 0x2f79cd65, 0x37be77a8,
0x14, 0x0, 0x0, 0x0, 0x1000}};
bignum256 bn_x;
memcpy(&bn_x.val, MP_OBJ_C_SCALAR(args[1 + off]), sizeof(bignum256modm));
bn_inverse(&bn_x, &bn_prime);
memcpy(MP_OBJ_SCALAR(res), bn_x.val, sizeof(bignum256modm));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_inv256_modm_obj, 1, 2,
mod_trezorcrypto_monero_inv256_modm);
/// def pack256_modm(
/// r: Optional[bytes], a: Sc25519, offset: Optional[int] = 0
/// ) -> bytes:
/// """
/// Scalar compression
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_pack256_modm(size_t n_args,
const mp_obj_t *args) {
if (n_args == 1 || args[0] == mp_const_none) {
assert_scalar(args[0]);
uint8_t buff[32];
contract256_modm(buff, MP_OBJ_C_SCALAR(args[0]));
return mp_obj_new_bytes(buff, 32);
} else {
mp_buffer_info_t bufm;
mp_get_buffer_raise(args[0], &bufm, MP_BUFFER_WRITE);
const mp_int_t offset = n_args >= 3 ? mp_obj_get_int(args[2]) : 0;
if (bufm.len < 32 + offset) {
mp_raise_ValueError("Buffer too small");
}
contract256_modm(((uint8_t *)bufm.buf) + offset, MP_OBJ_C_SCALAR(args[1]));
return args[0];
}
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_pack256_modm_obj, 1, 3,
mod_trezorcrypto_monero_pack256_modm);
/// def unpack256_modm(
/// r: Optional[Sc25519], a: bytes, offset: int = 0
/// ) -> Sc25519:
/// """
/// Scalar decompression
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_unpack256_modm(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args >= 2;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_scalar_r(res_arg ? args[0] : mp_const_none);
const mp_int_t offset = n_args >= 3 ? mp_obj_get_int(args[2]) : 0;
mp_unpack_scalar(MP_OBJ_SCALAR(res), args[1 + off], offset);
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_unpack256_modm_obj, 1, 3,
mod_trezorcrypto_monero_unpack256_modm);
/// def unpack256_modm_noreduce(
/// r: Optional[Sc25519], a: bytes, offset: int = 0
/// ) -> Sc25519:
/// """
/// Scalar decompression, raw, without modular reduction
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_unpack256_modm_noreduce(
size_t n_args, const mp_obj_t *args) {
const bool res_arg = n_args >= 2;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_scalar_r(res_arg ? args[0] : mp_const_none);
const mp_int_t offset = n_args >= 3 ? mp_obj_get_int(args[2]) : 0;
mp_buffer_info_t buff;
mp_get_buffer_raise(args[1 + off], &buff, MP_BUFFER_READ);
if (buff.len != 32 + offset) {
mp_raise_ValueError("Invalid length of secret key");
}
expand_raw256_modm(MP_OBJ_SCALAR(res), ((uint8_t *)buff.buf) + offset);
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_unpack256_modm_noreduce_obj, 1, 3,
mod_trezorcrypto_monero_unpack256_modm_noreduce);
//
// GE25519 Defs
//
/// def ge25519_set_neutral(r: Optional[Ge25519]) -> Ge25519:
/// """
/// Sets neutral point
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519_set_neutral(
size_t n_args, const mp_obj_t *args) {
mp_obj_t res = mp_obj_new_ge25519_r(n_args == 1 ? args[0] : mp_const_none);
ge25519_set_neutral(&MP_OBJ_GE25519(res));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_ge25519_set_neutral_obj, 0, 1,
mod_trezorcrypto_monero_ge25519_set_neutral);
/// def ge25519_set_xmr_h(r: Optional[Ge25519]) -> Ge25519:
/// """
/// Sets H point
/// """
STATIC mp_obj_t
mod_trezorcrypto_monero_ge25519_set_xmr_h(size_t n_args, const mp_obj_t *args) {
mp_obj_t res = mp_obj_new_ge25519_r(n_args == 1 ? args[0] : mp_const_none);
ge25519_set_xmr_h(&MP_OBJ_GE25519(res));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_ge25519_set_xmr_h_obj, 0, 1,
mod_trezorcrypto_monero_ge25519_set_xmr_h);
/// def ge25519_check(r: Ge25519):
/// """
/// Checks point, throws if not on curve
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519_check(const mp_obj_t arg) {
assert_ge25519(arg);
if (ge25519_check(&MP_OBJ_C_GE25519(arg)) != 1) {
mp_raise_ValueError("Ed25519 point not on curve");
}
return mp_const_none;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_trezorcrypto_monero_ge25519_check_obj,
mod_trezorcrypto_monero_ge25519_check);
/// def ge25519_eq(a: Ge25519, b: Ge25519) -> bool:
/// """
/// Compares EC points
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519_eq(const mp_obj_t a,
const mp_obj_t b) {
assert_ge25519(a);
assert_ge25519(b);
int r = ge25519_eq(&MP_OBJ_C_GE25519(a), &MP_OBJ_C_GE25519(b));
return MP_OBJ_NEW_SMALL_INT(r);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_2(mod_trezorcrypto_monero_ge25519_eq_obj,
mod_trezorcrypto_monero_ge25519_eq);
/// def ge25519_add(r: Optional[Ge25519], a: Ge25519, b: Ge25519) -> Ge25519:
/// """
/// Adds EC points
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519_add(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 3;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
assert_ge25519(args[1 + off]);
assert_ge25519(args[2 + off]);
ge25519_add(&MP_OBJ_GE25519(res), &MP_OBJ_C_GE25519(args[1 + off]),
&MP_OBJ_C_GE25519(args[2 + off]), 0);
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_ge25519_add_obj, 2, 3,
mod_trezorcrypto_monero_ge25519_add);
/// def ge25519_sub(r: Optional[Ge25519], a: Ge25519, b: Ge25519) -> Ge25519:
/// """
/// Subtracts EC points
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519_sub(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 3;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
assert_ge25519(args[1 + off]);
assert_ge25519(args[2 + off]);
ge25519_add(&MP_OBJ_GE25519(res), &MP_OBJ_C_GE25519(args[1 + off]),
&MP_OBJ_C_GE25519(args[2 + off]), 1);
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_ge25519_sub_obj, 2, 3,
mod_trezorcrypto_monero_ge25519_sub);
/// def ge25519_double(r: Optional[Ge25519], p: Ge25519) -> Ge25519:
/// """
/// EC point doubling
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519_double(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 2;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
mp_obj_t src = res_arg ? args[1] : args[0];
assert_ge25519(src);
ge25519_double(&MP_OBJ_GE25519(res), &MP_OBJ_C_GE25519(src));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_ge25519_double_obj, 1, 2,
mod_trezorcrypto_monero_ge25519_double);
/// def ge25519_mul8(r: Optional[Ge25519], p: Ge25519) -> Ge25519:
/// """
/// EC point * 8
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519_mul8(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 2;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
mp_obj_t src = res_arg ? args[1] : args[0];
assert_ge25519(src);
ge25519_mul8(&MP_OBJ_GE25519(res), &MP_OBJ_C_GE25519(src));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_ge25519_mul8_obj, 1, 2,
mod_trezorcrypto_monero_ge25519_mul8);
/// def ge25519_double_scalarmult_vartime(
/// r: Optional[Ge25519], p1: Ge25519, s1: Sc25519, s2: Sc25519
/// ) -> Ge25519:
/// """
/// s1 * G + s2 * p1
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519_double_scalarmult_vartime(
size_t n_args, const mp_obj_t *args) {
const bool res_arg = n_args == 4;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
assert_ge25519(args[1 + off]);
assert_scalar(args[2 + off]);
assert_scalar(args[3 + off]);
ge25519_double_scalarmult_vartime(
&MP_OBJ_GE25519(res), &MP_OBJ_C_GE25519(args[1 + off]),
MP_OBJ_C_SCALAR(args[2 + off]), MP_OBJ_C_SCALAR(args[3 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_ge25519_double_scalarmult_vartime_obj, 3, 4,
mod_trezorcrypto_monero_ge25519_double_scalarmult_vartime);
/// def ge25519_double_scalarmult_vartime2(
/// r: Optional[Ge25519],
/// p1: Ge25519,
/// s1: Sc25519,
/// p2: Ge25519,
/// s2: Sc25519,
/// ) -> Ge25519:
/// """
/// s1 * p1 + s2 * p2
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519_double_scalarmult_vartime2(
size_t n_args, const mp_obj_t *args) {
const bool res_arg = n_args == 5;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
assert_ge25519(args[1 + off]);
assert_scalar(args[2 + off]);
assert_ge25519(args[3 + off]);
assert_scalar(args[4 + off]);
ge25519_double_scalarmult_vartime2(
&MP_OBJ_GE25519(res), &MP_OBJ_C_GE25519(args[1 + off]),
MP_OBJ_C_SCALAR(args[2 + off]), &MP_OBJ_C_GE25519(args[3 + off]),
MP_OBJ_C_SCALAR(args[4 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_ge25519_double_scalarmult_vartime2_obj, 4, 5,
mod_trezorcrypto_monero_ge25519_double_scalarmult_vartime2);
/// def ge25519_scalarmult_base(
/// r: Optional[Ge25519], s: Union[Sc25519, int]
/// ) -> Ge25519:
/// """
/// s * G
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519_scalarmult_base(
size_t n_args, const mp_obj_t *args) {
const bool res_arg = n_args == 2;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
if (MP_OBJ_IS_SCALAR(args[1 + off])) {
ge25519_scalarmult_base_wrapper(&MP_OBJ_GE25519(res),
MP_OBJ_C_SCALAR(args[1 + off]));
} else if (mp_obj_is_integer(args[1 + off])) {
bignum256modm mlt;
set256_modm(mlt, mp_obj_get_int(args[1 + off]));
ge25519_scalarmult_base_wrapper(&MP_OBJ_GE25519(res), mlt);
} else {
mp_raise_ValueError("unknown base mult type");
}
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_ge25519_scalarmult_base_obj, 1, 2,
mod_trezorcrypto_monero_ge25519_scalarmult_base);
/// def ge25519_scalarmult(
/// r: Optional[Ge25519], p: Ge25519, s: Union[Sc25519, int]
/// ) -> Ge25519:
/// """
/// s * p
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519_scalarmult(
size_t n_args, const mp_obj_t *args) {
const bool res_arg = n_args == 3;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
assert_ge25519(args[1 + off]);
if (MP_OBJ_IS_SCALAR(args[2 + off])) {
ge25519_scalarmult(&MP_OBJ_GE25519(res), &MP_OBJ_C_GE25519(args[1 + off]),
MP_OBJ_C_SCALAR(args[2 + off]));
} else if (mp_obj_is_integer(args[2 + off])) {
bignum256modm mlt;
set256_modm(mlt, mp_obj_get_int(args[2 + off]));
ge25519_scalarmult(&MP_OBJ_GE25519(res), &MP_OBJ_C_GE25519(args[1 + off]),
mlt);
} else {
mp_raise_ValueError("unknown mult type");
}
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_ge25519_scalarmult_obj, 2, 3,
mod_trezorcrypto_monero_ge25519_scalarmult);
/// def ge25519_pack(r: bytes, p: Ge25519, offset: int = 0) -> bytes:
/// """
/// Point compression
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519_pack(size_t n_args,
const mp_obj_t *args) {
if (n_args == 1 || args[0] == mp_const_none) {
assert_ge25519(args[0]);
uint8_t buff[32];
ge25519_pack(buff, &MP_OBJ_C_GE25519(args[0]));
return mp_obj_new_bytes(buff, 32);
} else {
mp_buffer_info_t bufm;
mp_get_buffer_raise(args[0], &bufm, MP_BUFFER_WRITE);
const mp_int_t offset = n_args >= 3 ? mp_obj_get_int(args[2]) : 0;
if (bufm.len < 32 + offset) {
mp_raise_ValueError("Buffer too small");
}
ge25519_pack(((uint8_t *)bufm.buf) + offset, &MP_OBJ_C_GE25519(args[1]));
return args[0];
}
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_ge25519_pack_obj, 1, 3,
mod_trezorcrypto_monero_ge25519_pack);
/// def ge25519_unpack_vartime(
/// r: Optional[Ge25519], buff: bytes, offset: int = 0
/// ) -> Ge25519:
/// """
/// Point decompression
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_ge25519_unpack_vartime(
size_t n_args, const mp_obj_t *args) {
const bool res_arg = n_args >= 2;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
const mp_int_t offset = n_args >= 3 ? mp_obj_get_int(args[2]) : 0;
mp_unpack_ge25519(&MP_OBJ_GE25519(res), args[1 + off], offset);
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_ge25519_unpack_vartime_obj, 1, 3,
mod_trezorcrypto_monero_ge25519_unpack_vartime);
//
// XMR defs
//
/// def base58_addr_encode_check(tag: int, buff: bytes) -> bytes:
/// """
/// Monero block base 58 encoding
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_base58_addr_encode_check(
size_t n_args, const mp_obj_t *args) {
uint8_t out[128];
mp_buffer_info_t data;
mp_get_buffer_raise(args[1], &data, MP_BUFFER_READ);
int sz = xmr_base58_addr_encode_check(mp_obj_get_int(args[0]), data.buf,
data.len, (char *)out, sizeof(out));
if (sz == 0) {
mp_raise_ValueError("b58 encoding error");
}
return mp_obj_new_bytes(out, sz);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_base58_addr_encode_check_obj, 2, 2,
mod_trezorcrypto_monero_xmr_base58_addr_encode_check);
/// def base58_addr_decode_check(buff: bytes) -> Tuple[bytes, int]:
/// """
/// Monero block base 58 decoding, returning (decoded, tag) or raising on
/// error.
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_base58_addr_decode_check(
size_t n_args, const mp_obj_t *args) {
uint8_t out[128];
uint64_t tag;
mp_buffer_info_t data;
mp_get_buffer_raise(args[0], &data, MP_BUFFER_READ);
int sz =
xmr_base58_addr_decode_check(data.buf, data.len, &tag, out, sizeof(out));
if (sz == 0) {
mp_raise_ValueError("b58 decoding error");
}
mp_obj_tuple_t *tuple = MP_OBJ_TO_PTR(mp_obj_new_tuple(2, NULL));
tuple->items[0] = mp_obj_new_bytes(out, sz);
tuple->items[1] = mp_obj_new_int_from_ull(tag);
return MP_OBJ_FROM_PTR(tuple);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_base58_addr_decode_check_obj, 1, 1,
mod_trezorcrypto_monero_xmr_base58_addr_decode_check);
/// def xmr_random_scalar(r: Optional[Sc25519] = None) -> Sc25519:
/// """
/// Generates a random scalar
/// """
STATIC mp_obj_t
mod_trezorcrypto_monero_xmr_random_scalar(size_t n_args, const mp_obj_t *args) {
mp_obj_t res = mp_obj_new_scalar_r(n_args == 1 ? args[0] : mp_const_none);
xmr_random_scalar(MP_OBJ_SCALAR(res));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_random_scalar_obj, 0, 1,
mod_trezorcrypto_monero_xmr_random_scalar);
/// def xmr_fast_hash(r: Optional[bytes], buff: bytes) -> bytes:
/// """
/// XMR fast hash
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_fast_hash(size_t n_args,
const mp_obj_t *args) {
const int off = n_args == 2 ? 0 : -1;
uint8_t buff[32];
uint8_t *buff_use = buff;
if (n_args > 1) {
mp_buffer_info_t odata;
mp_get_buffer_raise(args[0], &odata, MP_BUFFER_WRITE);
if (odata.len < 32) {
mp_raise_ValueError("Output buffer too small");
}
buff_use = odata.buf;
}
mp_buffer_info_t data;
mp_get_buffer_raise(args[1 + off], &data, MP_BUFFER_READ);
xmr_fast_hash(buff_use, data.buf, data.len);
return n_args == 2 ? args[0] : mp_obj_new_bytes(buff, 32);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_fast_hash_obj, 1, 2,
mod_trezorcrypto_monero_xmr_fast_hash);
/// def xmr_hash_to_ec(r: Optional[Ge25519], buff: bytes) -> Ge25519:
/// """
/// XMR hashing to EC point
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_hash_to_ec(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 2;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
mp_buffer_info_t data;
mp_get_buffer_raise(args[1 + off], &data, MP_BUFFER_READ);
xmr_hash_to_ec(&MP_OBJ_GE25519(res), data.buf, data.len);
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_hash_to_ec_obj, 1, 2,
mod_trezorcrypto_monero_xmr_hash_to_ec);
/// def xmr_hash_to_scalar(r: Optional[Sc25519], buff: bytes) -> Sc25519:
/// """
/// XMR hashing to EC scalar
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_hash_to_scalar(
size_t n_args, const mp_obj_t *args) {
const bool res_arg = n_args == 2;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_scalar_r(res_arg ? args[0] : mp_const_none);
mp_buffer_info_t data;
mp_get_buffer_raise(args[1 + off], &data, MP_BUFFER_READ);
xmr_hash_to_scalar(MP_OBJ_SCALAR(res), data.buf, data.len);
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_hash_to_scalar_obj, 1, 2,
mod_trezorcrypto_monero_xmr_hash_to_scalar);
/// def xmr_derivation_to_scalar(
/// r: Optional[Sc25519], p: Ge25519, output_index: int
/// ) -> Sc25519:
/// """
/// H_s(derivation || varint(output_index))
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_derivation_to_scalar(
size_t n_args, const mp_obj_t *args) {
const bool res_arg = n_args == 3;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_scalar_r(res_arg ? args[0] : mp_const_none);
assert_ge25519(args[1 + off]);
xmr_derivation_to_scalar(MP_OBJ_SCALAR(res), &MP_OBJ_C_GE25519(args[1 + off]),
mp_obj_get_int(args[2 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_derivation_to_scalar_obj, 2, 3,
mod_trezorcrypto_monero_xmr_derivation_to_scalar);
/// def xmr_generate_key_derivation(
/// r: Optional[Ge25519], A: Ge25519, b: Sc25519
/// ) -> Ge25519:
/// """
/// 8*(key2*key1)
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_generate_key_derivation(
size_t n_args, const mp_obj_t *args) {
const bool res_arg = n_args == 3;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
assert_ge25519(args[1 + off]);
assert_scalar(args[2 + off]);
xmr_generate_key_derivation(&MP_OBJ_GE25519(res),
&MP_OBJ_C_GE25519(args[1 + off]),
MP_OBJ_C_SCALAR(args[2 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_generate_key_derivation_obj, 2, 3,
mod_trezorcrypto_monero_xmr_generate_key_derivation);
/// def xmr_derive_private_key(
/// r: Optional[Sc25519], deriv: Ge25519, idx: int, base: Sc25519
/// ) -> Sc25519:
/// """
/// base + H_s(derivation || varint(output_index))
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_derive_private_key(
size_t n_args, const mp_obj_t *args) {
const bool res_arg = n_args == 4;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_scalar_r(res_arg ? args[0] : mp_const_none);
assert_ge25519(args[1 + off]);
assert_scalar(args[3 + off]);
xmr_derive_private_key(MP_OBJ_SCALAR(res), &MP_OBJ_C_GE25519(args[1 + off]),
mp_obj_get_int(args[2 + off]),
MP_OBJ_C_SCALAR(args[3 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_derive_private_key_obj, 3, 4,
mod_trezorcrypto_monero_xmr_derive_private_key);
/// def xmr_derive_public_key(
/// r: Optional[Ge25519], deriv: Ge25519, idx: int, base: Ge25519
/// ) -> Ge25519:
/// """
/// H_s(derivation || varint(output_index))G + base
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_derive_public_key(
size_t n_args, const mp_obj_t *args) {
const bool res_arg = n_args == 4;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
assert_ge25519(args[1 + off]);
assert_ge25519(args[3 + off]);
xmr_derive_public_key(&MP_OBJ_GE25519(res), &MP_OBJ_C_GE25519(args[1 + off]),
mp_obj_get_int(args[2 + off]),
&MP_OBJ_C_GE25519(args[3 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_derive_public_key_obj, 3, 4,
mod_trezorcrypto_monero_xmr_derive_public_key);
/// def xmr_add_keys2(
/// r: Optional[Ge25519], a: Sc25519, b: Sc25519, B: Ge25519
/// ) -> Ge25519:
/// """
/// aG + bB, G is basepoint
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_add_keys2(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 4;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
assert_scalar(args[1 + off]);
assert_scalar(args[2 + off]);
assert_ge25519(args[3 + off]);
xmr_add_keys2(&MP_OBJ_GE25519(res), MP_OBJ_SCALAR(args[1 + off]),
MP_OBJ_SCALAR(args[2 + off]), &MP_OBJ_C_GE25519(args[3 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_add_keys2_obj, 3, 4,
mod_trezorcrypto_monero_xmr_add_keys2);
/// def xmr_add_keys2_vartime(
/// r: Optional[Ge25519], a: Sc25519, b: Sc25519, B: Ge25519
/// ) -> Ge25519:
/// """
/// aG + bB, G is basepoint
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_add_keys2_vartime(
size_t n_args, const mp_obj_t *args) {
const bool res_arg = n_args == 4;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
assert_scalar(args[1 + off]);
assert_scalar(args[2 + off]);
assert_ge25519(args[3 + off]);
xmr_add_keys2_vartime(&MP_OBJ_GE25519(res), MP_OBJ_SCALAR(args[1 + off]),
MP_OBJ_SCALAR(args[2 + off]),
&MP_OBJ_C_GE25519(args[3 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_add_keys2_vartime_obj, 3, 4,
mod_trezorcrypto_monero_xmr_add_keys2_vartime);
/// def xmr_add_keys3(
/// r: Optional[Ge25519], a: Sc25519, A: Ge25519, b: Sc25519, B: Ge25519
/// ) -> Ge25519:
/// """
/// aA + bB
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_add_keys3(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 5;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
assert_scalar(args[1 + off]);
assert_ge25519(args[2 + off]);
assert_scalar(args[3 + off]);
assert_ge25519(args[4 + off]);
xmr_add_keys3(&MP_OBJ_GE25519(res), MP_OBJ_SCALAR(args[1 + off]),
&MP_OBJ_C_GE25519(args[2 + off]), MP_OBJ_SCALAR(args[3 + off]),
&MP_OBJ_C_GE25519(args[4 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_add_keys3_obj, 4, 5,
mod_trezorcrypto_monero_xmr_add_keys3);
/// def xmr_add_keys3_vartime(
/// r: Optional[Ge25519], a: Sc25519, A: Ge25519, b: Sc25519, B: Ge25519
/// ) -> Ge25519:
/// """
/// aA + bB
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_add_keys3_vartime(
size_t n_args, const mp_obj_t *args) {
const bool res_arg = n_args == 5;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
assert_scalar(args[1 + off]);
assert_ge25519(args[2 + off]);
assert_scalar(args[3 + off]);
assert_ge25519(args[4 + off]);
xmr_add_keys3_vartime(&MP_OBJ_GE25519(res), MP_OBJ_SCALAR(args[1 + off]),
&MP_OBJ_C_GE25519(args[2 + off]),
MP_OBJ_SCALAR(args[3 + off]),
&MP_OBJ_C_GE25519(args[4 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_add_keys3_vartime_obj, 4, 5,
mod_trezorcrypto_monero_xmr_add_keys3_vartime);
/// def xmr_get_subaddress_secret_key(
/// r: Optional[Sc25519], major: int, minor: int, m: Sc25519
/// ) -> Sc25519:
/// """
/// Hs(SubAddr || a || index_major || index_minor)
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_get_subaddress_secret_key(
size_t n_args, const mp_obj_t *args) {
const bool res_arg = n_args == 4;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_scalar_r(res_arg ? args[0] : mp_const_none);
assert_scalar(args[3 + off]);
xmr_get_subaddress_secret_key(
MP_OBJ_SCALAR(res), mp_obj_get_int(args[1 + off]),
mp_obj_get_int(args[2 + off]), MP_OBJ_C_SCALAR(args[3 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_get_subaddress_secret_key_obj, 3, 4,
mod_trezorcrypto_monero_xmr_get_subaddress_secret_key);
/// def xmr_gen_c(r: Optional[Ge25519], a: Sc25519, amount: int) -> Ge25519:
/// """
/// aG + amount * H
/// """
STATIC mp_obj_t mod_trezorcrypto_monero_xmr_gen_c(size_t n_args,
const mp_obj_t *args) {
const bool res_arg = n_args == 3;
const int off = res_arg ? 0 : -1;
mp_obj_t res = mp_obj_new_ge25519_r(res_arg ? args[0] : mp_const_none);
assert_scalar(args[1 + off]);
xmr_gen_c(&MP_OBJ_GE25519(res), MP_OBJ_C_SCALAR(args[1 + off]),
mp_obj_get_uint64(args[2 + off]));
return res;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_xmr_gen_c_obj, 2, 3,
mod_trezorcrypto_monero_xmr_gen_c);
/// def ct_equals(a: bytes, b: bytes) -> bool:
/// """
/// Constant time buffer comparison
/// """
STATIC mp_obj_t mod_trezorcrypto_ct_equals(const mp_obj_t a, const mp_obj_t b) {
mp_buffer_info_t buff_a, buff_b;
mp_get_buffer_raise(a, &buff_a, MP_BUFFER_READ);
mp_get_buffer_raise(b, &buff_b, MP_BUFFER_READ);
if (buff_a.len != buff_b.len) {
return MP_OBJ_NEW_SMALL_INT(0);
}
int r = ed25519_verify(buff_a.buf, buff_b.buf, buff_a.len);
return MP_OBJ_NEW_SMALL_INT(r);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_2(mod_trezorcrypto_ct_equals_obj,
mod_trezorcrypto_ct_equals);
// Hasher
STATIC mp_obj_t mod_trezorcrypto_monero_hasher_update(mp_obj_t self,
const mp_obj_t arg) {
mp_obj_hasher_t *o = MP_OBJ_TO_PTR(self);
mp_buffer_info_t buff;
mp_get_buffer_raise(arg, &buff, MP_BUFFER_READ);
if (buff.len > 0) {
xmr_hasher_update(&o->h, buff.buf, buff.len);
}
return mp_const_none;
}
STATIC MP_DEFINE_CONST_FUN_OBJ_2(mod_trezorcrypto_monero_hasher_update_obj,
mod_trezorcrypto_monero_hasher_update);
STATIC mp_obj_t mod_trezorcrypto_monero_hasher_digest(size_t n_args,
const mp_obj_t *args) {
mp_obj_hasher_t *o = MP_OBJ_TO_PTR(args[0]);
Hasher ctx;
memcpy(&ctx, &(o->h), sizeof(Hasher));
uint8_t out[SHA3_256_DIGEST_LENGTH];
xmr_hasher_final(&ctx, out);
memzero(&ctx, sizeof(SHA3_CTX));
if (n_args == 1 || args[1] == mp_const_none) {
return mp_obj_new_bytes(out, sizeof(out));
} else {
mp_buffer_info_t bufm;
mp_get_buffer_raise(args[1], &bufm, MP_BUFFER_WRITE);
const mp_int_t offset = n_args >= 3 ? mp_obj_get_int(args[2]) : 0;
if (bufm.len < 32 + offset) {
mp_raise_ValueError("Buffer too small");
}
memcpy((uint8_t *)bufm.buf + offset, out, SHA3_256_DIGEST_LENGTH);
return args[1];
}
}
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(
mod_trezorcrypto_monero_hasher_digest_obj, 1, 3,
mod_trezorcrypto_monero_hasher_digest);
STATIC mp_obj_t mod_trezorcrypto_monero_hasher_copy(mp_obj_t self) {
mp_obj_hasher_t *o = MP_OBJ_TO_PTR(self);
mp_obj_hasher_t *cp = m_new_obj(mp_obj_hasher_t);
cp->base.type = o->base.type;
memcpy(&(cp->h), &(o->h), sizeof(Hasher));
return MP_OBJ_FROM_PTR(o);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_trezorcrypto_monero_hasher_copy_obj,
mod_trezorcrypto_monero_hasher_copy);
//
// Type defs
//
STATIC const mp_rom_map_elem_t
mod_trezorcrypto_monero_ge25519_locals_dict_table[] = {
{MP_ROM_QSTR(MP_QSTR___del__),
MP_ROM_PTR(&mod_trezorcrypto_monero_ge25519___del___obj)},
};
STATIC MP_DEFINE_CONST_DICT(mod_trezorcrypto_monero_ge25519_locals_dict,
mod_trezorcrypto_monero_ge25519_locals_dict_table);
STATIC const mp_obj_type_t mod_trezorcrypto_monero_ge25519_type = {
{&mp_type_type},
.name = MP_QSTR_Ge25519,
.make_new = mod_trezorcrypto_monero_ge25519_make_new,
.locals_dict = (void *)&mod_trezorcrypto_monero_ge25519_locals_dict,
};
STATIC const mp_rom_map_elem_t
mod_trezorcrypto_monero_bignum256modm_locals_dict_table[] = {
{MP_ROM_QSTR(MP_QSTR___del__),
MP_ROM_PTR(&mod_trezorcrypto_monero_bignum256modm___del___obj)},
};
STATIC MP_DEFINE_CONST_DICT(
mod_trezorcrypto_monero_bignum256modm_locals_dict,
mod_trezorcrypto_monero_bignum256modm_locals_dict_table);
STATIC const mp_obj_type_t mod_trezorcrypto_monero_bignum256modm_type = {
{&mp_type_type},
.name = MP_QSTR_Sc25519,
.make_new = mod_trezorcrypto_monero_bignum256modm_make_new,
.locals_dict = (void *)&mod_trezorcrypto_monero_bignum256modm_locals_dict,
};
STATIC const mp_rom_map_elem_t
mod_trezorcrypto_monero_hasher_locals_dict_table[] = {
{MP_ROM_QSTR(MP_QSTR_update),
MP_ROM_PTR(&mod_trezorcrypto_monero_hasher_update_obj)},
{MP_ROM_QSTR(MP_QSTR_digest),
MP_ROM_PTR(&mod_trezorcrypto_monero_hasher_digest_obj)},
{MP_ROM_QSTR(MP_QSTR_copy),
MP_ROM_PTR(&mod_trezorcrypto_monero_hasher_copy_obj)},
{MP_ROM_QSTR(MP_QSTR___del__),
MP_ROM_PTR(&mod_trezorcrypto_monero_hasher___del___obj)},
{MP_ROM_QSTR(MP_QSTR_block_size), MP_ROM_INT(SHA3_256_BLOCK_LENGTH)},
{MP_ROM_QSTR(MP_QSTR_digest_size), MP_ROM_INT(SHA3_256_DIGEST_LENGTH)},
};
STATIC MP_DEFINE_CONST_DICT(mod_trezorcrypto_monero_hasher_locals_dict,
mod_trezorcrypto_monero_hasher_locals_dict_table);
STATIC const mp_obj_type_t mod_trezorcrypto_monero_hasher_type = {
{&mp_type_type},
.name = MP_QSTR_hasher,
.make_new = mod_trezorcrypto_monero_hasher_make_new,
.locals_dict = (void *)&mod_trezorcrypto_monero_hasher_locals_dict,
};
STATIC const mp_rom_map_elem_t mod_trezorcrypto_monero_globals_table[] = {
{MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_monero)},
{MP_ROM_QSTR(MP_QSTR_init256_modm),
MP_ROM_PTR(&mod_trezorcrypto_monero_init256_modm_obj)},
{MP_ROM_QSTR(MP_QSTR_check256_modm),
MP_ROM_PTR(&mod_trezorcrypto_monero_check256_modm_obj)},
{MP_ROM_QSTR(MP_QSTR_iszero256_modm),
MP_ROM_PTR(&mod_trezorcrypto_monero_iszero256_modm_obj)},
{MP_ROM_QSTR(MP_QSTR_eq256_modm),
MP_ROM_PTR(&mod_trezorcrypto_monero_eq256_modm_obj)},
{MP_ROM_QSTR(MP_QSTR_get256_modm),
MP_ROM_PTR(&mod_trezorcrypto_monero_get256_modm_obj)},
{MP_ROM_QSTR(MP_QSTR_add256_modm),
MP_ROM_PTR(&mod_trezorcrypto_monero_add256_modm_obj)},
{MP_ROM_QSTR(MP_QSTR_sub256_modm),
MP_ROM_PTR(&mod_trezorcrypto_monero_sub256_modm_obj)},
{MP_ROM_QSTR(MP_QSTR_mul256_modm),
MP_ROM_PTR(&mod_trezorcrypto_monero_mul256_modm_obj)},
{MP_ROM_QSTR(MP_QSTR_mulsub256_modm),
MP_ROM_PTR(&mod_trezorcrypto_monero_mulsub256_modm_obj)},
{MP_ROM_QSTR(MP_QSTR_muladd256_modm),
MP_ROM_PTR(&mod_trezorcrypto_monero_muladd256_modm_obj)},
{MP_ROM_QSTR(MP_QSTR_inv256_modm),
MP_ROM_PTR(&mod_trezorcrypto_monero_inv256_modm_obj)},
{MP_ROM_QSTR(MP_QSTR_pack256_modm),
MP_ROM_PTR(&mod_trezorcrypto_monero_pack256_modm_obj)},
{MP_ROM_QSTR(MP_QSTR_unpack256_modm),
MP_ROM_PTR(&mod_trezorcrypto_monero_unpack256_modm_obj)},
{MP_ROM_QSTR(MP_QSTR_unpack256_modm_noreduce),
MP_ROM_PTR(&mod_trezorcrypto_monero_unpack256_modm_noreduce_obj)},
{MP_ROM_QSTR(MP_QSTR_ge25519_set_neutral),
MP_ROM_PTR(&mod_trezorcrypto_monero_ge25519_set_neutral_obj)},
{MP_ROM_QSTR(MP_QSTR_ge25519_set_h),
MP_ROM_PTR(&mod_trezorcrypto_monero_ge25519_set_xmr_h_obj)},
{MP_ROM_QSTR(MP_QSTR_ge25519_pack),
MP_ROM_PTR(&mod_trezorcrypto_monero_ge25519_pack_obj)},
{MP_ROM_QSTR(MP_QSTR_ge25519_unpack_vartime),
MP_ROM_PTR(&mod_trezorcrypto_monero_ge25519_unpack_vartime_obj)},
{MP_ROM_QSTR(MP_QSTR_ge25519_check),
MP_ROM_PTR(&mod_trezorcrypto_monero_ge25519_check_obj)},
{MP_ROM_QSTR(MP_QSTR_ge25519_eq),
MP_ROM_PTR(&mod_trezorcrypto_monero_ge25519_eq_obj)},
{MP_ROM_QSTR(MP_QSTR_ge25519_add),
MP_ROM_PTR(&mod_trezorcrypto_monero_ge25519_add_obj)},
{MP_ROM_QSTR(MP_QSTR_ge25519_sub),
MP_ROM_PTR(&mod_trezorcrypto_monero_ge25519_sub_obj)},
{MP_ROM_QSTR(MP_QSTR_ge25519_double),
MP_ROM_PTR(&mod_trezorcrypto_monero_ge25519_double_obj)},
{MP_ROM_QSTR(MP_QSTR_ge25519_mul8),
MP_ROM_PTR(&mod_trezorcrypto_monero_ge25519_mul8_obj)},
{MP_ROM_QSTR(MP_QSTR_ge25519_double_scalarmult_vartime),
MP_ROM_PTR(
&mod_trezorcrypto_monero_ge25519_double_scalarmult_vartime_obj)},
{MP_ROM_QSTR(MP_QSTR_ge25519_double_scalarmult_vartime2),
MP_ROM_PTR(
&mod_trezorcrypto_monero_ge25519_double_scalarmult_vartime2_obj)},
{MP_ROM_QSTR(MP_QSTR_ge25519_scalarmult_base),
MP_ROM_PTR(&mod_trezorcrypto_monero_ge25519_scalarmult_base_obj)},
{MP_ROM_QSTR(MP_QSTR_ge25519_scalarmult),
MP_ROM_PTR(&mod_trezorcrypto_monero_ge25519_scalarmult_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_base58_addr_encode_check),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_base58_addr_encode_check_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_base58_addr_decode_check),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_base58_addr_decode_check_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_random_scalar),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_random_scalar_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_fast_hash),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_fast_hash_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_hash_to_ec),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_hash_to_ec_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_hash_to_scalar),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_hash_to_scalar_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_derivation_to_scalar),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_derivation_to_scalar_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_generate_key_derivation),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_generate_key_derivation_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_derive_private_key),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_derive_private_key_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_derive_public_key),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_derive_public_key_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_add_keys2),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_add_keys2_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_add_keys2_vartime),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_add_keys2_vartime_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_add_keys3),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_add_keys3_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_add_keys3_vartime),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_add_keys3_vartime_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_get_subaddress_secret_key),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_get_subaddress_secret_key_obj)},
{MP_ROM_QSTR(MP_QSTR_xmr_gen_c),
MP_ROM_PTR(&mod_trezorcrypto_monero_xmr_gen_c_obj)},
{MP_ROM_QSTR(MP_QSTR_ct_equals),
MP_ROM_PTR(&mod_trezorcrypto_ct_equals_obj)},
};
STATIC MP_DEFINE_CONST_DICT(mod_trezorcrypto_monero_globals,
mod_trezorcrypto_monero_globals_table);
STATIC const mp_obj_module_t mod_trezorcrypto_monero_module = {
.base = {&mp_type_module},
.globals = (mp_obj_dict_t *)&mod_trezorcrypto_monero_globals,
};
Reference in new issue View Git Blame Copy Permalink