mirror of https://github.com/sipwise/kamailio.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
289 lines
6.4 KiB
289 lines
6.4 KiB
/*
|
|
* $Id$
|
|
*
|
|
* Copyright (C) 2001-2003 FhG Fokus
|
|
*
|
|
* 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
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
/*
|
|
* History:
|
|
* --------
|
|
* 2004-06-06 updated to the new DB api (andrei)
|
|
*/
|
|
|
|
#include "../../mem/shm_mem.h"
|
|
#include "../../lib/srdb1/db.h"
|
|
#include "../../dprint.h"
|
|
#include "../../str.h"
|
|
#include "cpl_db.h"
|
|
|
|
#define TABLE_VERSION 1
|
|
|
|
static db1_con_t* db_hdl=0;
|
|
static db_func_t cpl_dbf;
|
|
|
|
str cpl_username_col = str_init("username");
|
|
str cpl_domain_col = str_init("domain");
|
|
str cpl_xml_col = str_init("cpl_xml");
|
|
str cpl_bin_col = str_init("cpl_bin");
|
|
|
|
|
|
int cpl_db_bind(const str* db_url, const str *db_table)
|
|
{
|
|
if (db_bind_mod(db_url, &cpl_dbf )) {
|
|
LM_CRIT("cannot bind to database module! "
|
|
"Did you forget to load a database module ?\n");
|
|
return -1;
|
|
}
|
|
|
|
/* CPL module uses all database functions */
|
|
if (!DB_CAPABILITY(cpl_dbf, DB_CAP_ALL)) {
|
|
LM_CRIT("Database modules does not "
|
|
"provide all functions needed by cpl-c module\n");
|
|
return -1;
|
|
}
|
|
|
|
if ( cpl_db_init( db_url, db_table) )
|
|
return -1;
|
|
|
|
if(db_check_table_version(&cpl_dbf, db_hdl, db_table, TABLE_VERSION) < 0) {
|
|
LM_ERR("error during table version check.\n");
|
|
cpl_db_close();
|
|
return -1;
|
|
}
|
|
|
|
cpl_db_close();
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
int cpl_db_init(const str* db_url, const str* db_table)
|
|
{
|
|
if (cpl_dbf.init==0){
|
|
LM_CRIT("BUG - unbound database module\n");
|
|
return -1;
|
|
}
|
|
|
|
db_hdl=cpl_dbf.init(db_url);
|
|
|
|
if (db_hdl==0){
|
|
LM_CRIT("cannot initialize database connection\n");
|
|
return -1;
|
|
}
|
|
|
|
if (cpl_dbf.use_table(db_hdl, db_table)<0) {
|
|
LM_CRIT("cannot select table \"%.*s\"\n",db_table->len, db_table->s);
|
|
cpl_db_close();
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
void cpl_db_close(void)
|
|
{
|
|
if (db_hdl && cpl_dbf.close){
|
|
cpl_dbf.close(db_hdl);
|
|
db_hdl=0;
|
|
}
|
|
}
|
|
|
|
|
|
/* gets from database the cpl script in binary format; the returned script is
|
|
* allocated in shared memory
|
|
* Returns: 1 - success
|
|
* -1 - error
|
|
*/
|
|
int get_user_script(str *username, str *domain, str *script, str* key)
|
|
{
|
|
db_key_t keys_cmp[2];
|
|
db_key_t keys_ret[1];
|
|
db_val_t vals[2];
|
|
db1_res_t *res = NULL;
|
|
int n;
|
|
|
|
keys_cmp[0] = &cpl_username_col;
|
|
keys_cmp[1] = &cpl_domain_col;
|
|
keys_ret[0] = key;
|
|
|
|
LM_DBG("fetching script for user <%.*s>\n",
|
|
username->len,username->s);
|
|
vals[0].type = DB1_STR;
|
|
vals[0].nul = 0;
|
|
vals[0].val.str_val = *username;
|
|
n = 1;
|
|
if (domain) {
|
|
vals[1].type = DB1_STR;
|
|
vals[1].nul = 0;
|
|
vals[1].val.str_val = *domain;
|
|
n++;
|
|
}
|
|
|
|
if (cpl_dbf.query(db_hdl, keys_cmp, 0, vals, keys_ret, n, 1, NULL, &res)
|
|
< 0){
|
|
LM_ERR("db_query failed\n");
|
|
goto error;
|
|
}
|
|
|
|
if (res->n==0) {
|
|
LM_DBG("user <%.*s> not found in db -> probably "
|
|
"he has no script\n",username->len, username->s);
|
|
script->s = 0;
|
|
script->len = 0;
|
|
} else {
|
|
if (res->rows[0].values[0].nul) {
|
|
LM_DBG("user <%.*s> has a NULL script\n",
|
|
username->len, username->s);
|
|
script->s = 0;
|
|
script->len = 0;
|
|
} else {
|
|
LM_DBG("we got the script len=%d\n",
|
|
res->rows[0].values[0].val.blob_val.len);
|
|
script->len = res->rows[0].values[0].val.blob_val.len;
|
|
script->s = shm_malloc( script->len );
|
|
if (!script->s) {
|
|
LM_ERR("no free sh_mem\n");
|
|
goto error;
|
|
}
|
|
memcpy( script->s, res->rows[0].values[0].val.blob_val.s,
|
|
script->len);
|
|
}
|
|
}
|
|
|
|
cpl_dbf.free_result( db_hdl, res);
|
|
return 1;
|
|
error:
|
|
if (res)
|
|
cpl_dbf.free_result( db_hdl, res);
|
|
script->s = 0;
|
|
script->len = 0;
|
|
return -1;
|
|
}
|
|
|
|
|
|
|
|
/* inserts into database a cpl script in XML format(xml) along with its binary
|
|
* format (bin)
|
|
* Returns: 1 - success
|
|
* -1 - error
|
|
*/
|
|
int write_to_db(str *username, str *domain, str *xml, str *bin)
|
|
{
|
|
db_key_t keys[4];
|
|
db_val_t vals[4];
|
|
db1_res_t *res = NULL;
|
|
int n;
|
|
|
|
/* lets see if the user is already in database */
|
|
keys[2] = &cpl_username_col;
|
|
vals[2].type = DB1_STR;
|
|
vals[2].nul = 0;
|
|
vals[2].val.str_val = *username;
|
|
n = 1;
|
|
if (domain) {
|
|
keys[3] = &cpl_domain_col;
|
|
vals[3].type = DB1_STR;
|
|
vals[3].nul = 0;
|
|
vals[3].val.str_val = *domain;
|
|
n++;
|
|
}
|
|
if (cpl_dbf.query(db_hdl, keys+2, 0, vals+2, keys+2, n, 1, NULL, &res)<0) {
|
|
LM_ERR("db_query failed\n");
|
|
goto error;
|
|
}
|
|
if (res->n>1) {
|
|
LM_ERR("Inconsistent CPL database:"
|
|
" %d records for user %.*s\n",res->n,username->len,username->s);
|
|
goto error;
|
|
}
|
|
|
|
/* cpl text */
|
|
keys[0] = &cpl_xml_col;
|
|
vals[0].type = DB1_BLOB;
|
|
vals[0].nul = 0;
|
|
vals[0].val.blob_val.s = xml->s;
|
|
vals[0].val.blob_val.len = xml->len;
|
|
n++;
|
|
/* cpl bin */
|
|
keys[1] = &cpl_bin_col;
|
|
vals[1].type = DB1_BLOB;
|
|
vals[1].nul = 0;
|
|
vals[1].val.blob_val.s = bin->s;
|
|
vals[1].val.blob_val.len = bin->len;
|
|
n++;
|
|
/* insert or update ? */
|
|
if (res->n==0) {
|
|
LM_DBG("no user %.*s in CPL database->insert\n",
|
|
username->len,username->s);
|
|
if (cpl_dbf.insert(db_hdl, keys, vals, n) < 0) {
|
|
LM_ERR("insert failed !\n");
|
|
goto error;
|
|
}
|
|
} else {
|
|
LM_DBG("user %.*s already in CPL database ->"
|
|
" update\n",username->len,username->s);
|
|
if (cpl_dbf.update(db_hdl, keys+2, 0, vals+2, keys, vals, n-2, 2) < 0) {
|
|
LM_ERR("update failed !\n");
|
|
goto error;
|
|
}
|
|
}
|
|
|
|
return 1;
|
|
error:
|
|
return -1;
|
|
}
|
|
|
|
|
|
|
|
/* delete from database the entity record for a given user - if a user has no
|
|
* script, he will be removed completely from db; users without script are not
|
|
* allowed into db ;-)
|
|
* Returns: 1 - success
|
|
* -1 - error
|
|
*/
|
|
int rmv_from_db(str *username, str *domain)
|
|
{
|
|
db_key_t keys[2];
|
|
db_val_t vals[2];
|
|
int n;
|
|
|
|
/* username */
|
|
keys[0] = &cpl_username_col;
|
|
vals[0].type = DB1_STR;
|
|
vals[0].nul = 0;
|
|
vals[0].val.str_val = *username;
|
|
n = 1;
|
|
if (domain) {
|
|
keys[1] = &cpl_domain_col;
|
|
vals[1].type = DB1_STR;
|
|
vals[1].nul = 0;
|
|
vals[1].val.str_val = *domain;
|
|
n++;
|
|
}
|
|
|
|
if (cpl_dbf.delete(db_hdl, keys, NULL, vals, n) < 0) {
|
|
LM_ERR("failed to delete script for "
|
|
"user \"%.*s\"\n",username->len,username->s);
|
|
return -1;
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|