/*pre-fork*/
int mln_pre_fork(void)
{
if (signal(SIGCHLD, SIG_IGN) == SIG_ERR) {
mln_log(error, "signal() to ignore SIGCHLD failed, %s\n", strerror(errno));
return -1;
}
if (signal(SIGPIPE, SIG_IGN) == SIG_ERR) {
mln_log(error, "signal() to ignore SIGPIPE failed, %s\n", strerror(errno));
return -1;
}
struct sigaction act;
memset(&act, 0, sizeof(act));
act.sa_handler = do_fork_sig_handler;
sigemptyset(&(act.sa_mask));
sigfillset(&(act.sa_mask));
if (sigaction(wait_signo, &act, NULL) < 0) {
mln_log(error, "sigaction error, signo:%d. %s\n", \
wait_signo, strerror(errno));
return -1;
}
if (mln_tcp_conn_init(&master_conn, -1) < 0) {
mln_log(error, "No memory.\n");
return -1;
}
child_state = STATE_IDLE;
child_error_bytes = 0;
cur_msg_len = 0;
child_msg_content = NULL;
cur_msg_type = 0;
struct mln_rbtree_attr rbattr;
rbattr.cmp = mln_fork_rbtree_cmp;
rbattr.data_free = NULL;
if ((master_ipc_tree = mln_rbtree_init(&rbattr)) < 0) {
mln_log(error, "No memory.\n");
if (mln_tcp_conn_get_fd(&master_conn) >= 0)
close(mln_tcp_conn_get_fd(&master_conn));
mln_tcp_conn_destroy(&master_conn);
return -1;
}
if ((worker_ipc_tree = mln_rbtree_init(&rbattr)) < 0) {
mln_log(error, "No memory.\n");
mln_rbtree_destroy(master_ipc_tree);
master_ipc_tree = NULL;
if (mln_tcp_conn_get_fd(&master_conn) >= 0)
close(mln_tcp_conn_get_fd(&master_conn));
mln_tcp_conn_destroy(&master_conn);
return -1;
}
mln_set_ipc_handlers();
return 0;
}
/*mln_conf_domain_t*/
static mln_conf_domain_t *
mln_conf_domain_init(mln_conf_t *cf, mln_string_t *domain_name)
{
mln_conf_domain_t *cd;
cd = (mln_conf_domain_t *)malloc(sizeof(mln_conf_domain_t));
if (cd == NULL) return NULL;
cd->search = mln_conf_search_cmd;
cd->domain_name = mln_string_dup(domain_name);
if (cd->domain_name == NULL) {
free(cd);
return NULL;
}
struct mln_rbtree_attr rbattr;
rbattr.cmp = mln_conf_cmd_cmp;
rbattr.data_free = mln_conf_cmd_destroy;
if ((cd->cmd = mln_rbtree_init(&rbattr)) == NULL) {
mln_string_free(cd->domain_name);
free(cd);
return NULL;
}
return cd;
}
/*mln_conf_t*/
static inline mln_conf_t *mln_conf_init(void)
{
mln_conf_lex_lex_dup(NULL, NULL);/*nothing to do, just get rid of compiler's warnging*/
mln_conf_t *cf;
mln_rbtree_node_t *rn;
cf = (mln_conf_t *)malloc(sizeof(mln_conf_t));
if (cf == NULL) {
fprintf(stderr, "No memory.\n");
return NULL;
}
cf->search = mln_conf_search_domain;
struct mln_rbtree_attr rbattr;
rbattr.cmp = mln_conf_domain_cmp;
rbattr.data_free = mln_conf_domain_destroy;
if ((cf->domain = mln_rbtree_init(&rbattr)) == NULL) {
fprintf(stderr, "No memory.\n");
free(cf);
return NULL;
}
mln_alloc_t *pool;
mln_string_t path;
char *conf_file_path;
mln_size_t path_len = strlen(mln_path());
mln_lex_hooks_t hooks;
struct mln_lex_attr lattr;
if ((pool = mln_alloc_init()) == NULL) {
fprintf(stderr, "No memory.\n");
mln_rbtree_destroy(cf->domain);
free(cf);
return NULL;
}
if ((conf_file_path = (char *)mln_alloc_m(pool, path_len + sizeof(conf_filename) + 1)) == NULL) {
fprintf(stderr, "No memory.\n");
mln_alloc_destroy(pool);
mln_rbtree_destroy(cf->domain);
free(cf);
return NULL;
}
memcpy(conf_file_path, mln_path(), path_len);
conf_file_path[path_len] = '/';
memcpy(conf_file_path+path_len+1, conf_filename, sizeof(conf_filename)-1);
mln_string_nSet(&path, conf_file_path, path_len + sizeof(conf_filename));
lattr.pool = pool;
lattr.keywords = conf_keywords;
memset(&hooks, 0, sizeof(hooks));
hooks.slash_handler = (lex_hook)mln_conf_lex_slash_handler;
hooks.sglq_handler = (lex_hook)mln_conf_lex_sglq_handler;
hooks.dblq_handler = (lex_hook)mln_conf_lex_dblq_handler;
lattr.hooks = &hooks;
lattr.preprocess = 1;
lattr.type = M_INPUT_T_FILE;
lattr.data = &path;
mln_lex_initWithHooks(mln_conf_lex, cf->lex, &lattr);
mln_alloc_free(conf_file_path);
if (cf->lex == NULL) {
fprintf(stderr, "No memory.\n");
mln_alloc_destroy(pool);
mln_rbtree_destroy(cf->domain);
free(cf);
return NULL;
}
mln_conf_domain_t *cd = mln_conf_domain_init(cf, &default_domain);
if (cd == NULL) {
fprintf(stderr, "No memory.\n");
mln_conf_destroy(cf);
return NULL;
}
if ((rn = mln_rbtree_new_node(cf->domain, cd)) == NULL) {
fprintf(stderr, "No memory.\n");
mln_conf_domain_destroy((void *)cd);
mln_conf_destroy(cf);
return NULL;
}
mln_rbtree_insert(cf->domain, rn);
return cf;
}
