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

Diff of /lbbs/src/fork.c

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

Revision 1.15 by sysadm, Tue May 6 05:31:26 2025 UTC Revision 1.26 by sysadm, Thu Jun 5 08:36:02 2025 UTC
# Line 16  Line 16 
16    
17  #include "common.h"  #include "common.h"
18  #include "bbs_main.h"  #include "bbs_main.h"
19    #include "bbs.h"
20  #include "log.h"  #include "log.h"
21  #include "io.h"  #include "io.h"
22  #include "fork.h"  #include "fork.h"
23    #include "menu.h"
24    #include "database.h"
25    #include "login.h"
26  #include <stdio.h>  #include <stdio.h>
27  #include <string.h>  #include <string.h>
28    #include <signal.h>
29  #include <unistd.h>  #include <unistd.h>
30  #include <stdlib.h>  #include <stdlib.h>
31    #include <errno.h>
32    #include <libssh/libssh.h>
33    #include <libssh/server.h>
34    #include <libssh/callbacks.h>
35    
36  int fork_server()  #define SSH_AUTH_MAX_DURATION 60 // seconds
37    
38    struct ssl_server_cb_data_t
39    {
40            int tries;
41            int error;
42    };
43    
44    static int auth_password(ssh_session session, const char *user,
45                                                     const char *password, void *userdata)
46    {
47            MYSQL *db;
48            struct ssl_server_cb_data_t *p_data = userdata;
49            int ret;
50    
51            if ((db = db_open()) == NULL)
52            {
53                    return SSH_AUTH_ERROR;
54            }
55    
56            if (strcmp(user, "guest") == 0)
57            {
58                    ret = load_guest_info(db);
59            }
60            else
61            {
62                    ret = check_user(db, user, password);
63            }
64    
65            mysql_close(db);
66    
67            if (ret == 0)
68            {
69                    return SSH_AUTH_SUCCESS;
70            }
71    
72            if ((++(p_data->tries)) >= BBS_login_retry_times)
73            {
74                    p_data->error = 1;
75            }
76    
77            return SSH_AUTH_DENIED;
78    }
79    
80    static int pty_request(ssh_session session, ssh_channel channel, const char *term,
81                                               int x, int y, int px, int py, void *userdata)
82    {
83            return 0;
84    }
85    
86    static int shell_request(ssh_session session, ssh_channel channel, void *userdata)
87    {
88            return 0;
89    }
90    
91    static struct ssh_channel_callbacks_struct channel_cb = {
92            .channel_pty_request_function = pty_request,
93            .channel_shell_request_function = shell_request};
94    
95    static ssh_channel new_session_channel(ssh_session session, void *userdata)
96    {
97            if (SSH_channel != NULL)
98            {
99                    return NULL;
100            }
101    
102            SSH_channel = ssh_channel_new(session);
103            ssh_callbacks_init(&channel_cb);
104            ssh_set_channel_callbacks(SSH_channel, &channel_cb);
105    
106            return SSH_channel;
107    }
108    
109    int fork_server(void)
110  {  {
111            ssh_event event;
112          int pid;          int pid;
113            int i;
114            int ret;
115    
116            struct ssl_server_cb_data_t cb_data = {
117                    .tries = 0,
118                    .error = 0,
119            };
120    
121            struct ssh_server_callbacks_struct cb = {
122                    .userdata = &cb_data,
123                    .auth_password_function = auth_password,
124                    .channel_open_request_session_function = new_session_channel,
125            };
126    
127          pid = fork();          pid = fork();
128    
129          if (pid > 0) // Parent process          if (pid > 0) // Parent process
130          {          {
131                  SYS_child_process_count++;                  SYS_child_process_count++;
132                  log_std("Child process (%d) start\n", pid);                  log_common("Child process (%d) start\n", pid);
133                  return 0;                  return pid;
134          }          }
135          else if (pid < 0) // Error          else if (pid < 0) // Error
136          {          {
137                    log_error("fork() error (%d)\n", errno);
138                  return -1;                  return -1;
139          }          }
140    
141          // Child process          // Child process
142          if (close(socket_server) == -1)  
143            if (close(socket_server[0]) == -1 || close(socket_server[1]) == -1)
144          {          {
145                  log_error("Close server socket failed\n");                  log_error("Close server socket failed\n");
146                  return -2;          }
147    
148            SSH_session = ssh_new();
149    
150            if (SSH_v2)
151            {
152                    if (ssh_bind_accept_fd(sshbind, SSH_session, socket_client) != SSH_OK)
153                    {
154                            log_error("ssh_bind_accept_fd() error: %s\n", ssh_get_error(SSH_session));
155                            goto cleanup;
156                    }
157    
158                    ssh_bind_free(sshbind);
159    
160                    ssh_callbacks_init(&cb);
161                    ssh_set_server_callbacks(SSH_session, &cb);
162    
163                    if (ssh_handle_key_exchange(SSH_session))
164                    {
165                            log_error("ssh_handle_key_exchange() error: %s\n", ssh_get_error(SSH_session));
166                            goto cleanup;
167                    }
168                    ssh_set_auth_methods(SSH_session, SSH_AUTH_METHOD_PASSWORD);
169    
170                    event = ssh_event_new();
171                    ssh_event_add_session(event, SSH_session);
172    
173                    for (i = 0; i < SSH_AUTH_MAX_DURATION && !SYS_server_exit && !cb_data.error && SSH_channel == NULL; i++)
174                    {
175                            ret = ssh_event_dopoll(event, 1000); // 1 second
176                            if (ret == SSH_ERROR)
177                            {
178                                    log_error("ssh_event_dopoll() error: %s\n", ssh_get_error(SSH_session));
179                                    goto cleanup;
180                            }
181                    }
182    
183                    if (cb_data.error)
184                    {
185                            log_error("SSH auth error, tried %d times\n", cb_data.tries);
186                            goto cleanup;
187                    }
188          }          }
189    
190          // Redirect Input          // Redirect Input
# Line 53  int fork_server() Line 192  int fork_server()
192          if (dup2(socket_client, STDIN_FILENO) == -1)          if (dup2(socket_client, STDIN_FILENO) == -1)
193          {          {
194                  log_error("Redirect stdin to client socket failed\n");                  log_error("Redirect stdin to client socket failed\n");
195                  return -3;                  goto cleanup;
196          }          }
197    
198          // Redirect Output          // Redirect Output
# Line 61  int fork_server() Line 200  int fork_server()
200          if (dup2(socket_client, STDOUT_FILENO) == -1)          if (dup2(socket_client, STDOUT_FILENO) == -1)
201          {          {
202                  log_error("Redirect stdout to client socket failed\n");                  log_error("Redirect stdout to client socket failed\n");
203                  return -4;                  goto cleanup;
204          }          }
205    
206            SYS_child_process_count = 0;
207    
208          bbs_main();          bbs_main();
209    
210          if (close(socket_client) == -1)  cleanup:
211            // Child process exit
212            SYS_server_exit = 1;
213    
214            if (SSH_v2)
215            {
216                    ssh_channel_free(SSH_channel);
217                    ssh_disconnect(SSH_session);
218            }
219            else if (close(socket_client) == -1)
220          {          {
221                  log_error("Close client socket failed\n");                  log_error("Close client socket failed\n");
222          }          }
223    
224            ssh_free(SSH_session);
225            ssh_finalize();
226    
227          // Close Input and Output for client          // Close Input and Output for client
228          close(STDIN_FILENO);          close(STDIN_FILENO);
229          close(STDOUT_FILENO);          close(STDOUT_FILENO);
230    
231          log_std("Process exit normally\n");          log_common("Process exit normally\n");
   
232          log_end();          log_end();
233    
234          // Exit child process normally          _exit(0);
         exit(0);  
235    
236          return 0;          return 0;
237  }  }

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