Fix checking of file operations in demo ('files') output format.

diff --git a/src/bin/pg_dump/pg_backup_files.c b/src/bin/pg_dump/pg_backup_files.c
index 9a5a7797a17faa6092f808d0b6e3ab54bf2cd128..40aace0f2859d3d9c49497afa12c6530e3baad48 100644 (file)
--- a/src/bin/pg_dump/pg_backup_files.c
+++ b/src/bin/pg_dump/pg_backup_files.c
@@ -20,7 +20,7 @@
  *
  *
  * IDENTIFICATION
- *             $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_backup_files.c,v 1.8 2001/03/22 04:00:13 momjian Exp $
+ *             $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_backup_files.c,v 1.9 2001/03/23 01:27:12 pjw Exp $
  *
  * Modifications - 28-Jun-2000 - [email protected]
  *
@@ -158,7 +158,9 @@ InitArchiveFmt_Files(ArchiveHandle *AH)
 
                ReadHead(AH);
                ReadToc(AH);
-               fclose(AH->FH);                 /* Nothing else in the file... */
+               /* Nothing else in the file... */
+               if (fclose(AH->FH) != 0)
+                       die_horribly(AH, "%s: Could not close TOC file (fclose failed).\n", progname);
        }
 
 }
@@ -268,7 +270,9 @@ _EndData(ArchiveHandle *AH, TocEntry *te)
        lclTocEntry *tctx = (lclTocEntry *) te->formatData;
 
        /* Close the file */
-       GZCLOSE(tctx->FH);
+       if (GZCLOSE(tctx->FH) != 0)
+               die_horribly(AH, "%s: could not close data file\n", progname);
+
        tctx->FH = NULL;
 }
 
@@ -299,7 +303,9 @@ _PrintFileData(ArchiveHandle *AH, char *filename, RestoreOptions *ropt)
                ahwrite(buf, 1, cnt, AH);
        }
 
-       GZCLOSE(AH->FH);
+       if (GZCLOSE(AH->FH) != 0)
+               die_horribly(AH, "%s: could not close data file after reading\n", progname);
+
 }
 
 
@@ -374,7 +380,8 @@ _LoadBlobs(ArchiveHandle *AH, RestoreOptions *ropt)
                _getBlobTocEntry(AH, &oid, fname);
        }
 
-       fclose(ctx->blobToc);
+       if (fclose(ctx->blobToc) != 0)
+               die_horribly(AH, "%s: could not close BLOB TOC file\n", progname);
 
        EndRestoreBlobs(AH);
 }
@@ -437,7 +444,8 @@ _CloseArchive(ArchiveHandle *AH)
        {
                WriteHead(AH);
                WriteToc(AH);
-               fclose(AH->FH);
+               if (fclose(AH->FH) != 0)
+                       die_horribly(AH, "%s: could not close TOC file\n", progname);
                WriteDataChunks(AH);
        }
 
@@ -524,7 +532,8 @@ _EndBlob(ArchiveHandle *AH, TocEntry *te, int oid)
 {
        lclTocEntry *tctx = (lclTocEntry *) te->formatData;
 
-       GZCLOSE(tctx->FH);
+       if (GZCLOSE(tctx->FH) != 0)
+               die_horribly(AH, "%s: could not close BLOB file\n", progname);
 }
 
 /*
@@ -541,6 +550,8 @@ _EndBlobs(ArchiveHandle *AH, TocEntry *te)
        /* Write out a fake zero OID to mark end-of-blobs. */
        /* WriteInt(AH, 0); */
 
-       fclose(ctx->blobToc);
+       if (fclose(ctx->blobToc) != 0)
+               die_horribly(AH, "%s: could not close BLOB TOC file\n", progname);
 
 }
+