| 23 |
#include "menu.h" |
#include "menu.h" |
| 24 |
#include "net_server.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 <libgen.h> |
#include <libgen.h> |
| 29 |
#include <signal.h> |
#include <signal.h> |
| 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) |
| 171 |
goto cleanup; |
goto cleanup; |
| 172 |
} |
} |
| 173 |
|
|
| 174 |
// Initialize data pools |
// Check section aid location dir |
| 175 |
if ((fp = fopen(VAR_TRIE_DICT_SHM, "w")) == NULL) |
ret = mkdir(VAR_SECTION_AID_LOC_DIR, 0750); |
| 176 |
|
if (ret == -1 && errno != EEXIST) |
| 177 |
{ |
{ |
| 178 |
log_error("fopen(%s) error\n", VAR_TRIE_DICT_SHM); |
log_error("mkdir(%s) error (%d)\n", VAR_SECTION_AID_LOC_DIR, errno); |
| 179 |
goto cleanup; |
goto cleanup; |
| 180 |
} |
} |
| 181 |
fclose(fp); |
|
| 182 |
|
// Initialize data pools |
| 183 |
if ((fp = fopen(VAR_ARTICLE_BLOCK_SHM, "w")) == NULL) |
if ((fp = fopen(VAR_ARTICLE_BLOCK_SHM, "w")) == NULL) |
| 184 |
{ |
{ |
| 185 |
log_error("fopen(%s) error\n", VAR_ARTICLE_BLOCK_SHM); |
log_error("fopen(%s) error\n", VAR_ARTICLE_BLOCK_SHM); |
| 192 |
goto cleanup; |
goto cleanup; |
| 193 |
} |
} |
| 194 |
fclose(fp); |
fclose(fp); |
| 195 |
|
if ((fp = fopen(VAR_TRIE_DICT_SHM, "w")) == NULL) |
| 196 |
|
{ |
| 197 |
|
log_error("fopen(%s) error\n", VAR_TRIE_DICT_SHM); |
| 198 |
|
goto cleanup; |
| 199 |
|
} |
| 200 |
|
fclose(fp); |
| 201 |
|
if ((fp = fopen(VAR_USER_LIST_SHM, "w")) == NULL) |
| 202 |
|
{ |
| 203 |
|
log_error("fopen(%s) error\n", VAR_USER_LIST_SHM); |
| 204 |
|
goto cleanup; |
| 205 |
|
} |
| 206 |
|
fclose(fp); |
| 207 |
|
|
| 208 |
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) |
| 209 |
{ |
{ |
| 210 |
printf("trie_dict_init failed\n"); |
printf("trie_dict_init(%s, %d) error\n", VAR_TRIE_DICT_SHM, TRIE_NODE_PER_POOL); |
| 211 |
goto cleanup; |
goto cleanup; |
| 212 |
} |
} |
| 213 |
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) |
| 228 |
} |
} |
| 229 |
|
|
| 230 |
// Load menus |
// Load menus |
| 231 |
p_bbs_menu = calloc(1, sizeof(MENU_SET)); |
if (load_menu(&bbs_menu, CONF_MENU) < 0) |
|
if (p_bbs_menu == NULL) |
|
| 232 |
{ |
{ |
| 233 |
log_error("OOM: calloc(MENU_SET)\n"); |
log_error("load_menu(%s) error\n", CONF_MENU); |
| 234 |
goto cleanup; |
goto cleanup; |
| 235 |
} |
} |
| 236 |
if (load_menu(p_bbs_menu, CONF_MENU) < 0) |
if (load_menu(&top10_menu, CONF_TOP10_MENU) < 0) |
| 237 |
{ |
{ |
| 238 |
|
log_error("load_menu(%s) error\n", CONF_TOP10_MENU); |
| 239 |
goto cleanup; |
goto cleanup; |
| 240 |
} |
} |
| 241 |
|
top10_menu.allow_exit = 1; |
| 242 |
|
|
| 243 |
// Load data files |
// Load data files |
| 244 |
if (file_loader_init() < 0) |
if (file_loader_init() < 0) |
| 250 |
{ |
{ |
| 251 |
if (load_file(data_files_load_startup[i]) < 0) |
if (load_file(data_files_load_startup[i]) < 0) |
| 252 |
{ |
{ |
| 253 |
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]); |
| 254 |
} |
} |
| 255 |
} |
} |
| 256 |
|
|
| 257 |
// Load section config |
// Load user_list and online_user_list |
| 258 |
if (load_section_config_from_db() < 0) |
if (user_list_pool_init(VAR_USER_LIST_SHM) < 0) |
| 259 |
|
{ |
| 260 |
|
log_error("user_list_pool_init(%s) error\n", VAR_USER_LIST_SHM); |
| 261 |
|
goto cleanup; |
| 262 |
|
} |
| 263 |
|
if (user_list_pool_reload(0) < 0) |
| 264 |
|
{ |
| 265 |
|
log_error("user_list_pool_reload(all_user) error\n"); |
| 266 |
|
goto cleanup; |
| 267 |
|
} |
| 268 |
|
if (user_list_pool_reload(1) < 0) |
| 269 |
|
{ |
| 270 |
|
log_error("user_list_pool_reload(online_user) error\n"); |
| 271 |
|
goto cleanup; |
| 272 |
|
} |
| 273 |
|
|
| 274 |
|
// Load section config and gen_ex |
| 275 |
|
if (load_section_config_from_db(1) < 0) |
| 276 |
{ |
{ |
| 277 |
log_error("load_section_config_from_db() error\n"); |
log_error("load_section_config_from_db(0) error\n"); |
| 278 |
goto cleanup; |
goto cleanup; |
| 279 |
} |
} |
| 280 |
|
|
| 295 |
|
|
| 296 |
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()); |
| 297 |
|
|
| 298 |
|
if ((ret = user_stat_update()) < 0) |
| 299 |
|
{ |
| 300 |
|
log_error("user_stat_update() error\n"); |
| 301 |
|
goto cleanup; |
| 302 |
|
} |
| 303 |
|
|
| 304 |
// Set signal handler |
// Set signal handler |
| 305 |
act.sa_handler = sig_hup_handler; |
act.sa_handler = sig_hup_handler; |
| 306 |
if (sigaction(SIGHUP, &act, NULL) == -1) |
if (sigaction(SIGHUP, &act, NULL) == -1) |
| 336 |
file_loader_cleanup(); |
file_loader_cleanup(); |
| 337 |
|
|
| 338 |
// Cleanup menu |
// Cleanup menu |
| 339 |
unload_menu(p_bbs_menu); |
unload_menu(&bbs_menu); |
| 340 |
free(p_bbs_menu); |
unload_menu(&top10_menu); |
|
p_bbs_menu = NULL; |
|
| 341 |
|
|
| 342 |
// Cleanup data pools |
// Cleanup data pools |
| 343 |
section_list_cleanup(); |
section_list_cleanup(); |
| 344 |
article_block_cleanup(); |
article_block_cleanup(); |
| 345 |
trie_dict_cleanup(); |
trie_dict_cleanup(); |
| 346 |
|
user_list_pool_cleanup(); |
| 347 |
|
|
|
if (unlink(VAR_TRIE_DICT_SHM) < 0) |
|
|
{ |
|
|
log_error("unlink(%s) error\n", VAR_TRIE_DICT_SHM); |
|
|
} |
|
| 348 |
if (unlink(VAR_ARTICLE_BLOCK_SHM) < 0) |
if (unlink(VAR_ARTICLE_BLOCK_SHM) < 0) |
| 349 |
{ |
{ |
| 350 |
log_error("unlink(%s) error\n", VAR_ARTICLE_BLOCK_SHM); |
log_error("unlink(%s) error\n", VAR_ARTICLE_BLOCK_SHM); |
| 353 |
{ |
{ |
| 354 |
log_error("unlink(%s) error\n", VAR_SECTION_LIST_SHM); |
log_error("unlink(%s) error\n", VAR_SECTION_LIST_SHM); |
| 355 |
} |
} |
| 356 |
|
if (unlink(VAR_TRIE_DICT_SHM) < 0) |
| 357 |
|
{ |
| 358 |
|
log_error("unlink(%s) error\n", VAR_TRIE_DICT_SHM); |
| 359 |
|
} |
| 360 |
|
if (unlink(VAR_USER_LIST_SHM) < 0) |
| 361 |
|
{ |
| 362 |
|
log_error("unlink(%s) error\n", VAR_SECTION_LIST_SHM); |
| 363 |
|
} |
| 364 |
|
|
| 365 |
log_common("Main process exit normally\n"); |
log_common("Main process exit normally\n"); |
| 366 |
|
|