From: Peter Eisentraut <peter_e@gmx.net>
Date: Mon, 13 Mar 2000 13:46:32 +0000 (+0000)
Subject: Fixed deficiency where an unterminated (no semicolon) command at end of
X-Git-Tag: REL7_0~448
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=a1642089bf4d4d85abec6cdde777471e97561657;p=thirdparty%2Fpostgresql.git

Fixed deficiency where an unterminated (no semicolon) command at end of
psql script would be dropped silently.
---

diff --git a/src/bin/psql/mainloop.c b/src/bin/psql/mainloop.c
index 4bd4608e5ce..4f71f3e4105 100644
--- a/src/bin/psql/mainloop.c
+++ b/src/bin/psql/mainloop.c
@@ -3,7 +3,7 @@
  *
  * Copyright 2000 by PostgreSQL Global Development Group
  *
- * $Header: /cvsroot/pgsql/src/bin/psql/mainloop.c,v 1.24 2000/03/05 13:30:19 petere Exp $
+ * $Header: /cvsroot/pgsql/src/bin/psql/mainloop.c,v 1.25 2000/03/13 13:46:32 petere Exp $
  */
 #include "postgres.h"
 #include "mainloop.h"
@@ -61,7 +61,7 @@ MainLoop(FILE *source)
 	bool		prev_cmd_interactive;
 
     unsigned int prev_lineno;
-	bool		die_on_error;
+	volatile bool die_on_error = false;
 
 
 	/* Save old settings */
@@ -395,6 +395,7 @@ MainLoop(FILE *source)
 						appendPQExpBufferChar(query_buf, '\n');
 					/* append the line to the query buffer */
 					appendPQExpBufferStr(query_buf, line + query_start);
+                    appendPQExpBufferChar(query_buf, ';');
 				}
 
 				/* execute query */
@@ -515,8 +516,21 @@ MainLoop(FILE *source)
 			successResult = EXIT_BADCONN;
 			break;
 		}
-	} /* while !endofprogram */
-
+	} /* while !endoffile/session */
+
+    /*
+     * Process query at the end of file without a semicolon
+     */
+    if (query_buf->len > 0 && !pset.cur_cmd_interactive)
+    {
+        success = SendQuery(query_buf->data);
+
+        if (!success && die_on_error)
+            successResult = EXIT_USER;
+        else if (pset.db == NULL)
+            successResult = EXIT_BADCONN;
+    }
+        
 	destroyPQExpBuffer(query_buf);
 	destroyPQExpBuffer(previous_buf);