Fix plpgsql's handling of -- comments following expressions.
authorTom Lane <[email protected]>
Wed, 10 Apr 2024 19:45:59 +0000 (15:45 -0400)
committerTom Lane <[email protected]>
Wed, 10 Apr 2024 19:45:59 +0000 (15:45 -0400)
commitdc5824a06e979e1f4412849d63d166ca16d1f84d
tree17e3c995be2b0efb6e74e780052afc3defa9abf3
parent2e56ad6a319e02f35444bec391cb777498796851
Fix plpgsql's handling of -- comments following expressions.

Up to now, read_sql_construct() has collected all the source text from
the statement or expression's initial token up to the character just
before the "until" token.  It normally tries to strip trailing
whitespace from that, largely for neatness.  If there was a "-- text"
comment after the expression, this resulted in removing the newline
that terminates the comment, which creates a hazard if we try to paste
the collected text into a larger SQL construct without inserting a
newline after it.  In particular this caused our handling of CASE
constructs to fail if there's a comment after a WHEN expression.

Commit 4adead1d2 noticed a similar problem with cursor arguments,
and worked around it through the rather crude hack of suppressing
the whitespace-trimming behavior for those.  Rather than do that
and leave the hazard open for future hackers to trip over, let's
fix it properly.  pl_scanner.c already has enough infrastructure
to report the end location of the expression's last token, so
we can copy up to that location and never collect any trailing
whitespace or comment to begin with.

Erik Wienhold and Tom Lane, per report from Michal Bartak.
Back-patch to all supported branches.

Discussion: https://postgr.es/m/CAAVzF_FjRoi8fOVuLCZhQJx6HATQ7MKm=aFOHWZODFnLmjX-xA@mail.gmail.com
src/pl/plpgsql/src/expected/plpgsql_control.out
src/pl/plpgsql/src/pl_gram.y
src/pl/plpgsql/src/pl_scanner.c
src/pl/plpgsql/src/plpgsql.h
src/pl/plpgsql/src/sql/plpgsql_control.sql
src/test/regress/expected/plpgsql.out
src/test/regress/sql/plpgsql.sql