mirror of
https://git.libssh.org/projects/libssh.git
synced 2025-07-28 01:41:48 +03:00
Signed-off-by: Jakub Jelen <jjelen@redhat.com> Reviewed-by: Andreas Schneider <asn@cryptomilk.org> Reviewed-by: Sahana Prasad <sahana@redhat.com>
177 lines
4.4 KiB
C
177 lines
4.4 KiB
C
#include "config.h"
|
|
|
|
#define LIBSSH_STATIC
|
|
|
|
#include "torture.h"
|
|
#include "libssh/bignum.h"
|
|
#include "libssh/string.h"
|
|
|
|
static void check_str(int n, ssh_string str)
|
|
{
|
|
if (n > 0 && n <= 127) {
|
|
assert_int_equal(1, ntohl(str->size));
|
|
assert_int_equal(n, str->data[0]);
|
|
} else if (n > 127 && n <= 255) {
|
|
assert_int_equal(2, ntohl(str->size));
|
|
assert_int_equal(0, str->data[0]);
|
|
assert_int_equal(n, str->data[1]);
|
|
} else if (n > 255 && n <= 32767) {
|
|
assert_int_equal(2, ntohl(str->size));
|
|
assert_int_equal(n >> 8, str->data[0]);
|
|
assert_int_equal(n & 0xFF, str->data[1]);
|
|
} else {
|
|
assert_int_equal(3, ntohl(str->size));
|
|
assert_int_equal(n >> 16, str->data[0]);
|
|
assert_int_equal((n >> 8) & 0xFF, str->data[1]);
|
|
assert_int_equal(n & 0xFF, str->data[2]);
|
|
}
|
|
}
|
|
|
|
static void check_padded_str(int n, ssh_string str)
|
|
{
|
|
assert_int_equal(4, ntohl(str->size));
|
|
if (n > 0 && n <= 255) {
|
|
assert_int_equal(0, str->data[0]);
|
|
assert_int_equal(0, str->data[1]);
|
|
assert_int_equal(0, str->data[2]);
|
|
assert_int_equal(n, str->data[3]);
|
|
} else if (n > 255 && n <= 65535) {
|
|
assert_int_equal(0, str->data[0]);
|
|
assert_int_equal(0, str->data[1]);
|
|
assert_int_equal(n >> 8, str->data[2]);
|
|
assert_int_equal(n & 0xFF, str->data[3]);
|
|
} else {
|
|
assert_int_equal(0, str->data[0]);
|
|
assert_int_equal(n >> 16, str->data[1]);
|
|
assert_int_equal((n >> 8) & 0xFF, str->data[2]);
|
|
assert_int_equal(n & 0xFF, str->data[3]);
|
|
}
|
|
}
|
|
|
|
static void check_bignum(int n, const char *nstr)
|
|
{
|
|
bignum num = NULL, num2 = NULL;
|
|
bignum num3 = NULL;
|
|
ssh_string str = NULL;
|
|
char *dec = NULL;
|
|
int rc;
|
|
|
|
num = bignum_new();
|
|
assert_non_null(num);
|
|
|
|
rc = bignum_set_word(num, n);
|
|
assert_int_equal(rc, 1);
|
|
|
|
ssh_print_bignum("num", num);
|
|
|
|
dec = bignum_bn2dec (num);
|
|
assert_non_null (dec);
|
|
assert_string_equal (nstr, dec);
|
|
ssh_crypto_free(dec);
|
|
|
|
/* ssh_make_bignum_string */
|
|
|
|
str = ssh_make_bignum_string(num);
|
|
assert_non_null(str);
|
|
|
|
check_str (n, str);
|
|
|
|
/* ssh_make_string_bn */
|
|
|
|
num2 = ssh_make_string_bn(str);
|
|
ssh_string_free (str);
|
|
assert_non_null(num2);
|
|
|
|
ssh_print_bignum("num2", num2);
|
|
|
|
assert_int_equal (0, bignum_cmp (num, num2));
|
|
|
|
dec = bignum_bn2dec (num2);
|
|
assert_non_null (dec);
|
|
assert_string_equal (nstr, dec);
|
|
ssh_crypto_free(dec);
|
|
|
|
bignum_dup(num, &num3);
|
|
assert_non_null(num3);
|
|
assert_int_equal(0, bignum_cmp(num, num3));
|
|
|
|
bignum_safe_free(num2);
|
|
|
|
/* ssh_make_padded_bignum_string */
|
|
|
|
str = ssh_make_padded_bignum_string(num, 4);
|
|
assert_non_null(str);
|
|
|
|
check_padded_str(n, str);
|
|
|
|
num2 = ssh_make_string_bn(str);
|
|
ssh_string_free(str);
|
|
assert_non_null(num2);
|
|
|
|
ssh_print_bignum("num2", num2);
|
|
|
|
assert_true(bignum_cmp(num, num2) == 0);
|
|
|
|
dec = bignum_bn2dec(num2);
|
|
assert_non_null(dec);
|
|
assert_string_equal(nstr, dec);
|
|
ssh_crypto_free(dec);
|
|
|
|
/* negative test */
|
|
str = ssh_make_padded_bignum_string(num, 2);
|
|
if (n > 65535) {
|
|
/* larger values need larger padding! */
|
|
assert_null(str);
|
|
} else {
|
|
assert_non_null(str);
|
|
assert_int_equal(2, ntohl(str->size));
|
|
if (n > 0 && n <= 255) {
|
|
assert_int_equal(0, str->data[0]);
|
|
assert_int_equal(n, str->data[1]);
|
|
} else {
|
|
assert_int_equal(n >> 8, str->data[0]);
|
|
assert_int_equal(n & 0xFF, str->data[1]);
|
|
}
|
|
ssh_string_free(str);
|
|
}
|
|
|
|
bignum_safe_free(num);
|
|
bignum_safe_free(num2);
|
|
bignum_safe_free(num3);
|
|
}
|
|
|
|
|
|
static void torture_bignum(void **state) {
|
|
(void) state; /* unused */
|
|
|
|
ssh_set_log_level(SSH_LOG_TRACE);
|
|
|
|
check_bignum (1, "1");
|
|
check_bignum (17, "17");
|
|
check_bignum (42, "42");
|
|
check_bignum (127, "127");
|
|
check_bignum (128, "128");
|
|
check_bignum (254, "254");
|
|
check_bignum (255, "255");
|
|
check_bignum (256, "256");
|
|
check_bignum (257, "257");
|
|
check_bignum (300, "300");
|
|
check_bignum (32767, "32767");
|
|
check_bignum (32768, "32768");
|
|
check_bignum (65535, "65535");
|
|
check_bignum (65536, "65536");
|
|
}
|
|
|
|
int torture_run_tests(void) {
|
|
int rc;
|
|
struct CMUnitTest tests[] = {
|
|
cmocka_unit_test(torture_bignum),
|
|
};
|
|
|
|
ssh_init();
|
|
torture_filter_tests(tests);
|
|
rc = cmocka_run_group_tests(tests, NULL, NULL);
|
|
ssh_finalize();
|
|
return rc;
|
|
}
|