static void torture_log_callback(void **state)
{
struct test_mock_state t = {
.executed = 0,
};
(void)state; /* unused */
ssh_set_log_callback(test_mock_ssh_logging_callback);
ssh_set_log_userdata(&t);
ssh_set_log_level(1);
expect_value(test_mock_ssh_logging_callback, priority, 1);
expect_string(test_mock_ssh_logging_callback, function, "torture_log_callback");
expect_string(test_mock_ssh_logging_callback, buffer, "torture_log_callback: test");
SSH_LOG(SSH_LOG_WARN, "test");
assert_int_equal(t.executed, 1);
}
static void cb1(ssh_session session, ssh_channel channel, void *userdata){
int *v = userdata;
(void) session;
(void) channel;
*v += 1;
}
static void torture_log_callback(void **state)
{
struct test_mock_state t = {
.executed = 0,
};
(void)state; /* unused */
ssh_set_log_callback(test_mock_ssh_logging_callback);
ssh_set_log_userdata(&t);
ssh_set_log_level(1);
expect_value(test_mock_ssh_logging_callback, priority, 1);
expect_string(test_mock_ssh_logging_callback, function, "torture_log_callback");
expect_string(test_mock_ssh_logging_callback, buffer, "torture_log_callback: test");
SSH_LOG(SSH_LOG_WARN, "test");
assert_int_equal(t.executed, 1);
}
int torture_run_tests(void) {
int rc;
UnitTest tests[] = {
unit_test_setup_teardown(torture_callbacks_size, setup, teardown),
unit_test_setup_teardown(torture_callbacks_exists, setup, teardown),
unit_test(torture_log_callback),
};
ssh_init();
torture_filter_tests(tests);
rc=run_tests(tests);
ssh_finalize();
return rc;
}
static void
cockpit_ssh_transport_class_init (CockpitSshTransportClass *klass)
{
GObjectClass *object_class = G_OBJECT_CLASS (klass);
CockpitTransportClass *transport_class = COCKPIT_TRANSPORT_CLASS (klass);
const gchar *env;
transport_class->send = cockpit_ssh_transport_send;
transport_class->close = cockpit_ssh_transport_close;
env = g_getenv ("G_MESSAGES_DEBUG");
if (env && strstr (env, "libssh"))
ssh_set_log_level (SSH_LOG_FUNCTIONS);
object_class->constructed = cockpit_ssh_transport_constructed;
object_class->get_property = cockpit_ssh_transport_get_property;
object_class->set_property = cockpit_ssh_transport_set_property;
object_class->dispose = cockpit_ssh_transport_dispose;
object_class->finalize = cockpit_ssh_transport_finalize;
g_object_class_install_property (object_class, PROP_HOST,
g_param_spec_string ("host", NULL, NULL, "localhost",
G_PARAM_WRITABLE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (object_class, PROP_PORT,
g_param_spec_uint ("port", NULL, NULL, 0, 65535, 0,
G_PARAM_WRITABLE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (object_class, PROP_COMMAND,
g_param_spec_string ("command", NULL, NULL, NULL,
G_PARAM_WRITABLE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (object_class, PROP_KNOWN_HOSTS,
g_param_spec_string ("known-hosts", NULL, NULL, NULL,
G_PARAM_WRITABLE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (object_class, PROP_HOST_KEY,
g_param_spec_string ("host-key", NULL, NULL, NULL,
G_PARAM_READWRITE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (object_class, PROP_HOST_FINGERPRINT,
g_param_spec_string ("host-fingerprint", NULL, NULL, NULL,
G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (object_class, PROP_IGNORE_KEY,
g_param_spec_boolean ("ignore-key", NULL, NULL, FALSE,
G_PARAM_WRITABLE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (object_class, PROP_CREDS,
g_param_spec_boxed ("creds", NULL, NULL, COCKPIT_TYPE_CREDS,
G_PARAM_WRITABLE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS));
g_signal_new ("result", COCKPIT_TYPE_SSH_TRANSPORT, G_SIGNAL_RUN_FIRST, 0, NULL, NULL,
g_cclosure_marshal_generic, G_TYPE_NONE, 1, G_TYPE_STRING);
g_object_class_override_property (object_class, PROP_NAME, "name");
}
static void torture_pki_write_privkey_rsa(void **state)
{
ssh_key origkey;
ssh_key privkey;
int rc;
(void) state; /* unused */
ssh_set_log_level(5);
rc = ssh_pki_import_privkey_file(LIBSSH_RSA_TESTKEY,
NULL,
NULL,
NULL,
&origkey);
assert_true(rc == 0);
unlink(LIBSSH_RSA_TESTKEY);
rc = ssh_pki_export_privkey_file(origkey,
"",
NULL,
NULL,
LIBSSH_RSA_TESTKEY);
assert_true(rc == 0);
rc = ssh_pki_import_privkey_file(LIBSSH_RSA_TESTKEY,
NULL,
NULL,
NULL,
&privkey);
assert_true(rc == 0);
rc = ssh_key_cmp(origkey, privkey, SSH_KEY_CMP_PRIVATE);
assert_true(rc == 0);
ssh_key_free(origkey);
ssh_key_free(privkey);
}
int
main (int argc,
char *argv[])
{
GOptionContext *context;
gchar *user = NULL;
gchar *password = NULL;
gchar *bind = NULL;
GError *error = NULL;
gboolean verbose = FALSE;
gboolean broken_auth = FALSE;
gboolean multi_step = FALSE;
gint port = 0;
int ret;
GOptionEntry entries[] = {
{ "user", 0, 0, G_OPTION_ARG_STRING, &user, "User name to expect", "name" },
{ "password", 0, 0, G_OPTION_ARG_STRING, &password, "Password to expect", "xxx" },
{ "bind", 0, 0, G_OPTION_ARG_STRING, &bind, "Address to bind to", "addr" },
{ "port", 'p', 0, G_OPTION_ARG_INT, &port, "Port to bind to", "NN" },
{ "verbose", 'v', 0, G_OPTION_ARG_NONE, &verbose, "Verbose info", NULL },
{ "multi-step", 'm', 0, G_OPTION_ARG_NONE, &multi_step, "Multi Step Auth", NULL },
{ "broken-auth", 0, 0, G_OPTION_ARG_NONE, &broken_auth, "Break authentication", NULL },
{ NULL }
};
#ifdef __linux
#include <sys/prctl.h>
prctl (PR_SET_PDEATHSIG, 15);
#endif
if (signal (SIGPIPE, SIG_IGN) == SIG_ERR)
g_assert_not_reached ();
ssh_init ();
context = g_option_context_new ("- mock ssh server");
g_option_context_add_main_entries (context, entries, "");
if (!g_option_context_parse (context, &argc, &argv, &error))
{
g_printerr ("mock-sshd: %s\n", error->message);
g_error_free (error);
ret = 2;
}
else if (argc != 1)
{
g_printerr ("mock-sshd: extra arguments on command line\n");
ret = 2;
}
else
{
if (broken_auth)
auth_methods = SSH_AUTH_METHOD_HOSTBASED;
if (verbose)
ssh_set_log_level (SSH_LOG_PROTOCOL);
ret = mock_ssh_server (bind, port, user, password, multi_step);
}
g_option_context_free (context);
g_free (password);
g_free (user);
g_free (bind);
ssh_finalize ();
return ret;
}
