ext/pdo_sqlite: Use new F ZPP (#14059)

closes #14059
This commit is contained in:
Saki Takamachi 2024-04-28 16:16:58 +09:00
parent 6303d1fc6a
commit cf92a191e8
No known key found for this signature in database
GPG key ID: E4A36F6D37931A8B
6 changed files with 212 additions and 79 deletions

View file

@ -335,25 +335,18 @@ PHP_METHOD(PdoSqlite, openBlob)
static int php_sqlite_collation_callback(void *context, int string1_len, const void *string1, static int php_sqlite_collation_callback(void *context, int string1_len, const void *string1,
int string2_len, const void *string2) int string2_len, const void *string2)
{ {
int ret; int ret = 0;
zval zargs[2]; zval zargs[2];
zval retval; zval retval;
struct pdo_sqlite_collation *collation = (struct pdo_sqlite_collation*) context; struct pdo_sqlite_collation *collation = (struct pdo_sqlite_collation*) context;
collation->fc.fci.size = sizeof(collation->fc.fci);
ZVAL_COPY_VALUE(&collation->fc.fci.function_name, &collation->callback);
collation->fc.fci.object = NULL;
collation->fc.fci.retval = &retval;
// Prepare the arguments. // Prepare the arguments.
ZVAL_STRINGL(&zargs[0], (char *) string1, string1_len); ZVAL_STRINGL(&zargs[0], (char *) string1, string1_len);
ZVAL_STRINGL(&zargs[1], (char *) string2, string2_len); ZVAL_STRINGL(&zargs[1], (char *) string2, string2_len);
collation->fc.fci.param_count = 2;
collation->fc.fci.params = zargs;
if ((ret = zend_call_function(&collation->fc.fci, &collation->fc.fcc)) == FAILURE) { zend_call_known_fcc(&collation->callback, &retval, /* argc */ 2, zargs, /* named_params */ NULL);
php_error_docref(NULL, E_WARNING, "An error occurred while invoking the callback");
} else if (!Z_ISUNDEF(retval)) { if (!Z_ISUNDEF(retval)) {
if (Z_TYPE(retval) != IS_LONG) { if (Z_TYPE(retval) != IS_LONG) {
zend_string *func_name = get_active_function_or_method_name(); zend_string *func_name = get_active_function_or_method_name();
zend_type_error("%s(): Return value of the callback must be of type int, %s returned", zend_type_error("%s(): Return value of the callback must be of type int, %s returned",
@ -361,7 +354,6 @@ static int php_sqlite_collation_callback(void *context, int string1_len, const v
zend_string_release(func_name); zend_string_release(func_name);
return FAILURE; return FAILURE;
} }
ret = 0;
if (Z_LVAL(retval) > 0) { if (Z_LVAL(retval) > 0) {
ret = 1; ret = 1;
} else if (Z_LVAL(retval) < 0) { } else if (Z_LVAL(retval) < 0) {

View file

@ -26,28 +26,23 @@ typedef struct {
char *errmsg; char *errmsg;
} pdo_sqlite_error_info; } pdo_sqlite_error_info;
struct pdo_sqlite_fci {
zend_fcall_info fci;
zend_fcall_info_cache fcc;
};
struct pdo_sqlite_func { struct pdo_sqlite_func {
struct pdo_sqlite_func *next; struct pdo_sqlite_func *next;
zval func, step, fini;
int argc; int argc;
const char *funcname; const char *funcname;
/* accelerated callback references */ /* accelerated callback references */
struct pdo_sqlite_fci afunc, astep, afini; zend_fcall_info_cache func;
zend_fcall_info_cache step;
zend_fcall_info_cache fini;
}; };
struct pdo_sqlite_collation { struct pdo_sqlite_collation {
struct pdo_sqlite_collation *next; struct pdo_sqlite_collation *next;
const char *name; const char *name;
zval callback; zend_fcall_info_cache callback;
struct pdo_sqlite_fci fc;
}; };
typedef struct { typedef struct {

View file

@ -112,14 +112,14 @@ static void pdo_sqlite_cleanup_callbacks(pdo_sqlite_db_handle *H)
} }
efree((char*)func->funcname); efree((char*)func->funcname);
if (!Z_ISUNDEF(func->func)) { if (ZEND_FCC_INITIALIZED(func->func)) {
zval_ptr_dtor(&func->func); zend_fcc_dtor(&func->func);
} }
if (!Z_ISUNDEF(func->step)) { if (ZEND_FCC_INITIALIZED(func->step)) {
zval_ptr_dtor(&func->step); zend_fcc_dtor(&func->step);
} }
if (!Z_ISUNDEF(func->fini)) { if (ZEND_FCC_INITIALIZED(func->fini)) {
zval_ptr_dtor(&func->fini); zend_fcc_dtor(&func->fini);
} }
efree(func); efree(func);
} }
@ -139,8 +139,8 @@ static void pdo_sqlite_cleanup_callbacks(pdo_sqlite_db_handle *H)
} }
efree((char*)collation->name); efree((char*)collation->name);
if (!Z_ISUNDEF(collation->callback)) { if (ZEND_FCC_INITIALIZED(collation->callback)) {
zval_ptr_dtor(&collation->callback); zend_fcc_dtor(&collation->callback);
} }
efree(collation); efree(collation);
} }
@ -309,12 +309,12 @@ typedef struct {
zend_long row; zend_long row;
} aggregate_context; } aggregate_context;
static int do_callback(struct pdo_sqlite_fci *fc, zval *cb, int argc, sqlite3_value **argv, sqlite3_context *context, int is_agg) static int do_callback(zend_fcall_info_cache *fcc, int argc, sqlite3_value **argv, sqlite3_context *context, int is_agg)
{ {
zval *zargs = NULL; zval *zargs = NULL;
zval retval; zval retval;
int i; int i;
int ret; int ret = SUCCESS;
int fake_argc; int fake_argc;
aggregate_context *agg_context = NULL; aggregate_context *agg_context = NULL;
@ -324,14 +324,7 @@ static int do_callback(struct pdo_sqlite_fci *fc, zval *cb, int argc, sqlite3_va
fake_argc = argc + is_agg; fake_argc = argc + is_agg;
fc->fci.size = sizeof(fc->fci);
ZVAL_COPY_VALUE(&fc->fci.function_name, cb);
fc->fci.object = NULL;
fc->fci.retval = &retval;
fc->fci.param_count = fake_argc;
/* build up the params */ /* build up the params */
if (fake_argc) { if (fake_argc) {
zargs = safe_emalloc(fake_argc, sizeof(zval), 0); zargs = safe_emalloc(fake_argc, sizeof(zval), 0);
} }
@ -372,11 +365,7 @@ static int do_callback(struct pdo_sqlite_fci *fc, zval *cb, int argc, sqlite3_va
} }
} }
fc->fci.params = zargs; zend_call_known_fcc(fcc, &retval, fake_argc, zargs, /* named_params */ NULL);
if ((ret = zend_call_function(&fc->fci, &fc->fcc)) == FAILURE) {
php_error_docref(NULL, E_WARNING, "An error occurred while invoking the callback");
}
/* clean up the params */ /* clean up the params */
if (zargs) { if (zargs) {
@ -445,41 +434,33 @@ static void php_sqlite3_func_step_callback(sqlite3_context *context, int argc, s
{ {
struct pdo_sqlite_func *func = (struct pdo_sqlite_func*)sqlite3_user_data(context); struct pdo_sqlite_func *func = (struct pdo_sqlite_func*)sqlite3_user_data(context);
do_callback(&func->astep, &func->step, argc, argv, context, 1); do_callback(&func->step, argc, argv, context, 1);
} }
static void php_sqlite3_func_final_callback(sqlite3_context *context) static void php_sqlite3_func_final_callback(sqlite3_context *context)
{ {
struct pdo_sqlite_func *func = (struct pdo_sqlite_func*)sqlite3_user_data(context); struct pdo_sqlite_func *func = (struct pdo_sqlite_func*)sqlite3_user_data(context);
do_callback(&func->afini, &func->fini, 0, NULL, context, 1); do_callback(&func->fini, 0, NULL, context, 1);
} }
static int php_sqlite3_collation_callback(void *context, int string1_len, const void *string1, int string2_len, const void *string2) static int php_sqlite3_collation_callback(void *context, int string1_len, const void *string1, int string2_len, const void *string2)
{ {
int ret; int ret = 0;
zval zargs[2]; zval zargs[2];
zval retval; zval retval;
struct pdo_sqlite_collation *collation = (struct pdo_sqlite_collation*) context; struct pdo_sqlite_collation *collation = (struct pdo_sqlite_collation*) context;
collation->fc.fci.size = sizeof(collation->fc.fci);
ZVAL_COPY_VALUE(&collation->fc.fci.function_name, &collation->callback);
collation->fc.fci.object = NULL;
collation->fc.fci.retval = &retval;
/* Prepare the arguments. */ /* Prepare the arguments. */
ZVAL_STRINGL(&zargs[0], (char *) string1, string1_len); ZVAL_STRINGL(&zargs[0], (char *) string1, string1_len);
ZVAL_STRINGL(&zargs[1], (char *) string2, string2_len); ZVAL_STRINGL(&zargs[1], (char *) string2, string2_len);
collation->fc.fci.param_count = 2;
collation->fc.fci.params = zargs;
if ((ret = zend_call_function(&collation->fc.fci, &collation->fc.fcc)) == FAILURE) { zend_call_known_fcc(&collation->callback, &retval, /* argc */ 2, zargs, /* named_params */ NULL);
php_error_docref(NULL, E_WARNING, "An error occurred while invoking the callback");
} else if (!Z_ISUNDEF(retval)) { if (!Z_ISUNDEF(retval)) {
if (Z_TYPE(retval) != IS_LONG) { if (Z_TYPE(retval) != IS_LONG) {
convert_to_long(&retval); convert_to_long(&retval);
} }
ret = 0;
if (Z_LVAL(retval) > 0) { if (Z_LVAL(retval) > 0) {
ret = 1; ret = 1;
} else if (Z_LVAL(retval) < 0) { } else if (Z_LVAL(retval) < 0) {
@ -498,14 +479,14 @@ static void php_sqlite3_func_callback(sqlite3_context *context, int argc, sqlite
{ {
struct pdo_sqlite_func *func = (struct pdo_sqlite_func*)sqlite3_user_data(context); struct pdo_sqlite_func *func = (struct pdo_sqlite_func*)sqlite3_user_data(context);
do_callback(&func->afunc, &func->func, argc, argv, context, 0); do_callback(&func->func, argc, argv, context, 0);
} }
void pdo_sqlite_create_function_internal(INTERNAL_FUNCTION_PARAMETERS) void pdo_sqlite_create_function_internal(INTERNAL_FUNCTION_PARAMETERS)
{ {
struct pdo_sqlite_func *func; struct pdo_sqlite_func *func;
zend_fcall_info fci; zend_fcall_info fci = empty_fcall_info;
zend_fcall_info_cache fcc; zend_fcall_info_cache fcc = empty_fcall_info_cache;
char *func_name; char *func_name;
size_t func_name_len; size_t func_name_len;
zend_long argc = -1; zend_long argc = -1;
@ -516,11 +497,11 @@ void pdo_sqlite_create_function_internal(INTERNAL_FUNCTION_PARAMETERS)
ZEND_PARSE_PARAMETERS_START(2, 4) ZEND_PARSE_PARAMETERS_START(2, 4)
Z_PARAM_STRING(func_name, func_name_len) Z_PARAM_STRING(func_name, func_name_len)
Z_PARAM_FUNC(fci, fcc) Z_PARAM_FUNC_NO_TRAMPOLINE_FREE(fci, fcc)
Z_PARAM_OPTIONAL Z_PARAM_OPTIONAL
Z_PARAM_LONG(argc) Z_PARAM_LONG(argc)
Z_PARAM_LONG(flags) Z_PARAM_LONG(flags)
ZEND_PARSE_PARAMETERS_END(); ZEND_PARSE_PARAMETERS_END_EX(goto error;);
dbh = Z_PDO_DBH_P(ZEND_THIS); dbh = Z_PDO_DBH_P(ZEND_THIS);
PDO_CONSTRUCT_CHECK; PDO_CONSTRUCT_CHECK;
@ -533,7 +514,7 @@ void pdo_sqlite_create_function_internal(INTERNAL_FUNCTION_PARAMETERS)
if (ret == SQLITE_OK) { if (ret == SQLITE_OK) {
func->funcname = estrdup(func_name); func->funcname = estrdup(func_name);
ZVAL_COPY(&func->func, &fci.function_name); zend_fcc_dup(&func->func, &fcc);
func->argc = argc; func->argc = argc;
@ -544,6 +525,9 @@ void pdo_sqlite_create_function_internal(INTERNAL_FUNCTION_PARAMETERS)
} }
efree(func); efree(func);
error:
zend_release_fcall_info_cache(&fcc);
RETURN_FALSE; RETURN_FALSE;
} }
@ -558,8 +542,10 @@ PHP_METHOD(PDO_SQLite_Ext, sqliteCreateFunction)
void pdo_sqlite_create_aggregate_internal(INTERNAL_FUNCTION_PARAMETERS) void pdo_sqlite_create_aggregate_internal(INTERNAL_FUNCTION_PARAMETERS)
{ {
struct pdo_sqlite_func *func; struct pdo_sqlite_func *func;
zend_fcall_info step_fci, fini_fci; zend_fcall_info step_fci = empty_fcall_info;
zend_fcall_info_cache step_fcc, fini_fcc; zend_fcall_info fini_fci = empty_fcall_info;
zend_fcall_info_cache step_fcc = empty_fcall_info_cache;
zend_fcall_info_cache fini_fcc = empty_fcall_info_cache;
char *func_name; char *func_name;
size_t func_name_len; size_t func_name_len;
zend_long argc = -1; zend_long argc = -1;
@ -569,11 +555,11 @@ void pdo_sqlite_create_aggregate_internal(INTERNAL_FUNCTION_PARAMETERS)
ZEND_PARSE_PARAMETERS_START(3, 4) ZEND_PARSE_PARAMETERS_START(3, 4)
Z_PARAM_STRING(func_name, func_name_len) Z_PARAM_STRING(func_name, func_name_len)
Z_PARAM_FUNC(step_fci, step_fcc) Z_PARAM_FUNC_NO_TRAMPOLINE_FREE(step_fci, step_fcc)
Z_PARAM_FUNC(fini_fci, fini_fcc) Z_PARAM_FUNC_NO_TRAMPOLINE_FREE(fini_fci, fini_fcc)
Z_PARAM_OPTIONAL Z_PARAM_OPTIONAL
Z_PARAM_LONG(argc) Z_PARAM_LONG(argc)
ZEND_PARSE_PARAMETERS_END(); ZEND_PARSE_PARAMETERS_END_EX(goto error;);
dbh = Z_PDO_DBH_P(ZEND_THIS); dbh = Z_PDO_DBH_P(ZEND_THIS);
PDO_CONSTRUCT_CHECK; PDO_CONSTRUCT_CHECK;
@ -587,9 +573,8 @@ void pdo_sqlite_create_aggregate_internal(INTERNAL_FUNCTION_PARAMETERS)
if (ret == SQLITE_OK) { if (ret == SQLITE_OK) {
func->funcname = estrdup(func_name); func->funcname = estrdup(func_name);
ZVAL_COPY(&func->step, &step_fci.function_name); zend_fcc_dup(&func->step, &step_fcc);
zend_fcc_dup(&func->fini, &fini_fcc);
ZVAL_COPY(&func->fini, &fini_fci.function_name);
func->argc = argc; func->argc = argc;
@ -600,6 +585,10 @@ void pdo_sqlite_create_aggregate_internal(INTERNAL_FUNCTION_PARAMETERS)
} }
efree(func); efree(func);
error:
zend_release_fcall_info_cache(&step_fcc);
zend_release_fcall_info_cache(&fini_fcc);
RETURN_FALSE; RETURN_FALSE;
} }
@ -631,8 +620,8 @@ PHP_METHOD(PDO_SQLite_Ext, sqliteCreateAggregate)
void pdo_sqlite_create_collation_internal(INTERNAL_FUNCTION_PARAMETERS, pdo_sqlite_create_collation_callback callback) void pdo_sqlite_create_collation_internal(INTERNAL_FUNCTION_PARAMETERS, pdo_sqlite_create_collation_callback callback)
{ {
struct pdo_sqlite_collation *collation; struct pdo_sqlite_collation *collation;
zend_fcall_info fci; zend_fcall_info fci = empty_fcall_info;
zend_fcall_info_cache fcc; zend_fcall_info_cache fcc = empty_fcall_info_cache;
char *collation_name; char *collation_name;
size_t collation_name_len; size_t collation_name_len;
pdo_dbh_t *dbh; pdo_dbh_t *dbh;
@ -641,7 +630,7 @@ void pdo_sqlite_create_collation_internal(INTERNAL_FUNCTION_PARAMETERS, pdo_sqli
ZEND_PARSE_PARAMETERS_START(2, 2) ZEND_PARSE_PARAMETERS_START(2, 2)
Z_PARAM_STRING(collation_name, collation_name_len) Z_PARAM_STRING(collation_name, collation_name_len)
Z_PARAM_FUNC(fci, fcc) Z_PARAM_FUNC_NO_TRAMPOLINE_FREE(fci, fcc)
ZEND_PARSE_PARAMETERS_END(); ZEND_PARSE_PARAMETERS_END();
dbh = Z_PDO_DBH_P(ZEND_THIS); dbh = Z_PDO_DBH_P(ZEND_THIS);
@ -655,7 +644,7 @@ void pdo_sqlite_create_collation_internal(INTERNAL_FUNCTION_PARAMETERS, pdo_sqli
if (ret == SQLITE_OK) { if (ret == SQLITE_OK) {
collation->name = estrdup(collation_name); collation->name = estrdup(collation_name);
ZVAL_COPY(&collation->callback, &fci.function_name); zend_fcc_dup(&collation->callback, &fcc);
collation->next = H->collations; collation->next = H->collations;
H->collations = collation; H->collations = collation;
@ -663,6 +652,8 @@ void pdo_sqlite_create_collation_internal(INTERNAL_FUNCTION_PARAMETERS, pdo_sqli
RETURN_TRUE; RETURN_TRUE;
} }
zend_release_fcall_info_cache(&fcc);
if (UNEXPECTED(EG(exception))) { if (UNEXPECTED(EG(exception))) {
RETURN_THROWS(); RETURN_THROWS();
} }
@ -706,15 +697,23 @@ static void pdo_sqlite_get_gc(pdo_dbh_t *dbh, zend_get_gc_buffer *gc_buffer)
struct pdo_sqlite_func *func = H->funcs; struct pdo_sqlite_func *func = H->funcs;
while (func) { while (func) {
zend_get_gc_buffer_add_zval(gc_buffer, &func->func); if (ZEND_FCC_INITIALIZED(func->func)) {
zend_get_gc_buffer_add_zval(gc_buffer, &func->step); zend_get_gc_buffer_add_fcc(gc_buffer, &func->func);
zend_get_gc_buffer_add_zval(gc_buffer, &func->fini); }
if (ZEND_FCC_INITIALIZED(func->step)) {
zend_get_gc_buffer_add_fcc(gc_buffer, &func->step);
}
if (ZEND_FCC_INITIALIZED(func->fini)) {
zend_get_gc_buffer_add_fcc(gc_buffer, &func->fini);
}
func = func->next; func = func->next;
} }
struct pdo_sqlite_collation *collation = H->collations; struct pdo_sqlite_collation *collation = H->collations;
while (collation) { while (collation) {
zend_get_gc_buffer_add_zval(gc_buffer, &collation->callback); if (ZEND_FCC_INITIALIZED(collation->callback)) {
zend_get_gc_buffer_add_fcc(gc_buffer, &collation->callback);
}
collation = collation->next; collation = collation->next;
} }
} }

View file

@ -0,0 +1,49 @@
--TEST--
Test PdoSqlite::createFunction() arguments error
--EXTENSIONS--
pdo_sqlite
--FILE--
<?php
declare(strict_types=1);
$db = new PdoSqlite('sqlite::memory:');
class TrampolineTest {
public function __call(string $name, array $arguments) {
return '';
}
}
try {
$db->createFunction(null, [new TrampolineTest(), 'strtoupper']);
} catch (Throwable $e) {
echo $e->getMessage() . "\n";
}
try {
$db->createFunction('strtoupper', null);
} catch (Throwable $e) {
echo $e->getMessage() . "\n";
}
try {
$db->createFunction('strtoupper', [new TrampolineTest(), 'strtoupper'], null);
} catch (Throwable $e) {
echo $e->getMessage() . "\n";
}
try {
$db->createFunction('strtoupper', [new TrampolineTest(), 'strtoupper'], 1, null);
} catch (Throwable $e) {
echo $e->getMessage() . "\n";
}
echo 'done!';
?>
--EXPECT--
PdoSqlite::createFunction(): Argument #1 ($function_name) must be of type string, null given
PdoSqlite::createFunction(): Argument #2 ($callback) must be a valid callback, no array or string given
PdoSqlite::createFunction(): Argument #3 ($num_args) must be of type int, null given
PdoSqlite::createFunction(): Argument #4 ($flags) must be of type int, null given
done!

View file

@ -0,0 +1,63 @@
--TEST--
Test PdoSqlite::createAggregate() arguments error
--EXTENSIONS--
pdo_sqlite
--FILE--
<?php
declare(strict_types=1);
$db = new PdoSqlite('sqlite::memory:');
class TrampolineTest {
public function __call(string $name, array $arguments) {
return '';
}
}
try {
$db->createAggregate(null, [new TrampolineTest(), 'step'], null, 1);
} catch (Throwable $e) {
echo $e->getMessage() . "\n";
}
try {
$db->createAggregate(null, null, [new TrampolineTest(), 'step'], 1);
} catch (Throwable $e) {
echo $e->getMessage() . "\n";
}
try {
$db->createAggregate(null, [new TrampolineTest(), 'step'], [new TrampolineTest(), 'step'], 1);
} catch (Throwable $e) {
echo $e->getMessage() . "\n";
}
try {
$db->createAggregate('S', null, [new TrampolineTest(), 'finalize'], 1);
} catch (Throwable $e) {
echo $e->getMessage() . "\n";
}
try {
$db->createAggregate('S', [new TrampolineTest(), 'step'], null, 1);
} catch (Throwable $e) {
echo $e->getMessage() . "\n";
}
try {
$db->createAggregate('S', [new TrampolineTest(), 'step'], [new TrampolineTest(), 'finalize'], null);
} catch (Throwable $e) {
echo $e->getMessage() . "\n";
}
echo 'done!';
?>
--EXPECT--
PdoSqlite::createAggregate(): Argument #1 ($name) must be of type string, null given
PdoSqlite::createAggregate(): Argument #1 ($name) must be of type string, null given
PdoSqlite::createAggregate(): Argument #1 ($name) must be of type string, null given
PdoSqlite::createAggregate(): Argument #2 ($step) must be a valid callback, no array or string given
PdoSqlite::createAggregate(): Argument #3 ($finalize) must be a valid callback, no array or string given
PdoSqlite::createAggregate(): Argument #4 ($numArgs) must be of type int, null given
done!

View file

@ -0,0 +1,35 @@
--TEST--
Test PdoSqlite::createCollation() arguments error
--EXTENSIONS--
pdo_sqlite
--FILE--
<?php
declare(strict_types=1);
$db = new PdoSqlite('sqlite::memory:');
class TrampolineTest {
public function __call(string $name, array $arguments) {
return '';
}
}
try {
$db->createCollation(null, [new TrampolineTest(), 'NAT']);
} catch (Throwable $e) {
echo $e->getMessage() . "\n";
}
try {
$db->createCollation('NAT', null);
} catch (Throwable $e) {
echo $e->getMessage() . "\n";
}
echo 'done!';
?>
--EXPECT--
PdoSqlite::createCollation(): Argument #1 ($name) must be of type string, null given
PdoSqlite::createCollation(): Argument #2 ($callback) must be a valid callback, no array or string given
done!