mirror of https://github.com/sipwise/kamailio.git
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.
1621 lines
32 KiB
1621 lines
32 KiB
/**
|
|
*
|
|
* Copyright (C) 2008 Elena-Ramona Modroiu (asipto.com)
|
|
*
|
|
* 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 <stddef.h>
|
|
#include <regex.h>
|
|
|
|
#include "../../mem/shm_mem.h"
|
|
#include "../../mem/mem.h"
|
|
#include "../../shm_init.h"
|
|
#include "../../dprint.h"
|
|
#include "../../parser/parse_param.h"
|
|
#include "../../hashes.h"
|
|
#include "../../ut.h"
|
|
#include "../../re.h"
|
|
#include "../../lib/kcore/faked_msg.h"
|
|
#include "../../action.h"
|
|
#include "../../route.h"
|
|
|
|
#include "ht_api.h"
|
|
#include "ht_db.h"
|
|
|
|
|
|
ht_t *_ht_root = NULL;
|
|
ht_cell_t *ht_expired_cell;
|
|
|
|
typedef struct _keyvalue {
|
|
str key;
|
|
str value;
|
|
int type;
|
|
union {
|
|
param_t *params;
|
|
} u;
|
|
} keyvalue_t;
|
|
|
|
|
|
#define KEYVALUE_TYPE_NONE 0
|
|
#define KEYVALUE_TYPE_PARAMS 1
|
|
|
|
/**
|
|
* parse a string like: 'key=>value'
|
|
* - the value can be parameter list: 'name1=value1;...;nameX=valueX'
|
|
*/
|
|
int keyvalue_parse_str(str *data, int type, keyvalue_t *res)
|
|
{
|
|
char *p;
|
|
str s;
|
|
str in;
|
|
param_hooks_t phooks;
|
|
|
|
if(data==NULL || data->s==NULL || data->len<=0 || res==NULL)
|
|
{
|
|
LM_ERR("invalid parameters\n");
|
|
return -1;
|
|
}
|
|
|
|
memset(res, 0, sizeof(keyvalue_t));
|
|
|
|
in.s = data->s;
|
|
in.len = data->len;
|
|
|
|
p = in.s;
|
|
while(p<in.s+in.len && (*p==' ' || *p=='\t' || *p=='\n' || *p=='\r'))
|
|
p++;
|
|
if(p>in.s+in.len || *p=='\0')
|
|
goto error;
|
|
res->key.s = p;
|
|
while(p < in.s + in.len)
|
|
{
|
|
if(*p=='=' || *p==' ' || *p=='\t' || *p=='\n' || *p=='\r')
|
|
break;
|
|
p++;
|
|
}
|
|
if(p>in.s+in.len || *p=='\0')
|
|
goto error;
|
|
res->key.len = (int)(p - res->key.s);
|
|
if(*p!='=')
|
|
{
|
|
while(p<in.s+in.len && (*p==' ' || *p=='\t' || *p=='\n' || *p=='\r'))
|
|
p++;
|
|
if(p>in.s+in.len || *p=='\0' || *p!='=')
|
|
goto error;
|
|
}
|
|
p++;
|
|
if(*p!='>')
|
|
goto error;
|
|
p++;
|
|
while(p<in.s+in.len && (*p==' ' || *p=='\t' || *p=='\n' || *p=='\r'))
|
|
p++;
|
|
|
|
s.s = p;
|
|
s.len = in.s + in.len - p;
|
|
res->value.s = s.s;
|
|
res->value.len = s.len;
|
|
res->type = type;
|
|
if(type==KEYVALUE_TYPE_PARAMS)
|
|
{
|
|
if(s.s[s.len-1]==';')
|
|
s.len--;
|
|
if (parse_params(&s, CLASS_ANY, &phooks, &res->u.params)<0)
|
|
{
|
|
LM_ERR("failed parsing params value\n");
|
|
goto error;
|
|
}
|
|
}
|
|
return 0;
|
|
error:
|
|
LM_ERR("invalid input parameter [%.*s] at [%d]\n", in.len, in.s,
|
|
(int)(p-in.s));
|
|
return -1;
|
|
}
|
|
|
|
void keyvalue_destroy(keyvalue_t *res)
|
|
{
|
|
if(res==NULL)
|
|
return;
|
|
if(res->type==KEYVALUE_TYPE_PARAMS)
|
|
{
|
|
if(res->u.params!=NULL)
|
|
free_params(res->u.params);
|
|
}
|
|
memset(res, 0, sizeof(keyvalue_t));
|
|
}
|
|
|
|
/**
|
|
* recursive/re-entrant lock of the slot in hash table
|
|
*/
|
|
void ht_slot_lock(ht_t *ht, int idx)
|
|
{
|
|
int mypid;
|
|
|
|
mypid = my_pid();
|
|
if (likely(atomic_get(&ht->entries[idx].locker_pid) != mypid)) {
|
|
lock_get(&ht->entries[idx].lock);
|
|
atomic_set(&ht->entries[idx].locker_pid, mypid);
|
|
} else {
|
|
/* locked within the same process that executed us */
|
|
ht->entries[idx].rec_lock_level++;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* recursive/re-entrant unlock of the slot in hash table
|
|
*/
|
|
void ht_slot_unlock(ht_t *ht, int idx)
|
|
{
|
|
if (likely(ht->entries[idx].rec_lock_level == 0)) {
|
|
atomic_set(&ht->entries[idx].locker_pid, 0);
|
|
lock_release(&ht->entries[idx].lock);
|
|
} else {
|
|
/* recursive locked => decrease lock count */
|
|
ht->entries[idx].rec_lock_level--;
|
|
}
|
|
}
|
|
|
|
ht_cell_t* ht_cell_new(str *name, int type, int_str *val, unsigned int cellid)
|
|
{
|
|
ht_cell_t *cell;
|
|
unsigned int msize;
|
|
|
|
msize = sizeof(ht_cell_t) + (name->len + 1)*sizeof(char);
|
|
|
|
if(type&AVP_VAL_STR)
|
|
msize += (val->s.len + 1)*sizeof(char);
|
|
|
|
cell = (ht_cell_t*)shm_malloc(msize);
|
|
if(cell==NULL)
|
|
{
|
|
LM_ERR("no more shm\n");
|
|
return NULL;
|
|
}
|
|
|
|
memset(cell, 0, msize);
|
|
cell->msize = msize;
|
|
cell->cellid = cellid;
|
|
cell->flags = type;
|
|
cell->name.len = name->len;
|
|
cell->name.s = (char*)cell + sizeof(ht_cell_t);
|
|
memcpy(cell->name.s, name->s, name->len);
|
|
cell->name.s[name->len] = '\0';
|
|
if(type&AVP_VAL_STR)
|
|
{
|
|
cell->value.s.s = (char*)cell->name.s + name->len + 1;
|
|
cell->value.s.len = val->s.len;
|
|
memcpy(cell->value.s.s, val->s.s, val->s.len);
|
|
cell->value.s.s[val->s.len] = '\0';
|
|
} else {
|
|
cell->value.n = val->n;
|
|
}
|
|
return cell;
|
|
}
|
|
|
|
int ht_cell_free(ht_cell_t *cell)
|
|
{
|
|
if(cell==NULL)
|
|
return -1;
|
|
shm_free(cell);
|
|
return 0;
|
|
}
|
|
|
|
int ht_cell_pkg_free(ht_cell_t *cell)
|
|
{
|
|
if(cell==NULL)
|
|
return -1;
|
|
pkg_free(cell);
|
|
return 0;
|
|
}
|
|
|
|
|
|
ht_t *ht_get_root(void)
|
|
{
|
|
return _ht_root;
|
|
}
|
|
|
|
ht_t* ht_get_table(str *name)
|
|
{
|
|
unsigned int htid;
|
|
ht_t *ht;
|
|
|
|
htid = ht_compute_hash(name);
|
|
|
|
/* does it exist */
|
|
ht = _ht_root;
|
|
while(ht!=NULL)
|
|
{
|
|
if(htid == ht->htid && name->len==ht->name.len
|
|
&& strncmp(name->s, ht->name.s, name->len)==0)
|
|
{
|
|
LM_DBG("htable found [%.*s]\n", name->len, name->s);
|
|
return ht;
|
|
}
|
|
ht = ht->next;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
int ht_add_table(str *name, int autoexp, str *dbtable, str *dbcols, int size,
|
|
int dbmode, int itype, int_str *ival, int updateexpire,
|
|
int dmqreplicate)
|
|
{
|
|
unsigned int htid;
|
|
ht_t *ht;
|
|
int c;
|
|
int i;
|
|
|
|
htid = ht_compute_hash(name);
|
|
|
|
/* does it exist */
|
|
ht = _ht_root;
|
|
while(ht!=NULL)
|
|
{
|
|
if(htid == ht->htid && name->len==ht->name.len
|
|
&& strncmp(name->s, ht->name.s, name->len)==0)
|
|
{
|
|
LM_ERR("htable already configured [%.*s]\n", name->len, name->s);
|
|
return -1;
|
|
}
|
|
ht = ht->next;
|
|
}
|
|
|
|
ht = (ht_t*)shm_malloc(sizeof(ht_t));
|
|
if(ht==NULL)
|
|
{
|
|
LM_ERR("no more shared memory\n");
|
|
return -1;
|
|
}
|
|
memset(ht, 0, sizeof(ht_t));
|
|
|
|
if(size<=1)
|
|
ht->htsize = 8;
|
|
else if(size>31)
|
|
ht->htsize = 1<<14;
|
|
else ht->htsize = 1<<size;
|
|
ht->htid = htid;
|
|
ht->htexpire = autoexp;
|
|
ht->updateexpire = updateexpire;
|
|
ht->name = *name;
|
|
if(dbtable!=NULL && dbtable->len>0)
|
|
ht->dbtable = *dbtable;
|
|
ht->dbmode = dbmode;
|
|
ht->flags = itype;
|
|
if(ival!=NULL)
|
|
ht->initval = *ival;
|
|
ht->dmqreplicate = dmqreplicate;
|
|
|
|
if(dbcols!=NULL && dbcols->s!=NULL && dbcols->len>0) {
|
|
ht->scols[0].s = (char*)shm_malloc((1+dbcols->len)*sizeof(char));
|
|
if(ht->scols[0].s==NULL) {
|
|
LM_ERR("no more shm memory\n");
|
|
shm_free(ht);
|
|
return -1;
|
|
}
|
|
memset(ht->scols[0].s, 0, (1+dbcols->len)*sizeof(char));
|
|
memcpy(ht->scols[0].s, dbcols->s, dbcols->len);
|
|
ht->scols[0].len = dbcols->len;
|
|
c = 0;
|
|
for(i=0; i<dbcols->len; i++) {
|
|
if(ht->scols[0].s[i]==',') {
|
|
ht->scols[c].len = (ht->scols[0].s + i - ht->scols[c].s);
|
|
LM_DBG("db table column[%d]='%.*s'\n", c,
|
|
ht->scols[c].len, ht->scols[c].s);
|
|
c++;
|
|
if(c>=HT_MAX_COLS) {
|
|
LM_ERR("too many columns %d\n", c);
|
|
shm_free(ht->scols[0].s);
|
|
shm_free(ht);
|
|
return -1;
|
|
}
|
|
ht->scols[c].s = ht->scols[0].s + i + 1;
|
|
ht->scols[c].len = ht->scols[0].s + dbcols->len - ht->scols[c].s;
|
|
}
|
|
}
|
|
LM_DBG("db table column[%d]='%.*s'\n", c,
|
|
ht->scols[c].len, ht->scols[c].s);
|
|
if(c==0) {
|
|
LM_ERR("there must be at least two columns (prefix, value)\n");
|
|
shm_free(ht->scols[0].s);
|
|
shm_free(ht);
|
|
return -1;
|
|
}
|
|
ht->ncols = c + 1;
|
|
ht->pack[0] = 'l';
|
|
ht->pack[1] = ',';
|
|
ht->pack[2] = '*';
|
|
}
|
|
|
|
ht->next = _ht_root;
|
|
_ht_root = ht;
|
|
return 0;
|
|
}
|
|
|
|
int ht_init_tables(void)
|
|
{
|
|
ht_t *ht;
|
|
int i;
|
|
char route_name[64];
|
|
|
|
ht = _ht_root;
|
|
|
|
while(ht)
|
|
{
|
|
LM_DBG("initializing htable [%.*s] with nr. of slots: %d\n",
|
|
ht->name.len, ht->name.s, ht->htsize);
|
|
if(ht->name.len + sizeof("htable:expired:") < 64)
|
|
{
|
|
strcpy(route_name, "htable:expired:");
|
|
strncat(route_name, ht->name.s, ht->name.len);
|
|
ht->evrt_expired = route_get(&event_rt, route_name);
|
|
|
|
if (ht->evrt_expired < 0
|
|
|| event_rt.rlist[ht->evrt_expired] == NULL)
|
|
{
|
|
ht->evrt_expired = -1;
|
|
LM_DBG("event route for expired items in [%.*s] does not exist\n",
|
|
ht->name.len, ht->name.s);
|
|
} else {
|
|
LM_DBG("event route for expired items in [%.*s] exists\n",
|
|
ht->name.len, ht->name.s);
|
|
}
|
|
} else {
|
|
LM_WARN("event route name for expired items in htable [%.*s]"
|
|
" is too long\n", ht->name.len, ht->name.s);
|
|
}
|
|
ht->entries = (ht_entry_t*)shm_malloc(ht->htsize*sizeof(ht_entry_t));
|
|
if(ht->entries==NULL)
|
|
{
|
|
LM_ERR("no more shm for [%.*s]\n", ht->name.len, ht->name.s);
|
|
shm_free(ht);
|
|
return -1;
|
|
}
|
|
memset(ht->entries, 0, ht->htsize*sizeof(ht_entry_t));
|
|
|
|
for(i=0; i<ht->htsize; i++)
|
|
{
|
|
if(lock_init(&ht->entries[i].lock)==0)
|
|
{
|
|
LM_ERR("cannot initialize lock[%d] in [%.*s]\n", i,
|
|
ht->name.len, ht->name.s);
|
|
i--;
|
|
while(i>=0)
|
|
{
|
|
lock_destroy(&ht->entries[i].lock);
|
|
i--;
|
|
}
|
|
shm_free(ht->entries);
|
|
shm_free(ht);
|
|
return -1;
|
|
|
|
}
|
|
}
|
|
ht = ht->next;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int ht_destroy(void)
|
|
{
|
|
int i;
|
|
ht_cell_t *it, *it0;
|
|
ht_t *ht;
|
|
ht_t *ht0;
|
|
|
|
if(_ht_root==NULL)
|
|
return -1;
|
|
|
|
ht = _ht_root;
|
|
while(ht)
|
|
{
|
|
ht0 = ht->next;
|
|
if(ht->entries!=NULL)
|
|
{
|
|
for(i=0; i<ht->htsize; i++)
|
|
{
|
|
/* free entries */
|
|
it = ht->entries[i].first;
|
|
while(it)
|
|
{
|
|
it0 = it;
|
|
it = it->next;
|
|
ht_cell_free(it0);
|
|
}
|
|
/* free locks */
|
|
lock_destroy(&ht->entries[i].lock);
|
|
}
|
|
shm_free(ht->entries);
|
|
}
|
|
shm_free(ht);
|
|
ht = ht0;
|
|
}
|
|
_ht_root = NULL;
|
|
return 0;
|
|
}
|
|
|
|
|
|
int ht_set_cell(ht_t *ht, str *name, int type, int_str *val, int mode)
|
|
{
|
|
unsigned int idx;
|
|
unsigned int hid;
|
|
ht_cell_t *it, *prev, *cell;
|
|
time_t now;
|
|
|
|
if(ht==NULL || ht->entries==NULL)
|
|
return -1;
|
|
|
|
hid = ht_compute_hash(name);
|
|
|
|
idx = ht_get_entry(hid, ht->htsize);
|
|
|
|
now = 0;
|
|
if(ht->htexpire>0)
|
|
now = time(NULL);
|
|
prev = NULL;
|
|
if(mode) ht_slot_lock(ht, idx);
|
|
it = ht->entries[idx].first;
|
|
while(it!=NULL && it->cellid < hid)
|
|
{
|
|
prev = it;
|
|
it = it->next;
|
|
}
|
|
while(it!=NULL && it->cellid == hid)
|
|
{
|
|
if(name->len==it->name.len
|
|
&& strncmp(name->s, it->name.s, name->len)==0)
|
|
{
|
|
/* update value */
|
|
if(it->flags&AVP_VAL_STR)
|
|
{
|
|
if(type&AVP_VAL_STR)
|
|
{
|
|
if(it->value.s.len >= val->s.len)
|
|
{
|
|
/* copy */
|
|
it->value.s.len = val->s.len;
|
|
memcpy(it->value.s.s, val->s.s, val->s.len);
|
|
it->value.s.s[it->value.s.len] = '\0';
|
|
|
|
if(ht->updateexpire)
|
|
it->expire = now + ht->htexpire;
|
|
} else {
|
|
/* new */
|
|
cell = ht_cell_new(name, type, val, hid);
|
|
if(cell == NULL)
|
|
{
|
|
LM_ERR("cannot create new cell\n");
|
|
if(mode) ht_slot_unlock(ht, idx);
|
|
return -1;
|
|
}
|
|
cell->next = it->next;
|
|
cell->prev = it->prev;
|
|
if(ht->updateexpire)
|
|
cell->expire = now + ht->htexpire;
|
|
else
|
|
cell->expire = it->expire;
|
|
if(it->prev)
|
|
it->prev->next = cell;
|
|
else
|
|
ht->entries[idx].first = cell;
|
|
if(it->next)
|
|
it->next->prev = cell;
|
|
ht_cell_free(it);
|
|
}
|
|
} else {
|
|
it->flags &= ~AVP_VAL_STR;
|
|
it->value.n = val->n;
|
|
|
|
if(ht->updateexpire)
|
|
it->expire = now + ht->htexpire;
|
|
}
|
|
if(mode) ht_slot_unlock(ht, idx);
|
|
return 0;
|
|
} else {
|
|
if(type&AVP_VAL_STR)
|
|
{
|
|
/* new */
|
|
cell = ht_cell_new(name, type, val, hid);
|
|
if(cell == NULL)
|
|
{
|
|
LM_ERR("cannot create new cell.\n");
|
|
if(mode) ht_slot_unlock(ht, idx);
|
|
return -1;
|
|
}
|
|
if(ht->updateexpire)
|
|
cell->expire = now + ht->htexpire;
|
|
else
|
|
cell->expire = it->expire;
|
|
cell->next = it->next;
|
|
cell->prev = it->prev;
|
|
if(it->prev)
|
|
it->prev->next = cell;
|
|
else
|
|
ht->entries[idx].first = cell;
|
|
if(it->next)
|
|
it->next->prev = cell;
|
|
ht_cell_free(it);
|
|
} else {
|
|
it->value.n = val->n;
|
|
|
|
if(ht->updateexpire)
|
|
it->expire = now + ht->htexpire;
|
|
}
|
|
if(mode) ht_slot_unlock(ht, idx);
|
|
return 0;
|
|
}
|
|
}
|
|
prev = it;
|
|
it = it->next;
|
|
}
|
|
/* add */
|
|
cell = ht_cell_new(name, type, val, hid);
|
|
if(cell == NULL)
|
|
{
|
|
LM_ERR("cannot create new cell.\n");
|
|
if(mode) ht_slot_unlock(ht, idx);
|
|
return -1;
|
|
}
|
|
cell->expire = now + ht->htexpire;
|
|
if(prev==NULL)
|
|
{
|
|
if(ht->entries[idx].first!=NULL)
|
|
{
|
|
cell->next = ht->entries[idx].first;
|
|
ht->entries[idx].first->prev = cell;
|
|
}
|
|
ht->entries[idx].first = cell;
|
|
} else {
|
|
cell->next = prev->next;
|
|
cell->prev = prev;
|
|
if(prev->next)
|
|
prev->next->prev = cell;
|
|
prev->next = cell;
|
|
}
|
|
ht->entries[idx].esize++;
|
|
if(mode) ht_slot_unlock(ht, idx);
|
|
return 0;
|
|
}
|
|
|
|
int ht_del_cell(ht_t *ht, str *name)
|
|
{
|
|
unsigned int idx;
|
|
unsigned int hid;
|
|
ht_cell_t *it;
|
|
|
|
if(ht==NULL || ht->entries==NULL)
|
|
return -1;
|
|
|
|
hid = ht_compute_hash(name);
|
|
|
|
idx = ht_get_entry(hid, ht->htsize);
|
|
|
|
/* head test and return */
|
|
if(ht->entries[idx].first==NULL)
|
|
return 0;
|
|
|
|
ht_slot_lock(ht, idx);
|
|
it = ht->entries[idx].first;
|
|
while(it!=NULL && it->cellid < hid)
|
|
it = it->next;
|
|
while(it!=NULL && it->cellid == hid)
|
|
{
|
|
if(name->len==it->name.len
|
|
&& strncmp(name->s, it->name.s, name->len)==0)
|
|
{
|
|
/* found */
|
|
if(it->prev==NULL)
|
|
ht->entries[idx].first = it->next;
|
|
else
|
|
it->prev->next = it->next;
|
|
if(it->next)
|
|
it->next->prev = it->prev;
|
|
ht->entries[idx].esize--;
|
|
ht_slot_unlock(ht, idx);
|
|
ht_cell_free(it);
|
|
return 0;
|
|
}
|
|
it = it->next;
|
|
}
|
|
ht_slot_unlock(ht, idx);
|
|
return 0;
|
|
}
|
|
|
|
ht_cell_t* ht_cell_value_add(ht_t *ht, str *name, int val, int mode,
|
|
ht_cell_t *old)
|
|
{
|
|
unsigned int idx;
|
|
unsigned int hid;
|
|
ht_cell_t *it, *prev, *cell;
|
|
time_t now;
|
|
int_str isval;
|
|
|
|
if(ht==NULL || ht->entries==NULL)
|
|
return NULL;
|
|
|
|
hid = ht_compute_hash(name);
|
|
|
|
idx = ht_get_entry(hid, ht->htsize);
|
|
|
|
now = 0;
|
|
if(ht->htexpire>0)
|
|
now = time(NULL);
|
|
prev = NULL;
|
|
if(mode) ht_slot_lock(ht, idx);
|
|
it = ht->entries[idx].first;
|
|
while(it!=NULL && it->cellid < hid)
|
|
{
|
|
prev = it;
|
|
it = it->next;
|
|
}
|
|
while(it!=NULL && it->cellid == hid)
|
|
{
|
|
if(name->len==it->name.len
|
|
&& strncmp(name->s, it->name.s, name->len)==0)
|
|
{
|
|
/* found */
|
|
if(now>0 && it->expire!=0 && it->expire<now) {
|
|
/* entry has expired */
|
|
ht_handle_expired_record(ht, it);
|
|
|
|
if(ht->flags==PV_VAL_INT) {
|
|
/* initval is integer, use it to create a fresh entry */
|
|
it->flags &= ~AVP_VAL_STR;
|
|
it->value.n = ht->initval.n;
|
|
/* increment will be done below */
|
|
} else {
|
|
/* delete expired entry */
|
|
if(it->prev==NULL)
|
|
ht->entries[idx].first = it->next;
|
|
else
|
|
it->prev->next = it->next;
|
|
if(it->next)
|
|
it->next->prev = it->prev;
|
|
ht->entries[idx].esize--;
|
|
if(mode) ht_slot_unlock(ht, idx);
|
|
ht_cell_free(it);
|
|
return NULL;
|
|
}
|
|
}
|
|
/* update value */
|
|
if(it->flags&AVP_VAL_STR)
|
|
{
|
|
/* string value cannot be incremented */
|
|
if(mode) ht_slot_unlock(ht, idx);
|
|
return NULL;
|
|
} else {
|
|
it->value.n += val;
|
|
if(ht->updateexpire)
|
|
it->expire = now + ht->htexpire;
|
|
if(old!=NULL)
|
|
{
|
|
if(old->msize>=it->msize)
|
|
{
|
|
memcpy(old, it, it->msize);
|
|
if(mode) ht_slot_unlock(ht, idx);
|
|
return old;
|
|
}
|
|
}
|
|
cell = (ht_cell_t*)pkg_malloc(it->msize);
|
|
if(cell!=NULL)
|
|
memcpy(cell, it, it->msize);
|
|
|
|
if(mode) ht_slot_unlock(ht, idx);
|
|
return cell;
|
|
}
|
|
}
|
|
prev = it;
|
|
it = it->next;
|
|
}
|
|
/* add val if htable has an integer init value */
|
|
if(ht->flags!=PV_VAL_INT)
|
|
{
|
|
if(mode) ht_slot_unlock(ht, idx);
|
|
return NULL;
|
|
}
|
|
isval.n = ht->initval.n + val;
|
|
it = ht_cell_new(name, 0, &isval, hid);
|
|
if(it == NULL)
|
|
{
|
|
LM_ERR("cannot create new cell.\n");
|
|
if(mode) ht_slot_unlock(ht, idx);
|
|
return NULL;
|
|
}
|
|
it->expire = now + ht->htexpire;
|
|
if(prev==NULL)
|
|
{
|
|
if(ht->entries[idx].first!=NULL)
|
|
{
|
|
it->next = ht->entries[idx].first;
|
|
ht->entries[idx].first->prev = it;
|
|
}
|
|
ht->entries[idx].first = it;
|
|
} else {
|
|
it->next = prev->next;
|
|
it->prev = prev;
|
|
if(prev->next)
|
|
prev->next->prev = it;
|
|
prev->next = it;
|
|
}
|
|
ht->entries[idx].esize++;
|
|
if(old!=NULL)
|
|
{
|
|
if(old->msize>=it->msize)
|
|
{
|
|
memcpy(old, it, it->msize);
|
|
if(mode) ht_slot_unlock(ht, idx);
|
|
return old;
|
|
}
|
|
}
|
|
cell = (ht_cell_t*)pkg_malloc(it->msize);
|
|
if(cell!=NULL)
|
|
memcpy(cell, it, it->msize);
|
|
|
|
if(mode) ht_slot_unlock(ht, idx);
|
|
return cell;
|
|
}
|
|
|
|
|
|
ht_cell_t* ht_cell_pkg_copy(ht_t *ht, str *name, ht_cell_t *old)
|
|
{
|
|
unsigned int idx;
|
|
unsigned int hid;
|
|
ht_cell_t *it, *cell;
|
|
|
|
if(ht==NULL || ht->entries==NULL)
|
|
return NULL;
|
|
|
|
hid = ht_compute_hash(name);
|
|
|
|
idx = ht_get_entry(hid, ht->htsize);
|
|
|
|
/* head test and return */
|
|
if(ht->entries[idx].first==NULL)
|
|
return NULL;
|
|
|
|
ht_slot_lock(ht, idx);
|
|
it = ht->entries[idx].first;
|
|
while(it!=NULL && it->cellid < hid)
|
|
it = it->next;
|
|
while(it!=NULL && it->cellid == hid)
|
|
{
|
|
if(name->len==it->name.len
|
|
&& strncmp(name->s, it->name.s, name->len)==0)
|
|
{
|
|
/* found */
|
|
if(ht->htexpire>0 && it->expire!=0 && it->expire<time(NULL)) {
|
|
/* entry has expired, delete it and return NULL */
|
|
ht_handle_expired_record(ht, it);
|
|
|
|
if(it->prev==NULL)
|
|
ht->entries[idx].first = it->next;
|
|
else
|
|
it->prev->next = it->next;
|
|
if(it->next)
|
|
it->next->prev = it->prev;
|
|
ht->entries[idx].esize--;
|
|
ht_slot_unlock(ht, idx);
|
|
ht_cell_free(it);
|
|
return NULL;
|
|
}
|
|
if(old!=NULL)
|
|
{
|
|
if(old->msize>=it->msize)
|
|
{
|
|
memcpy(old, it, it->msize);
|
|
ht_slot_unlock(ht, idx);
|
|
return old;
|
|
}
|
|
}
|
|
cell = (ht_cell_t*)pkg_malloc(it->msize);
|
|
if(cell!=NULL)
|
|
memcpy(cell, it, it->msize);
|
|
ht_slot_unlock(ht, idx);
|
|
return cell;
|
|
}
|
|
it = it->next;
|
|
}
|
|
ht_slot_unlock(ht, idx);
|
|
return NULL;
|
|
}
|
|
|
|
int ht_dbg(void)
|
|
{
|
|
int i;
|
|
ht_cell_t *it;
|
|
ht_t *ht;
|
|
|
|
ht = _ht_root;
|
|
while(ht)
|
|
{
|
|
LM_ERR("===== htable[%.*s] hid: %u exp: %u>\n", ht->name.len,
|
|
ht->name.s, ht->htid, ht->htexpire);
|
|
for(i=0; i<ht->htsize; i++)
|
|
{
|
|
ht_slot_lock(ht, i);
|
|
LM_ERR("htable[%d] -- <%d>\n", i, ht->entries[i].esize);
|
|
it = ht->entries[i].first;
|
|
while(it)
|
|
{
|
|
LM_ERR("\tcell: %.*s\n", it->name.len, it->name.s);
|
|
LM_ERR("\thid: %u msize: %u flags: %d expire: %u\n", it->cellid,
|
|
it->msize, it->flags, (unsigned int)it->expire);
|
|
if(it->flags&AVP_VAL_STR)
|
|
LM_ERR("\tv-s:%.*s\n", it->value.s.len, it->value.s.s);
|
|
else
|
|
LM_ERR("\tv-i:%d\n", it->value.n);
|
|
it = it->next;
|
|
}
|
|
ht_slot_unlock(ht, i);
|
|
}
|
|
ht = ht->next;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int ht_table_spec(char *spec)
|
|
{
|
|
keyvalue_t kval;
|
|
str name;
|
|
str dbtable = {0, 0};
|
|
str dbcols = {0, 0};
|
|
unsigned int autoexpire = 0;
|
|
unsigned int size = 4;
|
|
unsigned int dbmode = 0;
|
|
unsigned int updateexpire = 1;
|
|
unsigned int dmqreplicate = 0;
|
|
str in;
|
|
str tok;
|
|
param_t *pit=NULL;
|
|
int_str ival;
|
|
int itype;
|
|
|
|
if(!shm_initialized())
|
|
{
|
|
LM_ERR("shared memory was not initialized\n");
|
|
return -1;
|
|
}
|
|
/* parse: name=>dbtable=abc;autoexpire=123;size=123 */
|
|
in.s = spec;
|
|
in.len = strlen(in.s);
|
|
if(keyvalue_parse_str(&in, KEYVALUE_TYPE_PARAMS, &kval)<0)
|
|
{
|
|
LM_ERR("failed parsing: %.*s\n", in.len, in.s);
|
|
return -1;
|
|
}
|
|
name = kval.key;
|
|
itype = PV_VAL_NONE;
|
|
memset(&ival, 0, sizeof(int_str));
|
|
|
|
for (pit = kval.u.params; pit; pit=pit->next)
|
|
{
|
|
tok = pit->body;
|
|
if(pit->name.len==7 && strncmp(pit->name.s, "dbtable", 7)==0) {
|
|
dbtable = tok;
|
|
LM_DBG("htable [%.*s] - dbtable [%.*s]\n", name.len, name.s,
|
|
dbtable.len, dbtable.s);
|
|
} else if(pit->name.len==4 && strncmp(pit->name.s, "cols", 4)==0) {
|
|
dbcols = tok;
|
|
LM_DBG("htable [%.*s] - dbcols [%.*s]\n", name.len, name.s,
|
|
dbcols.len, dbcols.s);
|
|
} else if(pit->name.len==10 && strncmp(pit->name.s, "autoexpire", 10)==0) {
|
|
if(str2int(&tok, &autoexpire)!=0)
|
|
goto error;
|
|
LM_DBG("htable [%.*s] - expire [%u]\n", name.len, name.s,
|
|
autoexpire);
|
|
} else if(pit->name.len==4 && strncmp(pit->name.s, "size", 4)==0) {
|
|
if(str2int(&tok, &size)!=0)
|
|
goto error;
|
|
LM_DBG("htable [%.*s] - size [%u]\n", name.len, name.s,
|
|
size);
|
|
} else if(pit->name.len==6 && strncmp(pit->name.s, "dbmode", 6)==0) {
|
|
if(str2int(&tok, &dbmode)!=0)
|
|
goto error;
|
|
LM_DBG("htable [%.*s] - dbmode [%u]\n", name.len, name.s,
|
|
dbmode);
|
|
} else if(pit->name.len==7 && strncmp(pit->name.s, "initval", 7)==0) {
|
|
if(str2sint(&tok, &ival.n)!=0)
|
|
goto error;
|
|
itype = PV_VAL_INT;
|
|
LM_DBG("htable [%.*s] - initval [%d]\n", name.len, name.s,
|
|
ival.n);
|
|
} else if(pit->name.len == 12 && strncmp(pit->name.s, "updateexpire", 12) == 0) {
|
|
if(str2int(&tok, &updateexpire) != 0)
|
|
goto error;
|
|
|
|
LM_DBG("htable [%.*s] - updateexpire [%u]\n", name.len, name.s, updateexpire);
|
|
} else if(pit->name.len == 12 && strncmp(pit->name.s, "dmqreplicate", 12) == 0) {
|
|
if(str2int(&tok, &dmqreplicate) != 0)
|
|
goto error;
|
|
|
|
LM_DBG("htable [%.*s] - dmqreplicate [%u]\n", name.len, name.s, dmqreplicate);
|
|
} else { goto error; }
|
|
}
|
|
|
|
return ht_add_table(&name, autoexpire, &dbtable, &dbcols, size, dbmode,
|
|
itype, &ival, updateexpire, dmqreplicate);
|
|
|
|
error:
|
|
LM_ERR("invalid htable parameter [%.*s]\n", in.len, in.s);
|
|
return -1;
|
|
}
|
|
|
|
int ht_db_load_tables(void)
|
|
{
|
|
ht_t *ht;
|
|
|
|
ht = _ht_root;
|
|
while(ht)
|
|
{
|
|
if(ht->dbtable.len>0)
|
|
{
|
|
LM_DBG("loading db table [%.*s] in ht [%.*s]\n",
|
|
ht->dbtable.len, ht->dbtable.s,
|
|
ht->name.len, ht->name.s);
|
|
if(ht_db_load_table(ht, &ht->dbtable, 0)!=0)
|
|
return -1;
|
|
}
|
|
ht = ht->next;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int ht_db_sync_tables(void)
|
|
{
|
|
ht_t *ht;
|
|
|
|
ht = _ht_root;
|
|
while(ht)
|
|
{
|
|
if(ht->dbtable.len>0 && ht->dbmode!=0 && ht->ncols==0)
|
|
{
|
|
LM_DBG("sync db table [%.*s] from ht [%.*s]\n",
|
|
ht->dbtable.len, ht->dbtable.s,
|
|
ht->name.len, ht->name.s);
|
|
ht_db_delete_records(&ht->dbtable);
|
|
if(ht_db_save_table(ht, &ht->dbtable)!=0)
|
|
LM_ERR("failed sync'ing hash table [%.*s] to db\n",
|
|
ht->name.len, ht->name.s);
|
|
}
|
|
ht = ht->next;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int ht_has_autoexpire(void)
|
|
{
|
|
ht_t *ht;
|
|
|
|
if(_ht_root==NULL)
|
|
return 0;
|
|
|
|
ht = _ht_root;
|
|
while(ht)
|
|
{
|
|
if(ht->htexpire>0)
|
|
return 1;
|
|
ht = ht->next;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
extern int ht_timer_procs;
|
|
|
|
void ht_timer(unsigned int ticks, void *param)
|
|
{
|
|
ht_t *ht;
|
|
ht_cell_t *it;
|
|
ht_cell_t *it0;
|
|
time_t now;
|
|
int i;
|
|
int istart;
|
|
int istep;
|
|
|
|
if(_ht_root==NULL)
|
|
return;
|
|
|
|
now = time(NULL);
|
|
|
|
istart = (int)(long)param;
|
|
if(ht_timer_procs<=0) istep = 1;
|
|
else istep = ht_timer_procs;
|
|
|
|
ht = _ht_root;
|
|
while(ht)
|
|
{
|
|
if(ht->htexpire>0)
|
|
{
|
|
for(i=istart; i<ht->htsize; i+=istep)
|
|
{
|
|
/* free entries */
|
|
ht_slot_lock(ht, i);
|
|
it = ht->entries[i].first;
|
|
while(it)
|
|
{
|
|
it0 = it->next;
|
|
if(it->expire!=0 && it->expire<now)
|
|
{
|
|
/* expired */
|
|
ht_handle_expired_record(ht, it);
|
|
if(it->prev==NULL)
|
|
ht->entries[i].first = it->next;
|
|
else
|
|
it->prev->next = it->next;
|
|
if(it->next)
|
|
it->next->prev = it->prev;
|
|
ht->entries[i].esize--;
|
|
ht_cell_free(it);
|
|
}
|
|
it = it0;
|
|
}
|
|
ht_slot_unlock(ht, i);
|
|
}
|
|
}
|
|
ht = ht->next;
|
|
}
|
|
return;
|
|
}
|
|
|
|
void ht_handle_expired_record(ht_t *ht, ht_cell_t *cell)
|
|
{
|
|
if(ht->evrt_expired<0)
|
|
return;
|
|
ht_expired_cell = cell;
|
|
|
|
LM_DBG("running event_route[htable:expired:%.*s]\n",
|
|
ht->name.len, ht->name.s);
|
|
ht_expired_run_event_route(ht->evrt_expired);
|
|
|
|
ht_expired_cell = NULL;
|
|
}
|
|
|
|
void ht_expired_run_event_route(int routeid)
|
|
{
|
|
int backup_rt;
|
|
sip_msg_t *fmsg;
|
|
|
|
if (routeid < 0 || event_rt.rlist[routeid] == NULL)
|
|
{
|
|
LM_DBG("route does not exist\n");
|
|
return;
|
|
}
|
|
|
|
if (faked_msg_init() < 0)
|
|
{
|
|
LM_ERR("faked_msg_init() failed\n");
|
|
return;
|
|
}
|
|
fmsg = faked_msg_next();
|
|
fmsg->parsed_orig_ruri_ok = 0;
|
|
|
|
backup_rt = get_route_type();
|
|
|
|
set_route_type(EVENT_ROUTE);
|
|
run_top_route(event_rt.rlist[routeid], fmsg, 0);
|
|
|
|
set_route_type(backup_rt);
|
|
}
|
|
|
|
int ht_set_cell_expire(ht_t *ht, str *name, int type, int_str *val)
|
|
{
|
|
unsigned int idx;
|
|
unsigned int hid;
|
|
ht_cell_t *it;
|
|
time_t now;
|
|
|
|
if(ht==NULL || ht->entries==NULL)
|
|
return -1;
|
|
|
|
/* str value - ignore */
|
|
if(type&AVP_VAL_STR)
|
|
return 0;
|
|
/* not auto-expire htable */
|
|
if(ht->htexpire==0)
|
|
return 0;
|
|
|
|
hid = ht_compute_hash(name);
|
|
|
|
idx = ht_get_entry(hid, ht->htsize);
|
|
|
|
now = 0;
|
|
if(val->n>0)
|
|
now = time(NULL) + val->n;
|
|
LM_DBG("set auto-expire to %u (%d)\n", (unsigned int)now,
|
|
val->n);
|
|
|
|
ht_slot_lock(ht, idx);
|
|
it = ht->entries[idx].first;
|
|
while(it!=NULL && it->cellid < hid)
|
|
it = it->next;
|
|
while(it!=NULL && it->cellid == hid)
|
|
{
|
|
if(name->len==it->name.len
|
|
&& strncmp(name->s, it->name.s, name->len)==0)
|
|
{
|
|
/* update value */
|
|
it->expire = now;
|
|
ht_slot_unlock(ht, idx);
|
|
return 0;
|
|
}
|
|
it = it->next;
|
|
}
|
|
ht_slot_unlock(ht, idx);
|
|
return 0;
|
|
}
|
|
|
|
int ht_get_cell_expire(ht_t *ht, str *name, unsigned int *val)
|
|
{
|
|
unsigned int idx;
|
|
unsigned int hid;
|
|
ht_cell_t *it;
|
|
time_t now;
|
|
|
|
if(ht==NULL || ht->entries==NULL)
|
|
return -1;
|
|
|
|
*val = 0;
|
|
/* not auto-expire htable */
|
|
if(ht->htexpire==0)
|
|
return 0;
|
|
|
|
hid = ht_compute_hash(name);
|
|
|
|
idx = ht_get_entry(hid, ht->htsize);
|
|
|
|
now = time(NULL);
|
|
ht_slot_lock(ht, idx);
|
|
it = ht->entries[idx].first;
|
|
while(it!=NULL && it->cellid < hid)
|
|
it = it->next;
|
|
while(it!=NULL && it->cellid == hid)
|
|
{
|
|
if(name->len==it->name.len
|
|
&& strncmp(name->s, it->name.s, name->len)==0)
|
|
{
|
|
/* update value */
|
|
*val = (unsigned int)(it->expire - now);
|
|
ht_slot_unlock(ht, idx);
|
|
return 0;
|
|
}
|
|
it = it->next;
|
|
}
|
|
ht_slot_unlock(ht, idx);
|
|
return 0;
|
|
}
|
|
|
|
int ht_rm_cell_re(str *sre, ht_t *ht, int mode)
|
|
{
|
|
ht_cell_t *it;
|
|
ht_cell_t *it0;
|
|
int i;
|
|
regex_t re;
|
|
int match;
|
|
regmatch_t pmatch;
|
|
|
|
if(sre==NULL || sre->len<=0 || ht==NULL)
|
|
return -1;
|
|
|
|
if (regcomp(&re, sre->s, REG_EXTENDED|REG_ICASE|REG_NEWLINE))
|
|
{
|
|
LM_ERR("bad re %s\n", sre->s);
|
|
return -1;
|
|
}
|
|
|
|
for(i=0; i<ht->htsize; i++)
|
|
{
|
|
/* free entries */
|
|
ht_slot_lock(ht, i);
|
|
it = ht->entries[i].first;
|
|
while(it)
|
|
{
|
|
it0 = it->next;
|
|
match = 0;
|
|
if(mode==0)
|
|
{
|
|
if (regexec(&re, it->name.s, 1, &pmatch, 0)==0)
|
|
match = 1;
|
|
} else {
|
|
if(it->flags&AVP_VAL_STR)
|
|
if (regexec(&re, it->value.s.s, 1, &pmatch, 0)==0)
|
|
match = 1;
|
|
}
|
|
if(match==1)
|
|
{
|
|
if(it->prev==NULL)
|
|
ht->entries[i].first = it->next;
|
|
else
|
|
it->prev->next = it->next;
|
|
if(it->next)
|
|
it->next->prev = it->prev;
|
|
ht->entries[i].esize--;
|
|
ht_cell_free(it);
|
|
}
|
|
it = it0;
|
|
}
|
|
ht_slot_unlock(ht, i);
|
|
}
|
|
regfree(&re);
|
|
return 0;
|
|
}
|
|
|
|
int ht_reset_content(ht_t *ht)
|
|
{
|
|
ht_cell_t *it;
|
|
ht_cell_t *it0;
|
|
int i;
|
|
|
|
if(ht==NULL)
|
|
return -1;
|
|
|
|
for(i=0; i<ht->htsize; i++)
|
|
{
|
|
/* free entries */
|
|
ht_slot_lock(ht, i);
|
|
it = ht->entries[i].first;
|
|
while(it)
|
|
{
|
|
it0 = it->next;
|
|
if(it->prev==NULL)
|
|
ht->entries[i].first = it->next;
|
|
else
|
|
it->prev->next = it->next;
|
|
if(it->next)
|
|
it->next->prev = it->prev;
|
|
ht->entries[i].esize--;
|
|
ht_cell_free(it);
|
|
it = it0;
|
|
}
|
|
ht_slot_unlock(ht, i);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int ht_count_cells_re(str *sre, ht_t *ht, int mode)
|
|
{
|
|
ht_cell_t *it;
|
|
ht_cell_t *it0;
|
|
int i;
|
|
regex_t re;
|
|
regmatch_t pmatch;
|
|
int cnt = 0;
|
|
int op = 0;
|
|
str sval;
|
|
str tval;
|
|
int ival = 0;
|
|
|
|
if(sre==NULL || sre->len<=0 || ht==NULL)
|
|
return 0;
|
|
|
|
if(sre->len>=2)
|
|
{
|
|
switch(sre->s[0]) {
|
|
case '~':
|
|
switch(sre->s[1]) {
|
|
case '~':
|
|
op = 1; /* regexp */
|
|
break;
|
|
case '%':
|
|
op = 2; /* rlike */
|
|
break;
|
|
}
|
|
break;
|
|
case '%':
|
|
switch(sre->s[1]) {
|
|
case '~':
|
|
op = 3; /* llike */
|
|
break;
|
|
}
|
|
break;
|
|
case '=':
|
|
switch(sre->s[1]) {
|
|
case '=':
|
|
op = 4; /* str eq */
|
|
break;
|
|
}
|
|
break;
|
|
case 'e':
|
|
switch(sre->s[1]) {
|
|
case 'q':
|
|
op = 5; /* int eq */
|
|
break;
|
|
}
|
|
break;
|
|
case '*':
|
|
switch(sre->s[1]) {
|
|
case '*':
|
|
op = 6; /* int eq */
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
if(op==6) {
|
|
/* count all */
|
|
for(i=0; i<ht->htsize; i++)
|
|
cnt += ht->entries[i].esize;
|
|
return cnt;
|
|
}
|
|
|
|
if(op > 0) {
|
|
if(sre->len<=2)
|
|
return 0;
|
|
sval = *sre;
|
|
sval.s += 2;
|
|
sval.len -= 2;
|
|
if(op==5) {
|
|
if(mode==0)
|
|
{
|
|
/* match by name */
|
|
return 0;
|
|
}
|
|
str2sint(&sval, &ival);
|
|
}
|
|
} else {
|
|
sval = *sre;
|
|
op = 1;
|
|
}
|
|
|
|
if(op==1)
|
|
{
|
|
if (regcomp(&re, sval.s, REG_EXTENDED|REG_ICASE|REG_NEWLINE))
|
|
{
|
|
LM_ERR("bad re %s\n", sre->s);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
for(i=0; i<ht->htsize; i++)
|
|
{
|
|
/* free entries */
|
|
ht_slot_lock(ht, i);
|
|
it = ht->entries[i].first;
|
|
while(it)
|
|
{
|
|
it0 = it->next;
|
|
if(op==5)
|
|
{
|
|
if(!(it->flags&AVP_VAL_STR))
|
|
if( it->value.n==ival)
|
|
cnt++;
|
|
} else {
|
|
tval.len = -1;
|
|
if(mode==0)
|
|
{
|
|
/* match by name */
|
|
tval = it->name;
|
|
} else {
|
|
if(it->flags&AVP_VAL_STR)
|
|
tval = it->value.s;
|
|
}
|
|
if(tval.len>-1) {
|
|
switch(op) {
|
|
case 1: /* regexp */
|
|
if (regexec(&re, tval.s, 1, &pmatch, 0)==0)
|
|
cnt++;
|
|
break;
|
|
case 2: /* rlike */
|
|
if(sval.len<=tval.len
|
|
&& strncmp(sval.s,
|
|
tval.s+tval.len-sval.len, sval.len)==0)
|
|
cnt++;
|
|
break;
|
|
case 3: /* llike */
|
|
if(sval.len<=tval.len
|
|
&& strncmp(sval.s, tval.s, sval.len)==0)
|
|
cnt++;
|
|
break;
|
|
case 4: /* str eq */
|
|
if(sval.len==tval.len
|
|
&& strncmp(sval.s, tval.s, sval.len)==0)
|
|
cnt++;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
it = it0;
|
|
}
|
|
ht_slot_unlock(ht, i);
|
|
}
|
|
if(op==1)
|
|
regfree(&re);
|
|
return cnt;
|
|
}
|
|
|
|
#define HT_ITERATOR_SIZE 4
|
|
#define HT_ITERATOR_NAME_SIZE 32
|
|
|
|
typedef struct ht_iterator {
|
|
str name;
|
|
char bname[HT_ITERATOR_NAME_SIZE];
|
|
ht_t *ht;
|
|
int slot;
|
|
ht_cell_t *it;
|
|
} ht_iterator_t;
|
|
|
|
static ht_iterator_t _ht_iterators[HT_ITERATOR_SIZE];
|
|
|
|
void ht_iterator_init(void)
|
|
{
|
|
memset(_ht_iterators, 0, HT_ITERATOR_SIZE*sizeof(ht_iterator_t));
|
|
}
|
|
|
|
int ht_iterator_start(str *iname, str *hname)
|
|
{
|
|
int i;
|
|
int k;
|
|
|
|
k = -1;
|
|
for(i=0; i<HT_ITERATOR_SIZE; i++)
|
|
{
|
|
if(_ht_iterators[i].name.len>0)
|
|
{
|
|
if(_ht_iterators[i].name.len==iname->len
|
|
&& strncmp(_ht_iterators[i].name.s, iname->s, iname->len)==0)
|
|
{
|
|
k = i;
|
|
break;
|
|
}
|
|
} else {
|
|
if(k==-1) k = i;
|
|
}
|
|
}
|
|
if(k==-1)
|
|
{
|
|
LM_ERR("no iterator available - max number is %d\n", HT_ITERATOR_SIZE);
|
|
return -1;
|
|
}
|
|
if(_ht_iterators[k].name.len>0)
|
|
{
|
|
if(_ht_iterators[k].ht!=NULL && _ht_iterators[k].it!=NULL)
|
|
{
|
|
if(_ht_iterators[k].slot>=0 && _ht_iterators[k].slot<_ht_iterators[k].ht->htsize)
|
|
{
|
|
ht_slot_unlock(_ht_iterators[k].ht, _ht_iterators[k].slot);
|
|
}
|
|
}
|
|
} else {
|
|
if(iname->len>=HT_ITERATOR_NAME_SIZE)
|
|
{
|
|
LM_ERR("iterator name is too big [%.*s] (max %d)\n",
|
|
iname->len, iname->s, HT_ITERATOR_NAME_SIZE);
|
|
return -1;
|
|
}
|
|
strncpy(_ht_iterators[k].bname, iname->s, iname->len);
|
|
_ht_iterators[k].bname[iname->len] = '\0';
|
|
_ht_iterators[k].name.len = iname->len;
|
|
_ht_iterators[k].name.s = _ht_iterators[k].bname;
|
|
}
|
|
_ht_iterators[k].it = NULL;
|
|
_ht_iterators[k].slot = 0;
|
|
_ht_iterators[k].ht = ht_get_table(hname);
|
|
if(_ht_iterators[k].ht==NULL)
|
|
{
|
|
LM_ERR("cannot get hash table [%.*s]\n", hname->len, hname->s);
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int ht_iterator_next(str *iname)
|
|
{
|
|
int i;
|
|
int k;
|
|
|
|
k = -1;
|
|
for(i=0; i<HT_ITERATOR_SIZE; i++)
|
|
{
|
|
if(_ht_iterators[i].name.len>0)
|
|
{
|
|
if(_ht_iterators[i].name.len==iname->len
|
|
&& strncmp(_ht_iterators[i].name.s, iname->s, iname->len)==0)
|
|
{
|
|
k = i;
|
|
break;
|
|
}
|
|
} else {
|
|
if(k==-1) k = i;
|
|
}
|
|
}
|
|
if(k==-1)
|
|
{
|
|
LM_ERR("iterator not found [%.*s]\n", iname->len, iname->s);
|
|
return -1;
|
|
}
|
|
if(_ht_iterators[k].ht==NULL)
|
|
{
|
|
LM_ERR("iterator not initialized [%.*s]\n", iname->len, iname->s);
|
|
return -1;
|
|
}
|
|
if(_ht_iterators[k].it==NULL)
|
|
{
|
|
/* first execution - start from first slot */
|
|
_ht_iterators[k].slot=0;
|
|
} else {
|
|
_ht_iterators[k].it = _ht_iterators[k].it->next;
|
|
if(_ht_iterators[k].it!=NULL)
|
|
{
|
|
/* next item is in the same slot */
|
|
return 0;
|
|
}
|
|
/* next is not in the same slot - release and try next one */
|
|
_ht_iterators[k].it = NULL;
|
|
ht_slot_unlock(_ht_iterators[k].ht, _ht_iterators[k].slot);
|
|
_ht_iterators[k].slot++;
|
|
}
|
|
|
|
for( ; _ht_iterators[k].slot<_ht_iterators[k].ht->htsize; _ht_iterators[k].slot++)
|
|
{
|
|
ht_slot_lock(_ht_iterators[k].ht, _ht_iterators[k].slot);
|
|
if(_ht_iterators[k].ht->entries[_ht_iterators[k].slot].first!=NULL)
|
|
{
|
|
_ht_iterators[k].it = _ht_iterators[k].ht->entries[_ht_iterators[k].slot].first;
|
|
return 0;
|
|
}
|
|
ht_slot_unlock(_ht_iterators[k].ht, _ht_iterators[k].slot);
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
int ht_iterator_end(str *iname)
|
|
{
|
|
int i;
|
|
|
|
for(i=0; i<HT_ITERATOR_SIZE; i++)
|
|
{
|
|
if(_ht_iterators[i].name.len>0)
|
|
{
|
|
if(_ht_iterators[i].name.len==iname->len
|
|
&& strncmp(_ht_iterators[i].name.s, iname->s, iname->len)==0)
|
|
{
|
|
if(_ht_iterators[i].ht!=NULL && _ht_iterators[i].it!=NULL)
|
|
{
|
|
if(_ht_iterators[i].slot>=0 && _ht_iterators[i].slot<_ht_iterators[i].ht->htsize)
|
|
{
|
|
ht_slot_unlock(_ht_iterators[i].ht, _ht_iterators[i].slot);
|
|
}
|
|
}
|
|
memset(&_ht_iterators[i], 0, sizeof(ht_iterator_t));
|
|
return 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
ht_cell_t* ht_iterator_get_current(str *iname)
|
|
{
|
|
int i;
|
|
if(iname==NULL || iname->len<=0)
|
|
return NULL;
|
|
|
|
for(i=0; i<HT_ITERATOR_SIZE; i++)
|
|
{
|
|
if(_ht_iterators[i].name.len>0)
|
|
{
|
|
if(_ht_iterators[i].name.len==iname->len
|
|
&& strncmp(_ht_iterators[i].name.s, iname->s, iname->len)==0)
|
|
{
|
|
return _ht_iterators[i].it;
|
|
}
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|