Improve fix for bug #81740

This commit is contained in:
Stanislav Malyshev 2022-12-18 23:20:29 -07:00
parent 5975f33c37
commit a6a80eefe0
3 changed files with 17 additions and 4 deletions

View File

@ -1161,7 +1161,7 @@ PHP_METHOD(PDO, query)
PHP_METHOD(PDO, quote)
{
pdo_dbh_t *dbh = Z_PDO_DBH_P(ZEND_THIS);
zend_string *str;
zend_string *str, *quoted;
zend_long paramtype = PDO_PARAM_STR;
ZEND_PARSE_PARAMETERS_START(1, 2)
@ -1177,8 +1177,14 @@ PHP_METHOD(PDO, quote)
pdo_raise_impl_error(dbh, NULL, "IM001", "driver does not support quoting");
RETURN_FALSE;
}
quoted = dbh->methods->quoter(dbh, str, paramtype);
RETURN_STR(dbh->methods->quoter(dbh, str, paramtype));
if (quoted == NULL) {
PDO_HANDLE_DBH_ERR();
RETURN_FALSE;
}
RETURN_STR(quoted);
}
/* }}} */

View File

@ -242,6 +242,13 @@ safe:
if (buf) {
zend_string_release_ex(buf, 0);
}
if (plc->quoted == NULL) {
/* bork */
ret = -1;
strncpy(stmt->error_code, stmt->dbh->error_code, 6);
goto clean_up;
}
} else {
pdo_raise_impl_error(stmt->dbh, stmt, "HY105", "Expected a stream resource");
ret = -1;

View File

@ -227,8 +227,8 @@ static zend_string *pdo_sqlite_last_insert_id(pdo_dbh_t *dbh, const zend_string
static zend_string* sqlite_handle_quoter(pdo_dbh_t *dbh, const zend_string *unquoted, enum pdo_param_type paramtype)
{
char *quoted;
if (unquotedlen > (INT_MAX - 3) / 2) {
return 0;
if (ZSTR_LEN(unquoted) > (INT_MAX - 3) / 2) {
return NULL;
}
quoted = safe_emalloc(2, ZSTR_LEN(unquoted), 3);
/* TODO use %Q format? */