| 1 |
|
/*************************************************************************** |
| 2 |
|
test_ssh_server.c - description |
| 3 |
|
------------------- |
| 4 |
|
Copyright : (C) 2004-2025 by Leaflet |
| 5 |
|
Email : leaflet@leafok.com |
| 6 |
|
***************************************************************************/ |
| 7 |
|
|
| 8 |
|
/*************************************************************************** |
| 9 |
|
* * |
| 10 |
|
* This program is free software; you can redistribute it and/or modify * |
| 11 |
|
* it under the terms of the GNU General Public License as published by * |
| 12 |
|
* the Free Software Foundation; either version 3 of the License, or * |
| 13 |
|
* (at your option) any later version. * |
| 14 |
|
* * |
| 15 |
|
***************************************************************************/ |
| 16 |
|
|
| 17 |
|
// This test was written based on libssh example/proxy.c |
| 18 |
|
|
| 19 |
#include "log.h" |
#include "log.h" |
| 20 |
#include <stdio.h> |
#include <stdio.h> |
| 21 |
|
#include <libssh/callbacks.h> |
| 22 |
#include <libssh/libssh.h> |
#include <libssh/libssh.h> |
| 23 |
#include <libssh/server.h> |
#include <libssh/server.h> |
|
#include <libssh/callbacks.h> |
|
| 24 |
|
|
| 25 |
#ifndef BUF_SIZE |
#ifndef BUF_SIZE |
| 26 |
#define BUF_SIZE 2048 |
#define BUF_SIZE 2048 |
| 27 |
#endif |
#endif |
| 28 |
|
|
| 29 |
#define SSH_HOST_KEYFILE "../conf/ssh_host_rsa_key" |
#define SSH_HOST_RSA_KEYFILE "../conf/ssh_host_rsa_key" |
| 30 |
|
|
| 31 |
#define USER "test" |
#define USER "test" |
| 32 |
#define PASSWORD "123456" |
#define PASSWORD "123456" |
| 33 |
|
|
| 34 |
static ssh_channel channel; |
static ssh_channel SSH_channel; |
| 35 |
static int authenticated = 0; |
static int authenticated = 0; |
| 36 |
static int tries = 0; |
static int tries = 0; |
| 37 |
static int error = 0; |
static int error = 0; |
| 41 |
{ |
{ |
| 42 |
(void)userdata; |
(void)userdata; |
| 43 |
|
|
| 44 |
log_std("Authenticating user %s pwd %s\n", user, password); |
log_common("Authenticating user %s pwd %s\n", user, password); |
| 45 |
if (strcmp(user, USER) == 0 && strcmp(password, PASSWORD) == 0) |
if (strcmp(user, USER) == 0 && strcmp(password, PASSWORD) == 0) |
| 46 |
{ |
{ |
| 47 |
authenticated = 1; |
authenticated = 1; |
| 48 |
log_std("Authenticated\n"); |
log_common("Authenticated\n"); |
| 49 |
return SSH_AUTH_SUCCESS; |
return SSH_AUTH_SUCCESS; |
| 50 |
} |
} |
| 51 |
if (tries >= 3) |
if (tries >= 3) |
| 70 |
(void)px; |
(void)px; |
| 71 |
(void)py; |
(void)py; |
| 72 |
(void)userdata; |
(void)userdata; |
| 73 |
log_std("Allocated terminal\n"); |
log_common("Allocated terminal\n"); |
| 74 |
return 0; |
return 0; |
| 75 |
} |
} |
| 76 |
|
|
| 79 |
(void)session; |
(void)session; |
| 80 |
(void)channel; |
(void)channel; |
| 81 |
(void)userdata; |
(void)userdata; |
| 82 |
log_std("Allocated shell\n"); |
log_common("Allocated shell\n"); |
| 83 |
return 0; |
return 0; |
| 84 |
} |
} |
| 85 |
|
|
| 86 |
struct ssh_channel_callbacks_struct channel_cb = { |
struct ssh_channel_callbacks_struct channel_cb = { |
| 87 |
.channel_pty_request_function = pty_request, |
.channel_pty_request_function = pty_request, |
| 88 |
.channel_shell_request_function = shell_request}; |
.channel_shell_request_function = shell_request}; |
| 92 |
(void)session; |
(void)session; |
| 93 |
(void)userdata; |
(void)userdata; |
| 94 |
|
|
| 95 |
if (channel != NULL) |
if (SSH_channel != NULL) |
| 96 |
return NULL; |
return NULL; |
| 97 |
|
|
| 98 |
log_std("Allocated session channel\n"); |
log_common("Allocated session channel\n"); |
| 99 |
channel = ssh_channel_new(session); |
SSH_channel = ssh_channel_new(session); |
| 100 |
ssh_callbacks_init(&channel_cb); |
ssh_callbacks_init(&channel_cb); |
| 101 |
ssh_set_channel_callbacks(channel, &channel_cb); |
ssh_set_channel_callbacks(SSH_channel, &channel_cb); |
| 102 |
|
|
| 103 |
return channel; |
return SSH_channel; |
| 104 |
} |
} |
| 105 |
|
|
| 106 |
int ssh_server(const char *hostaddr, unsigned int port) |
int ssh_server(const char *hostaddr, unsigned int port) |
| 107 |
{ |
{ |
|
ssh_session session; |
|
| 108 |
ssh_bind sshbind; |
ssh_bind sshbind; |
| 109 |
|
ssh_session session; |
| 110 |
ssh_event event; |
ssh_event event; |
| 111 |
|
|
| 112 |
struct ssh_server_callbacks_struct cb = { |
struct ssh_server_callbacks_struct cb = { |
| 114 |
.auth_password_function = auth_password, |
.auth_password_function = auth_password, |
| 115 |
.channel_open_request_session_function = new_session_channel}; |
.channel_open_request_session_function = new_session_channel}; |
| 116 |
|
|
| 117 |
|
long int ssh_timeout = 0; |
| 118 |
|
|
| 119 |
char buf[BUF_SIZE]; |
char buf[BUF_SIZE]; |
| 120 |
char host[128] = ""; |
char host[128] = ""; |
| 121 |
int i, r; |
int i, r; |
| 122 |
|
|
| 123 |
int ssh_log_level = SSH_LOG_WARNING; |
int ssh_log_level = SSH_LOG_PROTOCOL; |
| 124 |
|
|
| 125 |
ssh_init(); |
ssh_init(); |
| 126 |
|
|
| 128 |
|
|
| 129 |
if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_BINDADDR, hostaddr) < 0 || |
if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_BINDADDR, hostaddr) < 0 || |
| 130 |
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_BINDPORT, &port) < 0 || |
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_BINDPORT, &port) < 0 || |
| 131 |
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY, SSH_HOST_KEYFILE) < 0 || |
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY, SSH_HOST_RSA_KEYFILE) < 0 || |
| 132 |
|
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY_ALGORITHMS, "ssh-rsa,rsa-sha2-512,rsa-sha2-256,ecdsa-sha2-nistp256") < 0 || |
| 133 |
|
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_PUBKEY_ACCEPTED_KEY_TYPES, "ssh-rsa,rsa-sha2-512,rsa-sha2-256,ecdsa-sha2-nistp256") < 0 || |
| 134 |
|
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_KEY_EXCHANGE, "curve25519-sha256@libssh.org,ecdh-sha2-nistp256,ecdh-sha2-nistp384,ecdh-sha2-nistp521,diffie-hellman-group-exchange-sha256,diffie-hellman-group16-sha512,diffie-hellman-group18-sha512,diffie-hellman-group14-sha256,diffie-hellman-group14-sha1") < 0 || |
| 135 |
|
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HMAC_C_S, "umac-64-etm@openssh.com,umac-128-etm@openssh.com,hmac-sha2-256-etm@openssh.com,hmac-sha2-512-etm@openssh.com,hmac-sha1-etm@openssh.com,umac-64@openssh.com,umac-128@openssh.com,hmac-sha2-256,hmac-sha2-512,hmac-sha1") < 0 || |
| 136 |
|
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HMAC_S_C, "umac-64-etm@openssh.com,umac-128-etm@openssh.com,hmac-sha2-256-etm@openssh.com,hmac-sha2-512-etm@openssh.com,hmac-sha1-etm@openssh.com,umac-64@openssh.com,umac-128@openssh.com,hmac-sha2-256,hmac-sha2-512,hmac-sha1") < 0 || |
| 137 |
|
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_CIPHERS_C_S, "chacha20-poly1305@openssh.com,aes128-ctr,aes192-ctr,aes256-ctr,aes128-gcm@openssh.com,aes256-gcm@openssh.com") < 0 || |
| 138 |
|
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_CIPHERS_S_C, "chacha20-poly1305@openssh.com,aes128-ctr,aes192-ctr,aes256-ctr,aes128-gcm@openssh.com,aes256-gcm@openssh.com") < 0 || |
| 139 |
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_LOG_VERBOSITY, &ssh_log_level) < 0) |
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_LOG_VERBOSITY, &ssh_log_level) < 0) |
| 140 |
{ |
{ |
| 141 |
log_error("Error setting SSH bind options: %s\n", ssh_get_error(sshbind)); |
log_error("Error setting SSH bind options: %s\n", ssh_get_error(sshbind)); |
| 165 |
ssh_callbacks_init(&cb); |
ssh_callbacks_init(&cb); |
| 166 |
ssh_set_server_callbacks(session, &cb); |
ssh_set_server_callbacks(session, &cb); |
| 167 |
|
|
| 168 |
|
ssh_timeout = 60; // second |
| 169 |
|
if (ssh_options_set(session, SSH_OPTIONS_TIMEOUT, &ssh_timeout) < 0) |
| 170 |
|
{ |
| 171 |
|
log_error("Error setting SSH options: %s\n", ssh_get_error(session)); |
| 172 |
|
ssh_disconnect(session); |
| 173 |
|
_exit(1); |
| 174 |
|
} |
| 175 |
|
|
| 176 |
if (ssh_handle_key_exchange(session)) |
if (ssh_handle_key_exchange(session)) |
| 177 |
{ |
{ |
| 178 |
log_error("ssh_handle_key_exchange: %s\n", ssh_get_error(session)); |
log_error("ssh_handle_key_exchange: %s\n", ssh_get_error(session)); |
| 179 |
return 1; |
ssh_disconnect(session); |
| 180 |
|
_exit(1); |
| 181 |
} |
} |
| 182 |
ssh_set_auth_methods(session, SSH_AUTH_METHOD_PASSWORD | SSH_AUTH_METHOD_GSSAPI_MIC); |
ssh_set_auth_methods(session, SSH_AUTH_METHOD_PASSWORD | SSH_AUTH_METHOD_GSSAPI_MIC); |
| 183 |
|
|
| 184 |
event = ssh_event_new(); |
event = ssh_event_new(); |
| 185 |
ssh_event_add_session(event, session); |
ssh_event_add_session(event, session); |
| 186 |
|
|
| 187 |
while (!(authenticated && channel != NULL)) |
while (!(authenticated && SSH_channel != NULL)) |
| 188 |
{ |
{ |
| 189 |
if (error) |
if (error) |
| 190 |
break; |
break; |
| 204 |
} |
} |
| 205 |
else |
else |
| 206 |
{ |
{ |
| 207 |
log_std("Authenticated and got a channel\n"); |
log_common("Authenticated and got a channel\n"); |
| 208 |
|
} |
| 209 |
|
|
| 210 |
|
ssh_timeout = 0; |
| 211 |
|
if (ssh_options_set(session, SSH_OPTIONS_TIMEOUT, &ssh_timeout) < 0) |
| 212 |
|
{ |
| 213 |
|
log_error("Error setting SSH options: %s\n", ssh_get_error(session)); |
| 214 |
|
ssh_disconnect(session); |
| 215 |
|
_exit(1); |
| 216 |
} |
} |
| 217 |
|
|
| 218 |
snprintf(buf, sizeof(buf), "Hello, welcome to the Sample SSH proxy.\r\nPlease select your destination: "); |
snprintf(buf, sizeof(buf), "Hello, welcome to the Sample SSH proxy.\r\nPlease select your destination: "); |
| 219 |
ssh_channel_write(channel, buf, (uint32_t)strlen(buf)); |
ssh_channel_write(SSH_channel, buf, (uint32_t)strlen(buf)); |
| 220 |
do |
do |
| 221 |
{ |
{ |
| 222 |
i = ssh_channel_read(channel, buf, sizeof(buf), 0); |
i = ssh_channel_read(SSH_channel, buf, sizeof(buf), 0); |
| 223 |
if (i > 0) |
if (i > 0) |
| 224 |
{ |
{ |
| 225 |
ssh_channel_write(channel, buf, (uint32_t)i); |
ssh_channel_write(SSH_channel, buf, (uint32_t)i); |
| 226 |
if (strlen(host) + (size_t)i < sizeof(host)) |
if (strlen(host) + (size_t)i < sizeof(host)) |
| 227 |
{ |
{ |
| 228 |
strncat(host, buf, (size_t)i); |
strncat(host, buf, (size_t)i); |
| 230 |
if (strchr(host, '\x0d')) |
if (strchr(host, '\x0d')) |
| 231 |
{ |
{ |
| 232 |
*strchr(host, '\x0d') = '\0'; |
*strchr(host, '\x0d') = '\0'; |
| 233 |
ssh_channel_write(channel, "\n", 1); |
ssh_channel_write(SSH_channel, "\n", 1); |
| 234 |
break; |
break; |
| 235 |
} |
} |
| 236 |
} |
} |
| 241 |
} |
} |
| 242 |
} while (i > 0); |
} while (i > 0); |
| 243 |
snprintf(buf, sizeof(buf), "Trying to connect to \"%s\"\r\n", host); |
snprintf(buf, sizeof(buf), "Trying to connect to \"%s\"\r\n", host); |
| 244 |
ssh_channel_write(channel, buf, (uint32_t)strlen(buf)); |
ssh_channel_write(SSH_channel, buf, (uint32_t)strlen(buf)); |
| 245 |
log_std("%s", buf); |
log_common("%s", buf); |
| 246 |
|
|
| 247 |
ssh_disconnect(session); |
ssh_disconnect(session); |
| 248 |
ssh_free(session); |
ssh_free(session); |
| 278 |
return -1; |
return -1; |
| 279 |
} |
} |
| 280 |
|
|
| 281 |
log_std_redirect(STDOUT_FILENO); |
log_common_redir(STDOUT_FILENO); |
| 282 |
log_err_redirect(STDERR_FILENO); |
log_error_redir(STDERR_FILENO); |
| 283 |
|
|
| 284 |
ssh_server("0.0.0.0", 2322); |
ssh_server("0.0.0.0", 2322); |
| 285 |
|
|