mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-18 05:28:40 +00:00
133c068f37
This adds an is_canonic parameter to all sign functions. This is a callback that determines if a signature corresponds to some coin specific rules. It is used, e. g., by ethereum (where the recovery byte must be 0 or 1, and not 2 or 3) and or steem signatures (which require both r and s to be between 2^248 and 2^255). This also separates the initialization and the step function of the random number generator, making it easy to restart the signature process with the next random number.
125 lines
3.2 KiB
C
125 lines
3.2 KiB
C
#include <stdio.h>
|
|
#include <assert.h>
|
|
#include <time.h>
|
|
#include <string.h>
|
|
#include <stdint.h>
|
|
#include "curves.h"
|
|
#include "ecdsa.h"
|
|
#include "secp256k1.h"
|
|
#include "nist256p1.h"
|
|
#include "ed25519.h"
|
|
|
|
uint8_t msg[32];
|
|
|
|
void prepare_msg(void)
|
|
{
|
|
for (size_t i = 0; i < sizeof(msg); i++) {
|
|
msg[i] = i * 1103515245;
|
|
}
|
|
}
|
|
|
|
void bench_secp256k1(void) {
|
|
uint8_t sig[64], pub[33], priv[32], pby;
|
|
|
|
const ecdsa_curve *curve = &secp256k1;
|
|
|
|
memcpy(priv, "\xc5\x5e\xce\x85\x8b\x0d\xdd\x52\x63\xf9\x68\x10\xfe\x14\x43\x7c\xd3\xb5\xe1\xfb\xd7\xc6\xa2\xec\x1e\x03\x1f\x05\xe8\x6d\x8b\xd5", 32);
|
|
ecdsa_get_public_key33(curve, priv, pub);
|
|
ecdsa_sign(curve, priv, msg, sizeof(msg), sig, &pby, NULL);
|
|
|
|
clock_t t = clock();
|
|
for (int i = 0 ; i < 500; i++) {
|
|
int res = ecdsa_verify(curve, pub, sig, msg, sizeof(msg));
|
|
assert(res == 0);
|
|
}
|
|
printf("SECP256k1 verifying speed: %0.2f sig/s\n", 500.0f / ((float)(clock() - t) / CLOCKS_PER_SEC));
|
|
}
|
|
|
|
void bench_nist256p1(void) {
|
|
uint8_t sig[64], pub[33], priv[32], pby;
|
|
|
|
const ecdsa_curve *curve = &nist256p1;
|
|
|
|
memcpy(priv, "\xc5\x5e\xce\x85\x8b\x0d\xdd\x52\x63\xf9\x68\x10\xfe\x14\x43\x7c\xd3\xb5\xe1\xfb\xd7\xc6\xa2\xec\x1e\x03\x1f\x05\xe8\x6d\x8b\xd5", 32);
|
|
ecdsa_get_public_key33(curve, priv, pub);
|
|
ecdsa_sign(curve, priv, msg, sizeof(msg), sig, &pby, NULL);
|
|
|
|
clock_t t = clock();
|
|
for (int i = 0 ; i < 500; i++) {
|
|
int res = ecdsa_verify(curve, pub, sig, msg, sizeof(msg));
|
|
assert(res == 0);
|
|
}
|
|
printf("NIST256p1 verifying speed: %0.2f sig/s\n", 500.0f / ((float)(clock() - t) / CLOCKS_PER_SEC));
|
|
}
|
|
|
|
void bench_ed25519(void) {
|
|
ed25519_public_key pk;
|
|
ed25519_secret_key sk;
|
|
ed25519_signature sig;
|
|
|
|
memcpy(pk, "\xc5\x5e\xce\x85\x8b\x0d\xdd\x52\x63\xf9\x68\x10\xfe\x14\x43\x7c\xd3\xb5\xe1\xfb\xd7\xc6\xa2\xec\x1e\x03\x1f\x05\xe8\x6d\x8b\xd5", 32);
|
|
ed25519_publickey(sk, pk);
|
|
ed25519_sign(msg, sizeof(msg), sk, pk, sig);
|
|
|
|
clock_t t = clock();
|
|
for (int i = 0 ; i < 500; i++) {
|
|
int res = ed25519_sign_open(msg, sizeof(msg), pk, sig);
|
|
assert(res == 0);
|
|
}
|
|
printf("Ed25519 verifying speed: %0.2f sig/s\n", 500.0f / ((float)(clock() - t) / CLOCKS_PER_SEC));
|
|
}
|
|
|
|
void test_verify_speed(void) {
|
|
prepare_msg();
|
|
bench_secp256k1();
|
|
bench_nist256p1();
|
|
bench_ed25519();
|
|
}
|
|
|
|
HDNode root;
|
|
|
|
void prepare_node(void)
|
|
{
|
|
hdnode_from_seed((uint8_t *)"NothingToSeeHere", 16, SECP256K1_NAME, &root);
|
|
}
|
|
|
|
void bench_ckd_normal(void) {
|
|
char addr[40];
|
|
clock_t t = clock();
|
|
for (int i = 0; i < 1000; i++) {
|
|
HDNode node = root;
|
|
hdnode_public_ckd(&node, i);
|
|
ecdsa_get_address(node.public_key, 0, addr, 40);
|
|
if (i == 0) {
|
|
printf("address = %s\n", addr);
|
|
}
|
|
}
|
|
printf("CKD normal speed: %0.2f iter/s\n", 1000.0f / ((float)(clock() - t) / CLOCKS_PER_SEC));
|
|
}
|
|
|
|
void bench_ckd_optimized(void) {
|
|
char addr[40];
|
|
curve_point pub;
|
|
ecdsa_read_pubkey(0, root.public_key, &pub);
|
|
clock_t t = clock();
|
|
for (int i = 0; i < 1000; i++) {
|
|
hdnode_public_ckd_address_optimized(&pub, root.public_key, root.chain_code, i, 0, addr, 40);
|
|
if (i == 0) {
|
|
printf("address = %s\n", addr);
|
|
}
|
|
}
|
|
printf("CKD optim speed: %0.2f iter/s\n", 1000.0f / ((float)(clock() - t) / CLOCKS_PER_SEC));
|
|
}
|
|
|
|
void test_ckd_speed(void) {
|
|
prepare_node();
|
|
bench_ckd_normal();
|
|
bench_ckd_optimized();
|
|
}
|
|
|
|
int main(void) {
|
|
test_verify_speed();
|
|
test_ckd_speed();
|
|
return 0;
|
|
}
|