modules_k/siputils/checks.c

Go to the documentation of this file.

/*
 * $Id$
 *
 * Various URI checks and URI manipulation
 *
 * Copyright (C) 2001-2003 FhG Fokus
 *
 * Copyright (C) 2004-2010 Juha Heinanen
 *
 * This file is part of SIP-router, a free SIP server.
 *
 * SIP-router 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
 *
 * SIP-router 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 * History:
 * --------
 * 2003-02-26: Created by janakj
 * 2004-03-20: has_totag introduced (jiri)
 * 2004-04-14: uri_param and add_uri_param introduced (jih)
 */

#include <string.h>
#include "../../str.h"
#include "../../dprint.h"               /* Debugging */
#include "../../mem/mem.h"
#include "../../parser/digest/digest.h" /* get_authorized_cred */
#include "../../parser/parse_from.h"
#include "../../parser/parse_uri.h"
#include "../../parser/parse_param.h"
#include "../../ut.h"                   /* Handy utilities */
#include "../../lib/srdb1/db.h"                /* Database API */
#include "../../dset.h"
#include "../../pvar.h"
#include "../../lvalue.h"
#include "../../sr_module.h"
#include "checks.h"

int w_is_request(struct sip_msg* msg, char *foo, char *bar)
{
        if(msg==NULL)
                return -1;

        if(msg->first_line.type == SIP_REQUEST)
                return 1;

        return -1;
}

int w_is_reply(struct sip_msg* msg, char *foo, char *bar)
{
        if(msg==NULL)
                return -1;

        if(msg->first_line.type == SIP_REPLY)
                return 1;

        return -1;
}

/*
 * Checks if From includes a To-tag -- good to identify
 * if a request creates a new dialog
 */
int has_totag(struct sip_msg* _m, char* _foo, char* _bar)
{
        str tag;

        if (!_m->to && parse_headers(_m, HDR_TO_F,0)==-1) {
                LM_ERR("To parsing failed\n");
                return -1;
        }
        if (!_m->to) {
                LM_ERR("no To\n");
                return -1;
        }
        tag=get_to(_m)->tag_value;
        if (tag.s==0 || tag.len==0) {
                LM_DBG("no totag\n");
                return -1;
        }
        LM_DBG("totag found\n");
        return 1;
}


/*
 * Check if the username matches the username in credentials
 */
int is_user(struct sip_msg* _m, char* _user, char* _str2)
{
        str* s;
        struct hdr_field* h;
        auth_body_t* c;

        s = (str*)_user;

        get_authorized_cred(_m->authorization, &h);
        if (!h) {
                get_authorized_cred(_m->proxy_auth, &h);
                if (!h) {
                        LM_ERR("no authorized credentials found (error in scripts)\n");
                        LM_ERR("Call {www,proxy}_authorize before calling is_user function !\n");
                        return -1;
                }
        }

        c = (auth_body_t*)(h->parsed);

        if (!c->digest.username.user.len) {
                LM_DBG("username not found in credentials\n");
                return -1;
        }

        if (s->len != c->digest.username.user.len) {
                LM_DBG("username length does not match\n");
                return -1;
        }

        if (!memcmp(s->s, c->digest.username.user.s, s->len)) {
                LM_DBG("username matches\n");
                return 1;
        } else {
                LM_DBG("username differs\n");
                return -1;
        }
}


/*
 * Find if Request URI has a given parameter with no value
 */
int uri_param_1(struct sip_msg* _msg, char* _param, char* _str2)
{
        return uri_param_2(_msg, _param, (char*)0);
}


/*
 * Find if Request URI has a given parameter with matching value
 */
