database/postgres
1
0
Fork 0
mirror of https://github.com/postgres/postgres.git synced 2025-08-24 09:27:52 +03:00
Code Activity

Fix pg_prepared_statements.result_types for DML statements

Browse Source 
Amendment to 84ad713cf8: Not all
prepared statements have a result descriptor.  As currently coded,
this would crash when reading pg_prepared_statements.  Make those
cases return null for result_types instead.  Also add a test case for
it.
This commit is contained in:
Peter Eisentraut
2022-07-05 10:26:36 +02:00
parent e3dd7c06e6
commit 6ffff0fd22
4 changed files with 26 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/backend/commands/prepare.c
View File

@@ -684,15 +684,10 @@ pg_prepared_statement(PG_FUNCTION_ARGS)
while ((prep_stmt = hash_seq_search(&hash_seq)) != NULL)
{
TupleDesc result_desc;
Oid *result_types;
Datum values[8];
bool nulls[8];
result_desc = prep_stmt->plansource->resultDesc;
result_types = (Oid *) palloc(result_desc->natts * sizeof(Oid));
for (int i = 0; i < result_desc->natts; i++)
result_types[i] = result_desc->attrs[i].atttypid;
MemSet(nulls, 0, sizeof(nulls));
@@ -701,7 +696,20 @@ pg_prepared_statement(PG_FUNCTION_ARGS)
values[2] = TimestampTzGetDatum(prep_stmt->prepare_time);
values[3] = build_regtype_array(prep_stmt->plansource->param_types,
prep_stmt->plansource->num_params);
values[4] = build_regtype_array(result_types, result_desc->natts);
if (result_desc)
{
Oid *result_types;
result_types = (Oid *) palloc(result_desc->natts * sizeof(Oid));
for (int i = 0; i < result_desc->natts; i++)
result_types[i] = result_desc->attrs[i].atttypid;
values[4] = build_regtype_array(result_types, result_desc->natts);
}
else
{
/* no result descriptor (for example, DML statement) */
nulls[4] = true;
}
values[5] = BoolGetDatum(prep_stmt->from_sql);
values[6] = Int64GetDatumFast(prep_stmt->plansource->num_generic_plans);
values[7] = Int64GetDatumFast(prep_stmt->plansource->num_custom_plans);