/[LeafOK_CVS]/lbbs/src/test_ssh_server.c
ViewVC logotype

Diff of /lbbs/src/test_ssh_server.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

Revision 1.7 by sysadm, Sat Jun 7 07:35:06 2025 UTC Revision 1.20 by sysadm, Fri Dec 19 06:30:40 2025 UTC
# Line 1  Line 1 
1  // This test was written base on libssh example/proxy.c  /* SPDX-License-Identifier: GPL-3.0-or-later */
2    /*
3     * test_ssh_server
4     *   - tester for network server with SSH support
5     *
6     * Copyright (C) 2004-2025  Leaflet <leaflet@leafok.com>
7     */
8    
9    // This test was written based on libssh example/proxy.c
10    
11    #ifdef HAVE_CONFIG_H
12    #include "config.h"
13    #endif
14    
15  #include "log.h"  #include "log.h"
16  #include <stdio.h>  #include <stdio.h>
17    #include <libssh/callbacks.h>
18  #include <libssh/libssh.h>  #include <libssh/libssh.h>
19  #include <libssh/server.h>  #include <libssh/server.h>
 #include <libssh/callbacks.h>  
20    
21  #ifndef BUF_SIZE  enum test_ssh_server_constant_t
22  #define BUF_SIZE 2048  {
23  #endif          BUF_SIZE = 2048,
24    };
25    
26  #define SSH_HOST_RSA_KEYFILE "../conf/ssh_host_rsa_key"  static const char SSH_HOST_RSA_KEY_FILE[] = "../conf/ssh_host_rsa_key";
27    static const char SSH_HOST_ED25519_KEY_FILE[] = "../conf/ssh_host_ed25519_key";
28    static const char SSH_HOST_ECDSA_KEY_FILE[] = "../conf/ssh_host_ecdsa_key";
29    
30  #define USER "test"  static const char USER[] = "test";
31  #define PASSWORD "123456"  static const char PASSWORD[] = "123456";
32    
33  static ssh_channel SSH_channel;  static ssh_channel SSH_channel;
34  static int authenticated = 0;  static int authenticated = 0;
# Line 25  static int auth_password(ssh_session ses Line 40  static int auth_password(ssh_session ses
40  {  {
41          (void)userdata;          (void)userdata;
42    
43          log_common("Authenticating user %s pwd %s\n", user, password);          log_common("Authenticating user %s pwd %s", user, password);
44          if (strcmp(user, USER) == 0 && strcmp(password, PASSWORD) == 0)          if (strcmp(user, USER) == 0 && strcmp(password, PASSWORD) == 0)
45          {          {
46                  authenticated = 1;                  authenticated = 1;
47                  log_common("Authenticated\n");                  log_common("Authenticated");
48                  return SSH_AUTH_SUCCESS;                  return SSH_AUTH_SUCCESS;
49          }          }
50          if (tries >= 3)          if (tries >= 3)
51          {          {
52                  log_error("Too many authentication tries\n");                  log_error("Too many authentication tries");
53                  ssh_disconnect(session);                  ssh_disconnect(session);
54                  error = 1;                  error = 1;
55                  return SSH_AUTH_DENIED;                  return SSH_AUTH_DENIED;
# Line 54  static int pty_request(ssh_session sessi Line 69  static int pty_request(ssh_session sessi
69          (void)px;          (void)px;
70          (void)py;          (void)py;
71          (void)userdata;          (void)userdata;
72          log_common("Allocated terminal\n");          log_common("Allocated terminal");
73          return 0;          return 0;
74  }  }
75    
# Line 63  static int shell_request(ssh_session ses Line 78  static int shell_request(ssh_session ses
78          (void)session;          (void)session;
79          (void)channel;          (void)channel;
80          (void)userdata;          (void)userdata;
81          log_common("Allocated shell\n");          log_common("Allocated shell");
82          return 0;          return 0;
83  }  }
84    
# Line 71  struct ssh_channel_callbacks_struct chan Line 86  struct ssh_channel_callbacks_struct chan
86          .channel_pty_request_function = pty_request,          .channel_pty_request_function = pty_request,
87          .channel_shell_request_function = shell_request};          .channel_shell_request_function = shell_request};
88    
89  static ssh_channel new_session_channel(ssh_session session, void *userdata)  static ssh_channel channel_open(ssh_session session, void *userdata)
90  {  {
91          (void)session;          (void)session;
92          (void)userdata;          (void)userdata;
# Line 79  static ssh_channel new_session_channel(s Line 94  static ssh_channel new_session_channel(s
94          if (SSH_channel != NULL)          if (SSH_channel != NULL)
95                  return NULL;                  return NULL;
96    
97          log_common("Allocated session channel\n");          log_common("Allocated session channel");
98          SSH_channel = ssh_channel_new(session);          SSH_channel = ssh_channel_new(session);
99          ssh_callbacks_init(&channel_cb);          ssh_callbacks_init(&channel_cb);
100          ssh_set_channel_callbacks(SSH_channel, &channel_cb);          ssh_set_channel_callbacks(SSH_channel, &channel_cb);
# Line 96  int ssh_server(const char *hostaddr, uns Line 111  int ssh_server(const char *hostaddr, uns
111          struct ssh_server_callbacks_struct cb = {          struct ssh_server_callbacks_struct cb = {
112                  .userdata = NULL,                  .userdata = NULL,
113                  .auth_password_function = auth_password,                  .auth_password_function = auth_password,
114                  .channel_open_request_session_function = new_session_channel};                  .channel_open_request_session_function = channel_open};
115    
116            long int ssh_timeout = 0;
117    
118          char buf[BUF_SIZE];          char buf[BUF_SIZE];
119          char host[128] = "";          char host[128] = "";
120          int i, r;          int i, r;
121    
122            int ssh_key_valid = 0;
123          int ssh_log_level = SSH_LOG_PROTOCOL;          int ssh_log_level = SSH_LOG_PROTOCOL;
124    
125          ssh_init();          ssh_init();
126    
127          sshbind = ssh_bind_new();          sshbind = ssh_bind_new();
128    
129            if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY, SSH_HOST_RSA_KEY_FILE) < 0)
130            {
131                    log_error("Error loading SSH RSA key: %s", SSH_HOST_RSA_KEY_FILE);
132            }
133            else
134            {
135                    ssh_key_valid = 1;
136            }
137            if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY, SSH_HOST_ED25519_KEY_FILE) < 0)
138            {
139                    log_error("Error loading SSH ED25519 key: %s", SSH_HOST_ED25519_KEY_FILE);
140            }
141            else
142            {
143                    ssh_key_valid = 1;
144            }
145            if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY, SSH_HOST_ECDSA_KEY_FILE) < 0)
146            {
147                    log_error("Error loading SSH ECDSA key: %s", SSH_HOST_ECDSA_KEY_FILE);
148            }
149            else
150            {
151                    ssh_key_valid = 1;
152            }
153    
154            if (!ssh_key_valid)
155            {
156                    log_error("Error: no valid SSH host key");
157                    ssh_bind_free(sshbind);
158                    return -1;
159            }
160    
161          if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_BINDADDR, hostaddr) < 0 ||          if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_BINDADDR, hostaddr) < 0 ||
162                  ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_BINDPORT, &port) < 0 ||                  ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_BINDPORT, &port) < 0 ||
163                  ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY, SSH_HOST_RSA_KEYFILE) < 0 ||                  ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY_ALGORITHMS, "+ssh-ed25519,ecdsa-sha2-nistp256,ssh-rsa") < 0 ||
                 ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY_ALGORITHMS, "ssh-rsa,rsa-sha2-512,rsa-sha2-256,ecdsa-sha2-nistp256") < 0 ||  
                 ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_PUBKEY_ACCEPTED_KEY_TYPES, "ssh-rsa,rsa-sha2-512,rsa-sha2-256,ecdsa-sha2-nistp256") < 0 ||  
                 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 ||  
                 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 ||  
                 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 ||  
                 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 ||  
                 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 ||  
