Skip to content

Commit 5f6ceaf

Browse files
committed
schnorrsig: allow setting MSGLEN != 32 in benchmark
1 parent fdd06b7 commit 5f6ceaf

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/bench_schnorrsig.c

+5-4
Original file line numberDiff line numberDiff line change
@@ -28,13 +28,13 @@ typedef struct {
2828
void bench_schnorrsig_sign(void* arg, int iters) {
2929
bench_schnorrsig_data *data = (bench_schnorrsig_data *)arg;
3030
int i;
31-
unsigned char msg[MSGLEN] = "benchmarkexamplemessagetemplate";
31+
unsigned char msg[MSGLEN] = {0};
3232
unsigned char sig[64];
3333

3434
for (i = 0; i < iters; i++) {
3535
msg[0] = i;
3636
msg[1] = i >> 8;
37-
CHECK(secp256k1_schnorrsig_sign(data->ctx, sig, msg, data->keypairs[i], NULL));
37+
CHECK(secp256k1_schnorrsig_sign_custom(data->ctx, sig, msg, MSGLEN, data->keypairs[i], NULL));
3838
}
3939
}
4040

@@ -60,6 +60,7 @@ int main(void) {
6060
data.msgs = (const unsigned char **)malloc(iters * sizeof(unsigned char *));
6161
data.sigs = (const unsigned char **)malloc(iters * sizeof(unsigned char *));
6262

63+
CHECK(MSGLEN >= 4);
6364
for (i = 0; i < iters; i++) {
6465
unsigned char sk[32];
6566
unsigned char *msg = (unsigned char *)malloc(MSGLEN);
@@ -71,7 +72,7 @@ int main(void) {
7172
msg[1] = sk[1] = i >> 8;
7273
msg[2] = sk[2] = i >> 16;
7374
msg[3] = sk[3] = i >> 24;
74-
memset(&msg[4], 'm', 28);
75+
memset(&msg[4], 'm', MSGLEN - 4);
7576
memset(&sk[4], 's', 28);
7677

7778
data.keypairs[i] = keypair;
@@ -80,7 +81,7 @@ int main(void) {
8081
data.sigs[i] = sig;
8182

8283
CHECK(secp256k1_keypair_create(data.ctx, keypair, sk));
83-
CHECK(secp256k1_schnorrsig_sign(data.ctx, sig, msg, keypair, NULL));
84+
CHECK(secp256k1_schnorrsig_sign_custom(data.ctx, sig, msg, MSGLEN, keypair, NULL));
8485
CHECK(secp256k1_keypair_xonly_pub(data.ctx, &pk, NULL, keypair));
8586
CHECK(secp256k1_xonly_pubkey_serialize(data.ctx, pk_char, &pk) == 1);
8687
}

0 commit comments

Comments
 (0)