database/postgres
database/postgres
1
0
Fork 0
mirror of https://github.com/postgres/postgres.git synced 2025-05-18 17:41:14 +03:00
Code

Fix off-by-one memory allocation, as reported by Rod Taylor. Also

Browse Source 
avoid repalloc'ing twice when once is sufficient.
This commit is contained in:
Tom Lane 2004-11-17 19:54:34 +00:00
parent b990232c5d
commit aed7855dd8
1 changed files with 18 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
src/backend/libpq/hba.c
View File

@ -10,7 +10,7 @@
* *
* *
* IDENTIFICATION * IDENTIFICATION
* $Header: /cvsroot/pgsql/src/backend/libpq/hba.c,v 1.116.2.3 2004/09/18 01:23:12 tgl Exp $ * $Header: /cvsroot/pgsql/src/backend/libpq/hba.c,v 1.116.2.4 2004/11/17 19:54:34 tgl Exp $
* *
*------------------------------------------------------------------------- *-------------------------------------------------------------------------
*/ */
@ -179,6 +179,7 @@ next_token_expand(FILE *file)
char *comma_str = pstrdup(""); char *comma_str = pstrdup("");
bool trailing_comma; bool trailing_comma;
char *incbuf; char *incbuf;
int needed;
do do
{ {
@ -200,16 +201,14 @@ next_token_expand(FILE *file)
else else
incbuf = pstrdup(buf); incbuf = pstrdup(buf);
comma_str = repalloc(comma_str, needed = strlen(comma_str) + strlen(incbuf) + 1;
strlen(comma_str) + strlen(incbuf) + 1); if (trailing_comma)
strcat(comma_str, incbuf); needed++;
pfree(incbuf); comma_str = repalloc(comma_str, needed);
strcat(comma_str, incbuf);
if (trailing_comma) if (trailing_comma)
{
comma_str = repalloc(comma_str, strlen(comma_str) + 1 + 1);
strcat(comma_str, MULTI_VALUE_SEP); strcat(comma_str, MULTI_VALUE_SEP);
} pfree(incbuf);
} while (trailing_comma); } while (trailing_comma);
return comma_str; return comma_str;
@ -270,7 +269,7 @@ tokenize_inc_file(const char *inc_filename)
pfree(inc_fullname); pfree(inc_fullname);
/* return empty string, it matches nothing */ /* return empty string, it matches nothing */
return pstrdup(""); return comma_str;
} }
pfree(inc_fullname); pfree(inc_fullname);
@ -278,7 +277,7 @@ tokenize_inc_file(const char *inc_filename)
inc_lines = tokenize_file(inc_file); inc_lines = tokenize_file(inc_file);
FreeFile(inc_file); FreeFile(inc_file);
/* Create comma-separate string from List */ /* Create comma-separated string from List */
foreach(line, inc_lines) foreach(line, inc_lines)
{ {
List *ln = lfirst(line); List *ln = lfirst(line);
@ -287,13 +286,15 @@ tokenize_inc_file(const char *inc_filename)
/* First entry is line number */ /* First entry is line number */
foreach(token, lnext(ln)) foreach(token, lnext(ln))
{ {
if (strlen(comma_str)) int oldlen = strlen(comma_str);
{ int needed;
comma_str = repalloc(comma_str, strlen(comma_str) + 1);
needed = oldlen + strlen(lfirst(token)) + 1;
if (oldlen > 0)
needed++;
comma_str = repalloc(comma_str, needed);
if (oldlen > 0)
strcat(comma_str, MULTI_VALUE_SEP); strcat(comma_str, MULTI_VALUE_SEP);
}
comma_str = repalloc(comma_str,
strlen(comma_str) + strlen(lfirst(token)) + 1);
strcat(comma_str, lfirst(token)); strcat(comma_str, lfirst(token));
} }
} }