src/modules/rtpproxy/rtpproxy_db.c
9a0e5866
 /*
  * rtpproxy module
  *
  * Copyright (c) 2013 Crocodile RCS Ltd
  *
  * This file is part of Kamailio, a free SIP server.
  *
  * Kamailio is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
  * the Free Software Foundation; either version 2 of the License, or
  * (at your option) any later version
  *
  * Kamailio is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  * GNU General Public License for more details.
  *
  * You should have received a copy of the GNU General Public License
  * along with this program; if not, write to the Free Software
9e1ff448
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
9a0e5866
  *
  */
 
 #include "../../lib/srdb1/db.h"
 #include "../../lib/srdb1/db_res.h"
 
cf83221d
 #include "../../core/parser/msg_parser.h"
9a0e5866
 #include "rtpproxy.h"
 
 #define RTPP_TABLE_VERSION 1
 
 static db_func_t rtpp_dbf;
 static db1_con_t *rtpp_db_handle = NULL;
 
 str rtpp_db_url = {NULL, 0};
 str rtpp_table_name = str_init("rtpproxy");
b8953438
 str rtpp_setid_col = str_init("setid");
9a0e5866
 str rtpp_url_col = str_init("url");
 str rtpp_weight_col = str_init("weight");
 str rtpp_flags_col = str_init("flags");
 
 static int rtpp_connect_db(void)
 {
 	if ((rtpp_db_url.s == NULL) || (rtpp_db_url.len == 0))
 		return -1;
 	if ((rtpp_db_handle = rtpp_dbf.init(&rtpp_db_url)) == NULL)
 	{
 		LM_ERR("Cannot initialize db connection\n");
 		return -1;
 	}
 	return 0;
 }
 
 static void rtpp_disconnect_db(void)
 {
 	if (rtpp_db_handle)
 	{
 		rtpp_dbf.close(rtpp_db_handle);
 		rtpp_db_handle = NULL;
 	}
 }
 
 static int rtpp_load_db(void)
 {
 	int i;
 	struct rtpp_set *rtpp_list = NULL;
 	db1_res_t *res = NULL;
 	db_val_t *values = NULL;
 	db_row_t *rows = NULL;
6fbf225f
 	db_key_t query_cols[] = {&rtpp_setid_col, &rtpp_url_col, &rtpp_weight_col,
 		&rtpp_flags_col};
9a0e5866
 
 	str set, url;
 	int weight, flags;
 	int n_rows = 0;
 	int n_cols = 4;
 
 	if (rtpp_db_handle == NULL)
 	{
 		LM_ERR("invalid db handle\n");
 		return -1;
 	}
 	if (rtpp_dbf.use_table(rtpp_db_handle, &rtpp_table_name) < 0)
 	{
6fbf225f
 		LM_ERR("unable to use table '%.*s'\n", rtpp_table_name.len,
 				rtpp_table_name.s);
9a0e5866
 		return -1;
 	}
 	if (rtpp_dbf.query(rtpp_db_handle, 0, 0, 0, query_cols, 0, n_cols, 0, &res) < 0)
 	{
 		LM_ERR("error while running db query\n");
 		return -1;
 	}
 
 	n_rows = RES_ROW_N(res);
 	rows = RES_ROWS(res);
 	if (n_rows == 0)
 	{
 		LM_WARN("No rtpproxy instances in database\n");
 		return 0;
 	}
 	for (i=0; i<n_rows; i++)
 	{
 		values = ROW_VALUES(rows + i);
 
 		set.s = VAL_STR(values).s;
 		set.len = strlen(set.s);
 		url.s = VAL_STR(values+1).s;
 		url.len = strlen(url.s);
 		weight = VAL_INT(values+2);
 		flags = VAL_INT(values+3);
 
 		if ((rtpp_list = get_rtpp_set(&set)) == NULL)
 		{
 			LM_ERR("error getting rtpp_list for set '%.*s'\n", set.len, set.s);
 			continue;
 		}
 		if (insert_rtpp_node(rtpp_list, &url, weight, flags) < 0)
 		{
6fbf225f
 			LM_ERR("error inserting '%.*s' into set '%.*s'\n", url.len, url.s,
 					set.len, set.s);
9a0e5866
 		}
 	}
 
 	rtpp_dbf.free_result(rtpp_db_handle, res);
 	return 0;
 }
 
 int init_rtpproxy_db(void)
 {
 	int ret;
 	if (rtpp_db_url.s == NULL)
 		/* Database not configured */
 		return 0;
 
 	if (db_bind_mod(&rtpp_db_url, &rtpp_dbf) < 0)
 	{
6fbf225f
 		LM_ERR("Unable to bind to db driver - %.*s\n",
 				rtpp_db_url.len, rtpp_db_url.s);
9a0e5866
 		return -1;
 	}
 	if (rtpp_connect_db() != 0)
 	{
 		LM_ERR("Unable to connect to db\n");
 		return -1;
 	}
 
97084998
 	if (db_check_table_version(&rtpp_dbf, rtpp_db_handle, &rtpp_table_name, RTPP_TABLE_VERSION) < 0)
9a0e5866
 	{
97084998
 		DB_TABLE_VERSION_ERROR(rtpp_table_name);
9a0e5866
 		ret = -1;
 		goto done;
 	}
 	ret = rtpp_load_db();
 
 done:
 	rtpp_disconnect_db();
 
 	return ret;
 }