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

Annotation of /lbbs/src/bbs_main.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.94 - (hide annotations)
Sat Nov 1 10:35:42 2025 UTC (4 months, 2 weeks ago) by sysadm
Branch: MAIN
Changes since 1.93: +18 -69 lines
Content type: text/x-csrc
Get online user/guest count from user_list instead of database

1 sysadm 1.1 /***************************************************************************
2 sysadm 1.31 bbs_main.c - description
3     -------------------
4 sysadm 1.36 Copyright : (C) 2004-2025 by Leaflet
5     Email : leaflet@leafok.com
6 sysadm 1.1 ***************************************************************************/
7    
8     /***************************************************************************
9     * *
10     * This program is free software; you can redistribute it and/or modify *
11     * it under the terms of the GNU General Public License as published by *
12 sysadm 1.36 * the Free Software Foundation; either version 3 of the License, or *
13 sysadm 1.1 * (at your option) any later version. *
14     * *
15     ***************************************************************************/
16    
17 sysadm 1.78 #include "article_favor.h"
18 sysadm 1.73 #include "article_view_log.h"
19     #include "bbs.h"
20     #include "bbs_cmd.h"
21 sysadm 1.32 #include "bbs_main.h"
22 sysadm 1.1 #include "common.h"
23 sysadm 1.35 #include "database.h"
24 sysadm 1.73 #include "editor.h"
25     #include "io.h"
26 sysadm 1.32 #include "log.h"
27 sysadm 1.73 #include "login.h"
28     #include "menu.h"
29 sysadm 1.32 #include "screen.h"
30 sysadm 1.55 #include "section_list.h"
31     #include "trie_dict.h"
32 sysadm 1.93 #include "user_list.h"
33 sysadm 1.73 #include "user_priv.h"
34 sysadm 1.74 #include <errno.h>
35     #include <signal.h>
36 sysadm 1.73 #include <stdlib.h>
37     #include <string.h>
38     #include <time.h>
39 sysadm 1.32 #include <unistd.h>
40 sysadm 1.16
41 sysadm 1.31 int bbs_info()
42 sysadm 1.5 {
43 sysadm 1.77 prints("欢迎光临 \033[1;33m%s \033[32m[%s] \033[37m( %s )\033[m\r\n",
44 sysadm 1.72 BBS_name, BBS_server, APP_INFO);
45 sysadm 1.16
46 sysadm 1.43 return iflush();
47     }
48    
49 sysadm 1.65 int bbs_welcome(void)
50 sysadm 1.43 {
51     char sql[SQL_BUFFER_LEN];
52    
53 sysadm 1.94 int u_online = 0;
54     int u_anonymous = 0;
55     int u_total = 0;
56     int u_login_count = 0;
57 sysadm 1.43
58 sysadm 1.65 MYSQL *db;
59 sysadm 1.43 MYSQL_RES *rs;
60     MYSQL_ROW row;
61    
62 sysadm 1.94 if (get_user_online_list_count(&u_online, &u_anonymous) < 0)
63 sysadm 1.43 {
64 sysadm 1.94 log_error("get_user_online_list_count() error\n");
65     u_online = 0;
66 sysadm 1.43 }
67 sysadm 1.94 u_online += u_anonymous;
68     u_online++; // current user
69     if (BBS_priv.uid == 0)
70 sysadm 1.43 {
71 sysadm 1.94 u_anonymous++;
72 sysadm 1.43 }
73    
74 sysadm 1.94 if (get_user_list_count(&u_total) < 0)
75 sysadm 1.43 {
76 sysadm 1.94 log_error("get_user_list_count() error\n");
77     u_total = 0;
78 sysadm 1.43 }
79    
80 sysadm 1.94 db = db_open();
81     if (db == NULL)
82 sysadm 1.43 {
83 sysadm 1.94 return -1;
84 sysadm 1.43 }
85    
86     snprintf(sql, sizeof(sql), "SELECT ID FROM user_login_log ORDER BY ID LIMIT 1");
87     if (mysql_query(db, sql) != 0)
88     {
89 sysadm 1.54 log_error("Query user_login_log error: %s\n", mysql_error(db));
90 sysadm 1.65 mysql_close(db);
91 sysadm 1.43 return -2;
92     }
93     if ((rs = mysql_store_result(db)) == NULL)
94     {
95     log_error("Get user_login_log data failed\n");
96 sysadm 1.65 mysql_close(db);
97 sysadm 1.43 return -2;
98     }
99     if ((row = mysql_fetch_row(rs)))
100     {
101 sysadm 1.94 u_login_count = atoi(row[0]);
102 sysadm 1.43 }
103     mysql_free_result(rs);
104    
105 sysadm 1.65 mysql_close(db);
106    
107 sysadm 1.43 // Display logo
108 sysadm 1.68 display_file(DATA_WELCOME, 2);
109 sysadm 1.43
110     // Display welcome message
111 sysadm 1.76 prints("\r\033[1;35m欢迎光临\033[33m 【 %s 】 \033[35mBBS\r\n"
112     "\033[32m目前上站人数 [\033[36m%d/%d\033[32m] "
113     "匿名游客[\033[36m%d\033[32m] "
114     "注册用户数[\033[36m%d/%d\033[32m]\r\n"
115     "从 [\033[36m%s\033[32m] 起,累计访问人次:[\033[36m%d\033[32m]\033[m\r\n",
116 sysadm 1.43 BBS_name, u_online, BBS_max_client, u_anonymous, u_total,
117 sysadm 1.93 BBS_max_user_count, BBS_start_dt, u_login_count);
118 sysadm 1.5
119 sysadm 1.64 iflush();
120    
121 sysadm 1.31 return 0;
122 sysadm 1.5 }
123    
124 sysadm 1.65 int bbs_logout(void)
125 sysadm 1.5 {
126 sysadm 1.65 MYSQL *db;
127    
128     db = db_open();
129     if (db == NULL)
130     {
131     return -1;
132     }
133    
134 sysadm 1.80 if (user_online_exp(db) < 0)
135     {
136     return -2;
137     }
138    
139 sysadm 1.43 if (user_online_del(db) < 0)
140     {
141 sysadm 1.80 return -3;
142 sysadm 1.43 }
143    
144 sysadm 1.65 mysql_close(db);
145    
146 sysadm 1.68 display_file(DATA_GOODBYE, 1);
147 sysadm 1.43
148 sysadm 1.91 log_common("User [%s] logout, idle for %ld seconds since last input\n", BBS_username, time(NULL) - BBS_last_access_tm);
149 sysadm 1.7
150 sysadm 1.31 return 0;
151 sysadm 1.7 }
152    
153 sysadm 1.31 int bbs_center()
154 sysadm 1.7 {
155 sysadm 1.39 int ch;
156 sysadm 1.31 time_t t_last_action;
157 sysadm 1.16
158 sysadm 1.70 BBS_last_access_tm = t_last_action = time(NULL);
159 sysadm 1.16
160 sysadm 1.31 clearscr();
161 sysadm 1.12
162 sysadm 1.58 show_top("", BBS_name, "");
163 sysadm 1.31 show_active_board();
164     show_bottom("");
165 sysadm 1.75 display_menu(&bbs_menu);
166 sysadm 1.50 iflush();
167 sysadm 1.16
168 sysadm 1.37 while (!SYS_server_exit)
169 sysadm 1.25 {
170 sysadm 1.44 ch = igetch(100);
171 sysadm 1.16
172 sysadm 1.85 if (ch != KEY_NULL && ch != KEY_TIMEOUT)
173     {
174     BBS_last_access_tm = time(NULL);
175     }
176 sysadm 1.82
177 sysadm 1.75 if (bbs_menu.choose_step == 0 && time(NULL) - t_last_action >= 10)
178 sysadm 1.31 {
179 sysadm 1.70 t_last_action = time(NULL);
180 sysadm 1.52
181 sysadm 1.31 show_active_board();
182     show_bottom("");
183 sysadm 1.75 display_menu_cursor(&bbs_menu, 1);
184 sysadm 1.50 iflush();
185 sysadm 1.31 }
186    
187 sysadm 1.71 if (user_online_update("MENU") < 0)
188     {
189     log_error("user_online_update(MENU) error\n");
190     }
191    
192 sysadm 1.31 switch (ch)
193     {
194 sysadm 1.42 case KEY_NULL: // broken pipe
195 sysadm 1.82 log_error("KEY_NULL\n");
196 sysadm 1.42 return 0;
197 sysadm 1.31 case KEY_TIMEOUT:
198 sysadm 1.70 if (time(NULL) - BBS_last_access_tm >= MAX_DELAY_TIME)
199 sysadm 1.31 {
200 sysadm 1.82 log_error("User input timeout\n");
201 sysadm 1.38 return 0;
202 sysadm 1.31 }
203 sysadm 1.39 continue;
204 sysadm 1.44 case CR:
205 sysadm 1.31 default:
206 sysadm 1.75 switch (menu_control(&bbs_menu, ch))
207 sysadm 1.31 {
208     case EXITBBS:
209 sysadm 1.74 case EXITMENU:
210 sysadm 1.31 return 0;
211     case REDRAW:
212 sysadm 1.70 t_last_action = time(NULL);
213 sysadm 1.39 clearscr();
214 sysadm 1.58 show_top("", BBS_name, "");
215 sysadm 1.39 show_active_board();
216     show_bottom("");
217 sysadm 1.75 display_menu(&bbs_menu);
218 sysadm 1.31 break;
219     case NOREDRAW:
220     case UNKNOWN_CMD:
221     default:
222     break;
223     }
224 sysadm 1.50 iflush();
225 sysadm 1.31 }
226 sysadm 1.16 }
227 sysadm 1.7
228 sysadm 1.31 return 0;
229 sysadm 1.15 }
230 sysadm 1.32
231 sysadm 1.87 int bbs_charset_select()
232     {
233     char msg[LINE_BUFFER_LEN];
234     int ch;
235    
236 sysadm 1.89 snprintf(msg, sizeof(msg),
237     "\rChoose character set in 5 seconds [UTF-8, GBK]: [U/g]");
238    
239 sysadm 1.88 while (!SYS_server_exit)
240     {
241     ch = press_any_key_ex(msg, 5);
242     switch (ch)
243 sysadm 1.87 {
244 sysadm 1.89 case KEY_NULL:
245     return -1;
246 sysadm 1.88 case KEY_TIMEOUT:
247     case CR:
248     case 'u':
249     case 'U':
250     return 0;
251     case 'g':
252     case 'G':
253 sysadm 1.92 if (io_conv_init("GBK") < 0)
254 sysadm 1.88 {
255 sysadm 1.92 log_error("io_conv_init(%s) error\n", "GBK");
256 sysadm 1.88 return -1;
257     }
258     return 0;
259     default:
260     continue;
261 sysadm 1.87 }
262     }
263    
264     return 0;
265     }
266    
267 sysadm 1.32 int bbs_main()
268     {
269 sysadm 1.74 struct sigaction act = {0};
270 sysadm 1.86 char msg[LINE_BUFFER_LEN];
271 sysadm 1.74
272     // Set signal handler
273     act.sa_handler = SIG_IGN;
274     if (sigaction(SIGHUP, &act, NULL) == -1)
275     {
276     log_error("set signal action of SIGHUP error: %d\n", errno);
277     goto cleanup;
278     }
279     act.sa_handler = SIG_DFL;
280     if (sigaction(SIGCHLD, &act, NULL) == -1)
281     {
282     log_error("set signal action of SIGCHLD error: %d\n", errno);
283     goto cleanup;
284     }
285    
286 sysadm 1.55 // Set data pools in shared memory readonly
287     if (set_trie_dict_shm_readonly() < 0)
288     {
289     goto cleanup;
290     }
291     if (set_article_block_shm_readonly() < 0)
292     {
293     goto cleanup;
294     }
295     if (set_section_list_shm_readonly() < 0)
296     {
297     goto cleanup;
298     }
299 sysadm 1.93 if (set_user_list_pool_shm_readonly() < 0)
300     {
301     goto cleanup;
302     }
303 sysadm 1.55
304     // Load menu in shared memory
305 sysadm 1.75 if (set_menu_shm_readonly(&bbs_menu) < 0)
306 sysadm 1.55 {
307     goto cleanup;
308     }
309 sysadm 1.32
310 sysadm 1.87 // Set default charset
311     if (io_conv_init(BBS_DEFAULT_CHARSET) < 0)
312     {
313     log_error("io_conv_init(%s) error\n", BBS_DEFAULT_CHARSET);
314     goto cleanup;
315     }
316    
317 sysadm 1.32 set_input_echo(0);
318    
319 sysadm 1.87 // Set user charset
320     bbs_charset_select();
321    
322 sysadm 1.43 // System info
323     if (bbs_info() < 0)
324     {
325 sysadm 1.55 goto cleanup;
326 sysadm 1.43 }
327    
328 sysadm 1.32 // Welcome
329 sysadm 1.65 if (bbs_welcome() < 0)
330 sysadm 1.43 {
331 sysadm 1.55 goto cleanup;
332 sysadm 1.43 }
333 sysadm 1.32
334 sysadm 1.43 // User login
335 sysadm 1.64 if (SSH_v2)
336     {
337 sysadm 1.86 snprintf(msg, sizeof(msg), "\033[1m%s 欢迎使用ssh方式访问 \033[1;33m按任意键继续...\033[m", BBS_username);
338 sysadm 1.87 press_any_key_ex(msg, 60);
339 sysadm 1.64 }
340 sysadm 1.65 else if (bbs_login() < 0)
341 sysadm 1.43 {
342 sysadm 1.55 goto cleanup;
343 sysadm 1.43 }
344 sysadm 1.83 log_common("User [%s] login\n", BBS_username);
345 sysadm 1.64
346 sysadm 1.67 // Load article view log
347     if (article_view_log_load(BBS_priv.uid, &BBS_article_view_log, 0) < 0)
348     {
349     log_error("article_view_log_load() error\n");
350     goto cleanup;
351     }
352 sysadm 1.65
353 sysadm 1.78 // Load article favorite
354     if (article_favor_load(BBS_priv.uid, &BBS_article_favor, 0) < 0)
355     {
356     log_error("article_favor_load() error\n");
357     goto cleanup;
358     }
359    
360 sysadm 1.69 // Init editor memory pool
361     if (editor_memory_pool_init() < 0)
362     {
363     log_error("editor_memory_pool_init() error\n");
364     goto cleanup;
365     }
366    
367 sysadm 1.32 clearscr();
368    
369     // BBS Top 10
370 sysadm 1.68 display_file(VAR_BBS_TOP, 1);
371 sysadm 1.32
372 sysadm 1.55 // Main
373     bbs_center();
374    
375     // Logout
376 sysadm 1.65 bbs_logout();
377 sysadm 1.55
378 sysadm 1.67 // Save incremental article view log
379     if (article_view_log_save_inc(&BBS_article_view_log) < 0)
380     {
381     log_error("article_view_log_save_inc() error\n");
382     }
383    
384 sysadm 1.78 // Save incremental article favorite
385     if (article_favor_save_inc(&BBS_article_favor) < 0)
386     {
387     log_error("article_favor_save_inc() error\n");
388     }
389    
390 sysadm 1.55 cleanup:
391 sysadm 1.87 // Cleanup iconv
392     io_conv_cleanup();
393    
394 sysadm 1.69 // Cleanup editor memory pool
395     editor_memory_pool_cleanup();
396 sysadm 1.71
397 sysadm 1.67 // Unload article view log
398     article_view_log_unload(&BBS_article_view_log);
399    
400 sysadm 1.78 // Unload article favor
401     article_favor_unload(&BBS_article_favor);
402    
403 sysadm 1.55 // Detach menu in shared memory
404 sysadm 1.75 detach_menu_shm(&bbs_menu);
405 sysadm 1.79 detach_menu_shm(&top10_menu);
406 sysadm 1.46
407 sysadm 1.55 // Detach data pools shm
408 sysadm 1.93 detach_user_list_pool_shm();
409 sysadm 1.55 detach_section_list_shm();
410     detach_article_block_shm();
411     detach_trie_dict_shm();
412 sysadm 1.35
413 sysadm 1.32 return 0;
414     }

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