int uri_param_2(struct sip_msg* _msg, char* _param, char* _value)
{
        str *param, *value, t;

        param_hooks_t hooks;
        param_t* params;

        param = (str*)_param;
        value = (str*)_value;

        if (parse_sip_msg_uri(_msg) < 0) {
                LM_ERR("ruri parsing failed\n");
                return -1;
        }

        t = _msg->parsed_uri.params;

        if (parse_params(&t, CLASS_ANY, &hooks, &params) < 0) {
                LM_ERR("ruri parameter parsing failed\n");
                return -1;
        }

        while (params) {
                if ((params->name.len == param->len) &&
                    (strncmp(params->name.s, param->s, param->len) == 0)) {
                        if (value) {
                                if ((value->len == params->body.len) &&
                                    strncmp(value->s, params->body.s, value->len) == 0) {
                                        goto ok;
                                } else {
                                        goto nok;
                                }
                        } else {
                                if (params->body.len > 0) {
                                        goto nok;
                                } else {
                                        goto ok;
                                }
                        }
                } else {
                        params = params->next;
                }
        }
        
nok:
        free_params(params);
        return -1;

ok:
        free_params(params);
        return 1;
}



/*
 * Adds a new parameter to Request URI
 */
int add_uri_param(struct sip_msg* _msg, char* _param, char* _s2)
{
        str *param, *cur_uri, new_uri;
        struct sip_uri *parsed_uri;
        char *at;

        param = (str*)_param;

        if (param->len == 0) {
                return 1;
        }

        if (parse_sip_msg_uri(_msg) < 0) {
                LM_ERR("ruri parsing failed\n");
                return -1;
        }

        parsed_uri = &(_msg->parsed_uri);

        /* if current ruri has no headers, pad param at the end */
        if (parsed_uri->headers.len == 0) {
                cur_uri =  GET_RURI(_msg);
                new_uri.len = cur_uri->len + param->len + 1;
                if (new_uri.len > MAX_URI_SIZE) {
                        LM_ERR("new ruri too long\n");
                        return -1;
                }
                new_uri.s = pkg_malloc(new_uri.len);
                if (new_uri.s == 0) {
                        LM_ERR("add_uri_param(): Memory allocation failure\n");
                        return -1;
                }
                memcpy(new_uri.s, cur_uri->s, cur_uri->len);
                *(new_uri.s + cur_uri->len) = ';';
                memcpy(new_uri.s + cur_uri->len + 1, param->s, param->len);
                if (rewrite_uri(_msg, &new_uri ) == 1) {
                        goto ok;
                } else {
                        goto nok;
                }
        }

        /* otherwise take the long path */
        new_uri.len = 4 +
                (parsed_uri->user.len ? parsed_uri->user.len + 1 : 0) +
                (parsed_uri->passwd.len ? parsed_uri->passwd.len + 1 : 0) +
                parsed_uri->host.len +
                (parsed_uri->port.len ? parsed_uri->port.len + 1 : 0) +
                parsed_uri->params.len + param->len + 1 +
                parsed_uri->headers.len + 1;
        if (new_uri.len > MAX_URI_SIZE) {
                LM_ERR("new ruri too long\n");
                return -1;
        }

        new_uri.s = pkg_malloc(new_uri.len);
        if (new_uri.s == 0) {
                LM_ERR("no more pkg memory\n");
                return -1;
        }

        at = new_uri.s;
        memcpy(at, "sip:", 4);
        at = at + 4;
        if (parsed_uri->user.len) {
                memcpy(at, parsed_uri->user.s, parsed_uri->user.len);
                if (parsed_uri->passwd.len) {
                        *at = ':';
                        at = at + 1;
                        memcpy(at, parsed_uri->passwd.s, parsed_uri->passwd.len);
                        at = at + parsed_uri->passwd.len;
                };
                *at = '@';
                at = at + 1;
        }
        memcpy(at, parsed_uri->host.s, parsed_uri->host.len);
        at = at + parsed_uri->host.len;
        if (parsed_uri->port.len) {
                *at = ':';
                at = at + 1;
                memcpy(at, parsed_uri->port.s, parsed_uri->port.len);
                at = at + parsed_uri->port.len;
        }
        memcpy(at, parsed_uri->params.s, parsed_uri->params.len);
        at = at + parsed_uri->params.len;
        *at = ';';
        at = at + 1;
        memcpy(at, param->s, param->len);
        at = at + param->len;
        *at = '?';
        at = at + 1;
        memcpy(at, parsed_uri->headers.s, parsed_uri->headers.len);

        if (rewrite_uri(_msg, &new_uri) == 1) {
                goto ok;
        }

nok:
        pkg_free(new_uri.s);
        return -1;

ok:
        pkg_free(new_uri.s);
        return 1;
}


