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

Diff of /lbbs/src/main.c

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

Revision 1.16 by sysadm, Sat May 7 12:15:30 2005 UTC Revision 1.94 by sysadm, Thu Jan 8 14:22:57 2026 UTC
# Line 1  Line 1 
1  /***************************************************************************  /* SPDX-License-Identifier: GPL-3.0-or-later */
2                            main.c  -  description  /*
3                               -------------------   * main
4      begin                : Mon Oct 11 2004   *   - entry of server program
5      copyright            : (C) 2004 by Leaflet   *
6      email                : leaflet@leafok.com   * Copyright (C) 2004-2026  Leaflet <leaflet@leafok.com>
7   ***************************************************************************/   */
8    
9  /***************************************************************************  #ifdef HAVE_CONFIG_H
10   *                                                                         *  #include "config.h"
11   *   This program is free software; you can redistribute it and/or modify  *  #endif
  *   it under the terms of the GNU General Public License as published by  *  
  *   the Free Software Foundation; either version 2 of the License, or     *  
  *   (at your option) any later version.                                   *  
  *                                                                         *  
  ***************************************************************************/  
12    
13  #include "bbs.h"  #include "bbs.h"
14    #include "bwf.h"
15  #include "common.h"  #include "common.h"
16    #include "file_loader.h"
17    #include "init.h"
18  #include "io.h"  #include "io.h"
19    #include "lml.h"
20    #include "log.h"
21  #include "menu.h"  #include "menu.h"
22    #include "net_server.h"
23    #include "section_list_loader.h"
24    #include "user_list.h"
25    #include <errno.h>
26    #include <libgen.h>
27    #include <locale.h>
28  #include <signal.h>  #include <signal.h>
29    #include <stdio.h>
30    #include <stdlib.h>
31  #include <string.h>  #include <string.h>
 #include <sys/types.h>  
32  #include <unistd.h>  #include <unistd.h>
33    #include <sys/stat.h>
34    #include <sys/types.h>
35    #include <sys/wait.h>
36    
37    typedef void (*arg_option_handler_t)(void);
38    
39    struct arg_option_t
40    {
41            const char *short_arg;
42            const char *long_arg;
43            const char *description;
44            arg_option_handler_t handler;
45    };
46    typedef struct arg_option_t ARG_OPTION;
47    
48    static void arg_foreground(void);
49    static void arg_help(void);
50    static void arg_version(void);
51    static void arg_display_log(void);
52    static void arg_display_error_log(void);
53    static void arg_compile_info(void);
54    static void arg_error(void);
55    
56    static const ARG_OPTION arg_options[] = {
57            {"-f", "--foreground", "Run in foreground", arg_foreground},
58            {"-h", "--help", "Display help message", arg_help},
59            {"-v", "--version", "Display version information", arg_version},
60            {NULL, "--display-log", "Display standard log information", arg_display_log},
61            {NULL, "--display-error-log", "Display error log information", arg_display_error_log},
62            {"-C", "--compile-config", "Display compile configuration", arg_compile_info}};
63    
64    static const int arg_options_count = sizeof(arg_options) / sizeof(ARG_OPTION);
65    
66    static int daemon_service = 1;
67    static int std_log_redir = 0;
68    static int error_log_redir = 0;
69    
70    static void arg_foreground(void)
71    {
72            daemon_service = 0;
73    }
74    
75    static void arg_usage(void)
76    {
77            fprintf(stderr, "Usage: bbsd [-fhvC] [...]\n\n");
78    
79            for (int i = 0; i < arg_options_count; i++)
80            {
81                    fprintf(stderr, "%s%*s%s%*s%s\n",
82                                    arg_options[i].short_arg ? arg_options[i].short_arg : "",
83                                    8 - (arg_options[i].short_arg ? (int)strlen(arg_options[i].short_arg) : 0), "",
84                                    arg_options[i].long_arg ? arg_options[i].long_arg : "",
85                                    24 - (arg_options[i].long_arg ? (int)strlen(arg_options[i].long_arg) : 0), "",
86                                    arg_options[i].description ? arg_options[i].description : "");
87            }
88    
89            fprintf(stderr, "\n    If meet any bug, please report to <leaflet@leafok.com>\n\n");
90    }
91    
92    static void arg_help(void)
93    {
94            arg_usage();
95    
96            exit(0);
97    }
98    
99    static void arg_version(void)
100    {
101            fprintf(stderr, "%s\n", APP_INFO);
102            fprintf(stderr, "%s\n", COPYRIGHT_INFO);
103    
104            exit(0);
105    }
106    
107  void  static void arg_display_log(void)
 app_help (void)  