164                  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)
165          {          {
166                  log_error("Error setting SSH bind options: %s\n", ssh_get_error(sshbind));                  log_error("Error setting SSH bind options: %s", ssh_get_error(sshbind));
167                  ssh_bind_free(sshbind);                  ssh_bind_free(sshbind);
168                  return -1;                  return -1;
169          }          }
170    
171          if (ssh_bind_listen(sshbind) < 0)          if (ssh_bind_listen(sshbind) < 0)
172          {          {
173                  log_error("Error listening at SSH server port: %s\n", ssh_get_error(sshbind));                  log_error("Error listening at SSH server port: %s", ssh_get_error(sshbind));
174                  ssh_bind_free(sshbind);                  ssh_bind_free(sshbind);
175                  return -1;                  return -1;
176          }          }
# Line 147  int ssh_server(const char *hostaddr, uns Line 190  int ssh_server(const char *hostaddr, uns
190                                  ssh_callbacks_init(&cb);                                  ssh_callbacks_init(&cb);
191                                  ssh_set_server_callbacks(session, &cb);                                  ssh_set_server_callbacks(session, &cb);
192    
193                                    ssh_timeout = 60; // second
194                                    if (ssh_options_set(session, SSH_OPTIONS_TIMEOUT, &ssh_timeout) < 0)
195                                    {
196                                            log_error("Error setting SSH options: %s", ssh_get_error(session));
197                                            ssh_disconnect(session);
198                                            _exit(1);
199                                    }
200    
201                                  if (ssh_handle_key_exchange(session))                                  if (ssh_handle_key_exchange(session))
202                                  {                                  {
203                                          log_error("ssh_handle_key_exchange: %s\n", ssh_get_error(session));                                          log_error("ssh_handle_key_exchange: %s", ssh_get_error(session));
204                                          return 1;                                          ssh_disconnect(session);
205                                            _exit(1);
206                                  }                                  }
207                                  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);
208    
# Line 164  int ssh_server(const char *hostaddr, uns Line 216  int ssh_server(const char *hostaddr, uns
216                                          r = ssh_event_dopoll(event, -1);                                          r = ssh_event_dopoll(event, -1);
217                                          if (r == SSH_ERROR)                                          if (r == SSH_ERROR)
218                                          {                                          {
219                                                  log_error("Error : %s\n", ssh_get_error(session));                                                  log_error("Error : %s", ssh_get_error(session));
220                                                  ssh_disconnect(session);                                                  ssh_disconnect(session);
221                                                  _exit(1);                                                  _exit(1);
222                                          }                                          }
# Line 172  int ssh_server(const char *hostaddr, uns Line 224  int ssh_server(const char *hostaddr, uns
224    
225                                  if (error)                                  if (error)
226                                  {                                  {
227                                          log_error("Error, exiting loop\n");                                          log_error("Error, exiting loop");
228                                          _exit(1);                                          _exit(1);
229                                  }                                  }
230                                  else                                  else
231                                  {                                  {
232                                          log_common("Authenticated and got a channel\n");                                          log_common("Authenticated and got a channel");
233                                    }
234    
235                                    ssh_timeout = 0;
236                                    if (ssh_options_set(session, SSH_OPTIONS_TIMEOUT, &ssh_timeout) < 0)
237                                    {
238                                            log_error("Error setting SSH options: %s", ssh_get_error(session));
239                                            ssh_disconnect(session);
240                                            _exit(1);
241                                  }                                  }
242    
243                                  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.\nPlease select your destination: ");
244                                  ssh_channel_write(SSH_channel, buf, (uint32_t)strlen(buf));                                  ssh_channel_write(SSH_channel, buf, (uint32_t)strlen(buf));
245                                  do                                  do
246                                  {                                  {
# Line 201  int ssh_server(const char *hostaddr, uns Line 261  int ssh_server(const char *hostaddr, uns
261                                          }                                          }
262                                          else                                          else
263                                          {                                          {
264                                                  log_error("Error: %s\n", ssh_get_error(session));                                                  log_error("Error: %s", ssh_get_error(session));
265                                                  _exit(1);                                                  _exit(1);
266                                          }                                          }
267                                  } while (i > 0);                                  } while (i > 0);
268                                  snprintf(buf, sizeof(buf), "Trying to connect to \"%s\"\r\n", host);                                  snprintf(buf, sizeof(buf), "Trying to connect to \"%s\"\n", host);
269                                  ssh_channel_write(SSH_channel, buf, (uint32_t)strlen(buf));                                  ssh_channel_write(SSH_channel, buf, (uint32_t)strlen(buf));
270                                  log_common("%s", buf);                                  log_common("%s", buf);
271    
# Line 214  int ssh_server(const char *hostaddr, uns Line 274  int ssh_server(const char *hostaddr, uns
274    
275                                  _exit(0);                                  _exit(0);
276                          case -1:                          case -1:
277                                  log_error("Failed to fork\n");                                  log_error("Failed to fork");
278                                  break;                                  break;
279                          }                          }
280                  }                  }
281                  else                  else
282                  {                  {
283                          log_error("%s\n", ssh_get_error(sshbind));                          log_error("%s", ssh_get_error(sshbind));
284                  }                  }
285    
286                  /* Since the session has been passed to a child fork, do some cleaning                  /* Since the session has been passed to a child fork, do some cleaning

Legend:
Removed lines/characters  
Changed lines/characters
  Added lines/characters
webmaster@leafok.com
 ViewVC Help
Powered by ViewVC 1.3.0-beta1