/*
 * Converts URI, if it is tel URI, to SIP URI.  Returns 1, if
 * conversion succeeded or if no conversion was needed, i.e., URI was not
 * tel URI.  Returns -1, if conversion failed.  Takes SIP URI hostpart from
 * second parameter and (if needed) writes the result to third paramater.
 */
int tel2sip(struct sip_msg* _msg, char* _uri, char* _hostpart, char* _res)
{
    str uri, hostpart, tel_uri, sip_uri;
    char *at;
    int i, j, in_tel_parameters = 0;
    pv_spec_t *res;
    pv_value_t res_val;

    /* get parameters */
    if (get_str_fparam(&uri, _msg, (fparam_t*)_uri) < 0) {
        LM_ERR("failed to get uri value\n");
    }
    if (get_str_fparam(&hostpart, _msg, (fparam_t*)_hostpart) < 0) {
        LM_ERR("failed to get hostpart value\n");
    }
    res = (pv_spec_t *)_res;

    /* check if anything needs to be done */
    if (uri.len < 4) return 1;
    if (strncasecmp(uri.s, "tel:", 4) != 0) return 1;
    
    /* reserve memory for clean tel uri */
    tel_uri.s = pkg_malloc(uri.len);
    if (tel_uri.s == 0) {
        LM_ERR("no more pkg memory\n");
        return -1;
    }
        
    /* Remove visual separators before converting to SIP URI. Don't remove 
       visual separators in TEL URI parameters (after the first ";") */
    for (i=0, j=0; i < uri.len; i++) {
        if (in_tel_parameters == 0) {
            if (uri.s[i] == ';')
                in_tel_parameters = 1;
        }
        if (in_tel_parameters == 0) {
            if ((uri.s[i] != '-') && (uri.s[i] != '.') && 
                (uri.s[i] != '(') && (uri.s[i] != ')'))
                tel_uri.s[j++] = tolower(uri.s[i]);
        } else {
            tel_uri.s[j++] = tolower(uri.s[i]);
        }
    }
    tel_uri.s[j] = '\0';
    tel_uri.len = strlen(tel_uri.s);

    /* reserve memory for resulting sip uri */
    sip_uri.len = 4 + tel_uri.len - 4 + 1 + hostpart.len + 1 + 10;
    sip_uri.s = pkg_malloc(sip_uri.len);
    if (sip_uri.s == 0) {
        LM_ERR("no more pkg memory\n");
        pkg_free(tel_uri.s);
        return -1;
    }

    /* create resulting sip uri */
    at = sip_uri.s;
    append_str(at, "sip:", 4);
    append_str(at, tel_uri.s + 4, tel_uri.len - 4);
    append_chr(at, '@');
    append_str(at, hostpart.s, hostpart.len);
    append_chr(at, ';');
    append_str(at, "user=phone", 10);

    /* tel_uri is not needed anymore */
    pkg_free(tel_uri.s);

    /* set result pv value and write sip uri to result pv */
    res_val.rs = sip_uri;
    res_val.flags = PV_VAL_STR;
    if (res->setf(_msg, &res->pvp, (int)EQ_T, &res_val) != 0) {
        LM_ERR("failed to set result pvar\n");
        pkg_free(sip_uri.s);
        return -1;
    }

    /* free allocated pkg memory and return */
    pkg_free(sip_uri.s);
    return 1;
}


/*
 * Check if parameter is an e164 number.
 */
