Fix for \g strdup error.

diff --git a/src/bin/psql/psql.c b/src/bin/psql/psql.c
index e7c13ba2a7f74524fc0256c4a72712af66780859..8b22ecf70cb7e36d505063067f2f3577b3821a03 100644 (file)
--- a/src/bin/psql/psql.c
+++ b/src/bin/psql/psql.c
@@ -7,7 +7,7 @@
  *
  *
  * IDENTIFICATION
- *    $Header: /cvsroot/pgsql/src/bin/psql/Attic/psql.c,v 1.31 1996/11/22 04:43:48 bryanh Exp $
+ *    $Header: /cvsroot/pgsql/src/bin/psql/Attic/psql.c,v 1.32 1996/11/22 06:45:14 momjian Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -1099,7 +1099,12 @@ HandleSlashCmds(PsqlSettings * settings,
            break;
        }
     case 'g':                  /* \g means send query */
-       settings->gfname = strdup(optarg);
+       if (!optarg)
+           settings->gfname = NULL;
+       else if (!(settings->gfname = strdup(optarg))) {
+           perror("malloc");
+           exit(1);
+       }
        status = 0;
        break;
     case 'h':                  /* help */