pg_restore: Fix memory and file descriptor leak with directory format

petere · petere · commit b6948e15464d · 2012-03-16T19:57:45.000+02:00
found by Coverity
diff --git a/src/bin/pg_dump/pg_backup_directory.c b/src/bin/pg_dump/pg_backup_directory.c
@@ -365,6 +365,9 @@ _PrintFileData(ArchiveHandle *AH, char *filename, RestoreOptions *ropt)
 		ahwrite(buf, 1, cnt, AH);
 
 	free(buf);
+	if (cfclose(cfp) != 0)
+		die_horribly(AH, modulename, "could not close data file: %s\n",
+					 strerror(errno));
 }
 
 /*

Original file line number	Diff line number	Diff line change
`@@ -365,6 +365,9 @@ _PrintFileData(ArchiveHandle AH, char filename, RestoreOptions *ropt)`
`365`	`365`	`ahwrite(buf, 1, cnt, AH);`
`366`	`366`
`367`	`367`	`free(buf);`
	`368`	`+ if (cfclose(cfp) != 0)`
	`369`	`+ die_horribly(AH, modulename, "could not close data file: %s\n",`
	`370`	`+ strerror(errno));`
`368`	`371`	`}`
`369`	`372`
`370`	`373`	`/*`