kenmirrors/asterisk
1
0
Fork 0
mirror of https://github.com/asterisk/asterisk.git synced 2025-11-02 03:48:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merged revisions 145076 via svnmerge from

Browse Source 
https://origsvn.digium.com/svn/asterisk/trunk

........
  r145076 | tilghman | 2008-09-28 16:39:07 -0500 (Sun, 28 Sep 2008) | 4 lines
  
  Change several improper "sizeof" to "strlen", as sizeof in that context would
  incorrectly use the size of a pointer, rather than the length of a string.
  (Closes issue #13574)
........


git-svn-id: https://origsvn.digium.com/svn/asterisk/branches/1.6.0@145123 65c4cc65-6c06-0410-ace0-fbb531ad65f3
This commit is contained in:
Tilghman Lesher
2008-09-29 00:13:00 +00:00
parent ae6e72ad45
commit e747412f1f
1 changed files with 56 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
res/res_config_pgsql.c
View File

@@ -470,14 +470,22 @@ static int update_pgsql(const char *database, const char *table, const char *key
return -1;
}
#define ESCAPE_STRING(buffer, stringname) \
do { \
int len; \
if ((len = strlen(stringname)) > (buffer->len - 1) / 2) { \
ast_str_make_space(&buffer, len * 2 + 1); \
} \
PQescapeStringConn(pgsqlConn, buffer->str, stringname, len, &pgresult); \
} while (0)
static int store_pgsql(const char *database, const char *table, va_list ap)
{
PGresult *result = NULL;
Oid insertid;
char sql[256];
char params[256];
char vals[256];
char buf[256];
struct ast_str *buf = ast_str_create(256);
struct ast_str *sql1 = ast_str_create(256);
struct ast_str *sql2 = ast_str_create(256);
int pgresult;
const char *newparam, *newval;
@@ -495,7 +503,7 @@ static int store_pgsql(const char *database, const char *table, va_list ap)
if (pgsqlConn) {
PQfinish(pgsqlConn);
pgsqlConn = NULL;
};
}
return -1;
}
@@ -508,28 +516,31 @@ static int store_pgsql(const char *database, const char *table, va_list ap)
/* Create the first part of the query using the first parameter/value pairs we just extracted
If there is only 1 set, then we have our query. Otherwise, loop thru the list and concat */
PQescapeStringConn(pgsqlConn, buf, newparam, sizeof(newparam), &pgresult);
snprintf(params, sizeof(params), "%s", buf);
PQescapeStringConn(pgsqlConn, buf, newval, sizeof(newval), &pgresult);
snprintf(vals, sizeof(vals), "'%s'", buf);
ESCAPE_STRING(buf, newparam);
ast_str_set(&sql1, 0, "INSERT INTO %s (%s", table, buf->str);
ESCAPE_STRING(buf, newval);
ast_str_set(&sql2, 0, ") VALUES ('%s'", buf->str);
while ((newparam = va_arg(ap, const char *))) {
newval = va_arg(ap, const char *);
PQescapeStringConn(pgsqlConn, buf, newparam, sizeof(newparam), &pgresult);
snprintf(params + strlen(params), sizeof(params) - strlen(params), ", %s", buf);
PQescapeStringConn(pgsqlConn, buf, newval, sizeof(newval), &pgresult);
snprintf(vals + strlen(vals), sizeof(vals) - strlen(vals), ", '%s'", buf);
ESCAPE_STRING(buf, newparam);
ast_str_append(&sql1, 0, ", %s", buf->str);
ESCAPE_STRING(buf, newval);
ast_str_append(&sql2, 0, ", '%s'", buf->str);
}
va_end(ap);
snprintf(sql, sizeof(sql), "INSERT INTO (%s) VALUES (%s)", params, vals);
ast_str_append(&sql1, 0, "%s)", sql2->str);
ast_debug(1, "PostgreSQL RealTime: Insert SQL: %s\n", sql);
ast_debug(1, "PostgreSQL RealTime: Insert SQL: %s\n", sql1->str);
if (!(result = PQexec(pgsqlConn, sql))) {
if (!(result = PQexec(pgsqlConn, sql1->str))) {
ast_log(LOG_WARNING,
"PostgreSQL RealTime: Failed to query database. Check debug for more info.\n");
ast_debug(1, "PostgreSQL RealTime: Query: %s\n", sql);
ast_debug(1, "PostgreSQL RealTime: Query: %s\n", sql1->str);
ast_debug(1, "PostgreSQL RealTime: Query Failed because: %s\n", PQerrorMessage(pgsqlConn));
ast_mutex_unlock(&pgsql_lock);
ast_free(sql1);
ast_free(sql2);
ast_free(buf);
return -1;
} else {
ExecStatusType result_status = PQresultStatus(result);
@@ -538,16 +549,22 @@ static int store_pgsql(const char *database, const char *table, va_list ap)
&& result_status != PGRES_NONFATAL_ERROR) {
ast_log(LOG_WARNING,
"PostgreSQL RealTime: Failed to query database. Check debug for more info.\n");
ast_debug(1, "PostgreSQL RealTime: Query: %s\n", sql);
ast_debug(1, "PostgreSQL RealTime: Query: %s\n", sql1->str);
ast_debug(1, "PostgreSQL RealTime: Query Failed because: %s (%s)\n",
PQresultErrorMessage(result), PQresStatus(result_status));
ast_mutex_unlock(&pgsql_lock);
ast_free(sql1);
ast_free(sql2);
ast_free(buf);
return -1;
}
}
insertid = PQoidValue(result);
ast_mutex_unlock(&pgsql_lock);
ast_free(sql1);
ast_free(sql2);
ast_free(buf);
ast_debug(1, "PostgreSQL RealTime: row inserted on table: %s, id: %u\n", table, insertid);
@@ -568,8 +585,8 @@ static int destroy_pgsql(const char *database, const char *table, const char *ke
PGresult *result = NULL;
int numrows = 0;
int pgresult;
char sql[256];
char buf[256], buf2[256];
struct ast_str *sql = ast_str_create(256);
struct ast_str *buf1 = ast_str_create(60), *buf2 = ast_str_create(60);
const char *newparam, *newval;
if (!table) {
@@ -602,25 +619,28 @@ static int destroy_pgsql(const char *database, const char *table, const char *ke
/* Create the first part of the query using the first parameter/value pairs we just extracted
If there is only 1 set, then we have our query. Otherwise, loop thru the list and concat */
PQescapeStringConn(pgsqlConn, buf, keyfield, sizeof(keyfield), &pgresult);
PQescapeStringConn(pgsqlConn, buf2, lookup, sizeof(lookup), &pgresult);
snprintf(sql, sizeof(sql), "DELETE FROM %s WHERE %s = '%s'", table, buf, buf2);
ESCAPE_STRING(buf1, keyfield);
ESCAPE_STRING(buf2, lookup);
ast_str_set(&sql, 0, "DELETE FROM %s WHERE %s = '%s'", table, buf1->str, buf2->str);
while ((newparam = va_arg(ap, const char *))) {
newval = va_arg(ap, const char *);
PQescapeStringConn(pgsqlConn, buf, newparam, sizeof(newparam), &pgresult);
PQescapeStringConn(pgsqlConn, buf2, newval, sizeof(newval), &pgresult);
snprintf(sql + strlen(sql), sizeof(sql) - strlen(sql), " AND %s = '%s'", buf, buf2);
ESCAPE_STRING(buf1, newparam);
ESCAPE_STRING(buf2, newval);
ast_str_append(&sql, 0, " AND %s = '%s'", buf1->str, buf2->str);
}
va_end(ap);
ast_debug(1, "PostgreSQL RealTime: Delete SQL: %s\n", sql);
ast_debug(1, "PostgreSQL RealTime: Delete SQL: %s\n", sql->str);
if (!(result = PQexec(pgsqlConn, sql))) {
if (!(result = PQexec(pgsqlConn, sql->str))) {
ast_log(LOG_WARNING,
"PostgreSQL RealTime: Failed to query database. Check debug for more info.\n");
ast_debug(1, "PostgreSQL RealTime: Query: %s\n", sql);
ast_debug(1, "PostgreSQL RealTime: Query: %s\n", sql->str);
ast_debug(1, "PostgreSQL RealTime: Query Failed because: %s\n", PQerrorMessage(pgsqlConn));
ast_mutex_unlock(&pgsql_lock);
ast_free(buf1);
ast_free(buf2);
ast_free(sql);
return -1;
} else {
ExecStatusType result_status = PQresultStatus(result);
@@ -629,16 +649,22 @@ static int destroy_pgsql(const char *database, const char *table, const char *ke
&& result_status != PGRES_NONFATAL_ERROR) {
ast_log(LOG_WARNING,
"PostgreSQL RealTime: Failed to query database. Check debug for more info.\n");
ast_debug(1, "PostgreSQL RealTime: Query: %s\n", sql);
ast_debug(1, "PostgreSQL RealTime: Query: %s\n", sql->str);
ast_debug(1, "PostgreSQL RealTime: Query Failed because: %s (%s)\n",
PQresultErrorMessage(result), PQresStatus(result_status));
ast_mutex_unlock(&pgsql_lock);
ast_free(buf1);
ast_free(buf2);
ast_free(sql);
return -1;
}
}
numrows = atoi(PQcmdTuples(result));
ast_mutex_unlock(&pgsql_lock);
ast_free(buf1);
ast_free(buf2);
ast_free(sql);
ast_debug(1, "PostgreSQL RealTime: Deleted %d rows on table: %s\n", numrows, table);