/[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.87 - (hide annotations)
Sat Oct 18 05:02:15 2025 UTC (4 months, 4 weeks ago) by sysadm
Branch: MAIN
Changes since 1.86: +43 -1 lines
Content type: text/x-csrc
Add user preferred charset selection and conversion

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.73 #include "user_priv.h"
33 sysadm 1.74 #include <errno.h>
34     #include <signal.h>
35 sysadm 1.73 #include <stdlib.h>
36     #include <string.h>
37     #include <time.h>
38 sysadm 1.32 #include <unistd.h>
39 sysadm 1.16
40 sysadm 1.31 int bbs_info()
41 sysadm 1.5 {
42 sysadm 1.77 prints("欢迎光临 \033[1;33m%s \033[32m[%s] \033[37m( %s )\033[m\r\n",
43 sysadm 1.72 BBS_name, BBS_server, APP_INFO);
44 sysadm 1.16
45 sysadm 1.43 return iflush();
46     }
47    
48 sysadm 1.65 int bbs_welcome(void)
49 sysadm 1.43 {
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 sysadm 1.65 MYSQL *db;
58 sysadm 1.43 MYSQL_RES *rs;
59     MYSQL_ROW row;
60    
61 sysadm 1.65 db = db_open();
62     if (db == NULL)
63     {
64     return -1;
65     }
66    
67 sysadm 1.43 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 sysadm 1.54 log_error("Query user_online error: %s\n", mysql_error(db));
75 sysadm 1.65 mysql_close(db);
76 sysadm 1.43 return -2;
77     }
78     if ((rs = mysql_store_result(db)) == NULL)
79     {
80     log_error("Get user_online data failed\n");
81 sysadm 1.65 mysql_close(db);
82 sysadm 1.43 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 sysadm 1.54 log_error("Query user_online error: %s\n", mysql_error(db));
98 sysadm 1.65 mysql_close(db);
99 sysadm 1.43 return -2;
100     }
101     if ((rs = mysql_store_result(db)) == NULL)
102     {
103     log_error("Get user_online data failed\n");
104 sysadm 1.65 mysql_close(db);
105 sysadm 1.43 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 sysadm 1.54 log_error("Query user_list error: %s\n", mysql_error(db));
117 sysadm 1.65 mysql_close(db);
118 sysadm 1.43 return -2;
119     }
120     if ((rs = mysql_store_result(db)) == NULL)
121     {
122     log_error("Get user_list data failed\n");
123 sysadm 1.65 mysql_close(db);
124 sysadm 1.43 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 sysadm 1.54 log_error("Query user_login_log error: %s\n", mysql_error(db));
136 sysadm 1.65 mysql_close(db);
137 sysadm 1.43 return -2;
138     }
139     if ((rs = mysql_store_result(db)) == NULL)
140     {
141     log_error("Get user_login_log data failed\n");
142 sysadm 1.65 mysql_close(db);
143 sysadm 1.43 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 sysadm 1.65 mysql_close(db);
152    
153 sysadm 1.43 // Count current user before login
154     u_online++;
155     u_anonymous++;
156    
157     // Display logo
158 sysadm 1.68 display_file(DATA_WELCOME, 2);
159 sysadm 1.43
160     // Display welcome message
161 sysadm 1.76 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 sysadm 1.43 BBS_name, u_online, BBS_max_client, u_anonymous, u_total,
167 sysadm 1.66 BBS_max_user, BBS_start_dt, u_login_count);
168 sysadm 1.5
169 sysadm 1.64 iflush();
170    
171 sysadm 1.31 return 0;
172 sysadm 1.5 }
173    
174 sysadm 1.65 int bbs_logout(void)
175 sysadm 1.5 {
176 sysadm 1.65 MYSQL *db;
177    
178     db = db_open();
179     if (db == NULL)
180     {
181     return -1;
182     }
183    
184 sysadm 1.80 if (user_online_exp(db) < 0)
185     {
186     return -2;
187     }
188    
189 sysadm 1.43 if (user_online_del(db) < 0)
190     {
191 sysadm 1.80 return -3;
192 sysadm 1.43 }
193    
194 sysadm 1.65 mysql_close(db);
195    
196 sysadm 1.68 display_file(DATA_GOODBYE, 1);
197 sysadm 1.43
198 sysadm 1.81 log_common("User [%s] logout\n", BBS_username);
199 sysadm 1.7
200 sysadm 1.31 return 0;
201 sysadm 1.7 }
202    
203 sysadm 1.31 int bbs_center()
204 sysadm 1.7 {
205 sysadm 1.39 int ch;
206 sysadm 1.31 time_t t_last_action;
207 sysadm 1.16
208 sysadm 1.70 BBS_last_access_tm = t_last_action = time(NULL);
209 sysadm 1.16
210 sysadm 1.31 clearscr();
211 sysadm 1.12
212 sysadm 1.58 show_top("", BBS_name, "");
213 sysadm 1.31 show_active_board();
214     show_bottom("");
215 sysadm 1.75 display_menu(&bbs_menu);
216 sysadm 1.50 iflush();
217 sysadm 1.16
218 sysadm 1.37 while (!SYS_server_exit)
219 sysadm 1.25 {
220 sysadm 1.44 ch = igetch(100);
221 sysadm 1.16
222 sysadm 1.85 if (ch != KEY_NULL && ch != KEY_TIMEOUT)
223     {
224     BBS_last_access_tm = time(NULL);
225     }
226 sysadm 1.82
227 sysadm 1.75 if (bbs_menu.choose_step == 0 && time(NULL) - t_last_action >= 10)
228 sysadm 1.31 {
229 sysadm 1.70 t_last_action = time(NULL);
230 sysadm 1.52
231 sysadm 1.31 show_active_board();
232     show_bottom("");
233 sysadm 1.75 display_menu_cursor(&bbs_menu, 1);
234 sysadm 1.50 iflush();
235 sysadm 1.31 }
236    
237 sysadm 1.71 if (user_online_update("MENU") < 0)
238     {
239     log_error("user_online_update(MENU) error\n");
240     }
241    
242 sysadm 1.31 switch (ch)
243     {
244 sysadm 1.42 case KEY_NULL: // broken pipe
245 sysadm 1.82 log_error("KEY_NULL\n");
246 sysadm 1.42 return 0;
247 sysadm 1.31 case KEY_TIMEOUT:
248 sysadm 1.70 if (time(NULL) - BBS_last_access_tm >= MAX_DELAY_TIME)
249 sysadm 1.31 {
250 sysadm 1.82 log_error("User input timeout\n");
251 sysadm 1.38 return 0;
252 sysadm 1.31 }
253 sysadm 1.39 continue;
254 sysadm 1.44 case CR:
255 sysadm 1.31 default:
256 sysadm 1.75 switch (menu_control(&bbs_menu, ch))
257 sysadm 1.31 {
258     case EXITBBS:
259 sysadm 1.74 case EXITMENU:
260 sysadm 1.31 return 0;
261     case REDRAW:
262 sysadm 1.70 t_last_action = time(NULL);
263 sysadm 1.39 clearscr();
264 sysadm 1.58 show_top("", BBS_name, "");
265 sysadm 1.39 show_active_board();
266     show_bottom("");
267 sysadm 1.75 display_menu(&bbs_menu);
268 sysadm 1.31 break;
269     case NOREDRAW:
270     case UNKNOWN_CMD:
271     default:
272     break;
273     }
274 sysadm 1.50 iflush();
275 sysadm 1.31 }
276 sysadm 1.16 }
277 sysadm 1.7
278 sysadm 1.31 return 0;
279 sysadm 1.15 }
280 sysadm 1.32
281 sysadm 1.87 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     break;
300     default:
301     log_error("Debug: %d\n", ch);
302     }
303    
304     prints("\r\n");
305     iflush();
306    
307     return 0;
308     }
309    
310 sysadm 1.32 int bbs_main()
311     {
312 sysadm 1.74 struct sigaction act = {0};
313 sysadm 1.86 char msg[LINE_BUFFER_LEN];
314 sysadm 1.74
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 sysadm 1.55 // 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 sysadm 1.75 if (set_menu_shm_readonly(&bbs_menu) < 0)
345 sysadm 1.55 {
346     goto cleanup;
347     }
348 sysadm 1.32
349 sysadm 1.87 // 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 sysadm 1.32 set_input_echo(0);
357    
358 sysadm 1.87 // Set user charset
359     bbs_charset_select();
360    
361 sysadm 1.43 // System info
362     if (bbs_info() < 0)
363     {
364 sysadm 1.55 goto cleanup;
365 sysadm 1.43 }
366    
367 sysadm 1.32 // Welcome
368 sysadm 1.65 if (bbs_welcome() < 0)
369 sysadm 1.43 {
370 sysadm 1.55 goto cleanup;
371 sysadm 1.43 }
372 sysadm 1.32
373 sysadm 1.43 // User login
374 sysadm 1.64 if (SSH_v2)
375     {
376 sysadm 1.86 snprintf(msg, sizeof(msg), "\033[1m%s 欢迎使用ssh方式访问 \033[1;33m按任意键继续...\033[m", BBS_username);
377 sysadm 1.87 press_any_key_ex(msg, 60);
378 sysadm 1.64 }
379 sysadm 1.65 else if (bbs_login() < 0)
380 sysadm 1.43 {
381 sysadm 1.55 goto cleanup;
382 sysadm 1.43 }
383 sysadm 1.83 log_common("User [%s] login\n", BBS_username);
384 sysadm 1.64
385 sysadm 1.67 // 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 sysadm 1.65
392 sysadm 1.78 // 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 sysadm 1.69 // 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    
406 sysadm 1.32 clearscr();
407    
408     // BBS Top 10
409 sysadm 1.68 display_file(VAR_BBS_TOP, 1);
410 sysadm 1.32
411 sysadm 1.55 // Main
412     bbs_center();
413    
414     // Logout
415 sysadm 1.65 bbs_logout();
416 sysadm 1.55
417 sysadm 1.67 // Save incremental article view log
418     if (article_view_log_save_inc(&BBS_article_view_log) < 0)
419     {
420     log_error("article_view_log_save_inc() error\n");
421     }
422    
423 sysadm 1.78 // 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 sysadm 1.55 cleanup:
430 sysadm 1.87 // Cleanup iconv
431     io_conv_cleanup();
432    
433 sysadm 1.69 // Cleanup editor memory pool
434     editor_memory_pool_cleanup();
435 sysadm 1.71
436 sysadm 1.67 // Unload article view log
437     article_view_log_unload(&BBS_article_view_log);
438    
439 sysadm 1.78 // Unload article favor
440     article_favor_unload(&BBS_article_favor);
441    
442 sysadm 1.55 // Detach menu in shared memory
443 sysadm 1.75 detach_menu_shm(&bbs_menu);
444 sysadm 1.79 detach_menu_shm(&top10_menu);
445 sysadm 1.46
446 sysadm 1.55 // Detach data pools shm
447     detach_section_list_shm();
448     detach_article_block_shm();
449     detach_trie_dict_shm();
450 sysadm 1.35
451 sysadm 1.32 return 0;
452     }

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