]> granicus.if.org Git - postgresql/commitdiff
Guard against duplicate IDs in input file in SortTocFromFile().
authorTom Lane <tgl@sss.pgh.pa.us>
Tue, 17 May 2005 17:31:15 +0000 (17:31 +0000)
committerTom Lane <tgl@sss.pgh.pa.us>
Tue, 17 May 2005 17:31:15 +0000 (17:31 +0000)
Per report from Brian Hackett.

src/bin/pg_dump/pg_backup_archiver.c

index 4afb011a4cf80d7b523de468f20ef769204567dc..474938dece75a464583a16cd0555b0a1c0499513 100644 (file)
@@ -15,7 +15,7 @@
  *
  *
  * IDENTIFICATION
- *             $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_backup_archiver.c,v 1.62.2.6 2005/04/30 08:42:17 neilc Exp $
+ *             $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_backup_archiver.c,v 1.62.2.7 2005/05/17 17:31:15 tgl Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -959,7 +959,7 @@ SortTocFromFile(Archive *AHX, RestoreOptions *ropt)
        if (!fh)
                die_horribly(AH, modulename, "could not open TOC file\n");
 
-       while (fgets(buf, 1024, fh) != NULL)
+       while (fgets(buf, sizeof(buf), fh) != NULL)
        {
                /* Find a comment */
                cmnt = strchr(buf, ';');
@@ -987,10 +987,13 @@ SortTocFromFile(Archive *AHX, RestoreOptions *ropt)
                if (!te)
                        die_horribly(AH, modulename, "could not find entry for id %d\n", id);
 
-               ropt->idWanted[id - 1] = 1;
+               if (!ropt->idWanted[id - 1])
+               {
+                       ropt->idWanted[id - 1] = 1;
 
-               _moveAfter(AH, tePrev, te);
-               tePrev = te;
+                       _moveAfter(AH, tePrev, te);
+                       tePrev = te;
+               }
        }
 
        if (fclose(fh) != 0)