| 15 |
***************************************************************************/ |
***************************************************************************/ |
| 16 |
|
|
| 17 |
#include "bbs.h" |
#include "bbs.h" |
|
#include "init.h" |
|
| 18 |
#include "common.h" |
#include "common.h" |
| 19 |
#include "net_server.h" |
#include "file_loader.h" |
| 20 |
#include "log.h" |
#include "init.h" |
| 21 |
#include "io.h" |
#include "io.h" |
| 22 |
|
#include "log.h" |
| 23 |
#include "menu.h" |
#include "menu.h" |
| 24 |
#include "file_loader.h" |
#include "net_server.h" |
| 25 |
#include "section_list_loader.h" |
#include "section_list_loader.h" |
| 26 |
|
#include "user_list.h" |
| 27 |
#include <errno.h> |
#include <errno.h> |
| 28 |
#include <stdlib.h> |
#include <libgen.h> |
| 29 |
#include <signal.h> |
#include <signal.h> |
| 30 |
|
#include <stdio.h> |
| 31 |
|
#include <stdlib.h> |
| 32 |
#include <string.h> |
#include <string.h> |
| 33 |
#include <unistd.h> |
#include <unistd.h> |
|
#include <libgen.h> |
|
|
#include <sys/types.h> |
|
| 34 |
#include <sys/stat.h> |
#include <sys/stat.h> |
| 35 |
|
#include <sys/types.h> |
| 36 |
|
|
| 37 |
void app_help(void) |
static void app_help(void) |
| 38 |
{ |
{ |
| 39 |
prints("Usage: bbsd [-fhv] [...]\n\n" |
fprintf(stderr, "Usage: bbsd [-fhv] [...]\n\n" |
| 40 |
"-f\t--foreground\t\tForce program run in foreground\n" |
"-f\t--foreground\t\tForce program run in foreground\n" |
| 41 |
"-h\t--help\t\t\tDisplay this help message\n" |
"-h\t--help\t\t\tDisplay this help message\n" |
| 42 |
"-v\t--version\t\tDisplay version information\n" |
"-v\t--version\t\tDisplay version information\n" |
| 43 |
"\t--display-log\t\tDisplay standard log information\n" |
"\t--display-log\t\tDisplay standard log information\n" |
| 44 |
"\t--display-error-log\tDisplay error log information\n" |
"\t--display-error-log\tDisplay error log information\n" |
| 45 |
"\n If meet any bug, please report to <leaflet@leafok.com>\n\n"); |
"\n If meet any bug, please report to <leaflet@leafok.com>\n\n"); |
| 46 |
} |
} |
| 47 |
|
|
| 48 |
void arg_error(void) |
static void arg_error(void) |
| 49 |
{ |
{ |
| 50 |
prints("Invalid arguments\n"); |
fprintf(stderr, "Invalid arguments\n"); |
| 51 |
app_help(); |
app_help(); |
| 52 |
} |
} |
| 53 |
|
|
| 129 |
strncpy(file_path_temp, argv[0], sizeof(file_path_temp) - 1); |
strncpy(file_path_temp, argv[0], sizeof(file_path_temp) - 1); |
| 130 |
file_path_temp[sizeof(file_path_temp) - 1] = '\0'; |
file_path_temp[sizeof(file_path_temp) - 1] = '\0'; |
| 131 |
|
|
| 132 |
chdir(dirname(file_path_temp)); |
if (chdir(dirname(file_path_temp)) < 0) |
| 133 |
chdir(".."); |
{ |
| 134 |
|
fprintf(stderr, "chdir(%s) error: %d\n", dirname(file_path_temp), errno); |
| 135 |
|
return -1; |
| 136 |
|
} |
| 137 |
|
if (chdir("..") < 0) |
| 138 |
|
{ |
| 139 |
|
fprintf(stderr, "chdir(..) error: %d\n", errno); |
| 140 |
|
return -1; |
| 141 |
|
} |
| 142 |
|
|
| 143 |
// Initialize log |
// Initialize log |
| 144 |
if (log_begin(LOG_FILE_INFO, LOG_FILE_ERROR) < 0) |
if (log_begin(LOG_FILE_INFO, LOG_FILE_ERROR) < 0) |
| 172 |
} |
} |
| 173 |
|
|
| 174 |
// Initialize data pools |
// Initialize data pools |
|
if ((fp = fopen(VAR_TRIE_DICT_SHM, "w")) == NULL) |
|
|
{ |
|
|
log_error("fopen(%s) error\n", VAR_TRIE_DICT_SHM); |
|
|
goto cleanup; |
|
|
} |
|
|
fclose(fp); |
|
| 175 |
if ((fp = fopen(VAR_ARTICLE_BLOCK_SHM, "w")) == NULL) |
if ((fp = fopen(VAR_ARTICLE_BLOCK_SHM, "w")) == NULL) |
| 176 |
{ |
{ |
| 177 |
log_error("fopen(%s) error\n", VAR_ARTICLE_BLOCK_SHM); |
log_error("fopen(%s) error\n", VAR_ARTICLE_BLOCK_SHM); |
| 184 |
goto cleanup; |
goto cleanup; |
| 185 |
} |
} |
| 186 |
fclose(fp); |
fclose(fp); |
| 187 |
|
if ((fp = fopen(VAR_TRIE_DICT_SHM, "w")) == NULL) |
| 188 |
|
{ |
| 189 |
|
log_error("fopen(%s) error\n", VAR_TRIE_DICT_SHM); |
| 190 |
|
goto cleanup; |
| 191 |
|
} |
| 192 |
|
fclose(fp); |
| 193 |
|
if ((fp = fopen(VAR_USER_LIST_SHM, "w")) == NULL) |
| 194 |
|
{ |
| 195 |
|
log_error("fopen(%s) error\n", VAR_USER_LIST_SHM); |
| 196 |
|
goto cleanup; |
| 197 |
|
} |
| 198 |
|
fclose(fp); |
| 199 |
|
|
| 200 |
if (trie_dict_init(VAR_TRIE_DICT_SHM, TRIE_NODE_PER_POOL) < 0) |
if (trie_dict_init(VAR_TRIE_DICT_SHM, TRIE_NODE_PER_POOL) < 0) |
| 201 |
{ |
{ |
| 202 |
printf("trie_dict_init failed\n"); |
printf("trie_dict_init(%s, %d) error\n", VAR_TRIE_DICT_SHM, TRIE_NODE_PER_POOL); |
| 203 |
goto cleanup; |
goto cleanup; |
| 204 |
} |
} |
| 205 |
if (article_block_init(VAR_ARTICLE_BLOCK_SHM, BBS_article_limit_per_section * BBS_max_section / ARTICLE_PER_BLOCK) < 0) |
if (article_block_init(VAR_ARTICLE_BLOCK_SHM, BBS_article_limit_per_section * BBS_max_section / ARTICLE_PER_BLOCK) < 0) |
| 220 |
} |
} |
| 221 |
|
|
| 222 |
// Load menus |
// Load menus |
| 223 |
p_bbs_menu = calloc(1, sizeof(MENU_SET)); |
if (load_menu(&bbs_menu, CONF_MENU) < 0) |
|
if (p_bbs_menu == NULL) |
|
| 224 |
{ |
{ |
| 225 |
log_error("OOM: calloc(MENU_SET)\n"); |
log_error("load_menu(%s) error\n", CONF_MENU); |
| 226 |
goto cleanup; |
goto cleanup; |
| 227 |
} |
} |
| 228 |
if (load_menu(p_bbs_menu, CONF_MENU) < 0) |
if (load_menu(&top10_menu, CONF_TOP10_MENU) < 0) |
| 229 |
{ |
{ |
| 230 |
|
log_error("load_menu(%s) error\n", CONF_TOP10_MENU); |
| 231 |
goto cleanup; |
goto cleanup; |
| 232 |
} |
} |
| 233 |
|
top10_menu.allow_exit = 1; |
| 234 |
|
|
| 235 |
// Load data files |
// Load data files |
| 236 |
if (file_loader_init() < 0) |
if (file_loader_init() < 0) |
| 242 |
{ |
{ |
| 243 |
if (load_file(data_files_load_startup[i]) < 0) |
if (load_file(data_files_load_startup[i]) < 0) |
| 244 |
{ |
{ |
| 245 |
log_error("load_file_mmap(%s) error\n", data_files_load_startup[i]); |
log_error("load_file(%s) error\n", data_files_load_startup[i]); |
| 246 |
} |
} |
| 247 |
} |
} |
| 248 |
|
|
| 249 |
// Load section config |
// Load user_list and online_user_list |
| 250 |
if (load_section_config_from_db() < 0) |
if (user_list_pool_init(VAR_USER_LIST_SHM) < 0) |
| 251 |
|
{ |
| 252 |
|
log_error("user_list_pool_init(%s) error\n", VAR_USER_LIST_SHM); |
| 253 |
|
goto cleanup; |
| 254 |
|
} |
| 255 |
|
if (user_list_pool_reload(0) < 0) |
| 256 |
|
{ |
| 257 |
|
log_error("user_list_pool_reload(all_user) error\n"); |
| 258 |
|
goto cleanup; |
| 259 |
|
} |
| 260 |
|
if (user_list_pool_reload(1) < 0) |
| 261 |
|
{ |
| 262 |
|
log_error("user_list_pool_reload(online_user) error\n"); |
| 263 |
|
goto cleanup; |
| 264 |
|
} |
| 265 |
|
|
| 266 |
|
// Load section config and gen_ex |
| 267 |
|
if (load_section_config_from_db(1) < 0) |
| 268 |
{ |
{ |
| 269 |
log_error("load_section_config_from_db() error\n"); |
log_error("load_section_config_from_db(0) error\n"); |
| 270 |
goto cleanup; |
goto cleanup; |
| 271 |
} |
} |
| 272 |
|
|
| 287 |
|
|
| 288 |
log_common("Initially load %d articles, last_aid = %d\n", article_block_article_count(), article_block_last_aid()); |
log_common("Initially load %d articles, last_aid = %d\n", article_block_article_count(), article_block_last_aid()); |
| 289 |
|
|
| 290 |
|
if ((ret = user_stat_update()) < 0) |
| 291 |
|
{ |
| 292 |
|
log_error("user_stat_update() error\n"); |
| 293 |
|
goto cleanup; |
| 294 |
|
} |
| 295 |
|
|
| 296 |
// Set signal handler |
// Set signal handler |
| 297 |
act.sa_handler = sig_hup_handler; |
act.sa_handler = sig_hup_handler; |
| 298 |
if (sigaction(SIGHUP, &act, NULL) == -1) |
if (sigaction(SIGHUP, &act, NULL) == -1) |
| 328 |
file_loader_cleanup(); |
file_loader_cleanup(); |
| 329 |
|
|
| 330 |
// Cleanup menu |
// Cleanup menu |
| 331 |
unload_menu(p_bbs_menu); |
unload_menu(&bbs_menu); |
| 332 |
free(p_bbs_menu); |
unload_menu(&top10_menu); |
|
p_bbs_menu = NULL; |
|
| 333 |
|
|
| 334 |
// Cleanup data pools |
// Cleanup data pools |
| 335 |
section_list_cleanup(); |
section_list_cleanup(); |
| 336 |
article_block_cleanup(); |
article_block_cleanup(); |
| 337 |
trie_dict_cleanup(); |
trie_dict_cleanup(); |
| 338 |
|
user_list_pool_cleanup(); |
| 339 |
|
|
|
if (unlink(VAR_TRIE_DICT_SHM) < 0) |
|
|
{ |
|
|
log_error("unlink(%s) error\n", VAR_TRIE_DICT_SHM); |
|
|
} |
|
| 340 |
if (unlink(VAR_ARTICLE_BLOCK_SHM) < 0) |
if (unlink(VAR_ARTICLE_BLOCK_SHM) < 0) |
| 341 |
{ |
{ |
| 342 |
log_error("unlink(%s) error\n", VAR_ARTICLE_BLOCK_SHM); |
log_error("unlink(%s) error\n", VAR_ARTICLE_BLOCK_SHM); |
| 345 |
{ |
{ |
| 346 |
log_error("unlink(%s) error\n", VAR_SECTION_LIST_SHM); |
log_error("unlink(%s) error\n", VAR_SECTION_LIST_SHM); |
| 347 |
} |
} |
| 348 |
|
if (unlink(VAR_TRIE_DICT_SHM) < 0) |
| 349 |
|
{ |
| 350 |
|
log_error("unlink(%s) error\n", VAR_TRIE_DICT_SHM); |
| 351 |
|
} |
| 352 |
|
if (unlink(VAR_USER_LIST_SHM) < 0) |
| 353 |
|
{ |
| 354 |
|
log_error("unlink(%s) error\n", VAR_SECTION_LIST_SHM); |
| 355 |
|
} |
| 356 |
|
|
| 357 |
log_common("Main process exit normally\n"); |
log_common("Main process exit normally\n"); |
| 358 |
|
|
| 359 |
|
log_end(); |
| 360 |
|
|
| 361 |
return 0; |
return 0; |
| 362 |
} |
} |