Repair OPEN cursor(args), which I broke on 11/29/01 with a change to

be smarter about parentheses in read_sql_construct().  Sigh.

diff --git a/src/pl/plpgsql/src/gram.y b/src/pl/plpgsql/src/gram.y
index 913d381144be286d64d32ee29d75b2c87edf205a..b3a86236e5ad50cebbd7d7209aa4a44bcea87f3d 100644 (file)
--- a/src/pl/plpgsql/src/gram.y
+++ b/src/pl/plpgsql/src/gram.y
@@ -4,7 +4,7 @@
  *                       procedural language
  *
  * IDENTIFICATION
- *   $Header: /cvsroot/pgsql/src/pl/plpgsql/src/gram.y,v 1.29 2001/11/29 22:57:37 tgl Exp $
+ *   $Header: /cvsroot/pgsql/src/pl/plpgsql/src/gram.y,v 1.29.2.1 2002/05/21 18:50:18 tgl Exp $
  *
  *   This software is copyrighted by Jan Wieck - Hamburg.
  *
@@ -1327,17 +1327,44 @@ stmt_open       : K_OPEN lno cursor_varptr
                                if (tok != '(')
                                {
                                    plpgsql_error_lineno = yylineno;
-                                   elog(ERROR, "cursor %s has arguments", $3->refname);
+                                   elog(ERROR, "cursor %s has arguments",
+                                        $3->refname);
                                }
 
+                               /*
+                                * Push back the '(', else read_sql_stmt
+                                * will complain about unbalanced parens.
+                                */
+                               plpgsql_push_back_token(tok);
+
                                new->argquery = read_sql_stmt("SELECT ");
-                               /* Remove the trailing right paren,
-                                 * because we want "select 1, 2", not
-                                 * "select (1, 2)".
+
+                               /*
+                                * Now remove the leading and trailing parens,
+                                * because we want "select 1, 2", not
+                                * "select (1, 2)".
                                 */
                                cp = new->argquery->query;
-                               cp += strlen(cp);
-                               --cp;
+
+                               if (strncmp(cp, "SELECT", 6) != 0)
+                               {
+                                   plpgsql_error_lineno = yylineno;
+                                   elog(ERROR, "expected 'SELECT (', got '%s' (internal error)",
+                                        new->argquery->query);
+                               }
+                               cp += 6;
+                               while (*cp == ' ') /* could be more than 1 space here */
+                                   cp++;
+                               if (*cp != '(')
+                               {
+                                   plpgsql_error_lineno = yylineno;
+                                   elog(ERROR, "expected 'SELECT (', got '%s' (internal error)",
+                                        new->argquery->query);
+                               }
+                               *cp = ' ';
+
+                               cp += strlen(cp) - 1;
+
                                if (*cp != ')')
                                {
                                    plpgsql_error_lineno = yylineno;