mirror of
https://github.com/asterisk/asterisk.git
synced 2025-10-18 02:32:36 +00:00
Ensure that all parts of SQL UPDATEs and DELETEs are encoded.
Patches: res_config_odbc.patch by John Hardin (License #6512) ........ Merged revisions 413304 from http://svn.asterisk.org/svn/asterisk/branches/1.8 ........ Merged revisions 413305 from http://svn.asterisk.org/svn/asterisk/branches/11 ........ Merged revisions 413306 from http://svn.asterisk.org/svn/asterisk/branches/12 git-svn-id: https://origsvn.digium.com/svn/asterisk/trunk@413307 65c4cc65-6c06-0410-ace0-fbb531ad65f3
This commit is contained in:
@@ -59,6 +59,26 @@ struct custom_prepare_struct {
|
||||
unsigned long long skip;
|
||||
};
|
||||
|
||||
#define ENCODE_CHUNK(buffer, s) \
|
||||
do { \
|
||||
char *eptr = buffer; \
|
||||
const char *vptr = s; \
|
||||
for (; *vptr && eptr < buffer + sizeof(buffer); vptr++) { \
|
||||
if (strchr("^;", *vptr)) { \
|
||||
/* We use ^XX, instead of %XX because '%' is a special character in SQL */ \
|
||||
snprintf(eptr, buffer + sizeof(buffer) - eptr, "^%02hhX", *vptr); \
|
||||
eptr += 3; \
|
||||
} else { \
|
||||
*eptr++ = *vptr; \
|
||||
} \
|
||||
} \
|
||||
if (eptr < buffer + sizeof(buffer)) { \
|
||||
*eptr = '\0'; \
|
||||
} else { \
|
||||
buffer[sizeof(buffer) - 1] = '\0'; \
|
||||
} \
|
||||
} while(0)
|
||||
|
||||
static void decode_chunk(char *chunk)
|
||||
{
|
||||
for (; *chunk; chunk++) {
|
||||
@@ -107,30 +127,23 @@ static SQLHSTMT custom_prepare(struct odbc_obj *obj, void *data)
|
||||
}
|
||||
ast_debug(1, "Parameter %d ('%s') = '%s'\n", x, field->name, newval);
|
||||
if (strchr(newval, ';') || strchr(newval, '^')) {
|
||||
char *eptr = encodebuf;
|
||||
const char *vptr = newval;
|
||||
for (; *vptr && eptr < encodebuf + sizeof(encodebuf); vptr++) {
|
||||
if (strchr("^;", *vptr)) {
|
||||
/* We use ^XX, instead of %XX because '%' is a special character in SQL */
|
||||
snprintf(eptr, encodebuf + sizeof(encodebuf) - eptr, "^%02hhX", *vptr);
|
||||
eptr += 3;
|
||||
} else {
|
||||
*eptr++ = *vptr;
|
||||
}
|
||||
}
|
||||
if (eptr < encodebuf + sizeof(encodebuf)) {
|
||||
*eptr = '\0';
|
||||
} else {
|
||||
encodebuf[sizeof(encodebuf) - 1] = '\0';
|
||||
}
|
||||
ENCODE_CHUNK(encodebuf, newval);
|
||||
ast_string_field_set(cps, encoding[x], encodebuf);
|
||||
newval = cps->encoding[x];
|
||||
}
|
||||
SQLBindParameter(stmt, x++, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, strlen(newval), 0, (void *)newval, 0, NULL);
|
||||
}
|
||||
|
||||
if (!ast_strlen_zero(cps->extra))
|
||||
SQLBindParameter(stmt, x++, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, strlen(cps->extra), 0, (void *)cps->extra, 0, NULL);
|
||||
if (!ast_strlen_zero(cps->extra)) {
|
||||
if (strchr(cps->extra, ';') || strchr(cps->extra, '^')) {
|
||||
ENCODE_CHUNK(encodebuf, cps->extra);
|
||||
SQLBindParameter(stmt, x++, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, strlen(encodebuf), 0, (void *)encodebuf, 0, NULL);
|
||||
}
|
||||
else {
|
||||
SQLBindParameter(stmt, x++, SQL_PARAM_INPUT, SQL_C_CHAR, SQL_CHAR, strlen(cps->extra), 0, (void *)cps->extra, 0, NULL);
|
||||
}
|
||||
}
|
||||
|
||||
return stmt;
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user