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 "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();

        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.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	sysadm	1.101	#ifdef _DEBUG
142			log_error("Debug: BBS_last_access_tm is updated\n");
143			#endif
144	sysadm	1.85	}
145	sysadm	1.82
146	sysadm	1.75	if (bbs_menu.choose_step == 0 && time(NULL) - t_last_action >= 10)
147	sysadm	1.31	{
148	sysadm	1.70	t_last_action = time(NULL);
149	sysadm	1.52
150	sysadm	1.31	show_active_board();
151			show_bottom("");
152	sysadm	1.75	display_menu_cursor(&bbs_menu, 1);
153	sysadm	1.50	iflush();
154	sysadm	1.31	}
155
156	sysadm	1.71	if (user_online_update("MENU") < 0)
157			{
158			log_error("user_online_update(MENU) error\n");
159			}
160
161	sysadm	1.31	switch (ch)
162			{
163	sysadm	1.42	case KEY_NULL: // broken pipe
164	sysadm	1.82	log_error("KEY_NULL\n");
165	sysadm	1.42	return 0;
166	sysadm	1.31	case KEY_TIMEOUT:
167	sysadm	1.99	if (time(NULL) - BBS_last_access_tm >= BBS_max_user_idle_time)
168	sysadm	1.31	{
169	sysadm	1.82	log_error("User input timeout\n");
170	sysadm	1.38	return 0;
171	sysadm	1.31	}
172	sysadm	1.39	continue;
173	sysadm	1.44	case CR:
174	sysadm	1.31	default:
175	sysadm	1.75	switch (menu_control(&bbs_menu, ch))
176	sysadm	1.31	{
177			case EXITBBS:
178	sysadm	1.74	case EXITMENU:
179	sysadm	1.31	return 0;
180			case REDRAW:
181	sysadm	1.70	t_last_action = time(NULL);
182	sysadm	1.39	clearscr();
183	sysadm	1.58	show_top("", BBS_name, "");
184	sysadm	1.39	show_active_board();
185			show_bottom("");
186	sysadm	1.75	display_menu(&bbs_menu);
187	sysadm	1.31	break;
188			case NOREDRAW:
189			case UNKNOWN_CMD:
190			default:
191			break;
192			}
193	sysadm	1.50	iflush();
194	sysadm	1.31	}
195	sysadm	1.16	}
196	sysadm	1.7
197	sysadm	1.31	return 0;
198	sysadm	1.15	}
199	sysadm	1.32
200	sysadm	1.87	int bbs_charset_select()
201			{
202			char msg[LINE_BUFFER_LEN];
203			int ch;
204
205	sysadm	1.89	snprintf(msg, sizeof(msg),
206			"\rChoose character set in 5 seconds [UTF-8, GBK]: [U/g]");
207
208	sysadm	1.88	while (!SYS_server_exit)
209			{
210			ch = press_any_key_ex(msg, 5);
211			switch (ch)
212	sysadm	1.87	{
213	sysadm	1.89	case KEY_NULL:
214			return -1;
215	sysadm	1.88	case KEY_TIMEOUT:
216			case CR:
217			case 'u':
218			case 'U':
219			return 0;
220			case 'g':
221			case 'G':
222	sysadm	1.92	if (io_conv_init("GBK") < 0)
223	sysadm	1.88	{
224	sysadm	1.92	log_error("io_conv_init(%s) error\n", "GBK");
225	sysadm	1.88	return -1;
226			}
227			return 0;
228			default:
229			continue;
230	sysadm	1.87	}
231			}
232
233			return 0;
234			}
235
236	sysadm	1.32	int bbs_main()
237			{
238	sysadm	1.74	struct sigaction act = {0};
239	sysadm	1.86	char msg[LINE_BUFFER_LEN];
240	sysadm	1.74
241			// Set signal handler
242			act.sa_handler = SIG_IGN;
243			if (sigaction(SIGHUP, &act, NULL) == -1)
244			{
245			log_error("set signal action of SIGHUP error: %d\n", errno);
246			goto cleanup;
247			}
248			act.sa_handler = SIG_DFL;
249			if (sigaction(SIGCHLD, &act, NULL) == -1)
250			{
251			log_error("set signal action of SIGCHLD error: %d\n", errno);
252			goto cleanup;
253			}
254
255	sysadm	1.55	// Set data pools in shared memory readonly
256			if (set_trie_dict_shm_readonly() < 0)
257			{
258			goto cleanup;
259			}
260			if (set_article_block_shm_readonly() < 0)
261			{
262			goto cleanup;
263			}
264			if (set_section_list_shm_readonly() < 0)
265			{
266			goto cleanup;
267			}
268	sysadm	1.93	if (set_user_list_pool_shm_readonly() < 0)
269			{
270			goto cleanup;
271			}
272	sysadm	1.55
273			// Load menu in shared memory
274	sysadm	1.75	if (set_menu_shm_readonly(&bbs_menu) < 0)
275	sysadm	1.55	{
276			goto cleanup;
277			}
278	sysadm	1.32
279	sysadm	1.87	// Set default charset
280	sysadm	1.100	if (io_conv_init(BBS_default_charset) < 0)
281	sysadm	1.87	{
282	sysadm	1.100	log_error("io_conv_init(%s) error\n", BBS_default_charset);
283	sysadm	1.87	goto cleanup;
284			}
285
286	sysadm	1.32	set_input_echo(0);
287
288	sysadm	1.87	// Set user charset
289			bbs_charset_select();
290
291	sysadm	1.43	// System info
292			if (bbs_info() < 0)
293			{
294	sysadm	1.55	goto cleanup;
295	sysadm	1.43	}
296
297	sysadm	1.32	// Welcome
298	sysadm	1.65	if (bbs_welcome() < 0)
299	sysadm	1.43	{
300	sysadm	1.55	goto cleanup;
301	sysadm	1.43	}
302	sysadm	1.32
303	sysadm	1.43	// User login
304	sysadm	1.64	if (SSH_v2)
305			{
306	sysadm	1.86	snprintf(msg, sizeof(msg), "\033[1m%s 欢迎使用ssh方式访问 \033[1;33m按任意键继续...\033[m", BBS_username);
307	sysadm	1.87	press_any_key_ex(msg, 60);
308	sysadm	1.64	}
309	sysadm	1.65	else if (bbs_login() < 0)
310	sysadm	1.43	{
311	sysadm	1.55	goto cleanup;
312	sysadm	1.43	}
313	sysadm	1.83	log_common("User [%s] login\n", BBS_username);
314	sysadm	1.64
315	sysadm	1.96	// Load section aid locations
316			if (section_aid_locations_load(BBS_priv.uid) < 0)
317			{
318			log_error("article_view_log_load() error\n");
319			goto cleanup;
320			}
321	sysadm	1.97
322	sysadm	1.67	// Load article view log
323			if (article_view_log_load(BBS_priv.uid, &BBS_article_view_log, 0) < 0)
324			{
325			log_error("article_view_log_load() error\n");
326			goto cleanup;
327			}
328	sysadm	1.65
329	sysadm	1.78	// Load article favorite
330			if (article_favor_load(BBS_priv.uid, &BBS_article_favor, 0) < 0)
331			{
332			log_error("article_favor_load() error\n");
333			goto cleanup;
334			}
335
336	sysadm	1.69	// Init editor memory pool
337			if (editor_memory_pool_init() < 0)
338			{
339			log_error("editor_memory_pool_init() error\n");
340			goto cleanup;
341			}
342
343	sysadm	1.32	clearscr();
344
345			// BBS Top 10
346	sysadm	1.68	display_file(VAR_BBS_TOP, 1);
347	sysadm	1.32
348	sysadm	1.55	// Main
349			bbs_center();
350
351			// Logout
352	sysadm	1.65	bbs_logout();
353	sysadm	1.55
354	sysadm	1.96	// Save section aid locations
355			if (section_aid_locations_save(BBS_priv.uid) < 0)
356			{
357			log_error("article_view_log_save() error\n");
358			}
359
360	sysadm	1.67	// Save incremental article view log
361			if (article_view_log_save_inc(&BBS_article_view_log) < 0)
362			{
363			log_error("article_view_log_save_inc() error\n");
364			}
365
366	sysadm	1.78	// Save incremental article favorite
367			if (article_favor_save_inc(&BBS_article_favor) < 0)
368			{
369			log_error("article_favor_save_inc() error\n");
370			}
371
372	sysadm	1.55	cleanup:
373	sysadm	1.87	// Cleanup iconv
374			io_conv_cleanup();
375
376	sysadm	1.69	// Cleanup editor memory pool
377			editor_memory_pool_cleanup();
378	sysadm	1.71
379	sysadm	1.67	// Unload article view log
380			article_view_log_unload(&BBS_article_view_log);
381
382	sysadm	1.78	// Unload article favor
383			article_favor_unload(&BBS_article_favor);
384
385	sysadm	1.55	// Detach menu in shared memory
386	sysadm	1.75	detach_menu_shm(&bbs_menu);
387	sysadm	1.79	detach_menu_shm(&top10_menu);
388	sysadm	1.46
389	sysadm	1.55	// Detach data pools shm
390	sysadm	1.93	detach_user_list_pool_shm();
391	sysadm	1.55	detach_section_list_shm();
392			detach_article_block_shm();
393			detach_trie_dict_shm();
394	sysadm	1.35
395	sysadm	1.32	return 0;
396			}
webmaster@leafok.com	ViewVC Help
Powered by ViewVC 1.3.0-beta1