132 строки
3.5 KiB
C
132 строки
3.5 KiB
C
#define LIBSSH_STATIC
|
|
|
|
#ifndef _WIN32
|
|
# include <pwd.h>
|
|
#endif
|
|
|
|
#include "torture.h"
|
|
#include <libssh/session.h>
|
|
#include <libssh/misc.h>
|
|
|
|
static void setup(void **state) {
|
|
ssh_session session = ssh_new();
|
|
*state = session;
|
|
}
|
|
|
|
static void teardown(void **state) {
|
|
ssh_free(*state);
|
|
}
|
|
|
|
static void torture_options_set_host(void **state) {
|
|
ssh_session session = *state;
|
|
int rc;
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_HOST, "localhost");
|
|
assert_true(rc == 0);
|
|
assert_string_equal(session->host, "localhost");
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_HOST, "guru@meditation");
|
|
assert_true(rc == 0);
|
|
assert_string_equal(session->host, "meditation");
|
|
assert_string_equal(session->username, "guru");
|
|
}
|
|
|
|
static void torture_options_set_port(void **state) {
|
|
ssh_session session = *state;
|
|
int rc;
|
|
unsigned int port = 42;
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_PORT, &port);
|
|
assert_true(rc == 0);
|
|
assert_true(session->port == port);
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_PORT_STR, "23");
|
|
assert_true(rc == 0);
|
|
assert_true(session->port == 23);
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_PORT_STR, "five");
|
|
assert_true(rc == 0);
|
|
assert_true(session->port == 0);
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_PORT, NULL);
|
|
assert_true(rc == 0);
|
|
assert_true(session->port == 22);
|
|
}
|
|
|
|
static void torture_options_set_fd(void **state) {
|
|
ssh_session session = *state;
|
|
socket_t fd = 42;
|
|
int rc;
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_FD, &fd);
|
|
assert_true(rc == 0);
|
|
assert_true(session->fd == fd);
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_FD, NULL);
|
|
assert_true(rc == 0);
|
|
assert_true(session->fd == SSH_INVALID_SOCKET);
|
|
}
|
|
|
|
static void torture_options_set_user(void **state) {
|
|
ssh_session session = *state;
|
|
int rc;
|
|
#ifndef _WIN32
|
|
# ifndef NSS_BUFLEN_PASSWD
|
|
# define NSS_BUFLEN_PASSWD 4096
|
|
# endif /* NSS_BUFLEN_PASSWD */
|
|
struct passwd pwd;
|
|
struct passwd *pwdbuf;
|
|
char buf[NSS_BUFLEN_PASSWD];
|
|
|
|
/* get local username */
|
|
rc = getpwuid_r(getuid(), &pwd, buf, NSS_BUFLEN_PASSWD, &pwdbuf);
|
|
assert_true(rc == 0);
|
|
#endif /* _WIN32 */
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_USER, "guru");
|
|
assert_true(rc == 0);
|
|
assert_string_equal(session->username, "guru");
|
|
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_USER, NULL);
|
|
assert_true(rc == 0);
|
|
|
|
#ifndef _WIN32
|
|
assert_string_equal(session->username, pwd.pw_name);
|
|
#endif
|
|
}
|
|
|
|
/* TODO */
|
|
#if 0
|
|
static voidtorture_options_set_sshdir)
|
|
{
|
|
}
|
|
END_TEST
|
|
#endif
|
|
|
|
static void torture_options_set_identity(void **state) {
|
|
ssh_session session = *state;
|
|
int rc;
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_ADD_IDENTITY, "identity1");
|
|
assert_true(rc == 0);
|
|
assert_string_equal(session->identity->root->data, "identity1");
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_IDENTITY, "identity2");
|
|
assert_true(rc == 0);
|
|
assert_string_equal(session->identity->root->data, "identity2");
|
|
assert_string_equal(session->identity->root->next->data, "identity1");
|
|
}
|
|
|
|
int torture_run_tests(void) {
|
|
const UnitTest tests[] = {
|
|
unit_test_setup_teardown(torture_options_set_host, setup, teardown),
|
|
unit_test_setup_teardown(torture_options_set_port, setup, teardown),
|
|
unit_test_setup_teardown(torture_options_set_fd, setup, teardown),
|
|
unit_test_setup_teardown(torture_options_set_user, setup, teardown),
|
|
unit_test_setup_teardown(torture_options_set_identity, setup, teardown),
|
|
};
|
|
|
|
return run_tests(tests);
|
|
}
|