Try to cleanup stuff

Signed-off-by: Adrien Gallouët <adrien@gallouet.fr>
This commit is contained in:
Adrien Gallouët 2020-09-13 22:13:44 +00:00
parent e9d2642efc
commit a58bc76804

26
totp.c
View file

@ -89,31 +89,39 @@ sha1(uint8_t *digest, uint8_t *buf, size_t len)
ebe32(&digest[i << 2], x[i]); ebe32(&digest[i << 2], x[i]);
} }
static void
erase(void *buf, size_t len)
{
volatile uint8_t *volatile x = (volatile uint8_t *volatile)buf;
for (size_t i = 0; i < len; i++)
x[i] = 0;
}
int int
main(int argc, char **argv) main(int argc, char **argv)
{ {
uint8_t key[64]; uint8_t h[20];
ssize_t len = read(0, key, sizeof(key)); uint8_t ki[64 + 8] = {0};
uint8_t ko[64 + 20] = {0};
ssize_t len = read(0, ki, 64);
if (len <= 0) if (len <= 0)
return -1; return -1;
uint8_t h[20]; memcpy(ko, ki, len);
uint8_t ki[64 + 8] = {0};
uint8_t ko[64 + 20] = {0};
memcpy(ki, key, len);
memcpy(ko, key, len);
for (int i = 0; i < 64; i++) { for (int i = 0; i < 64; i++) {
ki[i] ^= 0x36; ki[i] ^= 0x36;
ko[i] ^= 0x5c; ko[i] ^= 0x5c;
} }
ebe64(&ki[64], ((uint64_t)time(NULL)) / 30); ebe64(&ki[64], ((uint64_t)time(NULL)) / 30);
sha1(&ko[64], ki, sizeof(ki)); sha1(&ko[64], ki, sizeof(ki));
sha1(h, ko, sizeof(ko)); sha1(h, ko, sizeof(ko));
erase(ki, sizeof(ki));
erase(ko, sizeof(ko));
uint32_t ret = (dbe32(&h[h[19] & 0xF]) & ~(UINT32_C(1) << 31)) uint32_t ret = (dbe32(&h[h[19] & 0xF]) & ~(UINT32_C(1) << 31))
% UINT32_C(1000000); % UINT32_C(1000000);