Fix some bogus direct uses of realloc().

tglsfdc · tglsfdc · commit 3aa3ae8e1d95 · 2014-11-18T13:28:09.000-05:00
pg_dump/parallel.c was using realloc() directly with no error check.
While the odds of an actual failure here seem pretty low, Coverity
complains about it, so fix by using pg_realloc() instead.

While looking for other instances, I noticed a couple of places in
psql that hadn't gotten the memo about the availability of pg_realloc.
These aren't bugs, since they did have error checks, but verbosely
inconsistent code is not a good thing.

Back-patch as far as 9.3.  9.2 did not have pg_dump/parallel.c, nor
did it have pg_realloc available in all frontend code.
diff --git a/src/bin/pg_dump/parallel.c b/src/bin/pg_dump/parallel.c
@@ -1303,15 +1303,15 @@ readMessageFromPipe(int fd)
 		{
 			/* could be any number */
 			bufsize += 16;
-			msg = (char *) realloc(msg, bufsize);
+			msg = (char *) pg_realloc(msg, bufsize);
 		}
 	}
 
 	/*
 	 * Worker has closed the connection, make sure to clean up before return
 	 * since we are not returning msg (but did allocate it).
 	 */
-	free(msg);
+	pg_free(msg);
 
 	return NULL;
 }
diff --git a/src/bin/psql/command.c b/src/bin/psql/command.c
@@ -1130,12 +1130,7 @@ exec_command(const char *cmd,
 			while ((opt = psql_scan_slash_option(scan_state,
 												 OT_NORMAL, NULL, false)))
 			{
-				newval = realloc(newval, strlen(newval) + strlen(opt) + 1);
-				if (!newval)
-				{
-					psql_error("out of memory\n");
-					exit(EXIT_FAILURE);
-				}
+				newval = pg_realloc(newval, strlen(newval) + strlen(opt) + 1);
 				strcat(newval, opt);
 				free(opt);
 			}
diff --git a/src/bin/psql/tab-complete.c b/src/bin/psql/tab-complete.c
@@ -3975,13 +3975,8 @@ complete_from_variables(const char *text, const char *prefix, const char *suffix
 		if (nvars >= maxvars)
 		{
 			maxvars *= 2;
-			varnames = (char **) realloc(varnames,
-										 (maxvars + 1) * sizeof(char *));
-			if (!varnames)
-			{
-				psql_error("out of memory\n");
-				exit(EXIT_FAILURE);
-			}
+			varnames = (char **) pg_realloc(varnames,
+											(maxvars + 1) * sizeof(char *));
 		}
 
 		varnames[nvars++] = psprintf("%s%s%s", prefix, ptr->name, suffix);

Original file line number	Diff line number	Diff line change
`@@ -1303,15 +1303,15 @@ readMessageFromPipe(int fd)`
`1303`	`1303`	`{`
`1304`	`1304`	`/* could be any number */`
`1305`	`1305`	`bufsize += 16;`
`1306`		`- msg = (char *) realloc(msg, bufsize);`
	`1306`	`+ msg = (char *) pg_realloc(msg, bufsize);`
`1307`	`1307`	`}`
`1308`	`1308`	`}`
`1309`	`1309`
`1310`	`1310`	`/*`
`1311`	`1311`	`* Worker has closed the connection, make sure to clean up before return`
`1312`	`1312`	`* since we are not returning msg (but did allocate it).`
`1313`	`1313`	`*/`
`1314`		`- free(msg);`
	`1314`	`+ pg_free(msg);`
`1315`	`1315`
`1316`	`1316`	`return NULL;`
`1317`	`1317`	`}`
Original file line number	Diff line number	Diff line change
`@@ -1130,12 +1130,7 @@ exec_command(const char *cmd,`
`1130`	`1130`	`while ((opt = psql_scan_slash_option(scan_state,`
`1131`	`1131`	`OT_NORMAL, NULL, false)))`
`1132`	`1132`	`{`
`1133`		`- newval = realloc(newval, strlen(newval) + strlen(opt) + 1);`
`1134`		`- if (!newval)`
`1135`		`- {`
`1136`		`- psql_error("out of memory\n");`
`1137`		`- exit(EXIT_FAILURE);`
`1138`		`- }`
	`1133`	`+ newval = pg_realloc(newval, strlen(newval) + strlen(opt) + 1);`
`1139`	`1134`	`strcat(newval, opt);`
`1140`	`1135`	`free(opt);`
`1141`	`1136`	`}`