sipwise
/
kamailio
mirror of https://github.com/sipwise/kamailio.git
1
0
Fork 0
Code Issues Releases Wiki Activity
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.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '4e6ea7e991'
${ noResults }
kamailio/modules/rls/resource_notify.c

1193 lines
31 KiB
Raw Blame History

/*
* rls module - resource list server
*
* Copyright (C) 2007 Voice Sistem S.R.L.
*
* 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
*
*/
#include <stdio.h>
#include <stdlib.h>
#include <libxml/parser.h>
#include "../../parser/parse_content.h"
#include "../../parser/parse_from.h"
#include "../../lib/kcore/cmpapi.h"
#include "../../hashes.h"
#include "../../trim.h"
#include "../pua/hash.h"
#include "rls.h"
#include "notify.h"
#include "resource_notify.h"
/* how to relate resource oriented dialogs to list_uri */
/* sol1: use the same callid in Subscribe requests
* sol2: include an extra header
* sol3: put the list_uri as the id of the record stored in
* pua and write a function to return that id
* winner: sol3
* */
static str su_200_rpl = str_init("OK");
static str pu_400_rpl = str_init("Bad Request");
static str pu_481_rpl = str_init("Call/Transaction Does Not Exist");
static str pu_489_rpl = str_init("Bad Event");
static str pu_500_rpl = str_init("Server Internal Error");
int subset = 0;
int parse_rlsubs_did(char* str_did, str* callid, str* from_tag, str* to_tag)
{
char* smc= NULL;
smc= strstr(str_did, RLS_DID_SEP);
if(smc== NULL)
{
LM_ERR("bad format for resource list Subscribe dialog"
" indentifier[rlsubs did]= %s\n", str_did);
return -1;
}
callid->s= str_did;
callid->len= smc- str_did;
from_tag->s= smc+ RLS_DID_SEP_LEN;
smc= strstr(from_tag->s, RLS_DID_SEP);
if(smc== NULL)
{
LM_ERR("bad format for resource list Subscribe dialog"
" indentifier(rlsubs did)= %s\n", str_did);
return -1;
}
from_tag->len= smc- from_tag->s;
to_tag->s= smc+ RLS_DID_SEP_LEN;
to_tag->len= strlen(str_did)- 2* RLS_DID_SEP_LEN- callid->len- from_tag->len;
return 0;
}
void get_dialog_from_did(char* did, subs_t **dialog, unsigned int *hash_code)
{
str callid, to_tag, from_tag;
subs_t* s;
*dialog= NULL;
/* search the subscription in rlsubs_table*/
if( parse_rlsubs_did(did, &callid, &from_tag, &to_tag)< 0)
{
LM_ERR("bad format for resource list Subscribe dialog "
"indentifier(rlsubs did)\n");
return;
}
if (dbmode == RLS_DB_ONLY)
{
*dialog = get_dialog_notify_rlsdb(callid,to_tag,from_tag);
if(*dialog==NULL)
{
LM_INFO("record not retrieved from db [rlsubs_did]= %s\n", did);
return;
}
}
else
{
*hash_code= core_hash(&callid, &to_tag, hash_size);
lock_get(&rls_table[*hash_code].lock);
s= pres_search_shtable(rls_table,callid,to_tag,from_tag,*hash_code);
if(s== NULL)
{
LM_INFO("record not found in hash_table [rlsubs_did]= %s\n",
did);
lock_release(&rls_table[*hash_code].lock);
return;
}
/* save dialog info */
*dialog= pres_copy_subs(s, PKG_MEM_TYPE);
if(*dialog== NULL)
{
LM_ERR("while copying subs_t structure\n");
lock_release(&rls_table[*hash_code].lock);
return;
}
}
if ((*dialog)->expires < (int)time(NULL))
(*dialog)->expires = 0;
else
(*dialog)->expires -= (int)time(NULL);
if (dbmode != RLS_DB_ONLY)
lock_release(&rls_table[*hash_code].lock);
}
int send_notify(xmlDocPtr * rlmi_doc, char * buf, int buf_len,
const str bstr, subs_t * dialog, unsigned int hash_code)
{
int result = 0;
str rlmi_cont= {0, 0}, multi_cont;
xmlDocDumpFormatMemory(*rlmi_doc,(xmlChar**)(void*)&rlmi_cont.s,
&rlmi_cont.len, 0);
multi_cont.s= buf;
multi_cont.len= buf_len;
result =agg_body_sendn_update(&dialog->pres_uri, bstr.s, &rlmi_cont,
(buf_len==0)?NULL:&multi_cont, dialog, hash_code);
xmlFree(rlmi_cont.s);
xmlFreeDoc(*rlmi_doc);
*rlmi_doc= NULL;
return result;
}
static void send_notifies(db1_res_t *result, int did_col, int resource_uri_col, int auth_state_col, int reason_col,
int pres_state_col, int content_type_col)
{
int i;
char* prev_did= NULL, * curr_did= NULL;
db_row_t *row;
db_val_t *row_vals;
char* resource_uri;
str pres_state = {0, 0};
xmlDocPtr rlmi_doc= NULL;
xmlNodePtr list_node= NULL, instance_node= NULL, resource_node;
unsigned int hash_code= 0;
int size= BUF_REALLOC_SIZE, buf_len= 0;
char* buf= NULL, *auth_state= NULL, *boundary_string= NULL;
str cid = {0,0};
str content_type= {0, 0};
int auth_state_flag;
int chunk_len=0;
str bstr= {0, 0};
subs_t* dialog= NULL;
int len_est = 0;
int resource_added = 0; /* Flag to indicate that we have added at least one resource */
/* generate the boundary string */
boundary_string= generate_string(BOUNDARY_STRING_LEN);
bstr.len= strlen(boundary_string);
bstr.s= (char*)pkg_malloc((bstr.len+ 1)* sizeof(char));
if(bstr.s== NULL)
{
ERR_MEM(PKG_MEM_STR);
}
memcpy(bstr.s, boundary_string, bstr.len);
bstr.s[bstr.len]= '\0';
/* Allocate an initial buffer for the multipart body.
* This buffer will be reallocated if neccessary */
buf= pkg_malloc(size* sizeof(char));
if(buf== NULL)
{
ERR_MEM(PKG_MEM_STR);
}
if (dbmode == RLS_DB_ONLY && rls_dbf.start_transaction)
{
if (rls_dbf.start_transaction(rls_db, DB_LOCKING_WRITE) < 0)
{
LM_ERR("in start_transaction\n");
goto error;
}
}
LM_DBG("found %d records with updated state\n", result->n);
for(i= 0; i< result->n; i++)
{
row = &result->rows[i];
row_vals = ROW_VALUES(row);
curr_did= (char*)row_vals[did_col].val.string_val;
resource_uri= (char*)row_vals[resource_uri_col].val.string_val;
auth_state_flag= row_vals[auth_state_col].val.int_val;
pres_state.s= (char*)row_vals[pres_state_col].val.string_val;
pres_state.len = strlen(pres_state.s);
trim(&pres_state);
/* If we have moved onto a new resource list Subscribe dialog indentifier,
send a NOTIFY for the previous ID and then drop the existing documents. */
if(prev_did!= NULL && strcmp(prev_did, curr_did))
{
if (send_notify(&rlmi_doc, buf, buf_len, bstr, dialog, hash_code))
{
LM_ERR("in send_notify\n");
goto error;
}
len_est = 0;
pkg_free(dialog);
dialog= NULL;
}
/*if first or different*/
if(prev_did==NULL || strcmp(prev_did, curr_did)!=0)
{
/* Work out a subscription from the did. */
get_dialog_from_did(curr_did, &dialog, &hash_code);
if(dialog== NULL)
{
prev_did = NULL;
LM_INFO("Dialog is NULL\n");
continue;
}
len_est = create_empty_rlmi_doc(&rlmi_doc, &list_node, &dialog->pres_uri, dialog->version, 0);
len_est += 2*strlen(boundary_string)+4+102+2+50+strlen(resource_uri)+20;
buf_len= 0;
resource_added = 0;
/* !!!! for now I will include the auth state without checking if
* it has changed - > in future chech if it works */
}
/* add a node in rlmi_doc and if any presence state registered add
* it in the buffer */
resource_node= xmlNewChild(list_node,NULL,BAD_CAST "resource", NULL);
if(resource_node== NULL)
{
LM_ERR("when adding resource child\n");
goto done;
}
xmlNewProp(resource_node, BAD_CAST "uri", BAD_CAST resource_uri);
len_est += strlen (resource_uri) + 35; /* <resource uri="[uri]"></resource>/r/n */
resource_added = 1;
/* there might be more records with the same uri- more instances-
* search and add them all */
while(1)
{
cid.s= NULL;
cid.len= 0;
auth_state= get_auth_string(auth_state_flag);
if(auth_state== NULL)
{
LM_ERR("bad authorization status flag\n");
goto error;
}
len_est += strlen(auth_state) + 38; /* <instance id="12345678" state="[auth_state]" />r/n */
if(auth_state_flag & ACTIVE_STATE)
{
cid.s= generate_cid(resource_uri, strlen(resource_uri));
cid.len = strlen(cid.s);
len_est += cid.len + 8; /* cid="[cid]" */
content_type.s = (char*)row_vals[content_type_col].val.string_val;
content_type.len = strlen(content_type.s);
chunk_len = 4 + bstr.len
+ 35
+ 16 + cid.len
+ 18 + content_type.len
+ 4 + pres_state.len + 8;
len_est += chunk_len;
}
else
if(auth_state_flag & TERMINATED_STATE)
{
len_est += strlen(row_vals[resource_uri_col].val.string_val) + 10; /* reason="[resaon]" */
}
if (rls_max_notify_body_len > 0 && len_est > rls_max_notify_body_len)
{
/* We have a limit on body length set, and we were about to exceed it */
if (resource_added == 1)
{
/* We added at least one resource. */
LM_DBG("timer_send_notify hit the size limit. len_est = %d\n", len_est);
if (send_notify(&rlmi_doc, buf, buf_len, bstr, dialog, hash_code))
{
LM_ERR("in send_notify\n");
goto error;
}
i --;
}
else
{
LM_DBG("timer_send_notify hit the size limit. NO RESOURCE ADDED len_est = %d\n", len_est);
}
len_est = 0;
pkg_free(dialog);
dialog= NULL;
curr_did=NULL;
break;
}
/* OK, we are happy this will fit */
instance_node= xmlNewChild(resource_node, NULL, BAD_CAST "instance", NULL);
if(instance_node== NULL)
{
LM_ERR("while adding instance child\n");
goto error;
}
/* Instance ID should be unique for each instance node
within a resource node. The same instance ID can be
used in different resource nodes. Instance ID needs
to remain the same for each resource instance in
future updates. We can just use a common string
here because you will only get multiple instances
for a resource when the back-end SUBSCRIBE is forked
and pua does not support this. If/when pua supports
forking of the SUBSCRIBEs it sends this will need to
be fixed properly. */
xmlNewProp(instance_node, BAD_CAST "id",
BAD_CAST instance_id);
if(auth_state_flag & ACTIVE_STATE)
{
xmlNewProp(instance_node, BAD_CAST "state", BAD_CAST auth_state);
}
else
if(auth_state_flag & TERMINATED_STATE)
{
xmlNewProp(instance_node, BAD_CAST "reason",
BAD_CAST row_vals[resource_uri_col].val.string_val);
}
xmlNewProp(instance_node, BAD_CAST "cid", BAD_CAST cid.s);
/* add in the multipart buffer */
if(cid.s)
{
while(buf_len + chunk_len >= size)
{
REALLOC_BUF
}
buf_len+= sprintf(buf+ buf_len, "--%.*s\r\n", bstr.len,
bstr.s);
buf_len+= sprintf(buf+ buf_len,
"Content-Transfer-Encoding: binary\r\n");
buf_len+= sprintf(buf+ buf_len, "Content-ID: <%.*s>\r\n",
cid.len, cid.s);
buf_len+= sprintf(buf+ buf_len, "Content-Type: %.*s\r\n\r\n",
content_type.len, content_type.s);
buf_len+= sprintf(buf+buf_len,"%.*s\r\n\r\n", pres_state.len,
pres_state.s);
}
i++;
if(i== result->n)
{
i--;
break;
}
row = &result->rows[i];
row_vals = ROW_VALUES(row);
if(strncmp(resource_uri, row_vals[resource_uri_col].val.string_val,
strlen(resource_uri))
|| strncmp(curr_did, row_vals[did_col].val.string_val,
strlen(curr_did)))
{
i--;
break;
}
resource_uri= (char*)row_vals[resource_uri_col].val.string_val;
auth_state_flag= row_vals[auth_state_col].val.int_val;
pres_state.s= (char*)row_vals[pres_state_col].val.string_val;
pres_state.len= strlen(pres_state.s);
trim(&pres_state);
}
prev_did= curr_did;
}
if(rlmi_doc)
{
LM_DBG("timer_send_notify at end len_est = %d resource_added = %d\n", len_est, resource_added);
if (resource_added == 1)
{
send_notify(&rlmi_doc, buf, buf_len, bstr, dialog, hash_code);
}
if(dialog)
{
pkg_free(dialog);
}
dialog= NULL;
}
done:
if (dbmode == RLS_DB_ONLY && rls_dbf.end_transaction)
{
if (rls_dbf.end_transaction(rls_db) < 0)
{
LM_ERR("in end_transaction\n");
goto error;
}
}
error:
if(bstr.s)
pkg_free(bstr.s);
if(buf)
pkg_free(buf);
if(dialog)
pkg_free(dialog);
if (dbmode == RLS_DB_ONLY && rls_dbf.abort_transaction)
{
if (rls_dbf.abort_transaction(rls_db) < 0)
LM_ERR("in abort_transaction\n");
}
return;
}
int parse_subs_state(str auth_state, str *reason, int *expires)
{
str str_exp;
char* smc= NULL;
int len, flag= -1;
if (strncmp(auth_state.s, "active", 6)== 0)
flag= ACTIVE_STATE;
if (strncmp(auth_state.s, "pending", 7)== 0)
flag= PENDING_STATE;
if (strncmp(auth_state.s, "terminated", 10)== 0)
{
smc= strchr(auth_state.s, ';');
if (smc== NULL)
{
LM_ERR("terminated state and no reason found");
return -1;
}
if (strncmp(smc+1, "reason=", 7))
{
LM_ERR("terminated state and no reason found");
return -1;
}
len= auth_state.len- 10- 1- 7;
reason->s = (char*) pkg_malloc(len* sizeof(char));
if (reason->s== NULL)
{
ERR_MEM(PKG_MEM_STR);
}
memcpy(reason->s, smc+ 8, len);
reason->len= len;
return TERMINATED_STATE;
}
if(flag> 0)
{
smc= strchr(auth_state.s, ';');
if(smc== NULL)
{
LM_ERR("active or pending state and no expires parameter found");
return -1;
}
if(strncmp(smc+1, "expires=", 8))
{
LM_ERR("active or pending state and no expires parameter found");
return -1;
}
str_exp.s= smc+ 9;
str_exp.len= auth_state.s+ auth_state.len- smc- 9;
if( str2int(&str_exp, (unsigned int*)expires)< 0)
{
LM_ERR("while getting int from str\n");
return -1;
}
return flag;
}
error:
if (reason->s) pkg_free(reason->s);
return -1;
}
int rls_handle_notify(struct sip_msg* msg, char* c1, char* c2)
{
struct to_body *pto, TO = {0}, *pfrom = NULL;
str body= {0, 0};
ua_pres_t dialog;
str* res_id= NULL;
db_key_t query_cols[8];
db_val_t query_vals[8];
int n_query_cols= 0;
str auth_state= {0, 0};
int found= 0;
str reason = {0, 0};
int auth_flag;
struct hdr_field* hdr= NULL;
int expires= -1;
str content_type= {0, 0};
int reply_code = 500;
str reply_str = pu_500_rpl;
LM_DBG("start\n");
/* extract the dialog information and check if an existing dialog*/
if( parse_headers(msg,HDR_EOH_F, 0)==-1 )
{
LM_ERR("parsing headers\n");
reply_code = 400;
reply_str = pu_400_rpl;
goto error;
}
if((!msg->event ) ||(msg->event->body.len<=0))
{
LM_ERR("Missing event header field value\n");
reply_code = 400;
reply_str = pu_400_rpl;
goto error;
}
if( msg->to==NULL || msg->to->body.s==NULL)
{
LM_ERR("cannot parse TO header\n");
reply_code = 400;
reply_str = pu_400_rpl;
goto error;
}
if(msg->to->parsed != NULL)
{
pto = (struct to_body*)msg->to->parsed;
LM_DBG("'To' header ALREADY PARSED: <%.*s>\n",
pto->uri.len, pto->uri.s );
}
else
{
parse_to(msg->to->body.s,msg->to->body.s + msg->to->body.len + 1, &TO);
if(TO.uri.len <= 0)
{
LM_ERR(" 'To' header NOT parsed\n");
reply_code = 400;
reply_str = pu_400_rpl;
goto error;
}
pto = &TO;
}
memset(&dialog, 0, sizeof(ua_pres_t));
dialog.watcher_uri= &pto->uri;
if (pto->tag_value.s==NULL || pto->tag_value.len==0 )
{
LM_ERR("to tag value not parsed\n");
reply_code = 400;
reply_str = pu_400_rpl;
goto error;
}
dialog.from_tag= pto->tag_value;
if( msg->callid==NULL || msg->callid->body.s==NULL)
{
LM_ERR("cannot parse callid header\n");
reply_code = 400;
reply_str = pu_400_rpl;
goto error;
}
dialog.call_id = msg->callid->body;
if (!msg->from || !msg->from->body.s)
{
LM_ERR("cannot find 'from' header!\n");
reply_code = 400;
reply_str = pu_400_rpl;
goto error;
}
if (msg->from->parsed == NULL)
{
LM_DBG("'From' header not parsed\n");
/* parsing from header */
if ( parse_from_header( msg )<0 )
{
LM_ERR("cannot parse From header\n");
reply_code = 400;
reply_str = pu_400_rpl;
goto error;
}
}
pfrom = (struct to_body*)msg->from->parsed;
dialog.pres_uri= &pfrom->uri;
if( pfrom->tag_value.s ==NULL || pfrom->tag_value.len == 0)
{
LM_ERR("no from tag value present\n");
reply_code = 400;
reply_str = pu_400_rpl;
goto error;
}
dialog.to_tag= pfrom->tag_value;
dialog.flag|= RLS_SUBSCRIBE;
dialog.event= get_event_flag(&msg->event->body);
if(dialog.event< 0)
{
LM_ERR("unrecognized event package\n");
reply_code = 489;
reply_str = pu_489_rpl;
goto error;
}
/* extract the subscription state */
hdr = msg->headers;
while (hdr!= NULL)
{
if(cmp_hdrname_strzn(&hdr->name, "Subscription-State", 18)==0)
{
found = 1;
break;
}
hdr = hdr->next;
}
if(found==0 )
{
LM_ERR("'Subscription-State' header not found\n");
goto error;
}
auth_state = hdr->body;
/* extract state and reason */
auth_flag= parse_subs_state(auth_state, &reason, &expires);
if(auth_flag< 0)
{
LM_ERR("while parsing 'Subscription-State' header\n");
goto error;
}
if(pua_get_record_id(&dialog, &res_id)< 0) /* verify if within a stored dialog */
{
LM_ERR("occured when trying to get record id\n");
goto error;
}
if(res_id==0)
{
LM_DBG("presence dialog record not found\n");
/* if it is a NOTIFY for a terminated SUBSCRIBE dialog in RLS, then
* the module might not have the dialog structure anymore
* - just send 200ok, it is harmless
*/
if(auth_flag==TERMINATED_STATE)
goto done;
LM_INFO("no presence dialog record for non-TERMINATED state uri pres_uri = %.*s watcher_uri = %.*s\n",
dialog.pres_uri->len, dialog.pres_uri->s, dialog.watcher_uri->len, dialog.watcher_uri->s);
reply_code = 481;
reply_str = pu_481_rpl;
goto error;
}
if(msg->content_type== NULL || msg->content_type->body.s== NULL)
{
LM_DBG("cannot find content type header header\n");
}
else
content_type= msg->content_type->body;
/*constructing the xml body*/
if(get_content_length(msg) == 0 )
{
goto done;
}
else
{
if(content_type.s== 0)
{
LM_ERR("content length != 0 and no content type header found\n");
goto error;
}
body.s=get_body(msg);
if (body.s== NULL)
{
LM_ERR("cannot extract body from msg\n");
goto error;
}
body.len = get_content_length( msg );
}
/* update in rlpres_table where rlsusb_did= res_id and resource_uri= from_uri*/
LM_DBG("body= %.*s\n", body.len, body.s);
query_cols[n_query_cols]= &str_rlsubs_did_col;
query_vals[n_query_cols].type = DB1_STR;
query_vals[n_query_cols].nul = 0;
query_vals[n_query_cols].val.str_val= *res_id;
n_query_cols++;
query_cols[n_query_cols]= &str_resource_uri_col;
query_vals[n_query_cols].type = DB1_STR;
query_vals[n_query_cols].nul = 0;
query_vals[n_query_cols].val.str_val= *dialog.pres_uri;
n_query_cols++;
query_cols[n_query_cols]= &str_updated_col;
query_vals[n_query_cols].type = DB1_INT;
query_vals[n_query_cols].nul = 0;
if (dbmode == RLS_DB_ONLY)
query_vals[n_query_cols].val.int_val=
core_hash(res_id, NULL, 0) %
(waitn_time * rls_notifier_poll_rate
* rls_notifier_processes);
else
query_vals[n_query_cols].val.int_val = UPDATED_TYPE;
n_query_cols++;
query_cols[n_query_cols]= &str_auth_state_col;
query_vals[n_query_cols].type = DB1_INT;
query_vals[n_query_cols].nul = 0;
query_vals[n_query_cols].val.int_val= auth_flag;
n_query_cols++;
query_cols[n_query_cols]= &str_reason_col;
query_vals[n_query_cols].type = DB1_STR;
query_vals[n_query_cols].nul = 0;
if(reason.len > 0)
{
query_vals[n_query_cols].val.str_val.s= reason.s;
query_vals[n_query_cols].val.str_val.len= reason.len;
}
else
{
query_vals[n_query_cols].val.str_val.s = "";
query_vals[n_query_cols].val.str_val.len = 0;
}
n_query_cols++;
query_cols[n_query_cols]= &str_content_type_col;
query_vals[n_query_cols].type = DB1_STR;
query_vals[n_query_cols].nul = 0;
query_vals[n_query_cols].val.str_val= content_type;
n_query_cols++;
query_cols[n_query_cols]= &str_presence_state_col;
query_vals[n_query_cols].type = DB1_STR;
query_vals[n_query_cols].nul = 0;
query_vals[n_query_cols].val.str_val= body;
n_query_cols++;
query_cols[n_query_cols]= &str_expires_col;
query_vals[n_query_cols].type = DB1_INT;
query_vals[n_query_cols].nul = 0;
query_vals[n_query_cols].val.int_val= expires+ (int)time(NULL);
n_query_cols++;
if (rlpres_dbf.use_table(rlpres_db, &rlpres_table) < 0)
{
LM_ERR("in use_table\n");
goto error;
}
if (dbmode == RLS_DB_ONLY && rlpres_dbf.start_transaction)
{
if (rlpres_dbf.start_transaction(rlpres_db, DB_LOCKING_WRITE) < 0)
{
LM_ERR("in start_transaction\n");
goto error;
}
}
if (rlpres_dbf.replace != NULL)
{
if(rlpres_dbf.replace(rlpres_db, query_cols, query_vals, n_query_cols,
2, 0)< 0)
{
LM_ERR("in sql replace\n");
goto error;
}
LM_DBG("Inserted/replace in database table new record\n");
}
else
{
if(rlpres_dbf.update(rlpres_db, query_cols, 0, query_vals, query_cols+2,
query_vals+2, 2, n_query_cols-2)< 0)
{
LM_ERR("in sql update\n");
goto error;
}
if (rlpres_dbf.affected_rows(rlpres_db) == 0)
{
if(rlpres_dbf.insert(rlpres_db, query_cols, query_vals, n_query_cols)< 0)
{
LM_ERR("in sql insert\n");
goto error;
}
LM_DBG("Inserted in database table new record\n");
}
}
if (dbmode == RLS_DB_ONLY && rlpres_dbf.end_transaction)
{
if (rlpres_dbf.end_transaction(rlpres_db) < 0)
{
LM_ERR("in end_transaction\n");
goto error;
}
}
LM_DBG("Updated rlpres_table\n");
/* reply 200OK */
done:
if(slb.freply(msg, 200, &su_200_rpl) < 0)
{
LM_ERR("while sending reply\n");
goto error;
}
if(res_id!=NULL)
{
pkg_free(res_id->s);
pkg_free(res_id);
}
if (reason.s) pkg_free(reason.s);
free_to_params(&TO);
return 1;
error:
if(slb.freply(msg, reply_code, &reply_str) < 0)
{
LM_ERR("failed sending reply\n");
}
if(res_id!=NULL)
{
pkg_free(res_id->s);
pkg_free(res_id);
}
if (reason.s) pkg_free(reason.s);
free_to_params(&TO);
if (dbmode == RLS_DB_ONLY && rlpres_dbf.abort_transaction)
{
if (rlpres_dbf.abort_transaction(rlpres_db) < 0)
LM_ERR("in abort_transaction\n");
}
return -1;
}
#define EXTRACT_STRING(strng, chars)\
do {\
strng.s = (char *) chars;\
strng.len = strlen(strng.s);\
} while(0);
static void timer_send_full_state_notifies(int round)
{
db_key_t query_cols[1], result_cols[22], update_cols[1];
db_val_t query_vals[1], update_vals[1], *values;
db_row_t *rows;
db1_res_t *result = NULL;
int n_result_cols = 0, i;
int pres_uri_col, tuser_col, tdomain_col, fuser_col, fdomain_col;
int wuser_col, wdomain_col, callid_col, to_tag_col, from_tag_col;
int sockinfo_col, lcontact_col, contact_col, rroute_col, event_id_col;
int reason_col, event_col, lcseq_col, rcseq_col, status_col;
int version_col, expires_col;
subs_t sub;
str ev_sname;
event_t parsed_event;
xmlDocPtr doc = NULL;
xmlNodePtr service_node = NULL;
int now = (int)time(NULL);
db_query_f query_fn = rls_dbf.query_lock ? rls_dbf.query_lock : rls_dbf.query;
query_cols[0] = &str_updated_col;
query_vals[0].type = DB1_INT;
query_vals[0].nul = 0;
query_vals[0].val.int_val = round;
result_cols[pres_uri_col = n_result_cols++] = &str_presentity_uri_col;
result_cols[tuser_col = n_result_cols++] = &str_to_user_col;
result_cols[tdomain_col = n_result_cols++] = &str_to_domain_col;
result_cols[fuser_col = n_result_cols++] = &str_from_user_col;
result_cols[fdomain_col = n_result_cols++] = &str_from_domain_col;
result_cols[wuser_col = n_result_cols++] = &str_watcher_username_col;
result_cols[wdomain_col = n_result_cols++] = &str_watcher_domain_col;
result_cols[callid_col = n_result_cols++] = &str_callid_col;
result_cols[to_tag_col = n_result_cols++] = &str_to_tag_col;
result_cols[from_tag_col = n_result_cols++] = &str_from_tag_col;
result_cols[sockinfo_col = n_result_cols++] = &str_socket_info_col;
result_cols[lcontact_col = n_result_cols++] = &str_local_contact_col;
result_cols[contact_col = n_result_cols++] = &str_contact_col;
result_cols[rroute_col = n_result_cols++] = &str_record_route_col;
result_cols[event_id_col = n_result_cols++] = &str_event_id_col;
result_cols[reason_col = n_result_cols++] = &str_reason_col;
result_cols[event_col = n_result_cols++] = &str_event_col;
result_cols[lcseq_col = n_result_cols++] = &str_local_cseq_col;
result_cols[rcseq_col = n_result_cols++] = &str_remote_cseq_col;
result_cols[status_col = n_result_cols++] = &str_status_col;
result_cols[version_col = n_result_cols++] = &str_version_col;
result_cols[expires_col = n_result_cols++] = &str_expires_col;
update_cols[0] = &str_updated_col;
update_vals[0].type = DB1_INT;
update_vals[0].nul = 0;
update_vals[0].val.int_val = NO_UPDATE_TYPE;
if (rls_dbf.use_table(rls_db, &rlsubs_table) < 0)
{
LM_ERR("use table failed\n");
goto done;
}
if (dbmode == RLS_DB_ONLY && rls_dbf.start_transaction)
{
if (rls_dbf.start_transaction(rls_db, DB_LOCKING_WRITE) < 0)
{
LM_ERR("in start_transaction\n");
goto done;
}
}
/* Step 1: Find rls_watchers that require full-state notification */
if (query_fn(rls_db, query_cols, 0, query_vals, result_cols,
1, n_result_cols, 0, &result) < 0)
{
LM_ERR("in sql query\n");
goto done;
}
if(result== NULL || result->n<= 0)
goto done;
/* Step 2: Reset the update flag so we do not full-state notify
these watchers again */
if(rls_dbf.update(rls_db, query_cols, 0, query_vals, update_cols,
update_vals, 1, 1)< 0)
{
LM_ERR("in sql update\n");
goto done;
}
if (dbmode == RLS_DB_ONLY && rls_dbf.end_transaction)
{
if (rls_dbf.end_transaction(rls_db) < 0)
{
LM_ERR("in end_transaction\n");
goto done;
}
}
/* Step 3: Full-state notify each watcher we found */
rows = RES_ROWS(result);
for (i = 0; i < RES_ROW_N(result); i++)
{
memset(&sub, 0, sizeof(subs_t));
values = ROW_VALUES(&rows[i]);
EXTRACT_STRING(sub.pres_uri, VAL_STRING(&values[pres_uri_col]));
EXTRACT_STRING(sub.to_user, VAL_STRING(&values[tuser_col]));
EXTRACT_STRING(sub.to_domain, VAL_STRING(&values[tdomain_col]));
EXTRACT_STRING(sub.from_user, VAL_STRING(&values[fuser_col]));
EXTRACT_STRING(sub.from_domain, VAL_STRING(&values[fdomain_col]));
EXTRACT_STRING(sub.watcher_user, VAL_STRING(&values[wuser_col]));
EXTRACT_STRING(sub.watcher_domain, VAL_STRING(&values[wdomain_col]));
EXTRACT_STRING(sub.callid, VAL_STRING(&values[callid_col]));
EXTRACT_STRING(sub.to_tag, VAL_STRING(&values[to_tag_col]));
EXTRACT_STRING(sub.from_tag, VAL_STRING(&values[from_tag_col]));
EXTRACT_STRING(sub.sockinfo_str, VAL_STRING(&values[sockinfo_col]));
EXTRACT_STRING(sub.local_contact, VAL_STRING(&values[lcontact_col]));
EXTRACT_STRING(sub.contact, VAL_STRING(&values[contact_col]));
EXTRACT_STRING(sub.record_route, VAL_STRING(&values[rroute_col]));
EXTRACT_STRING(sub.event_id, VAL_STRING(&values[event_id_col]));
EXTRACT_STRING(sub.reason, VAL_STRING(&values[reason_col]));
EXTRACT_STRING(ev_sname, VAL_STRING(&values[event_col]));
sub.event = pres_contains_event(&ev_sname, &parsed_event);
if (sub.event == NULL)
{
LM_ERR("event not found and set to NULL\n");
goto done;
}
sub.local_cseq = VAL_INT(&values[lcseq_col]);
sub.remote_cseq = VAL_INT(&values[rcseq_col]);
sub.status = VAL_INT(&values[status_col]);
sub.version = VAL_INT(&values[version_col]);
if (VAL_INT(&values[expires_col]) > now + rls_expires_offset)
{
sub.expires = VAL_INT(&values[expires_col]) - now;
if (rls_get_service_list(&sub.pres_uri, &sub.watcher_user,
&sub.watcher_domain, &service_node, &doc) < 0)
{
LM_ERR("failed getting resource list\n");
goto done;
}
if (doc == NULL)
{
LM_WARN("no document returned for uri <%.*s>\n",
sub.pres_uri.len, sub.pres_uri.s);
goto done;
}
if (send_full_notify(&sub, service_node, &sub.pres_uri, 0) < 0)
{
LM_ERR("failed sending full state notify\n");
goto done;
}
xmlFreeDoc(doc);
doc = NULL;
}
else
{
sub.expires = 0;
rls_send_notify(&sub, NULL, NULL, NULL);
delete_rlsdb(&sub.callid, &sub.to_tag, &sub.from_tag);
}
}
done:
if (result != NULL)
rls_dbf.free_result(rls_db, result);
if (doc != NULL)
xmlFreeDoc(doc);
if (dbmode == RLS_DB_ONLY && rls_dbf.abort_transaction)
{
if (rls_dbf.abort_transaction(rls_db) < 0)
LM_ERR("in abort_transaction\n");
}
}
static void timer_send_update_notifies(int round)
{
db_key_t query_cols[1], update_cols[1], result_cols[6];
db_val_t query_vals[1], update_vals[1];
int did_col, resource_uri_col, auth_state_col, reason_col,
pres_state_col, content_type_col;
int n_result_cols= 0;
db1_res_t *result= NULL;
db_query_f query_fn = rlpres_dbf.query_lock ? rlpres_dbf.query_lock : rlpres_dbf.query;
query_cols[0]= &str_updated_col;
query_vals[0].type = DB1_INT;
query_vals[0].nul = 0;
query_vals[0].val.int_val= round;
result_cols[did_col= n_result_cols++]= &str_rlsubs_did_col;
result_cols[resource_uri_col= n_result_cols++]= &str_resource_uri_col;
result_cols[auth_state_col= n_result_cols++]= &str_auth_state_col;
result_cols[content_type_col= n_result_cols++]= &str_content_type_col;
result_cols[reason_col= n_result_cols++]= &str_reason_col;
result_cols[pres_state_col= n_result_cols++]= &str_presence_state_col;
update_cols[0]= &str_updated_col;
update_vals[0].type = DB1_INT;
update_vals[0].nul = 0;
update_vals[0].val.int_val= NO_UPDATE_TYPE;
/* query in alphabetical order after rlsusbs_did
* (resource list Subscribe dialog indentifier)*/
if (rlpres_dbf.use_table(rlpres_db, &rlpres_table) < 0)
{
LM_ERR("in use_table\n");
goto done;
}
if (dbmode == RLS_DB_ONLY && rlpres_dbf.start_transaction)
{
if (rlpres_dbf.start_transaction(rlpres_db, DB_LOCKING_WRITE) < 0)
{
LM_ERR("in start_transaction\n");
goto done;
}
}
if(query_fn(rlpres_db, query_cols, 0, query_vals, result_cols,
1, n_result_cols, &str_rlsubs_did_col, &result)< 0)
{
LM_ERR("in sql query\n");
goto done;
}
if(result == NULL || result->n <= 0)
goto done;
if(rlpres_dbf.update(rlpres_db, query_cols, 0, query_vals, update_cols,
update_vals, 1, 1)< 0)
{
LM_ERR("in sql update\n");
goto done;
}
if (dbmode == RLS_DB_ONLY && rlpres_dbf.end_transaction)
{
if (rlpres_dbf.end_transaction(rlpres_db) < 0)
{
LM_ERR("in end_transaction\n");
goto done;
}
}
send_notifies(result, did_col, resource_uri_col, auth_state_col, reason_col,
pres_state_col, content_type_col);
done:
if(result)
rlpres_dbf.free_result(rlpres_db, result);
if (dbmode == RLS_DB_ONLY && rls_dbf.abort_transaction)
{
if (rlpres_dbf.abort_transaction(rlpres_db) < 0)
LM_ERR("in abort_transaction\n");
}
}
void timer_send_notify(unsigned int ticks,void *param)
{
if (dbmode == RLS_DB_ONLY)
{
int process_num = *((int *) param);
int round = subset + (waitn_time * rls_notifier_poll_rate * process_num);
if (++subset > (waitn_time * rls_notifier_poll_rate) - 1) subset = 0;
timer_send_full_state_notifies(round);
timer_send_update_notifies(round);
}
else
timer_send_update_notifies(UPDATED_TYPE);
}
/* function to periodicaly clean the rls_presentity table */
void rls_presentity_clean(unsigned int ticks,void *param)
{
db_key_t query_cols[1];
db_op_t query_ops[1];
db_val_t query_vals[1];
query_cols[0]= &str_expires_col;
query_ops[0]= OP_LT;
query_vals[0].nul= 0;
query_vals[0].type= DB1_INT;
query_vals[0].val.int_val= (int)time(NULL) - rls_expires_offset;
if (rlpres_dbf.use_table(rlpres_db, &rlpres_table) < 0)
{
LM_ERR("in use_table\n");
return ;
}
if(rlpres_dbf.delete(rlpres_db, query_cols, query_ops, query_vals, 1)< 0)
{
LM_ERR("in sql delete\n");
return ;
}
}
Reference in new issue View Git Blame Copy Permalink