static inline int e164_check(str* _user)
{
    int i;
    char c;
    
    if ((_user->len > 2) && (_user->len < 17) && ((_user->s)[0] == '+')) {
        for (i = 1; i <= _user->len; i++) {
            c = (_user->s)[i];
            if (c < '0' && c > '9') return -1;
        }
        return 1;
    }
    return -1;
}


/*
 * Check if user part of URI in pseudo variable is an e164 number
 */
int is_e164(struct sip_msg* _m, char* _sp, char* _s2)
{
    pv_spec_t *sp;
    pv_value_t pv_val;

    sp = (pv_spec_t *)_sp;

    if (sp && (pv_get_spec_value(_m, sp, &pv_val) == 0)) {
        if (pv_val.flags & PV_VAL_STR) {
            if (pv_val.rs.len == 0 || pv_val.rs.s == NULL) {
                LM_DBG("missing argument\n");
                return -1;
            }
            return e164_check(&(pv_val.rs));
        } else {
            LM_ERR("pseudo variable value is not string\n");
            return -1;
        }
    } else {
        LM_ERR("failed to get pseudo variable value\n");
        return -1;
    }
}


/*
 * Check if user part of URI in pseudo variable is an e164 number
 */
int w_is_uri_user_e164(struct sip_msg* _m, char* _sp, char* _s2)
{
    pv_spec_t *sp;
    pv_value_t pv_val;

    sp = (pv_spec_t *)_sp;

    if (sp && (pv_get_spec_value(_m, sp, &pv_val) == 0)) {
        if (pv_val.flags & PV_VAL_STR) {
            if (pv_val.rs.len == 0 || pv_val.rs.s == NULL) {
                LM_DBG("missing uri\n");
                return -1;
            }
            return is_uri_user_e164(&pv_val.rs);
        } else {
            LM_ERR("pseudo variable value is not string\n");
            return -1;
        }
    } else {
        LM_ERR("failed to get pseudo variable value\n");
        return -1;
    }
}


int is_uri_user_e164(str *uri)
{
    char *chr;
    str user;

    chr = memchr(uri->s, ':', uri->len);
    if (chr == NULL) {
        LM_ERR("parsing URI failed\n");
        return -1;
    };
    user.s = chr + 1;
    chr = memchr(user.s, '@', uri->len - (user.s - uri->s));
    if (chr == NULL) return -1;
    user.len = chr - user.s;

    return e164_check(&user);
}

/*
 * Set userpart of URI
 */
int set_uri_user(struct sip_msg* _m, char* _uri, char* _value)
{
    pv_spec_t *uri_pv, *value_pv;
    pv_value_t uri_val, value_val, res_val;
    str uri, value;
    char *at, *colon, *c;
    char new_uri[MAX_URI_SIZE + 1];

    uri_pv = (pv_spec_t *)_uri;
    if (uri_pv && (pv_get_spec_value(_m, uri_pv, &uri_val) == 0)) {
        if (uri_val.flags & PV_VAL_STR) {
            if (uri_val.rs.len == 0 || uri_val.rs.s == NULL) {
                LM_ERR("missing uri value\n");
                return -1;
            }
        } else {
            LM_ERR("uri value is not string\n");
            return -1;
        }
    } else {
        LM_ERR("failed to get uri value\n");
        return -1;
    }
    uri = uri_val.rs;

    value_pv = (pv_spec_t *)_value;
    if (value_pv && (pv_get_spec_value(_m, value_pv, &value_val) == 0)) {
        if (value_val.flags & PV_VAL_STR) {
            if (value_val.rs.s == NULL) {
                LM_ERR("missing uriuser value\n");
                return -1;
            }
        } else {
            LM_ERR("uriuser value is not string\n");
            return -1;
        }
    } else {
        LM_ERR("failed to get uriuser value\n");
        return -1;
    }
    value = value_val.rs;

    colon = strchr(uri.s, ':');
    if (colon == NULL) {
        LM_ERR("uri does not contain ':' character\n");
        return -1;
    }
    at = strchr(uri.s, '@');
    c = &(new_uri[0]);
    if (at == NULL) {
        if (value.len == 0) return 1;
        if (uri.len + value.len > MAX_URI_SIZE) {
            LM_ERR("resulting uri would be too large\n");
            return -1;
        }
        append_str(c, uri.s, colon - uri.s + 1);
        append_str(c, value.s, value.len);
        append_chr(c, '@');
        append_str(c, colon + 1, uri.len - (colon - uri.s + 1));
        res_val.rs.len = uri.len + value.len + 1;
    } else {
        if (value.len == 0) {
            append_str(c, uri.s, colon - uri.s + 1);
            append_str(c, at + 1, uri.len - (at - uri.s + 1));
            res_val.rs.len = uri.len - (at - colon);
        } else {
            if (uri.len + value.len - (at - colon - 1) > MAX_URI_SIZE) {
                LM_ERR("resulting uri would be too large\n");
                return -1;
            }
            append_str(c, uri.s, colon - uri.s + 1);
            append_str(c, value.s, value.len);
            append_str(c, at, uri.len - (at - uri.s));
            res_val.rs.len = uri.len + value.len - (at - colon - 1);
        }
    }

    res_val.rs.s = &(new_uri[0]);
    LM_DBG("resulting uri: %.*s\n", res_val.rs.len, res_val.rs.s);
    res_val.flags = PV_VAL_STR;
    uri_pv->setf(_m, &uri_pv->pvp, (int)EQ_T, &res_val);

    return 1;
}

