struct dlr_storage *dlr_init_sdb(Cfg* cfg)
{
CfgGroup *grp;
List *grplist;
Octstr *sdb_url, *sdb_id;
Octstr *p = NULL;
long pool_size;
DBConf *db_conf = NULL;
/*
* check for all mandatory directives that specify the field names
* of the used table
*/
if (!(grp = cfg_get_single_group(cfg, octstr_imm("dlr-db"))))
panic(0, "DLR: SDB: group 'dlr-db' is not specified!");
if (!(sdb_id = cfg_get(grp, octstr_imm("id"))))
panic(0, "DLR: SDB: directive 'id' is not specified!");
fields = dlr_db_fields_create(grp);
gw_assert(fields != NULL);
/*
* now grap the required information from the 'mysql-connection' group
* with the sdb-id we just obtained
*
* we have to loop through all available SDB connection definitions
* and search for the one we are looking for
*/
grplist = cfg_get_multi_group(cfg, octstr_imm("sdb-connection"));
while (grplist && (grp = gwlist_extract_first(grplist)) != NULL) {
p = cfg_get(grp, octstr_imm("id"));
if (p != NULL && octstr_compare(p, sdb_id) == 0) {
goto found;
}
if (p != NULL) octstr_destroy(p);
}
panic(0, "DLR: SDB: connection settings for id '%s' are not specified!",
octstr_get_cstr(sdb_id));
found:
octstr_destroy(p);
gwlist_destroy(grplist, NULL);
if (cfg_get_integer(&pool_size, grp, octstr_imm("max-connections")) == -1 || pool_size == 0)
pool_size = 1;
if (!(sdb_url = cfg_get(grp, octstr_imm("url"))))
panic(0, "DLR: SDB: directive 'url' is not specified!");
if (octstr_search(sdb_url, octstr_imm("oracle:"), 0) == 0)
sdb_conn_type = SDB_ORACLE;
else if (octstr_search(sdb_url, octstr_imm("mysql:"), 0) == 0) {
warning(0, "DLR[sdb]: Please use native MySQL support, instead of libsdb.");
sdb_conn_type = SDB_MYSQL;
}
else if (octstr_search(sdb_url, octstr_imm("postgres:"), 0) == 0) {
sdb_conn_type = SDB_POSTGRES;
}
else
sdb_conn_type = SDB_OTHER;
/*
* ok, ready to connect
*/
info(0,"Connecting to sdb resource <%s>.", octstr_get_cstr(sdb_url));
db_conf = gw_malloc(sizeof(DBConf));
gw_assert(db_conf != NULL);
db_conf->sdb = gw_malloc(sizeof(SDBConf));
gw_assert(db_conf->sdb != NULL);
db_conf->sdb->url = sdb_url;
pool = dbpool_create(DBPOOL_SDB, db_conf, pool_size);
gw_assert(pool != NULL);
/*
* XXX should a failing connect throw panic?!
*/
if (dbpool_conn_count(pool) == 0)
panic(0,"DLR: SDB: database pool has no connections!");
return &handles;
}
struct dlr_storage *dlr_init_pgsql(Cfg *cfg)
{
CfgGroup *grp;
List *grplist;
Octstr *pgsql_host, *pgsql_user, *pgsql_pass, *pgsql_db, *pgsql_id;
long pgsql_port = 0;
Octstr *p = NULL;
long pool_size;
DBConf *db_conf = NULL;
/*
* check for all mandatory directives that specify the field names
* of the table used
*/
if (!(grp = cfg_get_single_group(cfg, octstr_imm("dlr-db"))))
panic(0, "DLR: PgSQL: group 'dlr-db' is not specified!");
if (!(pgsql_id = cfg_get(grp, octstr_imm("id"))))
panic(0, "DLR: PgSQL: directive 'id' is not specified!");
fields = dlr_db_fields_create(grp);
gw_assert(fields != NULL);
/*
* now grap the required information from the 'pgsql-connection' group
* with the pgsql-id we just obtained
*
* we have to loop through all available PostgreSQL connection definitions
* and search for the one we are looking for
*/
grplist = cfg_get_multi_group(cfg, octstr_imm("pgsql-connection"));
while (grplist && (grp = gwlist_extract_first(grplist)) != NULL) {
p = cfg_get(grp, octstr_imm("id"));
if (p != NULL && octstr_compare(p, pgsql_id) == 0) {
goto found;
}
if (p != NULL)
octstr_destroy(p);
}
panic(0, "DLR: PgSQL: connection settings for id '%s' are not specified!",
octstr_get_cstr(pgsql_id));
found:
octstr_destroy(p);
gwlist_destroy(grplist, NULL);
if (cfg_get_integer(&pool_size, grp, octstr_imm("max-connections")) == -1 || pool_size == 0)
pool_size = 1;
if (!(pgsql_host = cfg_get(grp, octstr_imm("host"))))
panic(0, "DLR: PgSQL: directive 'host' is not specified!");
if (!(pgsql_user = cfg_get(grp, octstr_imm("username"))))
panic(0, "DLR: PgSQL: directive 'username' is not specified!");
if (!(pgsql_pass = cfg_get(grp, octstr_imm("password"))))
panic(0, "DLR: PgSQL: directive 'password' is not specified!");
if (!(pgsql_db = cfg_get(grp, octstr_imm("database"))))
panic(0, "DLR: PgSQL: directive 'database' is not specified!");
cfg_get_integer(&pgsql_port, grp, octstr_imm("port")); /* optional */
/*
* ok, ready to connect to the database
*/
db_conf = gw_malloc(sizeof(DBConf));
gw_assert(db_conf != NULL);
db_conf->pgsql = gw_malloc(sizeof(PgSQLConf));
gw_assert(db_conf->pgsql != NULL);
db_conf->pgsql->host = pgsql_host;
db_conf->pgsql->port = pgsql_port;
db_conf->pgsql->username = pgsql_user;
db_conf->pgsql->password = pgsql_pass;
db_conf->pgsql->database = pgsql_db;
pool = dbpool_create(DBPOOL_PGSQL, db_conf, pool_size);
gw_assert(pool != NULL);
/*
* XXX should a failing connect throw panic?!
*/
if (dbpool_conn_count(pool) == 0)
panic(0,"DLR: PgSQL: database pool has no connections!");
octstr_destroy(pgsql_id);
return &handles;
}
struct dlr_storage *dlr_init_redis(Cfg *cfg)
{
CfgGroup *grp;
List *grplist;
Octstr *redis_host, *redis_pass, *redis_id;
long redis_port = 0, redis_database = -1, redis_idle_timeout = -1;
Octstr *p = NULL;
long pool_size;
DBConf *db_conf = NULL;
/*
* Check for all mandatory directives that specify the field names
* of the used Redis key
*/
if (!(grp = cfg_get_single_group(cfg, octstr_imm("dlr-db"))))
panic(0, "DLR: Redis: group 'dlr-db' is not specified!");
if (!(redis_id = cfg_get(grp, octstr_imm("id"))))
panic(0, "DLR: Redis: directive 'id' is not specified!");
fields = dlr_db_fields_create(grp);
gw_assert(fields != NULL);
/*
* Escaping special quotes for field/table names
*/
octstr_replace(fields->table, octstr_imm("`"), octstr_imm("``"));
octstr_replace(fields->field_smsc, octstr_imm("`"), octstr_imm("``"));
octstr_replace(fields->field_ts, octstr_imm("`"), octstr_imm("``"));
octstr_replace(fields->field_src, octstr_imm("`"), octstr_imm("``"));
octstr_replace(fields->field_dst, octstr_imm("`"), octstr_imm("``"));
octstr_replace(fields->field_serv, octstr_imm("`"), octstr_imm("``"));
octstr_replace(fields->field_url, octstr_imm("`"), octstr_imm("``"));
octstr_replace(fields->field_mask, octstr_imm("`"), octstr_imm("``"));
octstr_replace(fields->field_status, octstr_imm("`"), octstr_imm("``"));
octstr_replace(fields->field_boxc, octstr_imm("`"), octstr_imm("``"));
/*
* Now grab the required information from the 'redis-connection' group
* with the redis-id we just obtained.
*
* We have to loop through all available Redis connection definitions
* and search for the one we are looking for.
*/
grplist = cfg_get_multi_group(cfg, octstr_imm("redis-connection"));
while (grplist && (grp = gwlist_extract_first(grplist)) != NULL) {
p = cfg_get(grp, octstr_imm("id"));
if (p != NULL && octstr_compare(p, redis_id) == 0) {
goto found;
}
if (p != NULL)
octstr_destroy(p);
}
panic(0, "DLR: Redis: connection settings for id '%s' are not specified!",
octstr_get_cstr(redis_id));
found:
octstr_destroy(p);
gwlist_destroy(grplist, NULL);
if (cfg_get_integer(&pool_size, grp, octstr_imm("max-connections")) == -1 || pool_size == 0)
pool_size = 1;
if (!(redis_host = cfg_get(grp, octstr_imm("host"))))
panic(0, "DLR: Redis: directive 'host' is not specified!");
if (cfg_get_integer(&redis_port, grp, octstr_imm("port")) == -1)
panic(0, "DLR: Redis: directive 'port' is not specified!");
redis_pass = cfg_get(grp, octstr_imm("password"));
cfg_get_integer(&redis_database, grp, octstr_imm("database"));
cfg_get_integer(&redis_idle_timeout, grp, octstr_imm("idle-timeout"));
/*
* Ok, ready to connect to Redis
*/
db_conf = gw_malloc(sizeof(DBConf));
gw_assert(db_conf != NULL);
db_conf->redis = gw_malloc(sizeof(RedisConf));
gw_assert(db_conf->redis != NULL);
db_conf->redis->host = redis_host;
db_conf->redis->port = redis_port;
db_conf->redis->password = redis_pass;
db_conf->redis->database = redis_database;
db_conf->redis->idle_timeout = redis_idle_timeout;
pool = dbpool_create(DBPOOL_REDIS, db_conf, pool_size);
gw_assert(pool != NULL);
/*
* Panic on failure to connect. Should we just try to reconnect?
*/
if (dbpool_conn_count(pool) == 0)
panic(0,"DLR: Redis: database pool has no connections!");
octstr_destroy(redis_id);
return &handles;
}
struct dlr_storage *dlr_init_oracle(Cfg *cfg)
{
CfgGroup *grp;
List *grplist;
long pool_size;
DBConf *db_conf = NULL;
Octstr *id, *username, *password, *tnsname;
int found;
if ((grp = cfg_get_single_group(cfg, octstr_imm("dlr-db"))) == NULL)
panic(0, "DLR: ORACLE: group 'dlr-db' is not specified!");
if (!(id = cfg_get(grp, octstr_imm("id"))))
panic(0, "DLR: ORACLE: directive 'id' is not specified!");
/* initialize database fields */
fields = dlr_db_fields_create(grp);
gw_assert(fields != NULL);
grplist = cfg_get_multi_group(cfg, octstr_imm("oracle-connection"));
found = 0;
while (grplist && (grp = gwlist_extract_first(grplist)) != NULL) {
Octstr *p = cfg_get(grp, octstr_imm("id"));
if (p != NULL && octstr_compare(p, id) == 0) {
found = 1;
}
if (p != NULL)
octstr_destroy(p);
if (found == 1)
break;
}
gwlist_destroy(grplist, NULL);
if (found == 0)
panic(0, "DLR: ORACLE: connection settings for id '%s' are not specified!",
octstr_get_cstr(id));
username = cfg_get(grp, octstr_imm("username"));
password = cfg_get(grp, octstr_imm("password"));
tnsname = cfg_get(grp, octstr_imm("tnsname"));
if (cfg_get_integer(&pool_size, grp, octstr_imm("max-connections")) == -1)
pool_size = 1;
if (username == NULL || password == NULL || tnsname == NULL)
panic(0, "DLR: ORACLE: connection settings missing for id '%s', please"
" check you configuration.",octstr_get_cstr(id));
/* ok we are ready to create dbpool */
db_conf = gw_malloc(sizeof(*db_conf));
db_conf->oracle = gw_malloc(sizeof(OracleConf));
db_conf->oracle->username = username;
db_conf->oracle->password = password;
db_conf->oracle->tnsname = tnsname;
pool = dbpool_create(DBPOOL_ORACLE, db_conf, pool_size);
gw_assert(pool != NULL);
if (dbpool_conn_count(pool) == 0)
panic(0, "DLR: ORACLE: Couldnot establish oracle connection(s).");
octstr_destroy(id);
return &handles;
}
struct dlr_storage *dlr_init_sqlite3(Cfg *cfg)
{
CfgGroup *grp;
List *grplist;
long pool_size;
DBConf *db_conf = NULL;
Octstr *id, *file;
int found;
if ((grp = cfg_get_single_group(cfg, octstr_imm("dlr-db"))) == NULL)
panic(0, "DLR: SQLite3: group 'dlr-db' is not specified!");
if (!(id = cfg_get(grp, octstr_imm("id"))))
panic(0, "DLR: SQLite3: directive 'id' is not specified!");
/* initialize database fields */
fields = dlr_db_fields_create(grp);
gw_assert(fields != NULL);
grplist = cfg_get_multi_group(cfg, octstr_imm("sqlite3-connection"));
found = 0;
while (grplist && (grp = gwlist_extract_first(grplist)) != NULL) {
Octstr *p = cfg_get(grp, octstr_imm("id"));
if (p != NULL && octstr_compare(p, id) == 0) {
found = 1;
}
if (p != NULL)
octstr_destroy(p);
if (found == 1)
break;
}
gwlist_destroy(grplist, NULL);
if (found == 0)
panic(0, "DLR: SQLite3: connection settings for id '%s' are not specified!",
octstr_get_cstr(id));
file = cfg_get(grp, octstr_imm("database"));
if (cfg_get_integer(&pool_size, grp, octstr_imm("max-connections")) == -1)
pool_size = 1;
if (file == NULL)
panic(0, "DLR: SQLite3: connection settings missing for id '%s', please"
" check you configuration.",octstr_get_cstr(id));
/* ok we are ready to create dbpool */
db_conf = gw_malloc(sizeof(*db_conf));
db_conf->sqlite3 = gw_malloc(sizeof(SQLite3Conf));
db_conf->sqlite3->file = file;
pool = dbpool_create(DBPOOL_SQLITE3, db_conf, pool_size);
gw_assert(pool != NULL);
if (dbpool_conn_count(pool) == 0)
panic(0, "DLR: SQLite3: Could not establish sqlite3 connection(s).");
octstr_destroy(id);
return &handles;
}
struct dlr_storage *dlr_init_mongodb(Cfg *cfg)
{
CfgGroup *grp;
List *grplist;
Octstr *mongodb_host, *mongodb_user, *mongodb_pass, *mongodb_db, *mongodb_id;
long pool_size;
long mongodb_port = 27017;
DBConf *db_conf = NULL;
int found;
if ((grp = cfg_get_single_group(cfg, octstr_imm("dlr-db"))) == NULL) {
panic(0, "DLR: MongoDB: group 'dlr-db' is not specified!");
}
if (!(mongodb_id = cfg_get(grp, octstr_imm("id")))) {
panic(0, "DLR: MongoDB: directive 'id' is not specified!");
}
/* initialize database fields */
fields = dlr_db_fields_create(grp);
gw_assert(fields != NULL);
grplist = cfg_get_multi_group(cfg, octstr_imm("mongodb-connection"));
found = 0;
while (grplist && (grp = gwlist_extract_first(grplist)) != NULL) {
Octstr *p = cfg_get(grp, octstr_imm("id"));
if (p != NULL && octstr_compare(p, mongodb_id) == 0) {
found = 1;
}
if (p != NULL) {
octstr_destroy(p);
}
if (found == 1) {
break;
}
}
gwlist_destroy(grplist, NULL);
if (found == 0) {
panic(0, "DLR: MongoDB: connection settings for id '%s' are not specified!",
octstr_get_cstr(mongodb_id));
}
if (!(mongodb_host = cfg_get(grp, octstr_imm("host")))) {
panic(0, "DLR: MongoDB: directive 'host' is not specified!");
}
if (!(mongodb_db = cfg_get(grp, octstr_imm("database")))) {
panic(0, "DLR: MongoDB: directive 'database' is not specified!");
}
/* Keep a global reference to the database and table */
mongodb_database = octstr_get_cstr(mongodb_db);
mongodb_table = octstr_get_cstr(fields->table);
mongodb_namespace = (char *)gw_malloc(strlen(mongodb_database) + strlen(mongodb_table) + 2); /* . and \0 */
sprintf(mongodb_namespace, "%s.%s", mongodb_database, mongodb_table);
mongodb_user = cfg_get(grp, octstr_imm("username"));
mongodb_pass = cfg_get(grp, octstr_imm("password"));
cfg_get_integer(&mongodb_port, grp, octstr_imm("port")); /* optional */
if (cfg_get_integer(&pool_size, grp, octstr_imm("max-connections")) == -1) {
pool_size = 1;
}
/* ok we are ready to create dbpool */
db_conf = gw_malloc(sizeof(*db_conf));
gw_assert(db_conf != NULL);
db_conf->mongodb = gw_malloc(sizeof(MongoDBConf));
gw_assert(db_conf->mongodb != NULL);
db_conf->mongodb->host = mongodb_host;
db_conf->mongodb->port = mongodb_port;
db_conf->mongodb->username = mongodb_user;
db_conf->mongodb->password = mongodb_pass;
db_conf->mongodb->database = mongodb_db;
pool = dbpool_create(DBPOOL_MONGODB, db_conf, pool_size);
gw_assert(pool != NULL);
if (dbpool_conn_count(pool) == 0) {
panic(0, "DLR: MongoDB: Could not establish connection(s).");
}
octstr_destroy(mongodb_id);
return &handles;
}
