lbbs/src/bbs_main.c

/* SPDX-License-Identifier: GPL-3.0-or-later */
/*
 * bbs_main
 *   - entry and major procedures of user interactive access
 *
 * Copyright (C) 2004-2025 by Leaflet <leaflet@leafok.com>
 */

#include "article_favor.h"
#include "article_view_log.h"
#include "bbs.h"
#include "bbs_cmd.h"
#include "bbs_main.h"
#include "common.h"
#include "database.h"
#include "editor.h"
#include "io.h"
#include "log.h"
#include "login.h"
#include "menu.h"
#include "screen.h"
#include "section_list.h"
#include "section_list_display.h"
#include "trie_dict.h"
#include "user_list.h"
#include "user_priv.h"
#include <errno.h>
#include <signal.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>

int bbs_info()
{
        prints("欢迎光临 \033[1;33m%s \033[32m[%s]  \033[37m( %s )\033[m\r\n",
                   BBS_name, BBS_server, APP_INFO);

        return iflush();
}

int bbs_welcome(void)
{
        int u_online = 0;
        int u_anonymous = 0;
        int u_total = 0;
        int u_login_count = 0;

        if (get_user_online_list_count(&u_online, &u_anonymous) < 0)
        {
                log_error("get_user_online_list_count() error\n");
                u_online = 0;
        }
        u_online += u_anonymous;
        u_online++; // current user
        if (BBS_priv.uid == 0)
        {
                u_anonymous++;
        }

        if (get_user_list_count(&u_total) < 0)
        {
                log_error("get_user_list_count() error\n");
                u_total = 0;
        }

        if (get_user_login_count(&u_login_count) < 0)
        {
                log_error("get_user_login_count() error\n");
                u_login_count = 0;
        }

        // Display logo
        display_file(DATA_WELCOME, 2);

        // Display welcome message
        prints("\r\033[1;35m欢迎光临\033[33m 【 %s 】 \033[35mBBS\r\n"
                   "\033[32m目前上站人数 [\033[36m%d/%d\033[32m] "
                   "匿名游客[\033[36m%d\033[32m] "
                   "注册用户数[\033[36m%d/%d\033[32m]\r\n"
                   "从 [\033[36m%s\033[32m] 起，累计访问人次：[\033[36m%d\033[32m]\033[m\r\n",
                   BBS_name, u_online, BBS_max_client, u_anonymous, u_total,
                   BBS_max_user_count, BBS_start_dt, u_login_count);

        iflush();

        return 0;
}

int bbs_logout(void)
{
        MYSQL *db;

        db = db_open();
        if (db == NULL)
        {
                return -1;
        }

        if (user_online_exp(db) < 0)
        {
                return -2;
        }

        if (user_online_del(db) < 0)
        {
                return -3;
        }

        mysql_close(db);

        display_file(DATA_GOODBYE, 1);

        log_common("User [%s] logout, idle for %ld seconds since last input\n", BBS_username, time(NULL) - BBS_last_access_tm);

        return 0;
}

int bbs_center()
{
        int ch;
        time_t t_last_action;

        BBS_last_access_tm = t_last_action = time(NULL);

        clearscr();

        show_top("", BBS_name, "");
        show_active_board();
        show_bottom("");
        display_menu(&bbs_menu);
        iflush();

        while (!SYS_server_exit)
        {
                ch = igetch(100);

                if (ch != KEY_NULL && ch != KEY_TIMEOUT)
                {
                        BBS_last_access_tm = time(NULL);
                }

                if (bbs_menu.choose_step == 0 && time(NULL) - t_last_action >= 10)
                {
                        t_last_action = time(NULL);

                        show_active_board();
                        show_bottom("");
                        display_menu_cursor(&bbs_menu, 1);
                        iflush();
                }

                if (user_online_update("MENU") < 0)
                {
                        log_error("user_online_update(MENU) error\n");
                }

                switch (ch)
                {
                case KEY_NULL: // broken pipe
                        log_error("KEY_NULL\n");
                        return 0;
                case KEY_TIMEOUT:
                        if (time(NULL) - BBS_last_access_tm >= MAX_DELAY_TIME)
                        {
                                log_error("User input timeout\n");
                                return 0;
                        }
                        continue;
                case CR:
                default:
                        switch (menu_control(&bbs_menu, ch))
                        {
                        case EXITBBS:
                        case EXITMENU:
                                return 0;
                        case REDRAW:
                                t_last_action = time(NULL);
                                clearscr();
                                show_top("", BBS_name, "");
                                show_active_board();
                                show_bottom("");
                                display_menu(&bbs_menu);
                                break;
                        case NOREDRAW:
                        case UNKNOWN_CMD:
                        default:
                                break;
                        }
                        iflush();
                }
        }

        return 0;
}

