In psql, avoid leaking a PGresult after a query is cancelled.
authorTom Lane <tgl@sss.pgh.pa.us>
Mon, 8 Apr 2024 21:00:07 +0000 (17:00 -0400)
committerTom Lane <tgl@sss.pgh.pa.us>
Mon, 8 Apr 2024 21:00:07 +0000 (17:00 -0400)
After a query cancel, the tail end of ExecQueryAndProcessResults
took care to clear any not-yet-read PGresults; but it forgot about
the one it has already read.  There would only be such a result
when handling a multi-command string made with "\;", so that you'd
have to cancel an earlier command in such a string to reach the
bug at all.  Even then, there would only be leakage of a single
PGresult per cancel, so it's not surprising nobody noticed this.
But a leak is a leak.

Noted while re-reviewing 90f517821, but this is independent of that:
it dates to 7844c9918.  Back-patch to v15 where that came in.

src/bin/psql/common.c

index 69063a6f785d542c78a240c4a3e65daec497ea03..fe8e049c4c1a243973d9be560ec001b78cce99aa 100644 (file)
@@ -1812,6 +1812,8 @@ ExecQueryAndProcessResults(const char *query,
 
        if (cancel_pressed)
        {
+           /* drop this next result, as well as any others not yet read */
+           ClearOrSaveResult(result);
            ClearOrSaveAllResults();
            break;
        }