/* pseudo-variable functions */
static int pv_get_rl_count(struct sip_msg *msg, pv_param_t *param,
pv_value_t *res)
{
int counter;
if (!param)
return pv_get_null(msg, param, res);
if(pv_get_spec_name(msg, param, res)!=0 || (!(res->flags&PV_VAL_STR))) {
LM_ERR("invalid name\n");
return -1;
}
counter = rl_get_counter_value(&res->rs);
if (counter < 0) {
return pv_get_null(msg, param, res);
}
return pv_get_uintval(msg, param, res, counter);
}
int ops_dbdelete_avps (struct sip_msg* msg, struct fis_param *sp,
struct db_param *dbp, struct db_url *url, int use_domain)
{
struct sip_uri uri;
int res;
str uuid;
pv_value_t xvalue;
str *s0, *s1, *s2;
s0 = s1 = s2 = NULL;
if (!((sp->opd&AVPOPS_VAL_PVAR)||(sp->opd&AVPOPS_VAL_STR))) {
LM_CRIT("invalid flag combination (%d/%d)\n", sp->opd, sp->ops);
goto error;
}
/* get uuid from avp */
if (sp->opd&AVPOPS_VAL_PVAR)
{
if(pv_get_spec_value(msg, &(sp->u.sval), &xvalue)!=0)
{
LM_CRIT("failed to get PVAR value (%d/%d)\n", sp->opd, sp->ops);
goto error;
}
if(xvalue.flags&(PV_VAL_NULL|PV_VAL_EMPTY))
{
LM_ERR("no value for first param\n");
goto error;
}
uuid = xvalue.rs;
} else {
uuid.s = sp->u.s.s;
uuid.len = sp->u.s.len;
}
if(sp->opd&AVPOPS_FLAG_UUID0)
{
s0 = &uuid;
} else {
/* parse uri */
if (parse_uri(uuid.s, uuid.len, &uri)<0)
{
LM_ERR("failed to parse uri\n");
goto error;
}
/* check uri */
if(!uri.user.s|| !uri.user.len|| !uri.host.len|| !uri.host.s)
{
LM_ERR("incomplet uri <%.*s>\n", uuid.len, uuid.s);
goto error;
}
if((sp->opd&AVPOPS_FLAG_URI0)||(sp->opd&AVPOPS_FLAG_USER0))
s1 = &uri.user;
if((sp->opd&AVPOPS_FLAG_URI0)||(sp->opd&AVPOPS_FLAG_DOMAIN0))
s2 = &uri.host;
}
/* is dynamic avp name ? */
if(dbp->a.type==AVPOPS_VAL_PVAR)
{
if(pv_has_dname(&dbp->a.u.sval))
{
if(pv_get_spec_name(msg, &(dbp->a.u.sval.pvp), &xvalue)!=0)
{
LM_CRIT("failed to get value for P2\n");
goto error;
}
if(xvalue.flags&(PV_VAL_NULL|PV_VAL_EMPTY))
{
LM_INFO("no value for p2\n");
goto error;
}
if(xvalue.flags&PV_VAL_STR)
{
if(xvalue.rs.len>=AVPOPS_ATTR_LEN)
{
LM_ERR("name too long [%d/%.*s...]\n",
xvalue.rs.len, 16, xvalue.rs.s);
goto error;
}
dbp->sa.s = avpops_attr_buf;
memcpy(dbp->sa.s, xvalue.rs.s, xvalue.rs.len);
dbp->sa.len = xvalue.rs.len;
dbp->sa.s[dbp->sa.len] = '\0';
} else {
LM_INFO("no string value for p2\n");
goto error;
}
}
}
/* do DB delete */
res = db_delete_avp( url, s0, s1,
(use_domain||(sp->opd&AVPOPS_FLAG_DOMAIN0))?s2:0,
dbp->sa.s, &dbp->table);
/* res ? */
if (res<0)
{
LM_ERR("db_delete failed\n");
goto error;
}
return 1;
error:
return -1;
}
int ops_dbstore_avps (struct sip_msg* msg, struct fis_param *sp,
struct db_param *dbp, struct db_url *url, int use_domain)
{
struct sip_uri uri;
struct usr_avp **avp_list;
struct usr_avp *avp;
unsigned short name_type;
int avp_name;
int_str i_s;
str uuid;
int keys_nr;
int n;
pv_value_t xvalue;
str *s0, *s1, *s2;
str *sn;
s0 = s1 = s2 = NULL;
name_type = 0;
if (!((sp->opd&AVPOPS_VAL_PVAR)||(sp->opd&AVPOPS_VAL_STR))) {
LM_CRIT("invalid flag combination (%d/%d)\n", sp->opd, sp->ops);
goto error;
}
keys_nr = 6; /* uuid, avp name, avp val, avp type, user, domain */
/* get uuid from avp */
if (sp->opd&AVPOPS_VAL_PVAR)
{
if(pv_get_spec_value(msg, &(sp->u.sval), &xvalue)!=0)
{
LM_CRIT("failed to get PVAR value (%d/%d)\n", sp->opd, sp->ops);
goto error;
}
if(xvalue.flags&(PV_VAL_NULL|PV_VAL_EMPTY))
{
LM_ERR("no value for first param\n");
goto error;
}
uuid = xvalue.rs;
} else {
uuid.s = sp->u.s.s;
uuid.len = sp->u.s.len;
}
if(sp->opd&AVPOPS_FLAG_UUID0)
{
s0 = &uuid;
} else {
/* parse uri */
if (parse_uri(uuid.s, uuid.len, &uri)<0)
{
LM_ERR("failed to parse uri\n");
goto error;
}
/* check uri */
if(!uri.user.s|| !uri.user.len|| !uri.host.len|| !uri.host.s)
{
LM_ERR("incomplet uri <%.*s>\n", uuid.len, uuid.s);
goto error;
}
if((sp->opd&AVPOPS_FLAG_URI0)||(sp->opd&AVPOPS_FLAG_USER0))
s1 = &uri.user;
if((sp->opd&AVPOPS_FLAG_URI0)||(sp->opd&AVPOPS_FLAG_DOMAIN0))
s2 = &uri.host;
}
/* set values for keys */
store_vals[0].val.str_val = (s0)?*s0:empty;
store_vals[4].val.str_val = (s1)?*s1:empty;
if (use_domain || sp->opd&AVPOPS_FLAG_DOMAIN0)
store_vals[5].val.str_val = (s2)?*s2:empty;
avp_name = -1;
/* is dynamic avp name ? */
if(dbp->a.type==AVPOPS_VAL_PVAR)
{
if(pv_has_dname(&dbp->a.u.sval))
{
/* TODO change here to be aware of the int name */
if(pv_get_spec_name(msg, &(dbp->a.u.sval.pvp), &xvalue)!=0)
{
LM_CRIT("failed to get value for P2\n");
goto error;
}
if(xvalue.flags&(PV_VAL_NULL|PV_VAL_EMPTY))
{
LM_INFO("no value for P2\n");
goto error;
}
if(xvalue.flags&PV_TYPE_INT)
{
name_type = 0;
avp_name = xvalue.ri;
} else {
name_type = AVP_NAME_STR;
avp_name = -1;
}
if(xvalue.flags&PV_VAL_STR)
{
if(xvalue.rs.len>=AVPOPS_ATTR_LEN)
{
LM_ERR("name too long [%d/%.*s...]\n",
xvalue.rs.len, 16, xvalue.rs.s);
goto error;
}
dbp->sa.s = avpops_attr_buf;
memcpy(dbp->sa.s, xvalue.rs.s, xvalue.rs.len);
dbp->sa.len = xvalue.rs.len;
dbp->sa.s[dbp->sa.len] = '\0';
avp_name = get_avp_id(&dbp->sa);
/* search for the id only once */
if (avp_name < 0) {
LM_ERR("cannot find avp\n");
goto error;
}
} else {
LM_INFO("no string value for p2\n");
goto error;
}
} else {
name_type = dbp->a.u.sval.pvp.pvn.u.isname.type;
avp_name = dbp->a.u.sval.pvp.pvn.u.isname.name.n;
}
} else {
LM_WARN("TODO: avp is not a dynamic name <%.*s> name is %d\n", dbp->sa.len, dbp->sa.s, avp_name);
avp_name = -1;
}
/* set the script flags */
if(dbp->a.type==AVPOPS_VAL_PVAR)
name_type |= dbp->a.u.sval.pvp.pvn.u.isname.type&0xff00;
/* set uuid/(username and domain) fields */
n =0 ;
if ((dbp->a.opd&AVPOPS_VAL_NONE)==0)
{
/* if avp wasn't found yet */
if (avp_name < 0) {
avp_name = get_avp_id(&dbp->sa);
/* search for the id only once */
if (avp_name < 0) {
LM_ERR("cannot find avp\n");
goto error;
}
}
/* avp name is known ->set it and its type */
store_vals[1].val.str_val = dbp->sa; /*attr name*/
avp = search_first_avp( 0, avp_name, &i_s, 0);
for( ; avp; avp=search_first_avp( 0, avp_name, &i_s, avp))
{
/* don't insert avps which were loaded */
if (avp->flags&AVP_IS_IN_DB)
continue;
/* set type */
store_vals[3].val.int_val =
(avp->flags&AVP_NAME_STR?0:AVPOPS_DB_NAME_INT)|
(avp->flags&AVP_VAL_STR?0:AVPOPS_DB_VAL_INT);
/* set value */
int_str2db_val( i_s, &store_vals[2].val.str_val,
avp->flags&AVP_VAL_STR);
/* save avp */
if (db_store_avp( url, store_keys, store_vals,
keys_nr, &dbp->table)==0 )
{
avp->flags |= AVP_IS_IN_DB;
n++;
}
}
} else {
/* avp name is unknown -> go through all list */
avp_list = get_avp_list();
avp = *avp_list;
for ( ; avp ; avp=avp->next )
{
/* don't insert avps which were loaded */
if (avp->flags&AVP_IS_IN_DB)
continue;
/* set attribute name and type */
if ( (sn=get_avp_name(avp))==0 )
i_s.n = avp->id;
else
i_s.s = *sn;
int_str2db_val( i_s, &store_vals[1].val.str_val, AVP_NAME_STR);
store_vals[3].val.int_val =
(avp->flags&AVP_NAME_STR?0:AVPOPS_DB_NAME_INT)|
(avp->flags&AVP_VAL_STR?0:AVPOPS_DB_VAL_INT);
/* set avp value */
get_avp_val( avp, &i_s);
int_str2db_val( i_s, &store_vals[2].val.str_val,
avp->flags&AVP_VAL_STR);
/* save avp */
if (db_store_avp( url, store_keys, store_vals,
keys_nr, &dbp->table)==0)
{
avp->flags |= AVP_IS_IN_DB;
n++;
}
}
}
LM_DBG(" %d avps were stored\n",n);
return n==0?-1:1;
error:
return -1;
}
int ops_dbload_avps (struct sip_msg* msg, struct fis_param *sp,
struct db_param *dbp, struct db_url *url, int use_domain, str *prefix)
{
struct sip_uri uri;
db_res_t *res = NULL;
str uuid;
int i, n, sh_flg;
str *s0, *s1, *s2;
int avp_name;
int avp_type = 0;
pv_value_t xvalue;
s0 = s1 = s2 = NULL;
if (!((sp->opd&AVPOPS_VAL_PVAR)||(sp->opd&AVPOPS_VAL_STR))) {
LM_CRIT("invalid flag combination (%d/%d)\n", sp->opd, sp->ops);
goto error;
}
/* get uuid from avp */
if (sp->opd&AVPOPS_VAL_PVAR)
{
if(pv_get_spec_value(msg, &(sp->u.sval), &xvalue)!=0)
{
LM_CRIT("failed to get PVAR value (%d/%d)\n", sp->opd, sp->ops);
goto error;
}
if(xvalue.flags&(PV_VAL_NULL|PV_VAL_EMPTY))
{
LM_ERR("no value for first param\n");
goto error;
}
uuid = xvalue.rs;
} else {
uuid.s = sp->u.s.s;
uuid.len = sp->u.s.len;
}
if(sp->opd&AVPOPS_FLAG_UUID0)
{
s0 = &uuid;
} else {
/* parse uri */
if (parse_uri(uuid.s, uuid.len, &uri)<0)
{
LM_ERR("failed to parse uri\n");
goto error;
}
/* check uri */
if(!uri.user.s|| !uri.user.len|| !uri.host.len|| !uri.host.s)
{
LM_ERR("incomplet uri <%.*s>\n", uuid.len, uuid.s);
goto error;
}
if((sp->opd&AVPOPS_FLAG_URI0)||(sp->opd&AVPOPS_FLAG_USER0))
s1 = &uri.user;
if((sp->opd&AVPOPS_FLAG_URI0)||(sp->opd&AVPOPS_FLAG_DOMAIN0))
s2 = &uri.host;
}
/* is dynamic avp name ? */
if(dbp->a.type==AVPOPS_VAL_PVAR)
{
if(pv_has_dname(&(dbp->a.u.sval)))
{
if(pv_get_spec_name(msg, &(dbp->a.u.sval.pvp), &xvalue)!=0)
{
LM_CRIT("failed to get value for P2\n");
goto error;
}
if(xvalue.flags&(PV_VAL_NULL|PV_VAL_EMPTY))
{
LM_ERR("no value for p2\n");
goto error;
}
if(xvalue.flags&PV_VAL_STR)
{
if(xvalue.rs.len>=AVPOPS_ATTR_LEN)
{
LM_ERR("name too long [%d/%.*s...]\n",
xvalue.rs.len, 16, xvalue.rs.s);
goto error;
}
dbp->sa.s = avpops_attr_buf;
memcpy(dbp->sa.s, xvalue.rs.s, xvalue.rs.len);
dbp->sa.len = xvalue.rs.len;
dbp->sa.s[dbp->sa.len] = '\0';
} else {
LM_INFO("no string value for p2\n");
goto error;
}
}
}
/* do DB query */
res = db_load_avp( url, s0, s1,
((use_domain)||(sp->opd&AVPOPS_FLAG_DOMAIN0))?s2:0,
dbp->sa.s, &dbp->table, dbp->scheme);
/* res query ? */
if (res==0)
{
LM_ERR("db_load failed\n");
goto error;
}
sh_flg = (dbp->scheme)?dbp->scheme->db_flags:-1;
/* validate row */
avp_name = -1;
if(dbp->a.type==AVPOPS_VAL_PVAR)
{
if(pv_has_dname(&dbp->a.u.sval))
{
if(xvalue.flags&PV_TYPE_INT)
{
avp_name = xvalue.ri;
} else {
if (prefix)
{
if (xvalue.rs.len + prefix->len > AVPOPS_ATTR_LEN)
{
LM_ERR("name too long [%d/%.*s...]\n",
prefix->len + xvalue.rs.len, 16, prefix->s);
goto error;
}
memcpy(avpops_attr_buf, prefix->s, prefix->len);
memcpy(avpops_attr_buf + prefix->len, xvalue.rs.s,
xvalue.rs.len);
xvalue.rs.s = avpops_attr_buf;
xvalue.rs.len = prefix->len + xvalue.rs.len;
}
avp_name = get_avp_id(&xvalue.rs);
if (avp_name < 0) {
LM_ERR("cannot get avp id\n");
return -1;
}
}
} else {
avp_name = dbp->a.u.sval.pvp.pvn.u.isname.name.n;
avp_type = dbp->a.u.sval.pvp.pvn.u.isname.type;
}
}
/* process the results */
for( n=0,i=0 ; i<res->n ; i++)
{
if (dbrow2avp(&res->rows[i], dbp, avp_name, avp_type, sh_flg, prefix) < 0)
continue;
n++;
}
db_close_query( url, res );
LM_DBG("loaded avps = %d\n",n);
return n?1:-1;
error:
return -1;
}