shinys000114/libssh
1
0
Fork 0
mirror of https://git.libssh.org/projects/libssh.git synced 2026-02-12 03:00:26 +09:00
Code Issues Packages Projects Releases Wiki Activity

tests: Migrated torture_auth to cmockery.

Browse Source
This commit is contained in:
Andreas Schneider
2010-12-28 16:41:14 +01:00
parent 0b2cf70622
commit 8bd29736d2
2 changed files with 94 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/client/CMakeLists.txt
View File

@@ -1,6 +1,6 @@
project(clienttests C) project(clienttests C)
add_cmockery_test(torture_algorithms torture_algorithms.c ${TORTURE_LIBRARY}) add_cmockery_test(torture_algorithms torture_algorithms.c ${TORTURE_LIBRARY})
#add_check_test(torture_auth torture_auth.c ${TORTURE_LIBRARY}) add_cmockery_test(torture_auth torture_auth.c ${TORTURE_LIBRARY})
#add_check_test(torture_knownhosts torture_knownhosts.c ${TORTURE_LIBRARY}) #add_check_test(torture_knownhosts torture_knownhosts.c ${TORTURE_LIBRARY})
#add_check_test(torture_proxycommand torture_proxycommand.c ${TORTURE_LIBRARY}) #add_check_test(torture_proxycommand torture_proxycommand.c ${TORTURE_LIBRARY})

154
tests/client/torture_auth.c
View File

