From: Robert Haas <rhaas@postgresql.org>
Date: Fri, 8 Apr 2011 04:30:54 +0000 (-0400)
Subject: Fix some sloppiness in new PL/python get_source_line() function.
X-Git-Tag: REL9_1_BETA1~140
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=b6bc481d5540a3ad0d39db1e9881e6bd52e54213;p=postgresql

Fix some sloppiness in new PL/python get_source_line() function.

Jan Urbański
---

diff --git a/src/pl/plpython/plpython.c b/src/pl/plpython/plpython.c
index 935258044d..47d898a976 100644
--- a/src/pl/plpython/plpython.c
+++ b/src/pl/plpython/plpython.c
@@ -4484,12 +4484,11 @@ cleanup:
 static char *
 get_source_line(const char *src, int lineno)
 {
-	const char *s;
-	const char *next;
-	int		current = 0;
+	const char	*s		 = NULL;
+	const char	*next	 = src;
+	int			 current = 0;
 
-	next = src;
-	while (current != lineno)
+	while (current < lineno)
 	{
 		s = next;
 		next = strchr(s + 1, '\n');
@@ -4501,7 +4500,7 @@ get_source_line(const char *src, int lineno)
 	if (current != lineno)
 		return NULL;
 
-	while (s && isspace((unsigned char) *s))
+	while (*s && isspace((unsigned char) *s))
 		s++;
 
 	if (next == NULL)