int bbs_charset_select()
{
        char msg[LINE_BUFFER_LEN];
        int ch;

        snprintf(msg, sizeof(msg),
                         "\rChoose character set in 5 seconds [UTF-8, GBK]: [U/g]");

        while (!SYS_server_exit)
        {
                ch = press_any_key_ex(msg, 5);
                switch (ch)
                {
                case KEY_NULL:
                        return -1;
                case KEY_TIMEOUT:
                case CR:
                case 'u':
                case 'U':
                        return 0;
                case 'g':
                case 'G':
                        if (io_conv_init("GBK") < 0)
                        {
                                log_error("io_conv_init(%s) error\n", "GBK");
                                return -1;
                        }
                        return 0;
                default:
                        continue;
                }
        }

        return 0;
}

int bbs_main()
{
        struct sigaction act = {0};
        char msg[LINE_BUFFER_LEN];

        // Set signal handler
        act.sa_handler = SIG_IGN;
        if (sigaction(SIGHUP, &act, NULL) == -1)
        {
                log_error("set signal action of SIGHUP error: %d\n", errno);
                goto cleanup;
        }
        act.sa_handler = SIG_DFL;
        if (sigaction(SIGCHLD, &act, NULL) == -1)
        {
                log_error("set signal action of SIGCHLD error: %d\n", errno);
                goto cleanup;
        }

        // Set data pools in shared memory readonly
        if (set_trie_dict_shm_readonly() < 0)
        {
                goto cleanup;
        }
        if (set_article_block_shm_readonly() < 0)
        {
                goto cleanup;
        }
        if (set_section_list_shm_readonly() < 0)
        {
                goto cleanup;
        }
        if (set_user_list_pool_shm_readonly() < 0)
        {
                goto cleanup;
        }

        // Load menu in shared memory
        if (set_menu_shm_readonly(&bbs_menu) < 0)
        {
                goto cleanup;
        }

        // Set default charset
        if (io_conv_init(BBS_DEFAULT_CHARSET) < 0)
        {
                log_error("io_conv_init(%s) error\n", BBS_DEFAULT_CHARSET);
                goto cleanup;
        }

        set_input_echo(0);

        // Set user charset
        bbs_charset_select();

        // System info
        if (bbs_info() < 0)
        {
                goto cleanup;
        }

        // Welcome
        if (bbs_welcome() < 0)
        {
                goto cleanup;
        }

        // User login
        if (SSH_v2)
        {
                snprintf(msg, sizeof(msg), "\033[1m%s 欢迎使用ssh方式访问 \033[1;33m按任意键继续...\033[m", BBS_username);
                press_any_key_ex(msg, 60);
        }
        else if (bbs_login() < 0)
        {
                goto cleanup;
        }
        log_common("User [%s] login\n", BBS_username);

        // Load section aid locations
        if (section_aid_locations_load(BBS_priv.uid) < 0)
        {
                log_error("article_view_log_load() error\n");
                goto cleanup;
        }

        // Load article view log
        if (article_view_log_load(BBS_priv.uid, &BBS_article_view_log, 0) < 0)
        {
                log_error("article_view_log_load() error\n");
                goto cleanup;
        }

        // Load article favorite
        if (article_favor_load(BBS_priv.uid, &BBS_article_favor, 0) < 0)
        {
                log_error("article_favor_load() error\n");
                goto cleanup;
        }

        // Init editor memory pool
        if (editor_memory_pool_init() < 0)
        {
                log_error("editor_memory_pool_init() error\n");
                goto cleanup;
        }

        clearscr();

        // BBS Top 10
        display_file(VAR_BBS_TOP, 1);

        // Main
        bbs_center();

        // Logout
        bbs_logout();

        // Save section aid locations
        if (section_aid_locations_save(BBS_priv.uid) < 0)
        {
                log_error("article_view_log_save() error\n");
        }

        // Save incremental article view log
        if (article_view_log_save_inc(&BBS_article_view_log) < 0)
        {
                log_error("article_view_log_save_inc() error\n");
        }

        // Save incremental article favorite
        if (article_favor_save_inc(&BBS_article_favor) < 0)
        {
                log_error("article_favor_save_inc() error\n");
        }

cleanup:
        // Cleanup iconv
        io_conv_cleanup();

        // Cleanup editor memory pool
        editor_memory_pool_cleanup();

        // Unload article view log
        article_view_log_unload(&BBS_article_view_log);

        // Unload article favor
        article_favor_unload(&BBS_article_favor);

        // Detach menu in shared memory
        detach_menu_shm(&bbs_menu);
        detach_menu_shm(&top10_menu);

        // Detach data pools shm
        detach_user_list_pool_shm();
        detach_section_list_shm();
        detach_article_block_shm();
        detach_trie_dict_shm();

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