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  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 "bwf.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);
#ifdef _DEBUG
                        log_error("Debug: BBS_last_access_tm is updated\n");
#endif
                }

                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 >= BBS_max_user_idle_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();

        // Cleanup BWF
        bwf_unload();

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