int pdt_print_tree(pdt_tree_t *pt)
{
int len;
if(pt == NULL)
{
LM_DBG("tree is empty\n");
return 0;
}
LM_DBG("[%.*s]\n", pt->sdomain.len, pt->sdomain.s);
len = 0;
pdt_print_node(pt->head, pdt_code_buf, len);
return pdt_print_tree(pt->next);
}
/**
* init module function
*/
static int mod_init(void)
{
LM_INFO("initializing...\n");
init_db_url( db_url , 0 /*cannot be null*/);
db_table.len = strlen(db_table.s);
sdomain_column.len = strlen(sdomain_column.s);
prefix_column.len = strlen(prefix_column.s);
domain_column.len = strlen(domain_column.s);
prefix.len = strlen(prefix.s);
pdt_char_list.len = strlen(pdt_char_list.s);
if(pdt_char_list.len<=0)
{
LM_ERR("invalid pdt char list\n");
return -1;
}
LM_INFO("pdt_char_list=%s \n",pdt_char_list.s);
/* binding to mysql module */
if(db_bind_mod(&db_url, &pdt_dbf))
{
LM_ERR("database module not found\n");
return -1;
}
if (!DB_CAPABILITY(pdt_dbf, DB_CAP_ALL))
{
LM_ERR("database module does not "
"implement all functions needed by the module\n");
return -1;
}
/* open a connection with the database */
db_con = pdt_dbf.init(&db_url);
if(db_con==NULL)
{
LM_ERR("failed to connect to the database\n");
return -1;
}
if (pdt_dbf.use_table(db_con, &db_table) < 0)
{
LM_ERR("failed to use_table\n");
goto error1;
}
LM_DBG("database connection opened successfully\n");
/* create & init lock */
if ((pdt_lock = lock_init_rw()) == NULL) {
LM_CRIT("failed to init lock\n");
goto error1;
}
/* tree pointer in shm */
_ptree = (pdt_tree_t**)shm_malloc( sizeof(pdt_tree_t*) );
if (_ptree==0) {
LM_ERR("out of shm mem for pdtree\n");
goto error1;
}
*_ptree=0;
/* loading all information from database */
if(pdt_load_db()!=0)
{
LM_ERR("cannot load info from database\n");
goto error1;
}
pdt_dbf.close(db_con);
db_con = 0;
#if 0
pdt_print_tree(*_ptree);
#endif
/* success code */
return 0;
error1:
if (pdt_lock)
{
lock_destroy_rw( pdt_lock );
pdt_lock = 0;
}
if(_ptree!=0)
shm_free(_ptree);
if(db_con!=NULL)
{
pdt_dbf.close(db_con);
db_con = 0;
}
return -1;
}
/**
* init module function
*/
static int mod_init(void)
{
if(pdt_init_rpc()<0)
{
LM_ERR("failed to register RPC commands\n");
return -1;
}
if(pdt_fetch_rows<=0)
pdt_fetch_rows = 1000;
pdt_char_list.len = strlen(pdt_char_list.s);
if(pdt_char_list.len<=0)
{
LM_ERR("invalid pdt char list\n");
return -1;
}
LM_INFO("pdt_char_list=%s \n",pdt_char_list.s);
/* binding to mysql module */
if(db_bind_mod(&db_url, &pdt_dbf))
{
LM_ERR("database module not found\n");
return -1;
}
if (!DB_CAPABILITY(pdt_dbf, DB_CAP_ALL))
{
LM_ERR("database module does not "
"implement all functions needed by the module\n");
return -1;
}
/* open a connection with the database */
db_con = pdt_dbf.init(&db_url);
if(db_con==NULL)
{
LM_ERR("failed to connect to the database\n");
return -1;
}
if (pdt_dbf.use_table(db_con, &db_table) < 0)
{
LM_ERR("failed to use_table\n");
goto error1;
}
LM_DBG("database connection opened successfully\n");
if ( (pdt_lock=lock_alloc())==0) {
LM_CRIT("failed to alloc lock\n");
goto error1;
}
if (lock_init(pdt_lock)==0 ) {
LM_CRIT("failed to init lock\n");
goto error1;
}
/* tree pointer in shm */
_ptree = (pdt_tree_t**)shm_malloc( sizeof(pdt_tree_t*) );
if (_ptree==0) {
LM_ERR("out of shm mem for pdtree\n");
goto error1;
}
*_ptree=0;
/* loading all information from database */
if(pdt_load_db()!=0)
{
LM_ERR("cannot load info from database\n");
goto error1;
}
pdt_dbf.close(db_con);
db_con = 0;
#if 0
pdt_print_tree(*_ptree);
#endif
/* success code */
return 0;
error1:
if (pdt_lock)
{
lock_destroy( pdt_lock );
lock_dealloc( pdt_lock );
pdt_lock = 0;
}
if(_ptree!=0)
{
shm_free(_ptree);
_ptree = 0;
}
if(db_con!=NULL)
{
pdt_dbf.close(db_con);
db_con = 0;
}
return -1;
}
/**
* init module function
*/
static int mod_init(void)
{
DBG("PDT: initializing...\n");
if(hs_two_pow<0)
{
LOG(L_ERR, "PDT:mod_init: hash_size_two_pow must be"
" positive and less than %d\n", MAX_HSIZE_TWO_POW);
return -1;
}
prefix.len = strlen(prefix.s);
/* binding to mysql module */
if(bind_dbmod(db_url, &pdt_dbf))
{
LOG(L_ERR, "PDT:mod_init: Database module not found\n");
return -1;
}
if (!DB_CAPABILITY(pdt_dbf, DB_CAP_ALL))
{
LOG(L_ERR, "PDT: mod_init: Database module does not "
"implement all functions needed by the module\n");
return -1;
}
/* open a connection with the database */
db_con = pdt_dbf.init(db_url);
if(db_con==NULL)
{
LOG(L_ERR,
"PDT: mod_init: Error while connecting to database\n");
return -1;
}
if (pdt_dbf.use_table(db_con, db_table) < 0)
{
LOG(L_ERR, "PDT: mod_init: Error in use_table\n");
goto error1;
}
DBG("PDT: mod_init: Database connection opened successfully\n");
/* init the hash and tree in share memory */
if( (_dhash = pdt_init_hash(hs_two_pow)) == NULL)
{
LOG(L_ERR, "PDT:mod_init: domain hash could not be allocated\n");
goto error1;
}
if( (_ptree = pdt_init_tree()) == NULL)
{
LOG(L_ERR, "PDT:mod_init: prefix tree could not be allocated\n");
goto error2;
}
/* loading all information from database */
if(pdt_load_db()!=0)
{
LOG(L_ERR, "PDT:mod_init: cannot load info from database\n");
goto error3;
}
pdt_dbf.close(db_con);
db_con = 0;
pdt_print_tree(_ptree);
DBG("PDT:mod_init: -------------------\n");
pdt_print_hash(_dhash);
last_sync = time(NULL);
register_timer(pdt_clean_cache, 0, clean_time);
/* success code */
return 0;
error3:
if(_ptree!=NULL)
{
pdt_free_tree(_ptree);
_ptree = 0;
}
error2:
if(_dhash!=NULL)
{
pdt_free_hash(_dhash);
_dhash = 0;
}
error1:
if(db_con!=NULL)
{
pdt_dbf.close(db_con);
db_con = 0;
}
return -1;
}
