/*
* Update some rows in the specified table
* _h: structure representing database connection
* _k: key names
* _o: operators
* _v: values of the keys that must match
* _uk: updated columns
* _uv: updated values of the columns
* _n: number of key=value pairs
* _un: number of columns to update
*/
int perlvdb_db_update(db_con_t* h, db_key_t* k, db_op_t* o, db_val_t* v,
db_key_t* uk, db_val_t* uv, int n, int un) {
AV *condarr;
AV *updatearr;
SV *condarrref;
SV *updatearrref;
SV *ret;
condarr = conds2perlarray(k, o, v, n);
updatearr = pairs2perlarray(uk, uv, un);
condarrref = newRV_noinc((SV*)condarr);
updatearrref = newRV_noinc((SV*)updatearr);
ret = perlvdb_perlmethod(getobj(h), PERL_VDB_UPDATEMETHOD,
condarrref, updatearrref, NULL, NULL);
av_undef(condarr);
av_undef(updatearr);
return IV2int(ret);
}
/*
* Query table for specified rows
* h: structure representing database connection
* k: key names
* op: operators
* v: values of the keys that must match
* c: column names to return
* n: number of key=values pairs to compare
* nc: number of columns to return
* o: order by the specified column
*/
int perlvdb_db_query(db_con_t* h, db_key_t* k, db_op_t* op, db_val_t* v,
db_key_t* c, int n, int nc,
db_key_t o, db_res_t** r) {
AV *condarr;
AV *retkeysarr;
SV *order;
SV *condarrref;
SV *retkeysref;
SV *resultset;
int retval = 0;
int i;
/* Create parameter set */
condarr = conds2perlarray(k, op, v, n);
retkeysarr = keys2perlarray(c, nc);
if (o) order = newSVpv(o->s, o->len);
else order = &PL_sv_undef;
condarrref = newRV_noinc((SV*)condarr);
retkeysref = newRV_noinc((SV*)retkeysarr);
/* Call perl method */
resultset = perlvdb_perlmethod(getobj(h), PERL_VDB_QUERYMETHOD,
condarrref, retkeysref, order, NULL);
SvREFCNT_dec(condarrref);
SvREFCNT_dec(retkeysref);
if(SvOK(order))
SvREFCNT_dec(order);
/* Transform perl result set to OpenSIPS result set */
if (!resultset) {
/* No results. */
retval = -1;
} else {
if (sv_isa(resultset, "OpenSIPS::VDB::Result")) {
retval = perlresult2dbres(resultset, r);
/* Nested refs are decreased/deleted inside the routine */
SvREFCNT_dec(resultset);
} else {
LM_ERR("invalid result set retrieved from perl call.\n");
retval = -1;
}
}
return retval;
}
/*
* Store name of table that will be used by
* subsequent database functions
*/
int perlvdb_use_table(db_con_t* h, const char* t) {
SV *ret;
if (!h || !t) {
LM_ERR("invalid parameter value\n");
return -1;
}
ret = perlvdb_perlmethod(getobj(h), PERL_VDB_USETABLEMETHOD,
sv_2mortal(newSVpv(t, 0)), NULL, NULL, NULL);
return IV2int(ret);
}
/*
* Delete a row from the specified table
* h: structure representing database connection
* k: key names
* o: operators
* v: values of the keys that must match
* n: number of key=value pairs
*/
int perlvdb_db_delete(db_con_t* h, db_key_t* k, db_op_t* o, db_val_t* v,
int n) {
AV *arr;
SV *arrref;
SV *ret;
arr = conds2perlarray(k, o, v, n);
arrref = newRV_noinc((SV*)arr);
ret = perlvdb_perlmethod(getobj(h), PERL_VDB_DELETEMETHOD,
arrref, NULL, NULL, NULL);
av_undef(arr);
return IV2int(ret);
}
/*
* Insert a row into specified table
* h: structure representing database connection
* k: key names
* v: values of the keys
* n: number of key=value pairs
*/
int perlvdb_db_insertreplace(db_con_t* h, db_key_t* k, db_val_t* v,
int n, char *insertreplace) {
AV *arr;
SV *arrref;
SV *ret;
arr = pairs2perlarray(k, v, n);
arrref = newRV_noinc((SV*)arr);
ret = perlvdb_perlmethod(getobj(h), insertreplace,
arrref, NULL, NULL, NULL);
av_undef(arr);
return IV2int(ret);
}
/*
* Store name of table that will be used by
* subsequent database functions
*/
int perlvdb_use_table(db_con_t* h, const str* t) {
SV *ret;
SV *table;
int res = -1;
if (!h || !t || !t->s) {
LM_ERR("invalid parameter value\n");
return -1;
}
table = newSVpv(t->s, t->len);
ret = perlvdb_perlmethod(getobj(h), PERL_VDB_USETABLEMETHOD,
table, NULL, NULL, NULL);
SvREFCNT_dec(table);
res = IV2int(ret);
return res;
}
int perlresult2dbres(SV *perlres, db1_res_t **r) {
SV *colarrayref = NULL;
AV *colarray = NULL;
SV *acol = NULL;
int colcount = 0;
SV *rowarrayref = NULL;
AV *rowarray = NULL;
int rowcount = 0;
SV *arowref = NULL;
AV *arow = NULL;
int arowlen = 0;
SV *aelement = NULL;
SV *atypesv = 0;
int atype = 0;
SV *aval = NULL;
char *charbuf;
char *currentstring;
int i, j;
int retval = 0;
STRLEN len;
SV *d1; /* helper variables */
/*db_val_t cur_val;*/ /* Abbreviation in "switch" below. The currently
modified db result value. */
if (!(SvROK(perlres) &&
(sv_derived_from(perlres, "Kamailio::VDB::Result")))) {
goto error;
}
/* Memory allocation for C side result structure */
*r = (db1_res_t *)pkg_malloc(sizeof(db1_res_t));
if (!(*r)) {
LM_ERR("no pkg memory left\n");
return -1;
}
memset(*r, 0, sizeof(db1_res_t));
/* Fetch column definitions */
colarrayref = perlvdb_perlmethod(perlres, PERL_VDB_COLDEFSMETHOD,
NULL, NULL, NULL, NULL);
if (!(SvROK(colarrayref))) goto error;
colarray = (AV *)SvRV(colarrayref);
if (!(SvTYPE(colarray) == SVt_PVAV)) goto error;
colcount = av_len(colarray) + 1;
/* Allocate col def memory */
(*r)->col.n = colcount;
(*r)->col.types = (db_type_t*)pkg_malloc(colcount*sizeof(db_type_t));
(*r)->col.names = (db_key_t*)pkg_malloc(colcount*sizeof(db_key_t));
/* reverse direction, as elements are removed by "SvREFCNT_dec" */
for (i = colcount-1; i >= 0; i--) {
acol = *av_fetch(colarray, i, 0);
d1 = perlvdb_perlmethod(acol, PERL_VDB_TYPEMETHOD,
NULL, NULL, NULL, NULL);
if (!SvIOK(d1)) goto error;
(*r)->col.types[i] = SvIV(d1);
SvREFCNT_dec(d1);
d1 = perlvdb_perlmethod(acol, PERL_VDB_NAMEMETHOD,
NULL, NULL, NULL, NULL);
if (!SvPOK(d1)) goto error;
currentstring = SvPV(d1, len);
charbuf = pkg_malloc(len+1);
strncpy(charbuf, currentstring, len+1);
(*r)->col.names[i]->s = charbuf;
(*r)->col.names[i]->len = strlen(charbuf);
SvREFCNT_dec(d1);
}
rowarrayref = perlvdb_perlmethod(perlres, PERL_VDB_ROWSMETHOD,
NULL, NULL, NULL, NULL);
if (!(SvROK(rowarrayref))) { /* Empty result set */
(*r)->n = 0;
(*r)->res_rows = 0;
(*r)->last_row = 0;
goto end;
}
rowarray = (AV *)SvRV(rowarrayref);
if (!(SvTYPE(rowarray) == SVt_PVAV)) goto error;
rowcount = av_len(rowarray) + 1;
(*r)->n = rowcount;
(*r)->res_rows = rowcount;
(*r)->last_row = rowcount;
(*r)->rows = (db_row_t *)pkg_malloc(rowcount*sizeof(db_row_t));
for (i = 0; i < rowcount; i++) {
arowref = *av_fetch(rowarray, 0, 0);
if (!SvROK(arowref)) goto error;
arow = (AV *)SvRV(arowref);
if (!(SvTYPE(colarray) == SVt_PVAV)) goto error;
arowlen = av_len(arow) + 1;
(*r)->rows[i].n = arowlen;
(*r)->rows[i].values =
(db_val_t *)pkg_malloc(arowlen*sizeof(db_val_t));
for (j = 0; j < arowlen; j++) {
aelement = *av_fetch(arow, j, 0);
#define cur_val (((*r)->rows)[i].values)[j]
/*cur_val = (((*r)->rows)[i].values)[j];*/
/* cur_val is just an "abbreviation" */
if (!(sv_isobject(aelement) &&
sv_derived_from(aelement, PERL_CLASS_VALUE))) {
cur_val.nul = 1;
continue;
}
atype = SvIV(atypesv = perlvdb_perlmethod(aelement,
PERL_VDB_TYPEMETHOD,
NULL, NULL, NULL, NULL));
aval = perlvdb_perlmethod(aelement, PERL_VDB_DATAMETHOD,
NULL, NULL, NULL, NULL);
(*r)->rows[i].values[j].type = atype;
if (!SvOK(aval)) {
cur_val.nul = 1;
} else {
switch (atype) {
case DB1_INT:
cur_val.val.int_val =
SvIV(aval);
cur_val.nul = 0;
break;
case DB1_DOUBLE:
cur_val.val.double_val =
SvNV(aval);
cur_val.nul = 0;
break;
case DB1_STRING:
case DB1_STR:
/* We dont support DB1_STR for now.
* Set DB1_STRING instead */
cur_val.type = DB1_STRING;
currentstring = SvPV(aval, len);
charbuf = pkg_malloc(len+1);
strncpy(charbuf, currentstring,
len+1);
cur_val.val.string_val =
charbuf;
cur_val.nul = 0;
break;
case DB1_DATETIME:
cur_val.val.time_val =
(time_t)SvIV(aval);
cur_val.nul = 0;
break;
case DB1_BLOB:
currentstring = SvPV(aval, len);
charbuf = pkg_malloc(len+1);
strncpy(charbuf, currentstring,
len+1);
cur_val.val.blob_val.s =
charbuf;
cur_val.val.blob_val.len = len;
cur_val.nul = 0;
break;
case DB1_BITMAP:
cur_val.val.bitmap_val =
SvIV(aval);
cur_val.nul = 0;
break;
default:
LM_CRIT("cannot handle this data type.\n");
return -1;
break;
}
}
SvREFCNT_dec(atypesv);
SvREFCNT_dec(aval);
}
}
end:
av_undef(colarray);
av_undef(rowarray);
return retval;
error:
LM_CRIT("broken result set. Exiting, leaving Kamailio in unknown state.\n");
return -1;
}
int perlresult2dbres(SV *perlres, db_res_t **r) {
HV * result = NULL;
SV *colarrayref = NULL;
AV *colarray = NULL;
SV *acol = NULL;
int colcount = 0;
SV *rowarrayref = NULL;
AV *rowarray = NULL;
int rowcount = 0;
SV *arowref = NULL;
AV *arow = NULL;
int arowlen = 0;
SV *aelement = NULL;
SV *atypesv = 0;
int atype = 0;
SV *aval = NULL;
char *charbuf;
char *currentstring;
int i, j;
int retval = 0;
STRLEN len;
SV *d1; /* helper variables */
/*db_val_t cur_val;*/ /* Abbreviation in "switch" below. The currently
modified db result value. */
if (!(SvROK(perlres) &&
(sv_derived_from(perlres, "OpenSIPS::VDB::Result")))) {
goto error;
}
result = (HV*)SvRV(perlres);
/* Memory allocation for C side result structure */
*r = db_new_result();
/* Fetch column definitions */
colarrayref = *hv_fetchs(result, PERL_VDB_COLDEFSMETHOD, 0);
/* colarrayref = perlvdb_perlmethod(perlres, PERL_VDB_COLDEFSMETHOD,
NULL, NULL, NULL, NULL); */
if (!(SvROK(colarrayref))) goto error;
colarray = (AV *)SvRV(colarrayref);
/* SvREFCNT_dec(colarray); */
if (!(SvTYPE(colarray) == SVt_PVAV)) goto error;
colcount = av_len(colarray) + 1;
RES_COL_N(*r) = colcount;
db_allocate_columns(*r, colcount);
/* reverse direction, as elements are removed by "SvREFCNT_dec" */
for (i = colcount-1; i >= 0; i--) {
acol = *av_fetch(colarray, i, 0);
d1 = perlvdb_perlmethod(acol, PERL_VDB_TYPEMETHOD,
NULL, NULL, NULL, NULL);
if (!SvIOK(d1)) goto error;
(*r)->col.types[i] = SvIV(d1);
SvREFCNT_dec(d1);
d1 = perlvdb_perlmethod(acol, PERL_VDB_NAMEMETHOD,
NULL, NULL, NULL, NULL);
if (!SvPOK(d1)) goto error;
currentstring = SvPV(d1, len);
charbuf = pkg_malloc(len+1);
/* Column names buffers are freed in the perlvdb free function */
strncpy(charbuf, currentstring, len+1);
(*r)->col.names[i]->s = charbuf;
(*r)->col.names[i]->len = strlen(charbuf);
SvREFCNT_dec(d1);
}
if(hv_exists(result, "rows", 4)){
rowarrayref =(SV*) hv_fetchs(result, "rows", 0);
}else{
(*r)->n = 0;
(*r)->res_rows = 0;
(*r)->last_row = 0;
goto end;
}
if(rowarrayref){
rowarrayref = *((SV**)rowarrayref);
}else{
(*r)->n = 0;
(*r)->res_rows = 0;
(*r)->last_row = 0;
goto end;
}
if (!(SvROK(rowarrayref))) { /* Empty result set */
(*r)->n = 0;
(*r)->res_rows = 0;
(*r)->last_row = 0;
goto end;
}
rowarray = (AV *)SvRV(rowarrayref);
if (!(SvTYPE(rowarray) == SVt_PVAV)) goto error;
rowcount = av_len(rowarray) + 1;
(*r)->n = rowcount;
(*r)->res_rows = rowcount;
(*r)->last_row = rowcount;
db_allocate_rows(*r, rowcount);
/* (rows * (sizeof(db_row_t) + sizeof(db_val_t) * RES_COL_N(_res)) */
/* LM_DBG("We got %d rows each row requres %d bytes because the row struct is %d and"
"the values in that row take up %d. That is %d values each size is %d\n",
rowcount, sizeof(db_row_t) + sizeof(db_val_t) * RES_COL_N(*r), sizeof(db_row_t), sizeof(db_val_t) * RES_COL_N(*r), RES_COL_N(*r), sizeof(db_val_t));
*/
for (i = 0; i < rowcount; i++) {
arowref = *av_fetch(rowarray, i, 0);
if (!SvROK(arowref)) goto error;
arow = (AV *)SvRV(arowref);
if (!(SvTYPE(colarray) == SVt_PVAV)) goto error;
arowlen = av_len(arow) + 1;
(*r)->rows[i].n = arowlen;
for (j = 0; j < arowlen; j++) {
aelement = *av_fetch(arow, j, 0);
#define cur_val (((*r)->rows)[i].values)[j]
/*cur_val = (((*r)->rows)[i].values)[j];*/
/* cur_val is just an "abbreviation" */
if (!(sv_isobject(aelement) &&
sv_derived_from(aelement, PERL_CLASS_VALUE))) {
cur_val.nul = 1;
continue;
}
atypesv = *hv_fetchs((HV*)SvRV(aelement),PERL_VDB_TYPEMETHOD,0); /*aelement->{type} */
atype = SvIV(atypesv);
/*atypesv = perlvdb_perlmethod(aelement,
PERL_VDB_TYPEMETHOD,
NULL, NULL, NULL, NULL);*/
aval = perlvdb_perlmethod(aelement, PERL_VDB_DATAMETHOD,
NULL, NULL, NULL, NULL);
(*r)->rows[i].values[j].type = atype;
/* SvREFCNT_dec(atypesv); */
if (!SvOK(aval)) {
cur_val.nul = 1;
} else {
switch (atype) {
case DB_INT:
cur_val.val.int_val =
SvIV(aval);
cur_val.nul = 0;
break;
case DB_DOUBLE:
cur_val.val.double_val =
SvNV(aval);
cur_val.nul = 0;
break;
case DB_STRING:
case DB_STR:
/* We dont support DB_STR for now.
* Set DB_STRING instead */
cur_val.type = DB_STRING;
currentstring = SvPV(aval, len);
charbuf = pkg_malloc(len+1);
strncpy(charbuf, currentstring,
len+1);
cur_val.val.string_val =
charbuf;
cur_val.nul = 0;
break;
case DB_DATETIME:
cur_val.val.time_val =
(time_t)SvIV(aval);
cur_val.nul = 0;
break;
case DB_BLOB:
currentstring = SvPV(aval, len);
charbuf = pkg_malloc(len+1);
strncpy(charbuf, currentstring,
len+1);
cur_val.val.blob_val.s =
charbuf;
cur_val.val.blob_val.len = len;
cur_val.nul = 0;
break;
case DB_BITMAP:
cur_val.val.bitmap_val =
SvIV(aval);
cur_val.nul = 0;
break;
default:
LM_CRIT("cannot handle this data type.\n");
return -1;
break;
}
}
SvREFCNT_dec(aval);
}
}
end:
return retval;
error:
LM_CRIT("broken result set. Exiting, leaving OpenSIPS in unknown state.\n");
return -1;
}
