/**
* enforce a response coming from a UE to contain the same Via headers sent in the corresponding request
* @param rpl - the SIP reply
* @param str1 - not used
* @param str2 - not used
* @returns true if ok, false if not or error
*/
int P_enforce_via_list(struct sip_msg *rpl,char *str1, char *str2)
{
static struct hdr_field * h = NULL;
str hdr;
cscf_del_all_headers(rpl, HDR_VIA_T);
struct sip_msg *req = cscf_get_request_from_reply(rpl);
if (!req){
LOG(L_ERR,"ERR:"M_NAME":P_enforce_via_list: No transactional request found.\n");
return CSCF_RETURN_FALSE;
}
h = cscf_get_next_via_hdr(req,0);
while (h)
{
hdr.len = h->body.len + via_hdr_s.len + via_hdr_e.len;
hdr.s = pkg_malloc(hdr.len);
if (!hdr.s)
{
LOG(L_ERR, "ERR:"M_NAME":P_enforce_via_list: cannot alloc bytes : %d", hdr.len);
}
hdr.len=0;
STR_APPEND(hdr, via_hdr_s);
STR_APPEND(hdr, h->body);
STR_APPEND(hdr, via_hdr_e);
cscf_add_header_first(rpl, &hdr, HDR_VIA_T);
h = cscf_get_next_via_hdr(req,h);
}
return CSCF_RETURN_TRUE;
}
/**
* Enforce a response coming from a UE to contain the same Record Route headers sent in the
* corresponding request.
* @param msg - the SIP reply
* @param str1 - not used
* @param str2 - not used
* @returns #CSCF_RETURN_TRUE on success, #CSCF_RETURN_ERROR on error
*/
int P_enforce_record_routes(struct sip_msg *msg,char *str1, char *str2)
{
str hdr = {0,0};
str rr_req = {0,0};
struct sip_msg *req = cscf_get_request_from_reply(msg);
LOG(L_INFO,"INF:"M_NAME":P_enforce_record_routes(): Enforcing RR in %d reply with the request ones\n",
msg->first_line.u.reply.statuscode);
if (!req){
LOG(L_ERR,"ERR:"M_NAME":P_enforce_record_routes(): No transactional request found.\n");
return CSCF_RETURN_ERROR;
}
if(!cscf_del_all_headers(msg, HDR_RECORDROUTE_T)){
LOG(L_ERR,"ERR:"M_NAME":P_enforce_record_routes(): error while deleting headers\n");
return CSCF_RETURN_ERROR;
}
rr_req = cscf_get_record_routes(req);
if(rr_req.len){
hdr.len = pcscf_record_route_mt.len + s_record_route_s.len + rr_req.len+s_record_route_e.len;
if(!(hdr.s = pkg_malloc(hdr.len))){
LOG(L_ERR,"ERR:"M_NAME":P_enforce_record_routes(): Unable to allocate memory for hdr\n");
goto out_of_memory;
}
hdr.len = 0;
STR_APPEND(hdr,pcscf_record_route_mt);
STR_APPEND(hdr,s_record_route_s);
STR_APPEND(hdr,rr_req);
STR_APPEND(hdr,s_record_route_e);
}else{
LOG(L_ERR,"ERR:"M_NAME":P_enforce_record_routes(): Unable to get record routes - the RR should not be empty...\n");
// still, let it continue, maybe it was empty on purpose
//return CSCF_RETURN_ERROR;
STR_PKG_DUP(hdr,pcscf_record_route_mt,"pkg");
}
if(!cscf_add_header_first(msg,&hdr,HDR_RECORDROUTE_T)){
LOG(L_ERR,"ERR:"M_NAME":P_enforce_record_routes(): Unable to add header\n");
if (hdr.s) pkg_free(hdr.s);
return CSCF_RETURN_FALSE;
}
return CSCF_RETURN_TRUE;
out_of_memory:
return CSCF_RETURN_ERROR;
}
/**
* Inserts the Route header containing the dialog routes to be enforced
* @param msg - the SIP message to add to
* @param str1 - the value to insert - !!! quoted if needed
* @param str2 - not used
* @returns #CSCF_RETURN_TRUE if ok, #CSCF_RETURN_FALSE if not or #CSCF_RETURN_BREAK on error
*/
int P_enforce_dialog_routes(struct sip_msg *msg,char *str1,char*str2)
{
int i;
str newuri={0,0};
p_dialog *d;
str call_id,host;
int port,transport;
str x;
enum p_dialog_direction dir;
dir = get_dialog_direction(str1);
if (!find_dialog_contact(msg,dir,&host,&port,&transport)){
LOG(L_ERR,"ERR:"M_NAME":P_enforce_dialog_routes(): Error retrieving %s contact\n",str1);
return CSCF_RETURN_BREAK;
}
call_id = cscf_get_call_id(msg,0);
if (!call_id.len)
return CSCF_RETURN_FALSE;
LOG(L_INFO,"DBG:"M_NAME":P_enforce_dialog_routes(%s): Call-ID <%.*s> %d://%.*s:%d\n",
str1,call_id.len,call_id.s,
transport,host.len,host.s,port);
d = get_p_dialog(call_id,host,port,transport,&dir);
if (!d)
d = get_p_dialog(call_id,host,port,transport,0);
if (!d){
LOG(L_ERR,"ERR:"M_NAME":P_enforce_dialog_routes: dialog does not exists!\n");
return CSCF_RETURN_FALSE;
}
if (!d->routes_cnt){
d_unlock(d->hash);
return CSCF_RETURN_TRUE;
}
x.len = route_s.len + route_e.len + (d->routes_cnt-1)*route_1.len;
for(i=0;i<d->routes_cnt;i++)
x.len+=d->routes[i].len;
x.s = pkg_malloc(x.len);
if (!x.s){
LOG(L_ERR, "ERR:"M_NAME":P_enforce_dialog_routes: Error allocating %d bytes\n",
x.len);
x.len=0;
d_unlock(d->hash);
return CSCF_RETURN_ERROR;
}
x.len=0;
STR_APPEND(x,route_s);
for(i=0;i<d->routes_cnt;i++){
if (i) STR_APPEND(x,route_1);
STR_APPEND(x,d->routes[i]);
}
STR_APPEND(x,route_e);
newuri.s = pkg_malloc(d->routes[0].len);
if (!newuri.s){
LOG(L_ERR, "ERR:"M_NAME":P_enforce_dialog_routes: Error allocating %d bytes\n",
d->routes[0].len);
d_unlock(d->hash);
return CSCF_RETURN_ERROR;
}
newuri.len = d->routes[0].len;
memcpy(newuri.s,d->routes[0].s,newuri.len);
if (msg->dst_uri.s) pkg_free(msg->dst_uri.s);
msg->dst_uri = newuri;
//LOG(L_ERR,"%.*s",x.len,x.s);
d_unlock(d->hash);
if (cscf_add_header_first(msg,&x,HDR_ROUTE_T)) {
if (cscf_del_all_headers(msg,HDR_ROUTE_T))
return CSCF_RETURN_TRUE;
else {
LOG(L_ERR,"ERR:"M_NAME":P_enforce_dialog_routes: new Route headers added, but failed to drop old ones.\n");
return CSCF_RETURN_ERROR;
}
}
else {
if (x.s) pkg_free(x.s);
return CSCF_RETURN_ERROR;
}
}
