txt_var.c

Go to the documentation of this file.

/**
 * $Id$
 *
 * Copyright (C) 2008 Elena-Ramona Modroiu (asipto.com)
 *
 * This file is part of kamailio, a free SIP server.
 *
 * openser 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
 *
 * openser 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
 */

#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <unistd.h>

#include "../../mem/mem.h"
#include "../../dprint.h"
#include "../../trim.h"
#include "../../re.h"

#include "txt_var.h"

#define is_in_str(p, in) (p<in->s+in->len && *p)

int tr_txt_eval_re(struct sip_msg *msg, tr_param_t *tp, int subtype,
      pv_value_t *val)
{
   struct subst_expr *se = NULL;
   int nmatches;
   str* result;
#define TR_TXT_BUF_SIZE 2048
   char tr_txt_buf[TR_TXT_BUF_SIZE];

   if(val==NULL || (!(val->flags&PV_VAL_STR)) || val->rs.len<=0)
      return -1;
   
   switch(subtype)
   {
      case TR_TXT_RE_SUBST:
         se = (struct subst_expr*)tp->v.data;
         if(se==NULL)
            return 0;
         if(val->rs.len>=TR_TXT_BUF_SIZE-1)
         {
            LM_ERR("PV value too big %d, increase buffer size\n",
                  val->rs.len);
            return -1;
         }
         memcpy(tr_txt_buf, val->rs.s, val->rs.len);
         tr_txt_buf[val->rs.len] = '\0';
         /* pkg malloc'ed result */
         result=subst_str(tr_txt_buf, msg, se, &nmatches);
         if (result == NULL)
         {
            if (nmatches<0)
               LM_ERR("subst failed\n");
            return -1;
         }
         if(result->len>=TR_TXT_BUF_SIZE-1)
         {
            LM_ERR("subst result too big %d, increase buffer size\n",
                  result->len);
            return -1;
         }
         memcpy(tr_txt_buf, result->s, result->len);
         tr_txt_buf[result->len] = '\0';
         memset(val, 0, sizeof(pv_value_t));
         val->flags = PV_VAL_STR;
         val->rs.s = tr_txt_buf;
         val->rs.len = result->len;
         pkg_free(result->s);
         pkg_free(result);
         break;
      default:
         LM_ERR("unknown subtype %d\n", subtype);
         return -1;
   }
   return 0;

}

char* tr_txt_parse_re(str *in, trans_t *t)
{
   char *p;
   str name;
   str tok;
   struct subst_expr *se = NULL;
   tr_param_t *tp = NULL;
   int n;


   if(in==NULL || t==NULL)
      return NULL;

   p = in->s;
   name.s = in->s;
   t->type = TR_TXT_RE;
   t->trf = tr_txt_eval_re;

   /* find next token */
   while(is_in_str(p, in) && *p!=TR_PARAM_MARKER && *p!=TR_RBRACKET) p++;
   if(*p=='\0')
      goto error;
   name.len = p - name.s;
   trim(&name);

   if(name.len==5 && strncasecmp(name.s, "subst", 5)==0)
   {
      t->subtype = TR_TXT_RE_SUBST;
      if(*p!=TR_PARAM_MARKER)
         goto error;
      p++;
      /* get trans here */
      n = 0;
      tok.s = p;
      while(is_in_str(p, in))
      {
         if(*p==TR_RBRACKET)
         {
            if(n==0)
                  break;
            n--;
         }
         if(*p == TR_LBRACKET)
            n++;
         p++;
      }
      if(!is_in_str(p, in))
         goto error;
      if(p==tok.s)
         goto error;
      tok.len = p - tok.s;
      tp = (tr_param_t*)pkg_malloc(sizeof(tr_param_t));
      if(tp==NULL)
      {
         LM_ERR("no more private memory!\n");
         goto error;
      }

      se=subst_parser(&tok);

      if (se==0)
         goto error;
      tp->type = TR_PARAM_SUBST;
      tp->v.data = (void*)se;
      t->params = tp;

      while(*p && (*p==' ' || *p=='\t' || *p=='\n')) p++;
      if(*p!=TR_RBRACKET)
         goto error;
      goto done;
   }

   LM_ERR("unknown transformation: %.*s/%.*s/%d!\n", in->len, in->s,
         name.len, name.s, name.len);
error:
   LM_ERR("invalid transformation [%.*s] <%d>\n", in->len, in->s,
         (int)(p-in->s));
   if(tp!=NULL)
      pkg_free(tp);
   if(se!=NULL)
      subst_expr_free(se);
   return NULL;
done:
   t->name = name;
   return p;

}