311 строки
8.5 KiB
C
311 строки
8.5 KiB
C
|
/*
|
||
|
|
* This file is part of the SSH Library
|
||
|
|
*
|
||
|
|
* Copyright (c) 2018 by Red Hat, Inc.
|
||
|
|
*
|
||
|
|
* Author: Anderson Toshiyuki Sasaki <ansasaki@redhat.com>
|
||
|
|
*
|
||
|
|
* The SSH Library is free software; you can redistribute it and/or modify
|
||
|
|
* it under the terms of the GNU Lesser General Public License as published by
|
||
|
|
* the Free Software Foundation; either version 2.1 of the License, or (at your
|
||
|
|
* option) any later version.
|
||
|
|
*
|
||
|
|
* The SSH Library is distributed in the hope that it will be useful, but
|
||
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
||
|
|
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
|
||
|
|
* License for more details.
|
||
|
|
*
|
||
|
|
* You should have received a copy of the GNU Lesser General Public License
|
||
|
|
* along with the SSH Library; see the file COPYING. If not, write to
|
||
|
|
* the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
|
||
|
|
* MA 02111-1307, USA.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "test_server.h"
|
||
|
|
|
||
|
|
#include <libssh/priv.h>
|
||
|
|
#include <libssh/libssh.h>
|
||
|
|
#include <libssh/server.h>
|
||
|
|
#include <libssh/callbacks.h>
|
||
|
|
|
||
|
|
#include <signal.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <stdbool.h>
|
||
|
|
#include <errno.h>
|
||
|
|
|
||
|
|
#include <sys/wait.h>
|
||
|
|
|
||
|
|
void free_server_state(struct server_state_st *state)
|
||
|
|
{
|
||
|
|
if (state == NULL) {
|
||
|
|
goto end;
|
||
|
|
}
|
||
|
|
|
||
|
|
SAFE_FREE(state->address);
|
||
|
|
|
||
|
|
SAFE_FREE(state->ecdsa_key);
|
||
|
|
SAFE_FREE(state->dsa_key);
|
||
|
|
SAFE_FREE(state->ed25519_key);
|
||
|
|
SAFE_FREE(state->rsa_key);
|
||
|
|
SAFE_FREE(state->host_key);
|
||
|
|
|
||
|
|
SAFE_FREE(state->pcap_file);
|
||
|
|
|
||
|
|
SAFE_FREE(state->expected_username);
|
||
|
|
SAFE_FREE(state->expected_password);
|
||
|
|
|
||
|
|
end:
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* SIGCHLD handler for cleaning up dead children. */
|
||
|
|
static void sigchld_handler(int signo) {
|
||
|
|
(void) signo;
|
||
|
|
while (waitpid(-1, NULL, WNOHANG) > 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
int run_server(struct server_state_st *state)
|
||
|
|
{
|
||
|
|
ssh_session session = NULL;
|
||
|
|
ssh_bind sshbind = NULL;
|
||
|
|
ssh_event event = NULL;
|
||
|
|
|
||
|
|
struct sigaction sa = {0};
|
||
|
|
|
||
|
|
int rc;
|
||
|
|
|
||
|
|
/* Check provided state */
|
||
|
|
if (state == NULL) {
|
||
|
|
fprintf(stderr, "Invalid state\n");
|
||
|
|
return SSH_ERROR;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Set up SIGCHLD handler. */
|
||
|
|
sa.sa_handler = sigchld_handler;
|
||
|
|
sigemptyset(&sa.sa_mask);
|
||
|
|
sa.sa_flags = SA_RESTART | SA_NOCLDSTOP;
|
||
|
|
|
||
|
|
if (sigaction(SIGCHLD, &sa, NULL) != 0) {
|
||
|
|
fprintf(stderr, "Failed to register SIGCHLD handler\n");
|
||
|
|
return SSH_ERROR;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (state->address == NULL) {
|
||
|
|
fprintf(stderr, "Missing bind address\n");
|
||
|
|
return SSH_ERROR;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (state->address == NULL) {
|
||
|
|
fprintf(stderr, "Missing bind address\n");
|
||
|
|
return SSH_ERROR;
|
||
|
|
}
|
||
|
|
|
||
|
|
sshbind = ssh_bind_new();
|
||
|
|
if (sshbind == NULL) {
|
||
|
|
fprintf(stderr, "Out of memory\n");
|
||
|
|
return SSH_ERROR;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (state->verbosity) {
|
||
|
|
rc = ssh_bind_options_set(sshbind,
|
||
|
|
SSH_BIND_OPTIONS_LOG_VERBOSITY,
|
||
|
|
&state->verbosity);
|
||
|
|
if (rc != 0) {
|
||
|
|
fprintf(stderr,
|
||
|
|
"Error setting verbosity level: %s\n",
|
||
|
|
ssh_get_error(sshbind));
|
||
|
|
goto free_sshbind;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
rc = ssh_bind_options_set(sshbind,
|
||
|
|
SSH_BIND_OPTIONS_BINDADDR,
|
||
|
|
state->address);
|
||
|
|
if (rc != 0) {
|
||
|
|
fprintf(stderr,
|
||
|
|
"Error setting bind address: %s\n",
|
||
|
|
ssh_get_error(sshbind));
|
||
|
|
goto free_sshbind;
|
||
|
|
}
|
||
|
|
|
||
|
|
rc = ssh_bind_options_set(sshbind,
|
||
|
|
SSH_BIND_OPTIONS_BINDPORT,
|
||
|
|
&(state->port));
|
||
|
|
if (rc != 0) {
|
||
|
|
fprintf(stderr,
|
||
|
|
"Error setting bind port: %s\n",
|
||
|
|
ssh_get_error(sshbind));
|
||
|
|
goto free_sshbind;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (state->dsa_key != NULL) {
|
||
|
|
rc = ssh_bind_options_set(sshbind,
|
||
|
|
SSH_BIND_OPTIONS_DSAKEY,
|
||
|
|
state->dsa_key);
|
||
|
|
if (rc != 0) {
|
||
|
|
fprintf(stderr,
|
||
|
|
"Error setting DSA key: %s\n",
|
||
|
|
ssh_get_error(sshbind));
|
||
|
|
goto free_sshbind;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (state->rsa_key != NULL) {
|
||
|
|
rc = ssh_bind_options_set(sshbind,
|
||
|
|
SSH_BIND_OPTIONS_RSAKEY,
|
||
|
|
state->rsa_key);
|
||
|
|
if (rc != 0) {
|
||
|
|
fprintf(stderr,
|
||
|
|
"Error setting RSA key: %s\n",
|
||
|
|
ssh_get_error(sshbind));
|
||
|
|
goto free_sshbind;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (state->ecdsa_key != NULL) {
|
||
|
|
rc = ssh_bind_options_set(sshbind,
|
||
|
|
SSH_BIND_OPTIONS_ECDSAKEY,
|
||
|
|
state->ecdsa_key);
|
||
|
|
if (rc != 0) {
|
||
|
|
fprintf(stderr,
|
||
|
|
"Error setting ECDSA key: %s\n",
|
||
|
|
ssh_get_error(sshbind));
|
||
|
|
goto free_sshbind;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (state->host_key) {
|
||
|
|
rc = ssh_bind_options_set(sshbind,
|
||
|
|
SSH_BIND_OPTIONS_HOSTKEY,
|
||
|
|
state->host_key);
|
||
|
|
if (rc) {
|
||
|
|
fprintf(stderr,
|
||
|
|
"Error setting hostkey: %s\n",
|
||
|
|
ssh_get_error(sshbind));
|
||
|
|
goto free_sshbind;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
rc = ssh_bind_listen(sshbind);
|
||
|
|
if (rc != 0) {
|
||
|
|
fprintf(stderr,
|
||
|
|
"Error listening to socket: %s\n",
|
||
|
|
ssh_get_error(sshbind));
|
||
|
|
goto free_sshbind;
|
||
|
|
}
|
||
|
|
|
||
|
|
printf("Started libssh test server on port %d\n", state->port);
|
||
|
|
|
||
|
|
for (;;) {
|
||
|
|
session = ssh_new();
|
||
|
|
if (session == NULL) {
|
||
|
|
fprintf(stderr, "Out of memory\n");
|
||
|
|
rc = SSH_ERROR;
|
||
|
|
goto free_sshbind;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Blocks until there is a new incoming connection. */
|
||
|
|
rc = ssh_bind_accept(sshbind, session);
|
||
|
|
if (rc != SSH_ERROR) {
|
||
|
|
pid_t pid = fork();
|
||
|
|
|
||
|
|
switch(pid) {
|
||
|
|
case 0:
|
||
|
|
/* Remove the SIGCHLD handler inherited from parent. */
|
||
|
|
sa.sa_handler = SIG_DFL;
|
||
|
|
sigaction(SIGCHLD, &sa, NULL);
|
||
|
|
/* Remove socket binding, which allows us to restart the
|
||
|
|
* parent process, without terminating existing sessions. */
|
||
|
|
ssh_bind_free(sshbind);
|
||
|
|
|
||
|
|
event = ssh_event_new();
|
||
|
|
if (event != NULL) {
|
||
|
|
/* Blocks until the SSH session ends by either
|
||
|
|
* child process exiting, or client disconnecting. */
|
||
|
|
state->handle_session(event, session, state);
|
||
|
|
ssh_event_free(event);
|
||
|
|
} else {
|
||
|
|
fprintf(stderr, "Could not create polling context\n");
|
||
|
|
}
|
||
|
|
ssh_disconnect(session);
|
||
|
|
ssh_free(session);
|
||
|
|
|
||
|
|
free_server_state(state);
|
||
|
|
|
||
|
|
exit(0);
|
||
|
|
case -1:
|
||
|
|
fprintf(stderr, "Failed to fork\n");
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
fprintf(stderr,
|
||
|
|
"Error accepting a connection: %s\n",
|
||
|
|
ssh_get_error(sshbind));
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Since the session has been passed to a child fork, do some cleaning
|
||
|
|
* up at the parent process. */
|
||
|
|
ssh_disconnect(session);
|
||
|
|
ssh_free(session);
|
||
|
|
}
|
||
|
|
|
||
|
|
rc = 0;
|
||
|
|
|
||
|
|
free_sshbind:
|
||
|
|
ssh_bind_free(sshbind);
|
||
|
|
return rc;
|
||
|
|
}
|
||
|
|
|
||
|
|
pid_t fork_run_server(struct server_state_st *state)
|
||
|
|
{
|
||
|
|
pid_t pid;
|
||
|
|
int rc;
|
||
|
|
|
||
|
|
char err_str[1024] = {0};
|
||
|
|
|
||
|
|
struct sigaction sa;
|
||
|
|
|
||
|
|
/* Check provided state */
|
||
|
|
if (state == NULL) {
|
||
|
|
fprintf(stderr, "Invalid state\n");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Set up SIGCHLD handler. */
|
||
|
|
sa.sa_handler = sigchld_handler;
|
||
|
|
sigemptyset(&sa.sa_mask);
|
||
|
|
sa.sa_flags = SA_RESTART | SA_NOCLDSTOP;
|
||
|
|
|
||
|
|
if (sigaction(SIGCHLD, &sa, NULL) != 0) {
|
||
|
|
strerror_r(errno, err_str, 1024);
|
||
|
|
fprintf(stderr, "Failed to register SIGCHLD handler: %s\n",
|
||
|
|
err_str);
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
pid = fork();
|
||
|
|
switch(pid) {
|
||
|
|
case 0:
|
||
|
|
/* Remove the SIGCHLD handler inherited from parent. */
|
||
|
|
sa.sa_handler = SIG_DFL;
|
||
|
|
sigaction(SIGCHLD, &sa, NULL);
|
||
|
|
|
||
|
|
/* The child process starts a server which will listen for connections */
|
||
|
|
rc = run_server(state);
|
||
|
|
if (rc != 0) {
|
||
|
|
exit(rc);
|
||
|
|
}
|
||
|
|
|
||
|
|
exit(0);
|
||
|
|
case -1:
|
||
|
|
strerror_r(errno, err_str, 1024);
|
||
|
|
fprintf(stderr, "Failed to fork: %s\n",
|
||
|
|
err_str);
|
||
|
|
return -1;
|
||
|
|
default:
|
||
|
|
/* Return the child pid */
|
||
|
|
return pid;
|
||
|
|
}
|
||
|
|
}
|