/[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.93 - (hide annotations)
Tue Oct 21 06:24:51 2025 UTC (4 months, 3 weeks ago) by sysadm
Branch: MAIN
Changes since 1.92: +7 -1 lines
Content type: text/x-csrc
Add user_list with data loader

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     u_int32_t u_online = 0;
54     u_int32_t u_anonymous = 0;
55     u_int32_t u_total = 0;
56     u_int32_t u_login_count = 0;
57    
58 sysadm 1.65 MYSQL *db;
59 sysadm 1.43 MYSQL_RES *rs;
60     MYSQL_ROW row;
61    
62 sysadm 1.65 db = db_open();
63     if (db == NULL)
64     {
65     return -1;
66     }
67    
68 sysadm 1.43 snprintf(sql, sizeof(sql),
69     "SELECT COUNT(*) AS cc FROM "
70     "(SELECT DISTINCT SID FROM user_online "
71     "WHERE last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND)) AS t1",
72     BBS_user_off_line);
73     if (mysql_query(db, sql) != 0)
74     {
75 sysadm 1.54 log_error("Query user_online error: %s\n", mysql_error(db));
76 sysadm 1.65 mysql_close(db);
77 sysadm 1.43 return -2;
78     }
79     if ((rs = mysql_store_result(db)) == NULL)
80     {
81     log_error("Get user_online data failed\n");
82 sysadm 1.65 mysql_close(db);
83 sysadm 1.43 return -2;
84     }
85     if ((row = mysql_fetch_row(rs)))
86     {
87     u_online = (u_int32_t)atoi(row[0]);
88     }
89     mysql_free_result(rs);
90    
91     snprintf(sql, sizeof(sql),
92     "SELECT COUNT(*) AS cc FROM "
93     "(SELECT DISTINCT SID FROM user_online "
94     "WHERE UID = 0 AND last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND)) AS t1",
95     BBS_user_off_line);
96     if (mysql_query(db, sql) != 0)
97     {
98 sysadm 1.54 log_error("Query user_online error: %s\n", mysql_error(db));
99 sysadm 1.65 mysql_close(db);
100 sysadm 1.43 return -2;
101     }
102     if ((rs = mysql_store_result(db)) == NULL)
103     {
104     log_error("Get user_online data failed\n");
105 sysadm 1.65 mysql_close(db);
106 sysadm 1.43 return -2;
107     }
108     if ((row = mysql_fetch_row(rs)))
109     {
110     u_anonymous = (u_int32_t)atoi(row[0]);
111     }
112     mysql_free_result(rs);
113    
114     snprintf(sql, sizeof(sql), "SELECT COUNT(UID) AS cc FROM user_list WHERE enable");
115     if (mysql_query(db, sql) != 0)
116     {
117 sysadm 1.54 log_error("Query user_list error: %s\n", mysql_error(db));
118 sysadm 1.65 mysql_close(db);
119 sysadm 1.43 return -2;
120     }
121     if ((rs = mysql_store_result(db)) == NULL)
122     {
123     log_error("Get user_list data failed\n");
124 sysadm 1.65 mysql_close(db);
125 sysadm 1.43 return -2;
126     }
127     if ((row = mysql_fetch_row(rs)))
128     {
129     u_total = (u_int32_t)atoi(row[0]);
130     }
131     mysql_free_result(rs);
132    
133     snprintf(sql, sizeof(sql), "SELECT ID FROM user_login_log ORDER BY ID LIMIT 1");
134     if (mysql_query(db, sql) != 0)
135     {
136 sysadm 1.54 log_error("Query user_login_log error: %s\n", mysql_error(db));
137 sysadm 1.65 mysql_close(db);
138 sysadm 1.43 return -2;
139     }
140     if ((rs = mysql_store_result(db)) == NULL)
141     {
142     log_error("Get user_login_log data failed\n");
143 sysadm 1.65 mysql_close(db);
144 sysadm 1.43 return -2;
145     }
146     if ((row = mysql_fetch_row(rs)))
147     {
148     u_login_count = (u_int32_t)atoi(row[0]);
149     }
150     mysql_free_result(rs);
151    
152 sysadm 1.65 mysql_close(db);
153    
154 sysadm 1.43 // Count current user before login
155     u_online++;
156     u_anonymous++;
157    
158     // Display logo
159 sysadm 1.68 display_file(DATA_WELCOME, 2);
160 sysadm 1.43
161     // Display welcome message
162 sysadm 1.76 prints("\r\033[1;35m欢迎光临\033[33m 【 %s 】 \033[35mBBS\r\n"
163     "\033[32m目前上站人数 [\033[36m%d/%d\033[32m] "
164     "匿名游客[\033[36m%d\033[32m] "
165     "注册用户数[\033[36m%d/%d\033[32m]\r\n"
166     "从 [\033[36m%s\033[32m] 起,累计访问人次:[\033[36m%d\033[32m]\033[m\r\n",
167 sysadm 1.43 BBS_name, u_online, BBS_max_client, u_anonymous, u_total,
168 sysadm 1.93 BBS_max_user_count, BBS_start_dt, u_login_count);
169 sysadm 1.5
170 sysadm 1.64 iflush();
171    
172 sysadm 1.31 return 0;
173 sysadm 1.5 }
174    
175 sysadm 1.65 int bbs_logout(void)
176 sysadm 1.5 {
177 sysadm 1.65 MYSQL *db;
178    
179     db = db_open();
180     if (db == NULL)
181     {
182     return -1;
183     }
184    
185 sysadm 1.80 if (user_online_exp(db) < 0)
186     {
187     return -2;
188     }
189    
190 sysadm 1.43 if (user_online_del(db) < 0)
191     {
192 sysadm 1.80 return -3;
193 sysadm 1.43 }
194    
195 sysadm 1.65 mysql_close(db);
196    
197 sysadm 1.68 display_file(DATA_GOODBYE, 1);
198 sysadm 1.43
199 sysadm 1.91 log_common("User [%s] logout, idle for %ld seconds since last input\n", BBS_username, time(NULL) - BBS_last_access_tm);
200 sysadm 1.7
201 sysadm 1.31 return 0;
202 sysadm 1.7 }
203    
204 sysadm 1.31 int bbs_center()
205 sysadm 1.7 {
206 sysadm 1.39 int ch;
207 sysadm 1.31 time_t t_last_action;
208 sysadm 1.16
209 sysadm 1.70 BBS_last_access_tm = t_last_action = time(NULL);
210 sysadm 1.16
211 sysadm 1.31 clearscr();
212 sysadm 1.12
213 sysadm 1.58 show_top("", BBS_name, "");
214 sysadm 1.31 show_active_board();
215     show_bottom("");
216 sysadm 1.75 display_menu(&bbs_menu);
217 sysadm 1.50 iflush();
218 sysadm 1.16
219 sysadm 1.37 while (!SYS_server_exit)
220 sysadm 1.25 {
221 sysadm 1.44 ch = igetch(100);
222 sysadm 1.16
223 sysadm 1.85 if (ch != KEY_NULL && ch != KEY_TIMEOUT)
224     {
225     BBS_last_access_tm = time(NULL);
226     }
227 sysadm 1.82
228 sysadm 1.75 if (bbs_menu.choose_step == 0 && time(NULL) - t_last_action >= 10)
229 sysadm 1.31 {
230 sysadm 1.70 t_last_action = time(NULL);
231 sysadm 1.52
232 sysadm 1.31 show_active_board();
233     show_bottom("");
234 sysadm 1.75 display_menu_cursor(&bbs_menu, 1);
235 sysadm 1.50 iflush();
236 sysadm 1.31 }
237    
238 sysadm 1.71 if (user_online_update("MENU") < 0)
239     {
240     log_error("user_online_update(MENU) error\n");
241     }
242    
243 sysadm 1.31 switch (ch)
244     {
245 sysadm 1.42 case KEY_NULL: // broken pipe
246 sysadm 1.82 log_error("KEY_NULL\n");
247 sysadm 1.42 return 0;
248 sysadm 1.31 case KEY_TIMEOUT:
249 sysadm 1.70 if (time(NULL) - BBS_last_access_tm >= MAX_DELAY_TIME)
250 sysadm 1.31 {
251 sysadm 1.82 log_error("User input timeout\n");
252 sysadm 1.38 return 0;
253 sysadm 1.31 }
254 sysadm 1.39 continue;
255 sysadm 1.44 case CR:
256 sysadm 1.31 default:
257 sysadm 1.75 switch (menu_control(&bbs_menu, ch))
258 sysadm 1.31 {
259     case EXITBBS:
260 sysadm 1.74 case EXITMENU:
261 sysadm 1.31 return 0;
262     case REDRAW:
263 sysadm 1.70 t_last_action = time(NULL);
264 sysadm 1.39 clearscr();
265 sysadm 1.58 show_top("", BBS_name, "");
266 sysadm 1.39 show_active_board();
267     show_bottom("");
268 sysadm 1.75 display_menu(&bbs_menu);
269 sysadm 1.31 break;
270     case NOREDRAW:
271     case UNKNOWN_CMD:
272     default:
273     break;
274     }
275 sysadm 1.50 iflush();
276 sysadm 1.31 }
277 sysadm 1.16 }
278 sysadm 1.7
279 sysadm 1.31 return 0;
280 sysadm 1.15 }
281 sysadm 1.32
282 sysadm 1.87 int bbs_charset_select()
283     {
284     char msg[LINE_BUFFER_LEN];
285     int ch;
286    
287 sysadm 1.89 snprintf(msg, sizeof(msg),
288     "\rChoose character set in 5 seconds [UTF-8, GBK]: [U/g]");
289    
290 sysadm 1.88 while (!SYS_server_exit)
291     {
292     ch = press_any_key_ex(msg, 5);
293     switch (ch)
294 sysadm 1.87 {
295 sysadm 1.89 case KEY_NULL:
296     return -1;
297 sysadm 1.88 case KEY_TIMEOUT:
298     case CR:
299     case 'u':
300     case 'U':
301     return 0;
302     case 'g':
303     case 'G':
304 sysadm 1.92 if (io_conv_init("GBK") < 0)
305 sysadm 1.88 {
306 sysadm 1.92 log_error("io_conv_init(%s) error\n", "GBK");
307 sysadm 1.88 return -1;
308     }
309     return 0;
310     default:
311     continue;
312 sysadm 1.87 }
313     }
314    
315     return 0;
316     }
317    
318 sysadm 1.32 int bbs_main()
319     {
320 sysadm 1.74 struct sigaction act = {0};
321 sysadm 1.86 char msg[LINE_BUFFER_LEN];
322 sysadm 1.74
323     // Set signal handler
324     act.sa_handler = SIG_IGN;
325     if (sigaction(SIGHUP, &act, NULL) == -1)
326     {
327     log_error("set signal action of SIGHUP error: %d\n", errno);
328     goto cleanup;
329     }
330     act.sa_handler = SIG_DFL;
331     if (sigaction(SIGCHLD, &act, NULL) == -1)
332     {
333     log_error("set signal action of SIGCHLD error: %d\n", errno);
334     goto cleanup;
335     }
336    
337 sysadm 1.55 // Set data pools in shared memory readonly
338     if (set_trie_dict_shm_readonly() < 0)
339     {
340     goto cleanup;
341     }
342     if (set_article_block_shm_readonly() < 0)
343     {
344     goto cleanup;
345     }
346     if (set_section_list_shm_readonly() < 0)
347     {
348     goto cleanup;
349     }
350 sysadm 1.93 if (set_user_list_pool_shm_readonly() < 0)
351     {
352     goto cleanup;
353     }
354 sysadm 1.55
355     // Load menu in shared memory
356 sysadm 1.75 if (set_menu_shm_readonly(&bbs_menu) < 0)
357 sysadm 1.55 {
358     goto cleanup;
359     }
360 sysadm 1.32
361 sysadm 1.87 // Set default charset
362     if (io_conv_init(BBS_DEFAULT_CHARSET) < 0)
363     {
364     log_error("io_conv_init(%s) error\n", BBS_DEFAULT_CHARSET);
365     goto cleanup;
366     }
367    
368 sysadm 1.32 set_input_echo(0);
369    
370 sysadm 1.87 // Set user charset
371     bbs_charset_select();
372    
373 sysadm 1.43 // System info
374     if (bbs_info() < 0)
375     {
376 sysadm 1.55 goto cleanup;
377 sysadm 1.43 }
378    
379 sysadm 1.32 // Welcome
380 sysadm 1.65 if (bbs_welcome() < 0)
381 sysadm 1.43 {
382 sysadm 1.55 goto cleanup;
383 sysadm 1.43 }
384 sysadm 1.32
385 sysadm 1.43 // User login
386 sysadm 1.64 if (SSH_v2)
387     {
388 sysadm 1.86 snprintf(msg, sizeof(msg), "\033[1m%s 欢迎使用ssh方式访问 \033[1;33m按任意键继续...\033[m", BBS_username);
389 sysadm 1.87 press_any_key_ex(msg, 60);
390 sysadm 1.64 }
391 sysadm 1.65 else if (bbs_login() < 0)
392 sysadm 1.43 {
393 sysadm 1.55 goto cleanup;
394 sysadm 1.43 }
395 sysadm 1.83 log_common("User [%s] login\n", BBS_username);
396 sysadm 1.64
397 sysadm 1.67 // Load article view log
398     if (article_view_log_load(BBS_priv.uid, &BBS_article_view_log, 0) < 0)
399     {
400     log_error("article_view_log_load() error\n");
401     goto cleanup;
402     }
403 sysadm 1.65
404 sysadm 1.78 // Load article favorite
405     if (article_favor_load(BBS_priv.uid, &BBS_article_favor, 0) < 0)
406     {
407     log_error("article_favor_load() error\n");
408     goto cleanup;
409     }
410    
411 sysadm 1.69 // Init editor memory pool
412     if (editor_memory_pool_init() < 0)
413     {
414     log_error("editor_memory_pool_init() error\n");
415     goto cleanup;
416     }
417    
418 sysadm 1.32 clearscr();
419    
420     // BBS Top 10
421 sysadm 1.68 display_file(VAR_BBS_TOP, 1);
422 sysadm 1.32
423 sysadm 1.55 // Main
424     bbs_center();
425    
426     // Logout
427 sysadm 1.65 bbs_logout();
428 sysadm 1.55
429 sysadm 1.67 // Save incremental article view log
430     if (article_view_log_save_inc(&BBS_article_view_log) < 0)
431     {
432     log_error("article_view_log_save_inc() error\n");
433     }
434    
435 sysadm 1.78 // Save incremental article favorite
436     if (article_favor_save_inc(&BBS_article_favor) < 0)
437     {
438     log_error("article_favor_save_inc() error\n");
439     }
440    
441 sysadm 1.55 cleanup:
442 sysadm 1.87 // Cleanup iconv
443     io_conv_cleanup();
444    
445 sysadm 1.69 // Cleanup editor memory pool
446     editor_memory_pool_cleanup();
447 sysadm 1.71
448 sysadm 1.67 // Unload article view log
449     article_view_log_unload(&BBS_article_view_log);
450    
451 sysadm 1.78 // Unload article favor
452     article_favor_unload(&BBS_article_favor);
453    
454 sysadm 1.55 // Detach menu in shared memory
455 sysadm 1.75 detach_menu_shm(&bbs_menu);
456 sysadm 1.79 detach_menu_shm(&top10_menu);
457 sysadm 1.46
458 sysadm 1.55 // Detach data pools shm
459 sysadm 1.93 detach_user_list_pool_shm();
460 sysadm 1.55 detach_section_list_shm();
461     detach_article_block_shm();
462     detach_trie_dict_shm();
463 sysadm 1.35
464 sysadm 1.32 return 0;
465     }

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