lbbs/src/bbs_main.c

/***************************************************************************
                                                  bbs_main.c  -  description
                                                         -------------------
        Copyright            : (C) 2004-2025 by Leaflet
        Email                : leaflet@leafok.com
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 3 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/

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