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