forked from mupq/pqm4
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtestvectors.c
124 lines (101 loc) · 2.66 KB
/
testvectors.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
/* Deterministic randombytes by Daniel J. Bernstein */
/* taken from SUPERCOP (https://bench.cr.yp.to) */
#include "api.h"
#include "stm32wrapper.h"
#include "randombytes.h"
#include <stdio.h>
#include <string.h>
#include <stdint.h>
#define MAXMLEN 2048
typedef uint32_t uint32;
static void printbytes(const unsigned char *x, unsigned long long xlen)
{
char out[3];
unsigned long long i;
for(i=0;i<xlen;i++) {
sprintf(out, "%02x", x[i]);
send_USART_bytes((unsigned char *)out, 2);
}
send_USART_str("");
}
static uint32 seed[32] = { 3,1,4,1,5,9,2,6,5,3,5,8,9,7,9,3,2,3,8,4,6,2,6,4,3,3,8,3,2,7,9,5 } ;
static uint32 in[12];
static uint32 out[8];
static int outleft = 0;
#define ROTATE(x,b) (((x) << (b)) | ((x) >> (32 - (b))))
#define MUSH(i,b) x = t[i] += (((x ^ seed[i]) + sum) ^ ROTATE(x,b));
static void surf(void)
{
uint32 t[12]; uint32 x; uint32 sum = 0;
int r; int i; int loop;
for (i = 0;i < 12;++i) t[i] = in[i] ^ seed[12 + i];
for (i = 0;i < 8;++i) out[i] = seed[24 + i];
x = t[11];
for (loop = 0;loop < 2;++loop) {
for (r = 0;r < 16;++r) {
sum += 0x9e3779b9;
MUSH(0,5) MUSH(1,7) MUSH(2,9) MUSH(3,13)
MUSH(4,5) MUSH(5,7) MUSH(6,9) MUSH(7,13)
MUSH(8,5) MUSH(9,7) MUSH(10,9) MUSH(11,13)
}
for (i = 0;i < 8;++i) out[i] ^= t[i + 4];
}
}
void randombytes(unsigned char *x,unsigned long long xlen)
{
while (xlen > 0) {
if (!outleft) {
if (!++in[0]) if (!++in[1]) if (!++in[2]) ++in[3];
surf();
outleft = 8;
}
*x = out[--outleft];
++x;
--xlen;
}
}
int main(void)
{
unsigned char sk[CRYPTO_SECRETKEYBYTES];
unsigned char pk[CRYPTO_PUBLICKEYBYTES];
unsigned char mi[MAXMLEN];
unsigned char sm[MAXMLEN+CRYPTO_BYTES];
unsigned long long smlen;
unsigned long long mlen;
int r;
unsigned long long i,j;
clock_setup(CLOCK_FAST);
gpio_setup();
usart_setup(115200);
rng_enable();
send_USART_str("==========================");
for(i=0; i<MAXMLEN; i=(i==0)?i+1:i<<1)
{
randombytes(mi,i);
crypto_sign_keypair(pk, sk);
printbytes(pk,CRYPTO_PUBLICKEYBYTES);
printbytes(sk,CRYPTO_SECRETKEYBYTES);
crypto_sign(sm, &smlen, mi, i, sk);
printbytes(sm, smlen);
// By relying on m == sm we prevent having to allocate CRYPTO_BYTES twice
r = crypto_sign_open(sm, &mlen, sm, smlen, pk);
if(r)
{
send_USART_str("ERROR: signature verification failed");
send_USART_str("#");
return -1;
}
for(j=0;j<i;j++)
{
if(sm[j]!=mi[j])
{
send_USART_str("ERROR: message recovery failed");
send_USART_str("#");
return -1;
}
}
}
send_USART_str("#");
while(1);
return 0;
}