/*
 * Set hostpart of URI
 */
int set_uri_host(struct sip_msg* _m, char* _uri, char* _value)
{
    pv_spec_t *uri_pv, *value_pv;
    pv_value_t uri_val, value_val, res_val;
    str uri, value;
    char *at, *colon, *c, *next;
    unsigned int host_len;
    char new_uri[MAX_URI_SIZE + 1];

    uri_pv = (pv_spec_t *)_uri;
    if (uri_pv && (pv_get_spec_value(_m, uri_pv, &uri_val) == 0)) {
        if (uri_val.flags & PV_VAL_STR) {
            if (uri_val.rs.len == 0 || uri_val.rs.s == NULL) {
                LM_ERR("missing uri value\n");
                return -1;
            }
        } else {
            LM_ERR("uri value is not string\n");
            return -1;
        }
    } else {
        LM_ERR("failed to get uri value\n");
        return -1;
    }
    uri = uri_val.rs;

    value_pv = (pv_spec_t *)_value;
    if (value_pv && (pv_get_spec_value(_m, value_pv, &value_val) == 0)) {
        if (value_val.flags & PV_VAL_STR) {
            if (value_val.rs.s == NULL) {
                LM_ERR("missing uri value\n");
                return -1;
            }
        } else {
            LM_ERR("uri value is not string\n");
            return -1;
        }
    } else {
        LM_ERR("failed to get uri value\n");
        return -1;
    }
    value = value_val.rs;

    if (value.len == 0) {
        LM_ERR("hostpart of uri cannot be empty\n");
        return -1;
    }
    if (uri.len + value.len > MAX_URI_SIZE) {
        LM_ERR("resulting uri would be too large\n");
        return -1;
    }

    colon = strchr(uri.s, ':');
    if (colon == NULL) {
        LM_ERR("uri does not contain ':' character\n");
        return -1;
    }
    c = &(new_uri[0]);
    at = strchr(colon + 1, '@');
    if (at == NULL) {
        next = colon + 1;
    } else {
        next = at + 1;
    }
    append_str(c, uri.s, next - uri.s);
    host_len = strcspn(next, ":;?");
    append_str(c, value.s, value.len);
    strcpy(c, next + host_len);
    res_val.rs.len = uri.len + value.len - host_len;
    res_val.rs.s = &(new_uri[0]);

    LM_DBG("resulting uri: %.*s\n", res_val.rs.len, res_val.rs.s);
    res_val.flags = PV_VAL_STR;
    uri_pv->setf(_m, &uri_pv->pvp, (int)EQ_T, &res_val);

    return 1;
}