--- lbbs/src/bbs_main.c 2025/05/27 00:54:01 1.55 +++ lbbs/src/bbs_main.c 2025/11/20 09:02:46 1.111 @@ -1,183 +1,133 @@ -/*************************************************************************** - 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. * - * * - ***************************************************************************/ +/* SPDX-License-Identifier: GPL-3.0-or-later */ +/* + * bbs_main + * - entry and major procedures of user interactive access + * + * Copyright (C) 2004-2025 Leaflet + */ + +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif -#include "bbs_main.h" +#include "article_favor.h" +#include "article_view_log.h" #include "bbs.h" -#include "login.h" -#include "user_priv.h" +#include "bbs_cmd.h" +#include "bbs_main.h" +#include "bwf.h" #include "common.h" #include "database.h" -#include "log.h" +#include "editor.h" #include "io.h" -#include "screen.h" +#include "log.h" +#include "login.h" #include "menu.h" -#include "bbs_cmd.h" +#include "screen.h" #include "section_list.h" +#include "section_list_display.h" +#include "str_process.h" #include "trie_dict.h" -#include -#include -#include +#include "user_list.h" +#include "user_priv.h" +#include +#include +#include #include +#include +#include +#include + +static void child_proc_sig_usr1_handler(int i) +{ + // Restart log + if (log_restart() < 0) + { + log_error("Restart logging failed\n"); + } +} int bbs_info() { - prints("欢迎光临 \033[1;33m%s \033[32m[%s] \033[37m( %s )\r\n", - BBS_name, BBS_server, app_version); + prints("\r\n娆㈣繋鍏変复 \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(MYSQL *db) +int bbs_welcome(void) { - char sql[SQL_BUFFER_LEN]; - - u_int32_t u_online = 0; - u_int32_t u_anonymous = 0; - u_int32_t u_total = 0; - u_int32_t max_u_online = 0; - u_int32_t u_login_count = 0; - - MYSQL_RES *rs; - MYSQL_ROW row; - - snprintf(sql, sizeof(sql), - "SELECT COUNT(*) AS cc FROM " - "(SELECT DISTINCT SID FROM user_online " - "WHERE last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND)) AS t1", - BBS_user_off_line); - if (mysql_query(db, sql) != 0) - { - log_error("Query user_online error: %s\n", mysql_error(db)); - return -2; - } - if ((rs = mysql_store_result(db)) == NULL) - { - log_error("Get user_online data failed\n"); - return -2; - } - if ((row = mysql_fetch_row(rs))) - { - u_online = (u_int32_t)atoi(row[0]); - } - mysql_free_result(rs); - - snprintf(sql, sizeof(sql), - "SELECT COUNT(*) AS cc FROM " - "(SELECT DISTINCT SID FROM user_online " - "WHERE UID = 0 AND last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND)) AS t1", - BBS_user_off_line); - if (mysql_query(db, sql) != 0) - { - log_error("Query user_online error: %s\n", mysql_error(db)); - return -2; - } - if ((rs = mysql_store_result(db)) == NULL) - { - log_error("Get user_online data failed\n"); - return -2; - } - if ((row = mysql_fetch_row(rs))) - { - u_anonymous = (u_int32_t)atoi(row[0]); - } - mysql_free_result(rs); + int u_online = 0; + int u_anonymous = 0; + int u_total = 0; + int u_login_count = 0; - snprintf(sql, sizeof(sql), "SELECT COUNT(UID) AS cc FROM user_list WHERE enable"); - if (mysql_query(db, sql) != 0) + if (get_user_online_list_count(&u_online, &u_anonymous) < 0) { - log_error("Query user_list error: %s\n", mysql_error(db)); - return -2; - } - if ((rs = mysql_store_result(db)) == NULL) - { - log_error("Get user_list data failed\n"); - return -2; + log_error("get_user_online_list_count() error\n"); + u_online = 0; } - if ((row = mysql_fetch_row(rs))) + u_online += u_anonymous; + u_online++; // current user + if (BBS_priv.uid == 0) { - u_total = (u_int32_t)atoi(row[0]); + u_anonymous++; } - mysql_free_result(rs); - snprintf(sql, sizeof(sql), "SELECT ID FROM user_login_log ORDER BY ID LIMIT 1"); - if (mysql_query(db, sql) != 0) + if (get_user_list_count(&u_total) < 0) { - log_error("Query user_login_log error: %s\n", mysql_error(db)); - return -2; + log_error("get_user_list_count() error\n"); + u_total = 0; } - if ((rs = mysql_store_result(db)) == NULL) - { - log_error("Get user_login_log data failed\n"); - return -2; - } - if ((row = mysql_fetch_row(rs))) - { - u_login_count = (u_int32_t)atoi(row[0]); - } - mysql_free_result(rs); - // Log max user_online - FILE *fin, *fout; - if ((fin = fopen(VAR_MAX_USER_ONLINE, "r")) != NULL) + if (get_user_login_count(&u_login_count) < 0) { - fscanf(fin, "%d", &max_u_online); - fclose(fin); - } - if (u_online > max_u_online) - { - max_u_online = u_online; - if ((fout = fopen(VAR_MAX_USER_ONLINE, "w")) == NULL) - { - log_error("Open max_user_online.dat failed\n"); - return -3; - } - fprintf(fout, "%d\n", max_u_online); - fclose(fout); + log_error("get_user_login_count() error\n"); + u_login_count = 0; } - // Count current user before login - u_online++; - u_anonymous++; - // Display logo - display_file_ex(DATA_WELCOME, 1, 0); + 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[36m%d\033[32m]\r\n", + 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, BBS_start_dt, max_u_online, u_login_count); + BBS_max_user_count, BBS_start_dt, u_login_count); + + iflush(); return 0; } -int bbs_logout(MYSQL *db) +int bbs_logout(void) { - if (user_online_del(db) < 0) + MYSQL *db; + + db = db_open(); + if (db == NULL) { return -1; } - display_file_ex(DATA_GOODBYE, 1, 1); + if (user_online_exp(db) < 0) + { + return -2; + } + + if (user_online_del(db) < 0) + { + return -3; + } + + mysql_close(db); - log_std("User logout\n"); + 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; } @@ -187,53 +137,69 @@ int bbs_center() int ch; time_t t_last_action; - BBS_last_access_tm = t_last_action = time(0); + BBS_last_access_tm = t_last_action = time(NULL); clearscr(); - show_top(""); + show_top("", BBS_name, ""); show_active_board(); show_bottom(""); - display_menu(p_bbs_menu); + display_menu(&bbs_menu); iflush(); while (!SYS_server_exit) { ch = igetch(100); - if (p_bbs_menu->choose_step == 0 && time(0) - t_last_action >= 10) + 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(0); + 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(0) - BBS_last_access_tm >= MAX_DELAY_TIME) + if (time(NULL) - BBS_last_access_tm >= BBS_max_user_idle_time) { + log_error("User input timeout\n"); return 0; } continue; case CR: - igetch_reset(); default: - switch (menu_control(p_bbs_menu, ch)) + switch (menu_control(&bbs_menu, ch)) { case EXITBBS: + case EXITMENU: return 0; case REDRAW: - t_last_action = time(0); + t_last_action = time(NULL); clearscr(); - show_top(""); + show_top("", BBS_name, ""); show_active_board(); show_bottom(""); - display_menu(p_bbs_menu); + display_menu(&bbs_menu); break; case NOREDRAW: case UNKNOWN_CMD: @@ -242,8 +208,67 @@ int bbs_center() } iflush(); } + } - BBS_last_access_tm = time(0); + return 0; +} + +int bbs_charset_select() +{ + char msg[LINE_BUFFER_LEN]; + int ch; + + snprintf(msg, sizeof(msg), + "\rChoose character set in 5 seconds, (U)UTF-8, (G)GBK [U]: "); + + while (!SYS_server_exit) + { + ch = press_any_key_ex(msg, 5); + switch (toupper(ch)) + { + case KEY_NULL: + return -1; + case KEY_TIMEOUT: + case CR: + case 'U': + break; + case 'G': + if (io_conv_init("GBK") < 0) + { + log_error("io_conv_init(%s) error\n", "GBK"); + return -1; + } + break; + default: + continue; + } + + break; + } + + snprintf(msg, sizeof(msg), + "\r璇峰湪5绉掑唴閫夋嫨瀹藉瓧绗︽樉绀鸿鍒, (V)鍙樺, (F)瀹氬? [V]: "); + + while (!SYS_server_exit) + { + ch = press_any_key_ex(msg, 5); + switch (toupper(ch)) + { + case KEY_NULL: + return -1; + case KEY_TIMEOUT: + case CR: + case 'V': + UTF8_fixed_width = 0; + break; + case 'F': + UTF8_fixed_width = 1; + break; + default: + continue; + } + + break; } return 0; @@ -251,7 +276,28 @@ int bbs_center() int bbs_main() { - MYSQL *db = NULL; + 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; + } + act.sa_handler = child_proc_sig_usr1_handler; + if (sigaction(SIGUSR1, &act, NULL) == -1) + { + log_error("set signal action of SIGUSR1 error: %d\n", errno); + goto cleanup; + } // Set data pools in shared memory readonly if (set_trie_dict_shm_readonly() < 0) @@ -266,67 +312,140 @@ int bbs_main() { goto cleanup; } +#ifdef HAVE_SYSTEM_V + if (set_user_list_pool_shm_readonly() < 0) + { + goto cleanup; + } +#endif // Load menu in shared memory - if (set_menu_shm_readonly(p_bbs_menu) < 0) + 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); - // System info - if (bbs_info() < 0) + // Set user charset + if (bbs_charset_select() < 0) { goto cleanup; } - db = db_open(); - if (db == NULL) + // System info + if (bbs_info() < 0) { - prints("无法连接数据库\n"); goto cleanup; } // Welcome - if (bbs_welcome(db) < 0) + if (bbs_welcome() < 0) { - mysql_close(db); goto cleanup; } // User login - if (bbs_login(db) < 0) + 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) { - mysql_close(db); + log_error("editor_memory_pool_init() error\n"); goto cleanup; } + clearscr(); // BBS Top 10 - display_file_ex(VAR_BBS_TOP, 1, 1); + display_file(VAR_BBS_TOP, 1); // Main bbs_center(); // Logout - bbs_logout(db); + bbs_logout(); -cleanup: - if (db != NULL) + // 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) { - mysql_close(db); + 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(p_bbs_menu); - free(p_bbs_menu); - p_bbs_menu = NULL; + 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; }