108  {  {
109    prints ("Usage: bbsd [-fhv] [...]\n\n"          std_log_redir = 1;
110            "-f\t--foreground\t\tForce program run in foreground\n"  }
111            "-h\t--help\t\t\tDisplay this help message\n"  
112            "-v\t--version\t\tDisplay version information\n"  static void arg_display_error_log(void)
113            "\t--display-log\t\tDisplay standard log information\n"  {
114            "\t--display-error-log\tDisplay error log information\n"          error_log_redir = 1;
115            "\n    If meet any bug, please report to <leaflet@leafok.com>\n\n");  }
116  }  
117    static void arg_compile_info(void)
118  void  {
119  arg_error (void)          fprintf(stderr, "%s\n"
120  {                                          "--enable-shared\t\t[%s]\n"
121    prints ("Invalid arguments\n");                                          "--enable-systemd\t[%s]\n"
122    app_help ();                                          "--with-debug\t\t[%s]\n"
123  }                                          "--with-epoll\t\t[%s]\n"
124                                            "--with-mariadb\t\t[%s]\n"
125  int                                          "--with-sysv\t\t[%s]\n",
126  main (int argc, char *argv[])                          APP_INFO,
127  {  #ifdef _ENABLE_SHARED
128    char log_dir[256], file_log_std[256], file_log_error[256], file_config[256];                          "yes",
129    int i, j;  #else
130    int daemon = 1, std_log_redir = 0, error_log_redir = 0;                          "no",
131    #endif
132    //Parse args  #ifdef HAVE_SYSTEMD_SD_DAEMON_H
133    for (i = 1; i < argc; i++)                          "yes",
134      {  #else
135        switch (argv[i][0])                          "no",
136          {  #endif
137          case '-':  #ifdef _DEBUG
138            if (argv[i][1] != '-')                          "yes",
139              {  #else
140                for (j = 1; j < strlen (argv[i]); j++)                          "no",
141                  {  #endif
142                    switch (argv[i][j])  #ifdef HAVE_SYS_EPOLL_H
143                      {                          "yes",
144                      case 'f':  #else
145                        daemon = 0;                          "no",
146                        break;  #endif
147                      case 'h':  #ifdef HAVE_MARIADB_CLIENT
148                        app_help ();                          "yes",
149                        exit (0);  #else
150                      case 'v':                          "no",
151                        puts (app_version);  #endif
152                        exit (0);  #ifdef HAVE_SYSTEM_V
153                      default:                          "yes"
154                        arg_error ();  #else
155                        exit (1);                          "no"
156                      }  #endif
157                  }          );
158              }  
159            else          exit(0);
160              {  }
161                if (strcmp (argv[i] + 2, "foreground") == 0)  
162                  {  static void arg_error(void)
163                    daemon = 0;  {
164                    break;          fprintf(stderr, "Invalid arguments\n");
165                  }          arg_usage();
166                if (strcmp (argv[i] + 2, "help") == 0)  
167                  {          exit(1);
168                    app_help ();  }
169                    exit (0);  
170                  }  int main(int argc, char *argv[])
171                if (strcmp (argv[i] + 2, "version") == 0)  {
172                  {          char file_path_temp[FILE_PATH_LEN];
173                    puts (app_version);          FILE *fp;
174                    exit (0);          int ret;
175                  }          int last_aid;
176                if (strcmp (argv[i] + 2, "display-log") == 0)          struct sigaction act = {0};
177                  {          int i;
178                    std_log_redir = 1;          int j;
179                  }          struct stat file_stat;
180                if (strcmp (argv[i] + 2, "display-error-log") == 0)  
181                  {          // Parse args
182                    error_log_redir = 1;          for (i = 1; i < argc; i++)
183                  }          {
184              }                  for (j = 0; j < arg_options_count; j++)
185            break;                  {
186          }                          if (arg_options[j].short_arg && strcmp(argv[i], arg_options[j].short_arg) == 0)
187      }                          {
188                                    arg_options[j].handler();
189    //Initialize daemon                                  break;
190    if (daemon)                          }
191      init_daemon ();                          else if (arg_options[j].long_arg && strcmp(argv[i], arg_options[j].long_arg) == 0)
192                            {
193    //Change current dir                                  arg_options[j].handler();
194    strncpy (app_home_dir, argv[0], rindex (argv[0], '/') - argv[0] + 1);                                  break;
195    strcat (app_home_dir, "../");                          }
196    chdir (app_home_dir);                  }
197                    if (j == arg_options_count)
198    //Initialize log                  {
199    strcpy (app_temp_dir, "/tmp/lbbs/");                          arg_error();
200    mkdir (app_temp_dir, 0750);                          return -1;
201    strcpy (log_dir, app_home_dir);                  }
202    strcat (log_dir, "log/");          }
203    strcpy (file_log_std, log_dir);  
204    strcpy (file_log_error, log_dir);          // Initialize daemon
205    strcat (file_log_std, "bbsd.log");          if (daemon_service)
206    strcat (file_log_error, "error.log");          {
207    mkdir (log_dir, 0750);                  init_daemon();
208    if (log_begin (file_log_std, file_log_error) < 0)          }
209      exit (-1);  
210            // Change current dir
211    if ((!daemon) && std_log_redir)          strncpy(file_path_temp, argv[0], sizeof(file_path_temp) - 1);
212      log_std_redirect (2);          file_path_temp[sizeof(file_path_temp) - 1] = '\0';
213    if ((!daemon) && error_log_redir)  
214      log_err_redirect (3);          if (chdir(dirname(file_path_temp)) < 0)
215            {
216    //Load configuration                  fprintf(stderr, "chdir(%s) error: %d\n", dirname(file_path_temp), errno);
217    strcpy (file_config, app_home_dir);                  return -1;
218    strcat (file_config, "conf/bbsd.conf");          }
219    if (load_conf (file_config) < 0)          if (chdir("..") < 0)
220      exit (-2);          {
221                    fprintf(stderr, "chdir(..) error: %d\n", errno);
222    //Load menus                  return -1;
223    strcpy (file_config, app_home_dir);          }
224    strcat (file_config, "conf/menu.conf");  
225    if (load_menu (&bbs_menu, file_config) < 0)          // Apply the specified locale
226      exit (-3);          if (setlocale(LC_ALL, "en_US.UTF-8") == NULL)
227            {
228    //Set signal handler                  fprintf(stderr, "setlocale(LC_ALL, en_US.UTF-8) error\n");
229    signal (SIGCHLD, child_exit);                  return -1;
230    signal (SIGTERM, system_exit);          }
231    signal (SIG_RELOAD_MENU, reload_bbs_menu);  
232              fprintf(stderr, "%s\n", APP_INFO);
233    //Initialize socket server  
234    net_server (BBS_address, BBS_port);          // Initialize log
235            if (log_begin(LOG_FILE_INFO, LOG_FILE_ERROR) < 0)
236    //Wait for child process exit          {
237    while (SYS_child_process_count > 0)                  return -1;
238    {          }
239      log_std (".");  
240      sleep(1);          if ((!daemon_service) && std_log_redir)
241    }          {
242                    log_common_redir(STDERR_FILENO);
243            }
244            if ((!daemon_service) && error_log_redir)
245            {
246                    log_error_redir(STDERR_FILENO);
247            }
248    
249            log_common("Starting %s", APP_INFO);
250    
251            // Load configuration
252            if (load_conf(CONF_BBSD) < 0)
253            {
254                    return -2;
255            }
256    
257            // Load BWF config
258            if (bwf_load(CONF_BWF) < 0)
259            {
260                    return -2;
261            }
262    
263            // Get EULA modification tm
264            if (stat(DATA_EULA, &file_stat) == -1)
265            {
266                    log_error("stat(%s) error", DATA_EULA, errno);
267                    goto cleanup;
268            }
269            BBS_eula_tm = file_stat.st_mtim.tv_sec;
270    
271            // Check article cache dir
272            ret = mkdir(VAR_ARTICLE_CACHE_DIR, 0750);
273            if (ret == -1 && errno != EEXIST)
274            {
275                    log_error("mkdir(%s) error (%d)", VAR_ARTICLE_CACHE_DIR, errno);
276                    goto cleanup;
277            }
278    
279            // Check section aid location dir
280            ret = mkdir(VAR_SECTION_AID_LOC_DIR, 0750);
281            if (ret == -1 && errno != EEXIST)
282            {
283                    log_error("mkdir(%s) error (%d)", VAR_SECTION_AID_LOC_DIR, errno);
284                    goto cleanup;
285            }
286    
287            // Initialize data pools
288            if ((fp = fopen(VAR_ARTICLE_BLOCK_SHM, "w")) == NULL)
289            {
290                    log_error("fopen(%s) error", VAR_ARTICLE_BLOCK_SHM);
291                    goto cleanup;
292            }
293            fclose(fp);
294            if ((fp = fopen(VAR_SECTION_LIST_SHM, "w")) == NULL)
295            {
296                    log_error("fopen(%s) error", VAR_SECTION_LIST_SHM);
297                    goto cleanup;
298            }
299            fclose(fp);
300            if ((fp = fopen(VAR_TRIE_DICT_SHM, "w")) == NULL)
301            {
302                    log_error("fopen(%s) error", VAR_TRIE_DICT_SHM);
303                    goto cleanup;
304            }
305            fclose(fp);
306            if ((fp = fopen(VAR_USER_LIST_SHM, "w")) == NULL)
307            {
308                    log_error("fopen(%s) error", VAR_USER_LIST_SHM);
309                    goto cleanup;
310            }
311            fclose(fp);
312    
313            if (trie_dict_init(VAR_TRIE_DICT_SHM, TRIE_NODE_PER_POOL) < 0)
314            {
315                    log_error("trie_dict_init(%s, %d) error", VAR_TRIE_DICT_SHM, TRIE_NODE_PER_POOL);
316                    goto cleanup;
317            }
318            if (article_block_init(VAR_ARTICLE_BLOCK_SHM, BBS_article_limit_per_section * BBS_max_section / BBS_article_count_per_block) < 0)
319            {
320                    log_error("article_block_init(%s, %d) error", VAR_ARTICLE_BLOCK_SHM, BBS_article_limit_per_section * BBS_max_section / BBS_article_count_per_block);
321                    goto cleanup;
322            }
323            if (section_list_init(VAR_SECTION_LIST_SHM) < 0)
324            {
325                    log_error("section_list_pool_init(%s) error", VAR_SECTION_LIST_SHM);
326                    goto cleanup;
327            }
328    
329            // Init LML module
330            if (lml_init() < 0)
331            {
332                    log_error("lml_init() error");
333                    goto cleanup;
334            }
335    
336            // Load BBS cmd
337            if (load_cmd() < 0)
338            {
339                    log_error("load_cmd() error");
340                    goto cleanup;
341            }
342    
343            // Load menus
344            if (load_menu(&bbs_menu, CONF_MENU) < 0)
345            {
346                    log_error("load_menu(bbs_menu) error");
347                    goto cleanup;
348            }
349            if (load_menu(&top10_menu, CONF_TOP10_MENU) < 0)
350            {
351                    log_error("load_menu(top10_menu) error");
352                    goto cleanup;
353            }
354            top10_menu.allow_exit = 1;
355    
356            // Load data files
357            for (int i = 0; i < data_files_load_startup_count; i++)
358            {
359                    if (load_file(data_files_load_startup[i]) < 0)
360                    {
361                            log_error("load_file(%s) error", data_files_load_startup[i]);
362                    }
363            }
364    
365            // Load user_list and online_user_list
366            if (user_list_pool_init(VAR_USER_LIST_SHM) < 0)
367            {
368                    log_error("user_list_pool_init(%s) error", VAR_USER_LIST_SHM);
369                    goto cleanup;
370            }
371            if (user_list_pool_reload(0) < 0)
372            {
373                    log_error("user_list_pool_reload(all_user) error");
374                    goto cleanup;
375            }
376            if (user_list_pool_reload(1) < 0)
377            {
378                    log_error("user_list_pool_reload(online_user) error");
379                    goto cleanup;
380            }
381    
382            // Load section config and gen_ex
383            if (load_section_config_from_db(1) < 0)
384            {
385                    log_error("load_section_config_from_db(0) error");
386                    goto cleanup;
387            }
388    
389            set_last_article_op_log_from_db();
390            last_aid = 0;
391    
392            // Load section articles
393            do
394            {
395                    if ((ret = append_articles_from_db(last_aid + 1, 1, LOAD_ARTICLE_COUNT_LIMIT)) < 0)
396                    {
397                            log_error("append_articles_from_db(0, 1, %d) error", LOAD_ARTICLE_COUNT_LIMIT);
398                            goto cleanup;
399                    }
400    
401                    last_aid = article_block_last_aid();
402            } while (ret == LOAD_ARTICLE_COUNT_LIMIT);
403    
404            log_common("Initially load %d articles, last_aid = %d", article_block_article_count(), article_block_last_aid());
405    
406            if ((ret = user_stat_update()) < 0)
407            {
408                    log_error("user_stat_update() error");
409                    goto cleanup;
410            }
411    
412            // Set signal handler
413            act.sa_handler = sig_hup_handler;
414            if (sigaction(SIGHUP, &act, NULL) == -1)
415            {
416                    log_error("set signal action of SIGHUP error: %d", errno);
417                    goto cleanup;
418            }
419            act.sa_handler = sig_chld_handler;
420            if (sigaction(SIGCHLD, &act, NULL) == -1)
421            {
422                    log_error("set signal action of SIGCHLD error: %d", errno);
423                    goto cleanup;
424            }
425            act.sa_handler = sig_term_handler;
426            if (sigaction(SIGTERM, &act, NULL) == -1)
427            {
428                    log_error("set signal action of SIGTERM error: %d", errno);
429                    goto cleanup;
430            }
431            act.sa_handler = SIG_IGN;
432            if (sigaction(SIGPIPE, &act, NULL) == -1)
433            {
434                    log_error("set signal action of SIGPIPE error: %d", errno);
435                    goto cleanup;
436            }
437            act.sa_handler = SIG_IGN;
438            if (sigaction(SIGUSR1, &act, NULL) == -1)
439            {
440                    log_error("set signal action of SIGUSR1 error: %d", errno);
441                    goto cleanup;
442            }
443    
444            // Launch section_list_loader process
445            if (section_list_loader_launch() < 0)
446            {
447                    log_error("section_list_loader_launch() error");
448                    goto cleanup;
449            }
450    
451            // Initialize socket server
452            net_server(BBS_address, BBS_port);
453    
454    cleanup:
455            // Cleanup loader process if still running
456            if (SYS_child_process_count > 0)
457            {
458                    SYS_child_exit = 0;
459    
460                    if (kill(section_list_loader_pid, SIGTERM) < 0)
461                    {
462                            log_error("Send SIGTERM signal failed (%d)", errno);
463                    }
464    
465                    for (i = 0; SYS_child_exit == 0 && i < 5; i++)
466                    {
467                            sleep(1); // second
468                    }
469    
470                    if ((ret = waitpid(section_list_loader_pid, NULL, WNOHANG)) < 0)
471                    {
472                            log_error("waitpid(%d) error (%d)", section_list_loader_pid, errno);
473                    }
474                    else if (ret == 0)
475                    {
476                            log_common("Force kill section_list_loader process (%d)", section_list_loader_pid);
477                            if (kill(section_list_loader_pid, SIGKILL) < 0)
478                            {
479                                    log_error("Send SIGKILL signal failed (%d)", errno);
480                            }
481                    }
482            }
483    
484            // Cleanup loaded data files
485            for (int i = 0; i < data_files_load_startup_count; i++)
486            {
487                    if (unload_file(data_files_load_startup[i]) < 0)
488                    {
489                            log_error("unload_file(%s) error", data_files_load_startup[i]);
490                    }
491            }
492    
493            // Cleanup menu
494            unload_menu(&bbs_menu);
495            unload_menu(&top10_menu);
496    
497            // Cleanup LML module
498            lml_cleanup();
499    
500            // Cleanup data pools
501            section_list_cleanup();
502            article_block_cleanup();
503            trie_dict_cleanup();
504            user_list_pool_cleanup();
505    
506            if (unlink(VAR_ARTICLE_BLOCK_SHM) < 0)
507            {
508                    log_error("unlink(%s) error", VAR_ARTICLE_BLOCK_SHM);
509            }
510            if (unlink(VAR_SECTION_LIST_SHM) < 0)
511            {
512                    log_error("unlink(%s) error", VAR_SECTION_LIST_SHM);
513            }
514            if (unlink(VAR_TRIE_DICT_SHM) < 0)
515            {
516                    log_error("unlink(%s) error", VAR_TRIE_DICT_SHM);
517            }
518            if (unlink(VAR_USER_LIST_SHM) < 0)
519            {
520                    log_error("unlink(%s) error", VAR_SECTION_LIST_SHM);
521            }
522    
523            log_common("Main process exit normally");
524    
525            log_end();
526    
527    return 0;          return 0;
528  }  }


Legend:
Removed lines/characters  
Changed lines/characters
  Added lines/characters

webmaster@leafok.com
ViewVC Help
Powered by ViewVC 1.3.0-beta1