/*
* Print set clause of update SQL statement
*/
int db_print_set(const db1_con_t* _c, char* _b, const int _l, const db_key_t* _k,
const db_val_t* _v, const int _n, int (*val2str)(const db1_con_t*,
const db_val_t*,char*, int*))
{
int i, l, ret, len = 0;
if (!_c || !_b || !_l || !_k || !_v || !_n) {
LM_ERR("Invalid parameter value\n");
return -1;
}
for(i = 0; i < _n; i++) {
ret = snprintf(_b + len, _l - len, "%s%.*s%s=",
CON_TQUOTESZ(_c), _k[i]->len, _k[i]->s, CON_TQUOTESZ(_c));
if (ret < 0 || ret >= (_l - len)) goto error;
len += ret;
l = _l - len;
if ( (*val2str)(_c, &(_v[i]), _b + len, &l) < 0) {
LM_ERR("Error while converting value to string\n");
return -1;
}
len += l;
if (i != (_n - 1)) {
if ((_l - len) >= 1) {
*(_b + len++) = ',';
}
}
}
return len;
error:
LM_ERR("Error in snprintf\n");
return -1;
}
/**
* Insert a row into a specified table, update on duplicate key.
* \param _h structure representing database connection
* \param _k key names
* \param _v values of the keys
* \param _n number of key=value pairs
*/
int db_mysql_insert_update(const db1_con_t* _h, const db_key_t* _k, const db_val_t* _v,
const int _n)
{
int off, ret;
static str sql_str;
if ((!_h) || (!_k) || (!_v) || (!_n)) {
LM_ERR("invalid parameter value\n");
return -1;
}
ret = snprintf(mysql_sql_buf, sql_buffer_size, "insert into %s%.*s%s (",
CON_TQUOTESZ(_h), CON_TABLE(_h)->len, CON_TABLE(_h)->s, CON_TQUOTESZ(_h));
if (ret < 0 || ret >= sql_buffer_size) goto error;
off = ret;
ret = db_print_columns(mysql_sql_buf + off, sql_buffer_size - off, _k, _n, CON_TQUOTESZ(_h));
if (ret < 0) return -1;
off += ret;
ret = snprintf(mysql_sql_buf + off, sql_buffer_size - off, ") values (");
if (ret < 0 || ret >= (sql_buffer_size - off)) goto error;
off += ret;
ret = db_print_values(_h, mysql_sql_buf + off, sql_buffer_size - off, _v, _n, db_mysql_val2str);
if (ret < 0) return -1;
off += ret;
*(mysql_sql_buf + off++) = ')';
ret = snprintf(mysql_sql_buf + off, sql_buffer_size - off, " on duplicate key update ");
if (ret < 0 || ret >= (sql_buffer_size - off)) goto error;
off += ret;
ret = db_print_set(_h, mysql_sql_buf + off, sql_buffer_size - off, _k, _v, _n, db_mysql_val2str);
if (ret < 0) return -1;
off += ret;
sql_str.s = mysql_sql_buf;
sql_str.len = off;
if (db_mysql_submit_query(_h, &sql_str) < 0) {
LM_ERR("error while submitting query\n");
return -2;
}
return 0;
error:
LM_ERR("error while preparing insert_update operation\n");
return -1;
}
int db_do_insert_cmd(const db1_con_t* _h, const db_key_t* _k, const db_val_t* _v,
const int _n, int (*val2str) (const db1_con_t*, const db_val_t*, char*, int*),
int (*submit_query)(const db1_con_t* _h, const str* _c), int mode)
{
int off, ret;
if (!_h || !_k || !_v || !_n || !val2str || !submit_query) {
LM_ERR("invalid parameter value\n");
return -1;
}
if(mode==1)
ret = snprintf(sql_buf, sql_buffer_size, "insert delayed into %s%.*s%s (",
CON_TQUOTESZ(_h), CON_TABLE(_h)->len, CON_TABLE(_h)->s, CON_TQUOTESZ(_h));
else
ret = snprintf(sql_buf, sql_buffer_size, "insert into %s%.*s%s (",
CON_TQUOTESZ(_h), CON_TABLE(_h)->len, CON_TABLE(_h)->s, CON_TQUOTESZ(_h));
if (ret < 0 || ret >= sql_buffer_size) goto error;
off = ret;
ret = db_print_columns(sql_buf + off, sql_buffer_size - off, _k, _n, CON_TQUOTESZ(_h));
if (ret < 0) return -1;
off += ret;
ret = snprintf(sql_buf + off, sql_buffer_size - off, ") values (");
if (ret < 0 || ret >= (sql_buffer_size - off)) goto error;
off += ret;
ret = db_print_values(_h, sql_buf + off, sql_buffer_size - off, _v, _n, val2str);
if (ret < 0) return -1;
off += ret;
if (off + 2 > sql_buffer_size) goto error;
sql_buf[off++] = ')';
sql_buf[off] = '\0';
sql_str.s = sql_buf;
sql_str.len = off;
if (db_do_submit_query(_h, &sql_str, submit_query) < 0) {
LM_ERR("error while submitting query\n");
return -2;
}
return 0;
error:
LM_ERR("error while preparing insert operation\n");
return -1;
}
int db_do_update(const db1_con_t* _h, const db_key_t* _k, const db_op_t* _o,
const db_val_t* _v, const db_key_t* _uk, const db_val_t* _uv, const int _n,
const int _un, int (*val2str) (const db1_con_t*, const db_val_t*, char*, int*),
int (*submit_query)(const db1_con_t* _h, const str* _c))
{
int off, ret;
if (!_h || !_uk || !_uv || !_un || !val2str || !submit_query) {
LM_ERR("invalid parameter value\n");
return -1;
}
ret = snprintf(sql_buf, sql_buffer_size, "update %s%.*s%s set ",
CON_TQUOTESZ(_h), CON_TABLE(_h)->len, CON_TABLE(_h)->s, CON_TQUOTESZ(_h));
if (ret < 0 || ret >= sql_buffer_size) goto error;
off = ret;
ret = db_print_set(_h, sql_buf + off, sql_buffer_size - off, _uk, _uv, _un, val2str);
if (ret < 0) return -1;
off += ret;
if (_n) {
ret = snprintf(sql_buf + off, sql_buffer_size - off, " where ");
if (ret < 0 || ret >= (sql_buffer_size - off)) goto error;
off += ret;
ret = db_print_where(_h, sql_buf + off, sql_buffer_size - off, _k, _o, _v, _n, val2str);
if (ret < 0) return -1;
off += ret;
}
if (off + 1 > sql_buffer_size) goto error;
sql_buf[off] = '\0';
sql_str.s = sql_buf;
sql_str.len = off;
if (db_do_submit_query(_h, &sql_str, submit_query) < 0) {
LM_ERR("error while submitting query\n");
return -2;
}
return 0;
error:
LM_ERR("error while preparing update operation\n");
return -1;
}
/*
* Print where clause of SQL statement
*/
int db_print_where(const db1_con_t* _c, char* _b, const int _l, const db_key_t* _k,
const db_op_t* _o, const db_val_t* _v, const int _n, int (*val2str)
(const db1_con_t*, const db_val_t*, char*, int*))
{
int i, l, ret, len = 0;
if (!_c || !_b || !_l || !_k || !_v || !_n) {
LM_ERR("Invalid parameter value\n");
return -1;
}
for(i = 0; i < _n; i++) {
if (_o && strncmp(_o[i], OP_BITWISE_AND, 1) == 0) {
char tmp_buf[16];
int tmp_len = 15;
memset(tmp_buf, 0, 16);
if ((*val2str)(_c, &(_v[i]), tmp_buf, &tmp_len) < 0) {
LM_ERR("Error while converting value to string\n");
return -1;
}
ret = snprintf(_b + len, _l - len, "%s%.*s%s&%.*s=%.*s", CON_TQUOTESZ(_c),
_k[i]->len, _k[i]->s, CON_TQUOTESZ(_c), tmp_len, tmp_buf, tmp_len, tmp_buf);
if (ret < 0 || ret >= (_l - len)) goto error;
len += ret;
} else {
if (_o) {
ret = snprintf(_b + len, _l - len, "%s%.*s%s%s", CON_TQUOTESZ(_c),
_k[i]->len, _k[i]->s, CON_TQUOTESZ(_c), _o[i]);
if (ret < 0 || ret >= (_l - len)) goto error;
len += ret;
} else {
ret = snprintf(_b + len, _l - len, "%s%.*s%s=", CON_TQUOTESZ(_c),
_k[i]->len, _k[i]->s, CON_TQUOTESZ(_c));
if (ret < 0 || ret >= (_l - len)) goto error;
len += ret;
}
l = _l - len;
if ( (*val2str)(_c, &(_v[i]), _b + len, &l) < 0) {
LM_ERR("Error while converting value to string\n");
return -1;
}
len += l;
}
if (i != (_n - 1)) {
ret = snprintf(_b + len, _l - len, " AND ");
if (ret < 0 || ret >= (_l - len)) goto error;
len += ret;
}
}
return len;
error:
LM_ERR("Error in snprintf\n");
return -1;
}
static int db_do_query_internal(const db1_con_t* _h, const db_key_t* _k, const db_op_t* _op,
const db_val_t* _v, const db_key_t* _c, const int _n, const int _nc,
const db_key_t _o, db1_res_t** _r, int (*val2str) (const db1_con_t*,
const db_val_t*, char*, int* _len), int (*submit_query)(const db1_con_t*,
const str*), int (*store_result)(const db1_con_t* _h, db1_res_t** _r), int _l)
{
int off, ret;
if (!_h || !val2str || !submit_query || !store_result) {
LM_ERR("invalid parameter value\n");
return -1;
}
if (!_c) {
ret = snprintf(sql_buf, sql_buffer_size, "select * from %s%.*s%s ",
CON_TQUOTESZ(_h), CON_TABLE(_h)->len, CON_TABLE(_h)->s, CON_TQUOTESZ(_h));
if (ret < 0 || ret >= sql_buffer_size) goto error;
off = ret;
} else {
ret = snprintf(sql_buf, sql_buffer_size, "select ");
if (ret < 0 || ret >= sql_buffer_size) goto error;
off = ret;
ret = db_print_columns(sql_buf + off, sql_buffer_size - off, _c, _nc, CON_TQUOTESZ(_h));
if (ret < 0) return -1;
off += ret;
ret = snprintf(sql_buf + off, sql_buffer_size - off, "from %s%.*s%s ",
CON_TQUOTESZ(_h), CON_TABLE(_h)->len, CON_TABLE(_h)->s, CON_TQUOTESZ(_h));
if (ret < 0 || ret >= (sql_buffer_size - off)) goto error;
off += ret;
}
if (_n) {
ret = snprintf(sql_buf + off, sql_buffer_size - off, "where ");
if (ret < 0 || ret >= (sql_buffer_size - off)) goto error;
off += ret;
ret = db_print_where(_h, sql_buf + off,
sql_buffer_size - off, _k, _op, _v, _n, val2str);
if (ret < 0) return -1;;
off += ret;
}
if (_o) {
ret = snprintf(sql_buf + off, sql_buffer_size - off, " order by %.*s", _o->len, _o->s);
if (ret < 0 || ret >= (sql_buffer_size - off)) goto error;
off += ret;
}
if (_l) {
ret = snprintf(sql_buf + off, sql_buffer_size - off, " for update");
if (ret < 0 || ret >= (sql_buffer_size - off)) goto error;
off += ret;
}
/*
* Null-terminate the string for the postgres driver. Its query function
* don't support a length parameter, so they need this for the correct
* function of strlen. This zero is not included in the 'str' length.
* We need to check the length here, otherwise we could overwrite the buffer
* boundaries if off is equal to sql_buffer_size.
*/
if (off + 1 >= sql_buffer_size) goto error;
sql_buf[off + 1] = '\0';
sql_str.s = sql_buf;
sql_str.len = off;
if (db_do_submit_query(_h, &sql_str, submit_query) < 0) {
LM_ERR("error while submitting query\n");
return -2;
}
if(_r) {
int tmp = store_result(_h, _r);
if (tmp < 0) {
LM_ERR("error while storing result");
return tmp;
}
}
return 0;
error:
LM_ERR("error while preparing query\n");
return -1;
}
