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:
@ -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;
|
||||
|
Reference in New Issue
Block a user