1
0
mirror of https://git.libssh.org/projects/libssh.git synced 2025-07-29 13:01:13 +03:00

tests: Convert torture_connect to a cwrap test

Signed-off-by: Andreas Schneider <asn@cryptomilk.org>
This commit is contained in:
Andreas Schneider
2014-10-13 10:00:25 +02:00
parent 14f1ce2e5b
commit d403c01a30
3 changed files with 81 additions and 28 deletions

View File

@ -28,35 +28,52 @@
#include <netinet/in.h>
#include <sys/types.h>
#include <sys/socket.h>
#define HOST "localhost"
/* Should work until Apnic decides to assign it :) */
#define BLACKHOLE "1.1.1.1"
static int setup(void **state) {
int verbosity=torture_libssh_verbosity();
ssh_session session = ssh_new();
ssh_options_set(session, SSH_OPTIONS_LOG_VERBOSITY, &verbosity);
*state = session;
static int sshd_setup(void **state)
{
torture_setup_sshd_server(state);
return 0;
}
static int teardown(void **state) {
ssh_session session = *state;
ssh_disconnect(session);
ssh_free(session);
static int sshd_teardown(void **state) {
torture_teardown_sshd_server(state);
return 0;
}
static int session_setup(void **state)
{
struct torture_state *s = *state;
int verbosity = torture_libssh_verbosity();
s->ssh.session = ssh_new();
assert_non_null(s->ssh.session);
ssh_options_set(s->ssh.session, SSH_OPTIONS_LOG_VERBOSITY, &verbosity);
return 0;
}
static int session_teardown(void **state)
{
struct torture_state *s = *state;
ssh_disconnect(s->ssh.session);
ssh_free(s->ssh.session);
return 0;
}
static void torture_connect_nonblocking(void **state) {
ssh_session session = *state;
struct torture_state *s = *state;
ssh_session session = s->ssh.session;
int rc;
rc = ssh_options_set(session, SSH_OPTIONS_HOST, HOST);
rc = ssh_options_set(session, SSH_OPTIONS_HOST, TORTURE_SSH_SERVER);
assert_true(rc == SSH_OK);
ssh_set_blocking(session,0);
@ -68,8 +85,10 @@ static void torture_connect_nonblocking(void **state) {
assert_true(rc == SSH_OK);
}
#if 0 /* This does not work with socket_wrapper */
static void torture_connect_timeout(void **state) {
ssh_session session = *state;
struct torture_state *s = *state;
ssh_session session = s->ssh.session;
struct timeval before, after;
int rc;
long timeout = 2;
@ -94,13 +113,15 @@ static void torture_connect_timeout(void **state) {
sec--;
assert_in_range(sec, 1, 3);
}
#endif
static void torture_connect_double(void **state) {
ssh_session session = *state;
struct torture_state *s = *state;
ssh_session session = s->ssh.session;
int rc;
rc = ssh_options_set(session, SSH_OPTIONS_HOST, HOST);
rc = ssh_options_set(session, SSH_OPTIONS_HOST, TORTURE_SSH_SERVER);
assert_true(rc == SSH_OK);
rc = ssh_connect(session);
assert_true(rc == SSH_OK);
@ -116,12 +137,15 @@ static void torture_connect_failure(void **state) {
* ssh_new/ssh_disconnect/ssh_free sequence doesn't crash/leak
* and the behavior of a double ssh_disconnect
*/
ssh_session session = *state;
struct torture_state *s = *state;
ssh_session session = s->ssh.session;
ssh_disconnect(session);
}
static void torture_connect_socket(void **state) {
ssh_session session = *state;
struct torture_state *s = *state;
ssh_session session = s->ssh.session;
int rc;
int sock_fd = 0;
@ -132,7 +156,7 @@ static void torture_connect_socket(void **state) {
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(22);
server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
server_addr.sin_addr.s_addr = inet_addr(TORTURE_SSH_SERVER);
rc = connect(sock_fd, (struct sockaddr *)&server_addr, sizeof(server_addr));
assert_true(rc == 0);
@ -146,17 +170,19 @@ static void torture_connect_socket(void **state) {
int torture_run_tests(void) {
int rc;
struct CMUnitTest tests[] = {
cmocka_unit_test_setup_teardown(torture_connect_nonblocking, setup, teardown),
cmocka_unit_test_setup_teardown(torture_connect_double, setup, teardown),
cmocka_unit_test_setup_teardown(torture_connect_failure, setup, teardown),
cmocka_unit_test_setup_teardown(torture_connect_timeout, setup, teardown),
cmocka_unit_test_setup_teardown(torture_connect_socket, setup, teardown),
cmocka_unit_test_setup_teardown(torture_connect_nonblocking, session_setup, session_teardown),
cmocka_unit_test_setup_teardown(torture_connect_double, session_setup, session_teardown),
cmocka_unit_test_setup_teardown(torture_connect_failure, session_setup, session_teardown),
#if 0
cmocka_unit_test_setup_teardown(torture_connect_timeout, session_setup, session_teardown),
#endif
cmocka_unit_test_setup_teardown(torture_connect_socket, session_setup, session_teardown),
};
ssh_init();
torture_filter_tests(tests);
rc = cmocka_run_group_tests(tests, NULL, NULL);
rc = cmocka_run_group_tests(tests, sshd_setup, sshd_teardown);
ssh_finalize();
return rc;