@@ -25,89 +25,109 @@
#include "libssh/libssh.h" #include "libssh/libssh.h"
#include "libssh/priv.h" #include "libssh/priv.h"
#include "libssh/session.h" #include "libssh/session.h"
ssh_session session;
static void setup(void) { static void setup(void **state) {
int verbosity=torture_libssh_verbosity(); int verbosity = torture_libssh_verbosity();
session = ssh_new(); ssh_session session = ssh_new();
ssh_options_set(session,SSH_OPTIONS_HOST,"localhost");
ssh_options_set(session,SSH_OPTIONS_LOG_VERBOSITY,&verbosity); ssh_options_set(session, SSH_OPTIONS_HOST, "localhost");
ssh_options_set(session, SSH_OPTIONS_LOG_VERBOSITY, &verbosity);
*state = session;
} }
static void teardown(void) { static void teardown(void **state) {
ssh_disconnect(session); ssh_disconnect(*state);
ssh_free(session); ssh_free(*state);
} }
START_TEST (torture_auth_kbdint) static void torture_auth_kbdint(void **state) {
{ ssh_session session = *state;
char *user = getenv("TORTURE_USER");
char *password = getenv("TORTURE_PASSWORD");
int rc; int rc;
char *user=getenv("TORTURE_USER");
char *password=getenv("TORTURE_PASSWORD");
ck_assert_msg(user != NULL, "Please set the environment variable TORTURE_USER"
" to enable this test");
ck_assert_msg(password != NULL, "Please set the environment variable "
"TORTURE_PASSWORD to enable this test");
ssh_options_set(session,SSH_OPTIONS_USER,user);
rc=ssh_connect(session);
ck_assert_msg(rc==SSH_OK,ssh_get_error(session));
rc=ssh_userauth_none(session,NULL); if (user == NULL) {
/* This request should return a SSH_REQUEST_DENIED error */ print_message("*** Please set the environment variable TORTURE_USER"
if(rc == SSH_ERROR){ " to enable this test!!\n");
ck_assert_msg(ssh_get_error_code(session)==SSH_REQUEST_DENIED, return;
ssh_get_error(session));
} }
ck_assert_msg(ssh_auth_list(session) & SSH_AUTH_METHOD_INTERACTIVE,
"SSH server doesn't allow keyboard-interactive"); if (password == NULL) {
rc=ssh_userauth_kbdint(session,NULL,NULL); print_message("*** Please set the environment variable "
ck_assert_msg(rc==SSH_AUTH_INFO,ssh_get_error(session)); "TORTURE_PASSWORD to enable this test!!\n");
ck_assert_int_eq(ssh_userauth_kbdint_getnprompts(session),1); return;
ssh_userauth_kbdint_setanswer(session,0,password); }
rc=ssh_userauth_kbdint(session,NULL,NULL);
rc = ssh_options_set(session, SSH_OPTIONS_USER, user);
assert_true(rc == SSH_OK);
rc = ssh_connect(session);
assert_true(rc == SSH_OK);
rc = ssh_userauth_none(session,NULL);
/* This request should return a SSH_REQUEST_DENIED error */
if (rc == SSH_ERROR) {
assert_true(ssh_get_error_code(session) == SSH_REQUEST_DENIED);
}
assert_true(ssh_auth_list(session) & SSH_AUTH_METHOD_INTERACTIVE);
rc = ssh_userauth_kbdint(session, NULL, NULL);
assert_true(rc == SSH_AUTH_INFO);
assert_int_equal(ssh_userauth_kbdint_getnprompts(session), 1);
rc = ssh_userauth_kbdint_setanswer(session, 0, password);
assert_true(rc < 0);
rc = ssh_userauth_kbdint(session, NULL, NULL);
/* Sometimes, SSH server send an empty query at the end of exchange */ /* Sometimes, SSH server send an empty query at the end of exchange */
if(rc==SSH_AUTH_INFO){ if(rc == SSH_AUTH_INFO) {
ck_assert_int_eq(ssh_userauth_kbdint_getnprompts(session),0); assert_int_equal(ssh_userauth_kbdint_getnprompts(session), 0);
rc=ssh_userauth_kbdint(session,NULL,NULL); rc = ssh_userauth_kbdint(session, NULL, NULL);
} }
ck_assert_msg(rc==SSH_AUTH_SUCCESS,ssh_get_error(session)); assert_true(rc == SSH_AUTH_SUCCESS);
} }
END_TEST
START_TEST (torture_auth_password) static void torture_auth_password(void **state) {
{ ssh_session session = *state;
char *user = getenv("TORTURE_USER");
char *password = getenv("TORTURE_PASSWORD");
int rc; int rc;
char *user=getenv("TORTURE_USER");
char *password=getenv("TORTURE_PASSWORD");
ck_assert_msg(user != NULL, "Please set the environment variable TORTURE_USER"
" to enable this test");
ck_assert_msg(password != NULL, "Please set the environment variable "
"TORTURE_PASSWORD to enable this test");
ssh_options_set(session,SSH_OPTIONS_USER,user);
rc=ssh_connect(session);
ck_assert_msg(rc==SSH_OK,ssh_get_error(session));
rc=ssh_userauth_none(session,NULL); if (user == NULL) {
/* This request should return a SSH_REQUEST_DENIED error */ print_message("*** Please set the environment variable TORTURE_USER"
if(rc == SSH_ERROR){ " to enable this test!!\n");
ck_assert_msg(ssh_get_error_code(session)==SSH_REQUEST_DENIED, return;
ssh_get_error(session));
} }
ck_assert_msg(ssh_auth_list(session) & SSH_AUTH_METHOD_PASSWORD,
"SSH server doesn't allow password authentication");
rc=ssh_userauth_password(session,NULL,password);
ck_assert_msg(rc==SSH_AUTH_SUCCESS,ssh_get_error(session));
}
END_TEST
Suite *torture_make_suite(void) { if (password == NULL) {
Suite *s = suite_create("libssh_auth"); print_message("*** Please set the environment variable "
"TORTURE_PASSWORD to enable this test!!\n");
return;
}
torture_create_case_fixture(s, "torture_auth_kbint", rc = ssh_options_set(session, SSH_OPTIONS_USER, user);
torture_auth_kbdint, setup, teardown); assert_true(rc == SSH_OK);
torture_create_case_fixture(s, "torture_auth_password",
torture_auth_password, setup, teardown); rc = ssh_connect(session);
return s; assert_true(rc == SSH_OK);
rc = ssh_userauth_none(session, NULL);
/* This request should return a SSH_REQUEST_DENIED error */
if (rc == SSH_ERROR) {
assert_true(ssh_get_error_code(session) == SSH_REQUEST_DENIED);
}
assert_true(ssh_auth_list(session) & SSH_AUTH_METHOD_PASSWORD);
rc = ssh_userauth_password(session, NULL, password);
assert_true(rc == SSH_AUTH_SUCCESS);
} }
int torture_run_tests(void) {
const UnitTest tests[] = {
unit_test_setup_teardown(torture_auth_kbdint, setup, teardown),
unit_test_setup_teardown(torture_auth_password, setup, teardown),
};
return run_tests(tests);
}