diff options
Diffstat (limited to '')
| -rw-r--r-- | librecrypt_get_encoding.c | 91 |
1 files changed, 90 insertions, 1 deletions
diff --git a/librecrypt_get_encoding.c b/librecrypt_get_encoding.c index 5867758..f253e70 100644 --- a/librecrypt_get_encoding.c +++ b/librecrypt_get_encoding.c @@ -36,16 +36,105 @@ librecrypt_get_encoding(const char *settings, size_t len, char *pad_out, int *st #else +#define NSA "$~no~such~algorithm~$" + + +static int +check_encoding_lut(const char *lut, const char *alpha) +{ + size_t i; + for (i = 0u; i < 256u; i++) + if (lut[i] != alpha[i % 64u]) + return 0; + return 1; +} + + +static int +check_decoding_lut(const unsigned char *lut, const char *alpha) +{ + size_t i, invalid_count = 0u; + for (i = 0u; i < 64u; i++) + if (lut[(unsigned char)alpha[i]] != (unsigned char)i) + return 0; + for (i = 0u; i < 256u; i++) { + if (lut[i] == 0xFFu) + invalid_count += 1u; + else + EXPECT(lut[i] < 64u); + } + return invalid_count == 256u - 64u; +} + + +#define CHECK(PREFIX, ALPHABET, PAD, STRICT_PAD) \ + do {\ + pad = (char)~(PAD);\ + strict_pad = -1;\ + elut = librecrypt_get_encoding(PREFIX, sizeof(PREFIX) - 1u, &pad, &strict_pad, 0);\ + EXPECT(elut != NULL);\ + EXPECT(pad == (PAD));\ + EXPECT(strict_pad == (STRICT_PAD));\ + EXPECT(check_encoding_lut(elut, ALPHABET));\ + \ + pad = (char)~(PAD);\ + strict_pad = -1;\ + dlut = librecrypt_get_encoding(PREFIX, sizeof(PREFIX) - 1u, &pad, &strict_pad, 1);\ + EXPECT(dlut != NULL);\ + EXPECT(pad == (PAD));\ + EXPECT(strict_pad == (STRICT_PAD));\ + EXPECT(check_decoding_lut(dlut, ALPHABET));\ + \ + pad = (char)~(PAD);\ + strict_pad = -1;\ + elut = librecrypt_get_encoding(NSA">"PREFIX, sizeof(NSA">"PREFIX) - 1u, &pad, &strict_pad, 0);\ + EXPECT(elut != NULL);\ + EXPECT(pad == (PAD));\ + EXPECT(strict_pad == (STRICT_PAD));\ + EXPECT(check_encoding_lut(elut, ALPHABET));\ + } while (0) + + +#define DIGIT "0123456789" +#define LOWER "abcdefghijklmnopqrstuvwxyz" +#define UPPER "ABCDEFGHIJKLMNOPQRSTUVWXYZ" + + int main(void) { + const char *elut; + const unsigned char *dlut; + char pad; + int strict_pad; + SET_UP_ALARM(); INIT_RESOURCE_TEST(); + errno = 0; + EXPECT(librecrypt_get_encoding(NSA, 1u, &pad, &strict_pad, 0) == NULL); + EXPECT(errno == ENOSYS); + + errno = 0; + EXPECT(librecrypt_get_encoding(NSA, 1u, &pad, &strict_pad, 1) == NULL); + EXPECT(errno == ENOSYS); + + errno = 0; + EXPECT(librecrypt_get_encoding(">"NSA, 1u, &pad, &strict_pad, 0) == NULL); + EXPECT(errno == ENOSYS); + + errno = 0; + EXPECT(librecrypt_get_encoding(">"NSA, 1u, &pad, &strict_pad, 1) == NULL); + EXPECT(errno == ENOSYS); + + IF__argon2i__SUPPORTED(CHECK("$argon2i$", UPPER LOWER DIGIT "+/", '=', 0);) + IF__argon2d__SUPPORTED(CHECK("$argon2d$", UPPER LOWER DIGIT "+/", '=', 0);) + IF__argon2id__SUPPORTED(CHECK("$argon2id$", UPPER LOWER DIGIT "+/", '=', 0);) + IF__argon2ds__SUPPORTED(CHECK("$argon2ds$", UPPER LOWER DIGIT "+/", '=', 0);) + STOP_RESOURCE_TEST(); return 0; } #endif -/* TODO test */ |