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

Diff of /lbbs/src/bbs_main.c

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

Revision 1.48 by sysadm, Thu May 15 14:34:13 2025 UTC Revision 1.89 by sysadm, Sat Oct 18 05:55:55 2025 UTC
# Line 14  Line 14 
14   *                                                                         *   *                                                                         *
15   ***************************************************************************/   ***************************************************************************/
16    
17  #include "bbs_main.h"  #include "article_favor.h"
18    #include "article_view_log.h"
19  #include "bbs.h"  #include "bbs.h"
20  #include "login.h"  #include "bbs_cmd.h"
21  #include "user_priv.h"  #include "bbs_main.h"
22  #include "common.h"  #include "common.h"
23  #include "database.h"  #include "database.h"
24  #include "log.h"  #include "editor.h"
25  #include "io.h"  #include "io.h"
26  #include "screen.h"  #include "log.h"
27    #include "login.h"
28  #include "menu.h"  #include "menu.h"
29  #include "bbs_cmd.h"  #include "screen.h"
30  #include <unistd.h>  #include "section_list.h"
31  #include <time.h>  #include "trie_dict.h"
32    #include "user_priv.h"
33    #include <errno.h>
34    #include <signal.h>
35    #include <stdlib.h>
36  #include <string.h>  #include <string.h>
37    #include <time.h>
38    #include <unistd.h>
39    
40  int bbs_info()  int bbs_info()
41  {  {
42          prints("ӭ \033[1;33m%s \033[32m[%s]  \033[37m( %s )\r\n",          prints("欢迎光临 \033[1;33m%s \033[32m[%s]  \033[37m( %s )\033[m\r\n",
43                     BBS_name, BBS_server, app_version);                     BBS_name, BBS_server, APP_INFO);
44    
45          return iflush();          return iflush();
46  }  }
47    
48  int bbs_welcome(MYSQL *db)  int bbs_welcome(void)
49  {  {
50          char sql[SQL_BUFFER_LEN];          char sql[SQL_BUFFER_LEN];
51    
52          u_int32_t u_online = 0;          u_int32_t u_online = 0;
53          u_int32_t u_anonymous = 0;          u_int32_t u_anonymous = 0;
54          u_int32_t u_total = 0;          u_int32_t u_total = 0;
         u_int32_t max_u_online = 0;  
55          u_int32_t u_login_count = 0;          u_int32_t u_login_count = 0;
56    
57            MYSQL *db;
58          MYSQL_RES *rs;          MYSQL_RES *rs;
59          MYSQL_ROW row;          MYSQL_ROW row;
60    
61            db = db_open();
62            if (db == NULL)
63            {
64                    return -1;
65            }
66    
67          snprintf(sql, sizeof(sql),          snprintf(sql, sizeof(sql),
68                           "SELECT COUNT(*) AS cc FROM "                           "SELECT COUNT(*) AS cc FROM "
69                           "(SELECT DISTINCT SID FROM user_online "                           "(SELECT DISTINCT SID FROM user_online "
# Line 57  int bbs_welcome(MYSQL *db) Line 71  int bbs_welcome(MYSQL *db)
71                           BBS_user_off_line);                           BBS_user_off_line);
72          if (mysql_query(db, sql) != 0)          if (mysql_query(db, sql) != 0)
73          {          {
74                  log_error("Query user_online failed\n");                  log_error("Query user_online error: %s\n", mysql_error(db));
75                    mysql_close(db);
76                  return -2;                  return -2;
77          }          }
78          if ((rs = mysql_store_result(db)) == NULL)          if ((rs = mysql_store_result(db)) == NULL)
79          {          {
80                  log_error("Get user_online data failed\n");                  log_error("Get user_online data failed\n");
81                    mysql_close(db);
82                  return -2;                  return -2;
83          }          }
84          if ((row = mysql_fetch_row(rs)))          if ((row = mysql_fetch_row(rs)))
# Line 78  int bbs_welcome(MYSQL *db) Line 94  int bbs_welcome(MYSQL *db)
94                           BBS_user_off_line);                           BBS_user_off_line);
95          if (mysql_query(db, sql) != 0)          if (mysql_query(db, sql) != 0)
96          {          {
97                  log_error("Query user_online failed\n");                  log_error("Query user_online error: %s\n", mysql_error(db));
98                    mysql_close(db);
99                  return -2;                  return -2;
100          }          }
101          if ((rs = mysql_store_result(db)) == NULL)          if ((rs = mysql_store_result(db)) == NULL)
102          {          {
103                  log_error("Get user_online data failed\n");                  log_error("Get user_online data failed\n");
104                    mysql_close(db);
105                  return -2;                  return -2;
106          }          }
107          if ((row = mysql_fetch_row(rs)))          if ((row = mysql_fetch_row(rs)))
# Line 95  int bbs_welcome(MYSQL *db) Line 113  int bbs_welcome(MYSQL *db)
113          snprintf(sql, sizeof(sql), "SELECT COUNT(UID) AS cc FROM user_list WHERE enable");          snprintf(sql, sizeof(sql), "SELECT COUNT(UID) AS cc FROM user_list WHERE enable");
114          if (mysql_query(db, sql) != 0)          if (mysql_query(db, sql) != 0)
115          {          {
116                  log_error("Query user_list failed\n");                  log_error("Query user_list error: %s\n", mysql_error(db));
117                    mysql_close(db);
118                  return -2;                  return -2;
119          }          }
120          if ((rs = mysql_store_result(db)) == NULL)          if ((rs = mysql_store_result(db)) == NULL)
121          {          {
122                  log_error("Get user_list data failed\n");                  log_error("Get user_list data failed\n");
123                    mysql_close(db);
124                  return -2;                  return -2;
125          }          }
126          if ((row = mysql_fetch_row(rs)))          if ((row = mysql_fetch_row(rs)))
# Line 112  int bbs_welcome(MYSQL *db) Line 132  int bbs_welcome(MYSQL *db)
132          snprintf(sql, sizeof(sql), "SELECT ID FROM user_login_log ORDER BY ID LIMIT 1");          snprintf(sql, sizeof(sql), "SELECT ID FROM user_login_log ORDER BY ID LIMIT 1");
133          if (mysql_query(db, sql) != 0)          if (mysql_query(db, sql) != 0)
134          {          {
135                  log_error("Query user_login_log failed\n");                  log_error("Query user_login_log error: %s\n", mysql_error(db));
136                    mysql_close(db);
137                  return -2;                  return -2;
138          }          }
139          if ((rs = mysql_store_result(db)) == NULL)          if ((rs = mysql_store_result(db)) == NULL)
140          {          {
141                  log_error("Get user_login_log data failed\n");                  log_error("Get user_login_log data failed\n");
142                    mysql_close(db);
143                  return -2;                  return -2;
144          }          }
145          if ((row = mysql_fetch_row(rs)))          if ((row = mysql_fetch_row(rs)))
# Line 126  int bbs_welcome(MYSQL *db) Line 148  int bbs_welcome(MYSQL *db)
148          }          }
149          mysql_free_result(rs);          mysql_free_result(rs);
150    
151          // Log max user_online          mysql_close(db);
         FILE *fin, *fout;  
         if ((fin = fopen(VAR_MAX_USER_ONLINE, "r")) != NULL)  
         {  
                 fscanf(fin, "%d", &max_u_online);  
                 fclose(fin);  
         }  
         if (u_online > max_u_online)  
         {  
                 max_u_online = u_online;  
                 if ((fout = fopen(VAR_MAX_USER_ONLINE, "w")) == NULL)  
                 {  
                         log_error("Open max_user_online.dat failed\n");  
                         return -3;  
                 }  
                 fprintf(fout, "%d\n", max_u_online);  
                 fclose(fout);  
         }  
