159 строки
3.7 KiB
C
159 строки
3.7 KiB
C
#define LIBSSH_STATIC
|
|
|
|
#include "torture.h"
|
|
#include "options.c"
|
|
|
|
ssh_session session;
|
|
|
|
static void setup(void) {
|
|
session = ssh_new();
|
|
}
|
|
|
|
static void teardown(void) {
|
|
ssh_free(session);
|
|
}
|
|
|
|
START_TEST (torture_options_set_host)
|
|
{
|
|
int rc;
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_HOST, "localhost");
|
|
ck_assert(rc == 0);
|
|
ck_assert_str_eq(session->host, "localhost");
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_HOST, "guru@meditation");
|
|
ck_assert(rc == 0);
|
|
ck_assert_str_eq(session->host, "meditation");
|
|
ck_assert_str_eq(session->username, "guru");
|
|
}
|
|
END_TEST
|
|
|
|
START_TEST (torture_options_set_port)
|
|
{
|
|
int rc;
|
|
unsigned int port = 42;
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_PORT, &port);
|
|
ck_assert(rc == 0);
|
|
ck_assert(session->port == port);
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_PORT_STR, "23");
|
|
ck_assert(rc == 0);
|
|
ck_assert(session->port == 23);
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_PORT_STR, "five");
|
|
ck_assert(rc == 0);
|
|
ck_assert(session->port == 0);
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_PORT, NULL);
|
|
ck_assert(rc == 0);
|
|
ck_assert(session->port == 22);
|
|
}
|
|
END_TEST
|
|
|
|
START_TEST (torture_options_set_fd)
|
|
{
|
|
socket_t fd = 42;
|
|
int rc;
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_FD, &fd);
|
|
ck_assert(rc == 0);
|
|
ck_assert(session->fd == fd);
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_FD, NULL);
|
|
ck_assert(rc == 0);
|
|
ck_assert(session->fd == -1);
|
|
}
|
|
END_TEST
|
|
|
|
START_TEST (torture_options_set_user)
|
|
{
|
|
int rc;
|
|
#ifndef _WIN32
|
|
struct passwd pwd;
|
|
struct passwd *pwdbuf;
|
|
char buf[NSS_BUFLEN_PASSWD];
|
|
#endif
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_USER, "guru");
|
|
ck_assert(rc == 0);
|
|
ck_assert_str_eq(session->username, "guru");
|
|
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_USER, NULL);
|
|
ck_assert(rc == 0);
|
|
|
|
#ifndef _WIN32
|
|
/* get local username */
|
|
rc = getpwuid_r(getuid(), &pwd, buf, NSS_BUFLEN_PASSWD, &pwdbuf);
|
|
ck_assert(rc == 0);
|
|
|
|
ck_assert_str_eq(session->username, pwd.pw_name);
|
|
#endif
|
|
}
|
|
END_TEST
|
|
|
|
/* TODO */
|
|
#if 0
|
|
START_TEST (torture_options_set_sshdir)
|
|
{
|
|
}
|
|
END_TEST
|
|
#endif
|
|
|
|
START_TEST (torture_options_set_identity)
|
|
{
|
|
int rc;
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_ADD_IDENTITY, "identity1");
|
|
ck_assert(rc == 0);
|
|
ck_assert_str_eq(session->identity->root->data, "identity1");
|
|
|
|
rc = ssh_options_set(session, SSH_OPTIONS_IDENTITY, "identity2");
|
|
ck_assert(rc == 0);
|
|
ck_assert_str_eq(session->identity->root->data, "identity2");
|
|
ck_assert_str_eq(session->identity->root->next->data, "identity1");
|
|
}
|
|
END_TEST
|
|
|
|
static Suite *torture_make_suite(void) {
|
|
Suite *s = suite_create("libssh_keyfiles");
|
|
|
|
torture_create_case_fixture(s, "torture_options_set_host",
|
|
torture_options_set_host, setup, teardown);
|
|
torture_create_case_fixture(s, "torture_options_set_port",
|
|
torture_options_set_port, setup, teardown);
|
|
torture_create_case_fixture(s, "torture_options_set_fd",
|
|
torture_options_set_fd, setup, teardown);
|
|
torture_create_case_fixture(s, "torture_options_set_user",
|
|
torture_options_set_user, setup, teardown);
|
|
torture_create_case_fixture(s, "torture_options_set_identity",
|
|
torture_options_set_identity, setup, teardown);
|
|
|
|
return s;
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
Suite *s = NULL;
|
|
SRunner *sr = NULL;
|
|
struct argument_s arguments;
|
|
int nf;
|
|
|
|
ZERO_STRUCT(arguments);
|
|
|
|
torture_cmdline_parse(argc, argv, &arguments);
|
|
|
|
s = torture_make_suite();
|
|
|
|
sr = srunner_create(s);
|
|
if (arguments.nofork) {
|
|
srunner_set_fork_status(sr, CK_NOFORK);
|
|
}
|
|
srunner_run_all(sr, CK_VERBOSE);
|
|
nf = srunner_ntests_failed(sr);
|
|
srunner_free(sr);
|
|
|
|
return (nf == 0) ? EXIT_SUCCESS : EXIT_FAILURE;
|
|
}
|
|
|