diff options
Diffstat (limited to 'tests/client/torture_sftp_read.c')
-rw-r--r-- | tests/client/torture_sftp_read.c | 69 |
1 files changed, 42 insertions, 27 deletions
diff --git a/tests/client/torture_sftp_read.c b/tests/client/torture_sftp_read.c index 35bb7e43..51578773 100644 --- a/tests/client/torture_sftp_read.c +++ b/tests/client/torture_sftp_read.c @@ -3,48 +3,61 @@ #include "torture.h" #include "sftp.c" +#include <sys/types.h> +#include <pwd.h> + #define MAX_XFER_BUF_SIZE 16384 -static int setup(void **state) +static int sshd_setup(void **state) { - ssh_session session; - struct torture_sftp *t; - const char *host; - const char *user; - const char *password; - - host = getenv("TORTURE_HOST"); - if (host == NULL) { - host = "localhost"; - } + torture_setup_sshd_server(state); - user = getenv("TORTURE_USER"); - password = getenv("TORTURE_PASSWORD"); - - session = torture_ssh_session(host, NULL, user, password); - assert_false(session == NULL); - t = torture_sftp_session(session); - assert_false(t == NULL); + return 0; +} - *state = t; +static int sshd_teardown(void **state) { + torture_teardown_sshd_server(state); return 0; } -static int teardown(void **state) +static int session_setup(void **state) { - struct torture_sftp *t = (struct torture_sftp*) *state; + struct torture_state *s = *state; + struct passwd *pwd; + + pwd = getpwnam("bob"); + assert_non_null(pwd); + setuid(pwd->pw_uid); - assert_false(t == NULL); + s->ssh.session = torture_ssh_session(TORTURE_SSH_SERVER, + NULL, + TORTURE_SSH_USER_ALICE, + NULL); + assert_non_null(s->ssh.session); - torture_rmdirs(t->testdir); - torture_sftp_close(t); + s->ssh.tsftp = torture_sftp_session(s->ssh.session); + assert_non_null(s->ssh.tsftp); + + return 0; +} + +static int session_teardown(void **state) +{ + struct torture_state *s = *state; + + torture_rmdirs(s->ssh.tsftp->testdir); + torture_sftp_close(s->ssh.tsftp); + ssh_disconnect(s->ssh.session); + ssh_free(s->ssh.session); return 0; } static void torture_sftp_read_blocking(void **state) { - struct torture_sftp *t = (struct torture_sftp*) *state; + struct torture_state *s = *state; + struct torture_sftp *t = s->ssh.tsftp; + char libssh_tmp_file[] = "/tmp/libssh_sftp_test_XXXXXX"; char buf[MAX_XFER_BUF_SIZE]; ssize_t bytesread; @@ -79,13 +92,15 @@ static void torture_sftp_read_blocking(void **state) { int torture_run_tests(void) { int rc; struct CMUnitTest tests[] = { - cmocka_unit_test_setup_teardown(torture_sftp_read_blocking, setup, teardown) + cmocka_unit_test_setup_teardown(torture_sftp_read_blocking, + 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; |