152    
153          // Count current user before login          // Count current user before login
154          u_online++;          u_online++;
155          u_anonymous++;          u_anonymous++;
156    
157          // Display logo          // Display logo
158          display_file_ex(DATA_WELCOME, 1, 0);          display_file(DATA_WELCOME, 2);
159    
160          // Display welcome message          // Display welcome message
161          prints("\r\033[1;35mӭ\033[33m %s \033[35mBBS\r\n"          prints("\r\033[1;35m欢迎光临\033[33m 【 %s 】 \033[35mBBS\r\n"
162                     "\033[32mĿǰվ [\033[36m%d/%d\033[32m] "                     "\033[32m目前上站人数 [\033[36m%d/%d\033[32m] "
163                     "ο[\033[36m%d\033[32m] "                     "匿名游客[\033[36m%d\033[32m] "
164                     "עû[\033[36m%d/%d\033[32m]\r\n"                     "注册用户数[\033[36m%d/%d\033[32m]\r\n"
165                     " [\033[36m%s\033[32m] ¼"                     "从 [\033[36m%s\033[32m] 起，累计访问人次：[\033[36m%d\033[32m]\033[m\r\n",
                    "[\033[36m%d\033[32m]ۼƷ˴Σ[\033[36m%d\033[32m]\r\n",  
166                     BBS_name, u_online, BBS_max_client, u_anonymous, u_total,                     BBS_name, u_online, BBS_max_client, u_anonymous, u_total,
167                     BBS_max_user, BBS_start_dt, max_u_online, u_login_count);                     BBS_max_user, BBS_start_dt, u_login_count);
168    
169            iflush();
170    
171          return 0;          return 0;
172  }  }
173    
174  int bbs_logout(MYSQL *db)  int bbs_logout(void)
175  {  {
176          if (user_online_del(db) < 0)          MYSQL *db;
177    
178            db = db_open();
179            if (db == NULL)
180          {          {
181                  return -1;                  return -1;
182          }          }
183    
184          display_file_ex(DATA_GOODBYE, 1, 0);          if (user_online_exp(db) < 0)
185            {
186                    return -2;
187            }
188    
189          log_std("User logout\n");          if (user_online_del(db) < 0)
190            {
191                    return -3;
192            }
193    
194            mysql_close(db);
195    
196            display_file(DATA_GOODBYE, 1);
197    
198            log_common("User [%s] logout\n", BBS_username);
199    
200          return 0;          return 0;
201  }  }
# Line 184  int bbs_center() Line 205  int bbs_center()
205          int ch;          int ch;
206          time_t t_last_action;          time_t t_last_action;
207    
208          BBS_last_access_tm = t_last_action = time(0);          BBS_last_access_tm = t_last_action = time(NULL);
209    
210          clearscr();          clearscr();
211    
212          show_top("");          show_top("", BBS_name, "");
213          show_active_board();          show_active_board();
214          show_bottom("");          show_bottom("");
215          display_menu(p_bbs_menu);          display_menu(&bbs_menu);
216            iflush();
217    
218          while (!SYS_server_exit)          while (!SYS_server_exit)
219          {          {
220                  ch = igetch(100);                  ch = igetch(100);
221    
222                  if (time(0) - t_last_action >= 10)                  if (ch != KEY_NULL && ch != KEY_TIMEOUT)
223                    {
224                            BBS_last_access_tm = time(NULL);
225                    }
226    
227                    if (bbs_menu.choose_step == 0 && time(NULL) - t_last_action >= 10)
228                  {                  {
229                          t_last_action = time(0);                          t_last_action = time(NULL);
230    
231                          show_active_board();                          show_active_board();
232                          show_bottom("");                          show_bottom("");
233                            display_menu_cursor(&bbs_menu, 1);
234                            iflush();
235                    }
236    
237                    if (user_online_update("MENU") < 0)
238                    {
239                            log_error("user_online_update(MENU) error\n");
240                  }                  }
241    
242                  switch (ch)                  switch (ch)
243                  {                  {
244                  case KEY_NULL: // broken pipe                  case KEY_NULL: // broken pipe
245                            log_error("KEY_NULL\n");
246                          return 0;                          return 0;
247                  case KEY_TIMEOUT:                  case KEY_TIMEOUT:
248                          if (time(0) - BBS_last_access_tm >= MAX_DELAY_TIME)                          if (time(NULL) - BBS_last_access_tm >= MAX_DELAY_TIME)
249                          {                          {
250                                    log_error("User input timeout\n");
251                                  return 0;                                  return 0;
252                          }                          }
253                          continue;                          continue;
254                  case CR:                  case CR:
                         igetch_reset();  
255                  default:                  default:
256                          switch (menu_control(p_bbs_menu, ch))                          switch (menu_control(&bbs_menu, ch))
257                          {                          {
258                          case EXITBBS:                          case EXITBBS:
259                            case EXITMENU:
260                                  return 0;                                  return 0;
261                          case REDRAW:                          case REDRAW:
262                                    t_last_action = time(NULL);
263                                  clearscr();                                  clearscr();
264                                  show_top("");                                  show_top("", BBS_name, "");
265                                  show_active_board();                                  show_active_board();
266                                  show_bottom("");                                  show_bottom("");
267                                  display_menu(p_bbs_menu);                                  display_menu(&bbs_menu);
268                                  break;                                  break;
269                          case NOREDRAW:                          case NOREDRAW:
270                          case UNKNOWN_CMD:                          case UNKNOWN_CMD:
271                          default:                          default:
272                                  break;                                  break;
273                          }                          }
274                            iflush();
275                    }
276            }
277    
278            return 0;
279    }
280    
281    int bbs_charset_select()
282    {
283            char msg[LINE_BUFFER_LEN];
284            int ch;
285    
286            snprintf(msg, sizeof(msg),
287                             "\rChoose character set in 5 seconds [UTF-8, GBK]: [U/g]");
288    
289            while (!SYS_server_exit)
290            {
291                    ch = press_any_key_ex(msg, 5);
292                    switch (ch)
293                    {
294                    case KEY_NULL:
295                            return -1;
296                    case KEY_TIMEOUT:
297                    case CR:
298                    case 'u':
299                    case 'U':
300                            return 0;
301                    case 'g':
302                    case 'G':
303                            if (io_conv_init("GBK") < 0)
304                            {
305                                    log_error("io_conv_init(%s) error\n", "GBK");
306                                    return -1;
307                            }
308                            return 0;
309                    default:
310                            continue;
311                  }                  }
                 BBS_last_access_tm = time(0);  
312          }          }
313    
314          return 0;          return 0;
# Line 242  int bbs_center() Line 316  int bbs_center()
316    
317  int bbs_main()  int bbs_main()
318  {  {
319          MYSQL *db;          struct sigaction act = {0};
320            char msg[LINE_BUFFER_LEN];
321    
322            // Set signal handler
323            act.sa_handler = SIG_IGN;
324            if (sigaction(SIGHUP, &act, NULL) == -1)
325            {
326                    log_error("set signal action of SIGHUP error: %d\n", errno);
327                    goto cleanup;
328            }
329            act.sa_handler = SIG_DFL;
330            if (sigaction(SIGCHLD, &act, NULL) == -1)
331            {
332                    log_error("set signal action of SIGCHLD error: %d\n", errno);
333                    goto cleanup;
334            }
335    
336            // Set data pools in shared memory readonly
337            if (set_trie_dict_shm_readonly() < 0)
338            {
339                    goto cleanup;
340            }
341            if (set_article_block_shm_readonly() < 0)
342            {
343                    goto cleanup;
344            }
345            if (set_section_list_shm_readonly() < 0)
346            {
347                    goto cleanup;
348            }
349    
350            // Load menu in shared memory
351            if (set_menu_shm_readonly(&bbs_menu) < 0)
352            {
353                    goto cleanup;
354            }
355    
356            // Set default charset
357            if (io_conv_init(BBS_DEFAULT_CHARSET) < 0)
358            {
359                    log_error("io_conv_init(%s) error\n", BBS_DEFAULT_CHARSET);
360                    goto cleanup;
361            }
362    
363          set_input_echo(0);          set_input_echo(0);
364    
365            // Set user charset
366            bbs_charset_select();
367    
368          // System info          // System info
369          if (bbs_info() < 0)          if (bbs_info() < 0)
370          {          {
371                  return -1;                  goto cleanup;
372          }          }
373    
374          db = db_open();          // Welcome
375          if (db == NULL)          if (bbs_welcome() < 0)
376          {          {
377                  prints("޷ݿ\n");                  goto cleanup;
                 return -2;  
378          }          }
379    
380          // Welcome          // User login
381          if (bbs_welcome(db) < 0)          if (SSH_v2)
382          {          {
383                  mysql_close(db);                  snprintf(msg, sizeof(msg), "\033[1m%s 欢迎使用ssh方式访问 \033[1;33m按任意键继续...\033[m", BBS_username);
384                  return -3;                  press_any_key_ex(msg, 60);
385          }          }
386            else if (bbs_login() < 0)
387            {
388                    goto cleanup;
389            }
390            log_common("User [%s] login\n", BBS_username);
391    
392          // User login          // Load article view log
393          if (bbs_login(db) < 0)          if (article_view_log_load(BBS_priv.uid, &BBS_article_view_log, 0) < 0)
394          {          {
395                  mysql_close(db);                  log_error("article_view_log_load() error\n");
396                  return -4;                  goto cleanup;
397          }          }
         clearscr();  
398    
399          // BBS Top 10          // Load article favorite
400          display_file_ex("./var/bbs_top.txt", 1, 1);          if (article_favor_load(BBS_priv.uid, &BBS_article_favor, 0) < 0)
401            {
402                    log_error("article_favor_load() error\n");
403                    goto cleanup;
404            }
405    
406          // Load menu in shared memory          // Init editor memory pool
407          if (load_menu_shm(p_bbs_menu) < 0)          if (editor_memory_pool_init() < 0)
408          {          {
409                  return -5;                  log_error("editor_memory_pool_init() error\n");
410                    goto cleanup;
411          }          }
412    
413            clearscr();
414    
415            // BBS Top 10
416            display_file(VAR_BBS_TOP, 1);
417    
418          // Main          // Main
419          bbs_center();          bbs_center();
420    
         // Unload menu in shared memory  
         unload_menu_shm(p_bbs_menu);  
         free(p_bbs_menu);  
         p_bbs_menu = NULL;  
   
421          // Logout          // Logout
422          bbs_logout(db);          bbs_logout();
423    
424          mysql_close(db);          // Save incremental article view log
425            if (article_view_log_save_inc(&BBS_article_view_log) < 0)
426            {
427                    log_error("article_view_log_save_inc() error\n");
428            }
429    
430            // Save incremental article favorite
431            if (article_favor_save_inc(&BBS_article_favor) < 0)
432            {
433                    log_error("article_favor_save_inc() error\n");
434            }
435    
436    cleanup:
437            // Cleanup iconv
438            io_conv_cleanup();
439    
440            // Cleanup editor memory pool
441            editor_memory_pool_cleanup();
442    
443            // Unload article view log
444            article_view_log_unload(&BBS_article_view_log);
445    
446            // Unload article favor
447            article_favor_unload(&BBS_article_favor);
448    
449            // Detach menu in shared memory
450            detach_menu_shm(&bbs_menu);
451            detach_menu_shm(&top10_menu);
452    
453            // Detach data pools shm
454            detach_section_list_shm();
455            detach_article_block_shm();
456            detach_trie_dict_shm();
457    
458          return 0;          return 0;
459  }  }

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