/[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.37 by sysadm, Fri May 9 11:21:11 2025 UTC Revision 1.87 by sysadm, Sat Oct 18 05:02:15 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 "welcome.h"  #include "bbs_cmd.h"
21  #include "login.h"  #include "bbs_main.h"
 #include "user_priv.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();
46    }
47    
48    int bbs_welcome(void)
49    {
50            char sql[SQL_BUFFER_LEN];
51    
52            u_int32_t u_online = 0;
53            u_int32_t u_anonymous = 0;
54            u_int32_t u_total = 0;
55            u_int32_t u_login_count = 0;
56    
57            MYSQL *db;
58            MYSQL_RES *rs;
59            MYSQL_ROW row;
60    
61            db = db_open();
62            if (db == NULL)
63            {
64                    return -1;
65            }
66    
67            snprintf(sql, sizeof(sql),
68                             "SELECT COUNT(*) AS cc FROM "
69                             "(SELECT DISTINCT SID FROM user_online "
70                             "WHERE last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND)) AS t1",
71                             BBS_user_off_line);
72            if (mysql_query(db, sql) != 0)
73            {
74                    log_error("Query user_online error: %s\n", mysql_error(db));
75                    mysql_close(db);
76                    return -2;
77            }
78            if ((rs = mysql_store_result(db)) == NULL)
79            {
80                    log_error("Get user_online data failed\n");
81                    mysql_close(db);
82                    return -2;
83            }
84            if ((row = mysql_fetch_row(rs)))
85            {
86                    u_online = (u_int32_t)atoi(row[0]);
87            }
88            mysql_free_result(rs);
89    
90            snprintf(sql, sizeof(sql),
91                             "SELECT COUNT(*) AS cc FROM "
92                             "(SELECT DISTINCT SID FROM user_online "
93                             "WHERE UID = 0 AND last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND)) AS t1",
94                             BBS_user_off_line);
95            if (mysql_query(db, sql) != 0)
96            {
97                    log_error("Query user_online error: %s\n", mysql_error(db));
98                    mysql_close(db);
99                    return -2;
100            }
101            if ((rs = mysql_store_result(db)) == NULL)
102            {
103                    log_error("Get user_online data failed\n");
104                    mysql_close(db);
105                    return -2;
106            }
107            if ((row = mysql_fetch_row(rs)))
108            {
109                    u_anonymous = (u_int32_t)atoi(row[0]);
110            }
111            mysql_free_result(rs);
112    
113            snprintf(sql, sizeof(sql), "SELECT COUNT(UID) AS cc FROM user_list WHERE enable");
114            if (mysql_query(db, sql) != 0)
115            {
116                    log_error("Query user_list error: %s\n", mysql_error(db));
117                    mysql_close(db);
118                    return -2;
119            }
120            if ((rs = mysql_store_result(db)) == NULL)
121            {
122                    log_error("Get user_list data failed\n");
123                    mysql_close(db);
124                    return -2;
125            }
126            if ((row = mysql_fetch_row(rs)))
127            {
128                    u_total = (u_int32_t)atoi(row[0]);
129            }
130            mysql_free_result(rs);
131    
132            snprintf(sql, sizeof(sql), "SELECT ID FROM user_login_log ORDER BY ID LIMIT 1");
133            if (mysql_query(db, sql) != 0)
134            {
135                    log_error("Query user_login_log error: %s\n", mysql_error(db));
136                    mysql_close(db);
137                    return -2;
138            }
139            if ((rs = mysql_store_result(db)) == NULL)
140            {
141                    log_error("Get user_login_log data failed\n");
142                    mysql_close(db);
143                    return -2;
144            }
145            if ((row = mysql_fetch_row(rs)))
146            {
147                    u_login_count = (u_int32_t)atoi(row[0]);
148            }
149            mysql_free_result(rs);
150    
151            mysql_close(db);
152    
153            // Count current user before login
154            u_online++;
155            u_anonymous++;
156    
157            // Display logo
158            display_file(DATA_WELCOME, 2);
159    
160            // Display welcome message
161            prints("\r\033[1;35m欢迎光临\033[33m 【 %s 】 \033[35mBBS\r\n"
162                       "\033[32m目前上站人数 [\033[36m%d/%d\033[32m] "
163                       "匿名游客[\033[36m%d\033[32m] "
164                       "注册用户数[\033[36m%d/%d\033[32m]\r\n"
165                       "从 [\033[36m%s\033[32m] 起，累计访问人次：[\033[36m%d\033[32m]\033[m\r\n",
166                       BBS_name, u_online, BBS_max_client, u_anonymous, u_total,
167                       BBS_max_user, BBS_start_dt, u_login_count);
168    
169          iflush();          iflush();
170    
171          return 0;          return 0;
172  }  }
173    
174  int bbs_exit()  int bbs_logout(void)
175  {  {
176          display_file_ex(DATA_GOODBYE, 1, 0);          MYSQL *db;
177    
178            db = db_open();
179            if (db == NULL)
180            {
181                    return -1;
182            }
183    
184            if (user_online_exp(db) < 0)
185            {
186                    return -2;
187            }
188    
189          sleep(1);          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  }  }
202    
203  int bbs_center()  int bbs_center()
204  {  {
205          int ch, redraw;          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(get_menu(&bbs_menu, "TOPMENU"));          display_menu(&bbs_menu);
216            iflush();
217    
218          while (!SYS_server_exit)          while (!SYS_server_exit)
219          {          {
220                  ch = igetch(0);                  ch = igetch(100);
221    
222                  if (time(0) - t_last_action >= 10)                  if (ch != KEY_NULL && ch != KEY_TIMEOUT)
223                  {                  {
224                          t_last_action = time(0);                          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(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:                  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                                  return -1;                                  log_error("User input timeout\n");
251                                    return 0;
252                          }                          }
253                          continue;                          continue;
254                    case CR:
255                  default:                  default:
                         redraw = 1;  
256                          switch (menu_control(&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();
264                                    show_top("", BBS_name, "");
265                                    show_active_board();
266                                    show_bottom("");
267                                    display_menu(&bbs_menu);
268                                  break;                                  break;
269                          case NOREDRAW:                          case NOREDRAW:
270                          case UNKNOWN_CMD:                          case UNKNOWN_CMD:
271                          default:                          default:
                                 redraw = 0;  
272                                  break;                                  break;
273                          }                          }
274                          if (redraw)                          iflush();
275                          {                  }
276                                  clearscr();          }
277                                  show_top("");  
278                                  show_active_board();          return 0;
279                                  show_bottom("");  }
280                                  display_current_menu(&bbs_menu);  
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            ch = press_any_key_ex(msg, 5);
290            switch (ch)
291            {
292            case 'g':
293            case 'G':
294                    if (io_conv_init("GBK") < 0)
295                    {
296                            log_error("io_conv_init(%s) error\n", "GBK");
297                            return -1;
298                  }                  }
299                  BBS_last_access_tm = time(0);                  break;
300            default:
301                    log_error("Debug: %d\n", ch);
302          }          }
303    
304            prints("\r\n");
305            iflush();
306    
307          return 0;          return 0;
308  }  }
309    
310  int bbs_main()  int bbs_main()
311  {  {
312          int ret;          struct sigaction act = {0};
313            char msg[LINE_BUFFER_LEN];
314    
315            // Set signal handler
316            act.sa_handler = SIG_IGN;
317            if (sigaction(SIGHUP, &act, NULL) == -1)
318            {
319                    log_error("set signal action of SIGHUP error: %d\n", errno);
320                    goto cleanup;
321            }
322            act.sa_handler = SIG_DFL;
323            if (sigaction(SIGCHLD, &act, NULL) == -1)
324            {
325                    log_error("set signal action of SIGCHLD error: %d\n", errno);
326                    goto cleanup;
327            }
328    
329            // Set data pools in shared memory readonly
330            if (set_trie_dict_shm_readonly() < 0)
331            {
332                    goto cleanup;
333            }
334            if (set_article_block_shm_readonly() < 0)
335            {
336                    goto cleanup;
337            }
338            if (set_section_list_shm_readonly() < 0)
339            {
340                    goto cleanup;
341            }
342    
343            // Load menu in shared memory
344            if (set_menu_shm_readonly(&bbs_menu) < 0)
345            {
346                    goto cleanup;
347            }
348    
349            // Set default charset
350            if (io_conv_init(BBS_DEFAULT_CHARSET) < 0)
351            {
352                    log_error("io_conv_init(%s) error\n", BBS_DEFAULT_CHARSET);
353                    goto cleanup;
354            }
355    
356          set_input_echo(0);          set_input_echo(0);
357    
358          bbs_info();          // Set user charset
359            bbs_charset_select();
360    
361            // System info
362            if (bbs_info() < 0)
363            {
364                    goto cleanup;
365            }
366    
367          // Welcome          // Welcome
368          bbs_welcome();          if (bbs_welcome() < 0)
369            {
370                    goto cleanup;
371            }
372    
373            // User login
374            if (SSH_v2)
375            {
376                    snprintf(msg, sizeof(msg), "\033[1m%s 欢迎使用ssh方式访问 \033[1;33m按任意键继续...\033[m", BBS_username);
377                    press_any_key_ex(msg, 60);
378            }
379            else if (bbs_login() < 0)
380            {
381                    goto cleanup;
382            }
383            log_common("User [%s] login\n", BBS_username);
384    
385            // Load article view log
386            if (article_view_log_load(BBS_priv.uid, &BBS_article_view_log, 0) < 0)
387            {
388                    log_error("article_view_log_load() error\n");
389                    goto cleanup;
390            }
391    
392            // Load article favorite
393            if (article_favor_load(BBS_priv.uid, &BBS_article_favor, 0) < 0)
394            {
395                    log_error("article_favor_load() error\n");
396                    goto cleanup;
397            }
398    
399            // Init editor memory pool
400            if (editor_memory_pool_init() < 0)
401            {
402                    log_error("editor_memory_pool_init() error\n");
403                    goto cleanup;
404            }
405    
         // Login  
         ret = bbs_login();  
         if (ret < 0)  
                 return -1;  
         log_std("User \"%s\"(%ld) login from %s:%d\n",  
                         BBS_username, BBS_priv.uid, hostaddr_client, port_client);  
406          clearscr();          clearscr();
407    
408          // BBS Top 10          // BBS Top 10
409          display_file_ex("./var/bbs_top.txt", 1, 1);          display_file(VAR_BBS_TOP, 1);
410    
411          // Main          // Main
412          bbs_center();          bbs_center();
413    
414          // Logout          // Logout
415          bbs_exit();          bbs_logout();
         log_std("User logout\n");  
416    
417          MYSQL *db = db_open();          // Save incremental article view log
418          if (db == NULL)          if (article_view_log_save_inc(&BBS_article_view_log) < 0)
419          {          {
420                  return -1;                  log_error("article_view_log_save_inc() error\n");
421          }          }
422    
423          user_online_del(db);          // Save incremental article favorite
424            if (article_favor_save_inc(&BBS_article_favor) < 0)
425            {
426                    log_error("article_favor_save_inc() error\n");
427            }
428    
429          mysql_close(db);  cleanup:
430            // Cleanup iconv
431            io_conv_cleanup();
432    
433            // Cleanup editor memory pool
434            editor_memory_pool_cleanup();
435    
436            // Unload article view log
437            article_view_log_unload(&BBS_article_view_log);
438    
439            // Unload article favor
440            article_favor_unload(&BBS_article_favor);
441    
442            // Detach menu in shared memory
443            detach_menu_shm(&bbs_menu);
444            detach_menu_shm(&top10_menu);
445    
446            // Detach data pools shm
447            detach_section_list_shm();
448            detach_article_block_shm();
449            detach_trie_dict_shm();
450    
451          return 0;          return 0;
452  }  }

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