129 lines
4.2 KiB
C
129 lines
4.2 KiB
C
#include "test.h"
|
|
|
|
#define RSA_MSGSIZE 78
|
|
|
|
|
|
int rsa_test(void)
|
|
{
|
|
unsigned char in[1024], out[1024], tmp[1024];
|
|
rsa_key key;
|
|
int hash_idx, prng_idx, stat, stat2;
|
|
unsigned long rsa_msgsize, len, len2;
|
|
static unsigned char lparam[] = { 0x01, 0x02, 0x03, 0x04 };
|
|
|
|
hash_idx = find_hash("sha1");
|
|
prng_idx = find_prng("yarrow");
|
|
if (hash_idx == -1 || prng_idx == -1) {
|
|
printf("rsa_test requires SHA1 and yarrow");
|
|
return 1;
|
|
}
|
|
|
|
/* make a random key */
|
|
DO(rsa_make_key(&test_yarrow, prng_idx, 1024/8, 65537, &key));
|
|
|
|
/* encrypt the key (without lparam) */
|
|
for (rsa_msgsize = 1; rsa_msgsize <= 86; rsa_msgsize++) {
|
|
/* make a random key/msg */
|
|
yarrow_read(in, rsa_msgsize, &test_yarrow);
|
|
|
|
len = sizeof(out);
|
|
len2 = rsa_msgsize;
|
|
|
|
DO(rsa_encrypt_key(in, rsa_msgsize, out, &len, NULL, 0, &test_yarrow, prng_idx, hash_idx, &key));
|
|
/* change a byte */
|
|
out[8] ^= 1;
|
|
DO(rsa_decrypt_key(out, len, tmp, &len2, NULL, 0, &test_yarrow, prng_idx, hash_idx, &stat2, &key));
|
|
/* change a byte back */
|
|
out[8] ^= 1;
|
|
if (len2 != rsa_msgsize) {
|
|
printf("\nrsa_decrypt_key mismatch len %lu (first decrypt)", len2);
|
|
return 1;
|
|
}
|
|
|
|
len2 = rsa_msgsize;
|
|
DO(rsa_decrypt_key(out, len, tmp, &len2, NULL, 0, &test_yarrow, prng_idx, hash_idx, &stat, &key));
|
|
if (!(stat == 1 && stat2 == 0)) {
|
|
printf("rsa_decrypt_key failed");
|
|
return 1;
|
|
}
|
|
if (len2 != rsa_msgsize || memcmp(tmp, in, rsa_msgsize)) {
|
|
int x;
|
|
printf("\nrsa_decrypt_key mismatch, len %lu (second decrypt)\n", len2);
|
|
printf("Original contents: \n");
|
|
for (x = 0; x < rsa_msgsize; ) {
|
|
printf("%02x ", in[x]);
|
|
if (!(++x % 16)) {
|
|
printf("\n");
|
|
}
|
|
}
|
|
printf("\n");
|
|
printf("Output contents: \n");
|
|
for (x = 0; x < rsa_msgsize; ) {
|
|
printf("%02x ", out[x]);
|
|
if (!(++x % 16)) {
|
|
printf("\n");
|
|
}
|
|
}
|
|
printf("\n");
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
/* encrypt the key (with lparam) */
|
|
for (rsa_msgsize = 1; rsa_msgsize <= 86; rsa_msgsize++) {
|
|
len = sizeof(out);
|
|
len2 = rsa_msgsize;
|
|
DO(rsa_encrypt_key(in, rsa_msgsize, out, &len, lparam, sizeof(lparam), &test_yarrow, prng_idx, hash_idx, &key));
|
|
/* change a byte */
|
|
out[8] ^= 1;
|
|
DO(rsa_decrypt_key(out, len, tmp, &len2, lparam, sizeof(lparam), &test_yarrow, prng_idx, hash_idx, &stat2, &key));
|
|
if (len2 != rsa_msgsize) {
|
|
printf("\nrsa_decrypt_key mismatch len %lu (first decrypt)", len2);
|
|
return 1;
|
|
}
|
|
/* change a byte back */
|
|
out[8] ^= 1;
|
|
|
|
len2 = rsa_msgsize;
|
|
DO(rsa_decrypt_key(out, len, tmp, &len2, lparam, sizeof(lparam), &test_yarrow, prng_idx, hash_idx, &stat, &key));
|
|
if (!(stat == 1 && stat2 == 0)) {
|
|
printf("rsa_decrypt_key failed");
|
|
return 1;
|
|
}
|
|
if (len2 != rsa_msgsize || memcmp(tmp, in, rsa_msgsize)) {
|
|
printf("rsa_decrypt_key mismatch len %lu", len2);
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
/* sign a message (unsalted, lower cholestorol and Atkins approved) now */
|
|
len = sizeof(out);
|
|
DO(rsa_sign_hash(in, 20, out, &len, &test_yarrow, prng_idx, hash_idx, 0, &key));
|
|
DO(rsa_verify_hash(out, len, in, 20, &test_yarrow, prng_idx, hash_idx, 0, &stat, &key));
|
|
/* change a byte */
|
|
in[0] ^= 1;
|
|
DO(rsa_verify_hash(out, len, in, 20, &test_yarrow, prng_idx, hash_idx, 0, &stat2, &key));
|
|
|
|
if (!(stat == 1 && stat2 == 0)) {
|
|
printf("rsa_verify_hash (unsalted) failed, %d, %d", stat, stat2);
|
|
return 1;
|
|
}
|
|
|
|
/* sign a message (salted) now */
|
|
len = sizeof(out);
|
|
DO(rsa_sign_hash(in, 20, out, &len, &test_yarrow, prng_idx, hash_idx, 8, &key));
|
|
DO(rsa_verify_hash(out, len, in, 20, &test_yarrow, prng_idx, hash_idx, 8, &stat, &key));
|
|
/* change a byte */
|
|
in[0] ^= 1;
|
|
DO(rsa_verify_hash(out, len, in, 20, &test_yarrow, prng_idx, hash_idx, 8, &stat2, &key));
|
|
|
|
if (!(stat == 1 && stat2 == 0)) {
|
|
printf("rsa_verify_hash (salted) failed, %d, %d", stat, stat2);
|
|
return 1;
|
|
}
|
|
|
|
/* free the key and return */
|
|
rsa_free(&key);
|
|
return 0;
|
|
}
|