Re:Re: Add support for specifying tables in pg_createsubscriber. - Mailing list pgsql-hackers

From tianbing
Subject Re:Re: Add support for specifying tables in pg_createsubscriber.
Date
Msg-id 3d33a85a.535b.19ae2bf9401.Coremail.tian_bing_0531@163.com
Whole thread Raw
In response to Re: Add support for specifying tables in pg_createsubscriber.  (Peter Smith <smithpb2250@gmail.com>)
List pgsql-hackers
Hi, Peter, 
I have reviewed the v21 patch and noticed that there seems to be a memory leak.

+static bool
+check_publication_exists(PGconn *conn, const char *pubname, const char *dbname)
+{
+ PGresult *res;
+ bool exists;
+ char *query;
+
+ query = psprintf("SELECT 1 FROM pg_publication WHERE pubname = %s",
+ PQescapeLiteral(conn, pubname, strlen(pubname)));
+ res = PQexec(conn, query);
+
+ if (PQresultStatus(res) != PGRES_TUPLES_OK)
+ pg_fatal("could not check for publication \"%s\" in database \"%s\": %s",
+ pubname, dbname, PQerrorMessage(conn));
+
+ exists = (PQntuples(res) == 1);
+
+ PQclear(res);
+ pg_free(query);
+ return exists;
+}

The PQescapeLiteral() function through malloc to allocate memmory,and should be free by PQfreemem。

I suggest making the following modifications:

+ char *pub = PQescapeLiteral(conn, pubname, strlen(pubname);
+ query = psprintf("SELECT 1 FROM pg_publication WHERE pubname = %s", pub);
......
+ PQfreemem(pub);


Best Regards.


pgsql-hackers by date:

Previous
From: Amit Kapila
Date:
Subject: Re: Allow GUC settings in CREATE SUBSCRIPTION CONNECTION to take effect
Next
From: Michael Paquier
Date:
Subject: Re: [Proposal] Adding callback support for custom statistics kinds