mirror/postgresql
2
0
Fork 0
mirror of https://git.postgresql.org/git/postgresql.git synced 2024-12-15 08:20:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix write-past-buffer-end in ldapServiceLookup().

Browse Source 
The code to assemble ldap_get_values_len's output into a single string
wrote the terminating null one byte past where it should.  Fix that,
and make some other cosmetic adjustments to make the code a trifle more
readable and more in line with usual Postgres coding style.

Also, free the "result" string when done with it, to avoid a permanent
memory leak.

Bug report and patch by Albe Laurenz, cosmetic adjustments by me.
This commit is contained in:
Tom Lane 2011-05-12 11:56:38 -04:00
parent fbc56f706c
commit 30cf86fdf4
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/interfaces/libpq/fe-connect.c
View File

@ -3339,10 +3339,11 @@ ldapServiceLookup(const char *purl, PQconninfoOption *options,
return 1; return 1;
} }
/* concatenate values to a single string */ /* concatenate values into a single string with newline terminators */
for (size = 0, i = 0; values[i] != NULL; ++i) size = 1; /* for the trailing null */
for (i = 0; values[i] != NULL; i++)
size += values[i]->bv_len + 1; size += values[i]->bv_len + 1;
if ((result = malloc(size + 1)) == NULL) if ((result = malloc(size)) == NULL)
{ {
printfPQExpBuffer(errorMessage, printfPQExpBuffer(errorMessage,
libpq_gettext("out of memory\n")); libpq_gettext("out of memory\n"));
@ -3350,14 +3351,14 @@ ldapServiceLookup(const char *purl, PQconninfoOption *options,
ldap_unbind(ld); ldap_unbind(ld);
return 3; return 3;
} }
for (p = result, i = 0; values[i] != NULL; ++i) p = result;
for (i = 0; values[i] != NULL; i++)
{ {
strncpy(p, values[i]->bv_val, values[i]->bv_len); memcpy(p, values[i]->bv_val, values[i]->bv_len);
p += values[i]->bv_len; p += values[i]->bv_len;
*(p++) = '\n'; *(p++) = '\n';
if (values[i + 1] == NULL)
*(p + 1) = '\0';
} }
*p = '\0';
ldap_value_free_len(values); ldap_value_free_len(values);
ldap_unbind(ld); ldap_unbind(ld);
@ -3386,6 +3387,7 @@ ldapServiceLookup(const char *purl, PQconninfoOption *options,
printfPQExpBuffer(errorMessage, libpq_gettext( printfPQExpBuffer(errorMessage, libpq_gettext(
"missing \"=\" after \"%s\" in connection info string\n"), "missing \"=\" after \"%s\" in connection info string\n"),
optname); optname);
free(result);
return 3; return 3;
} }
else if (*p == '=') else if (*p == '=')
@ -3404,6 +3406,7 @@ ldapServiceLookup(const char *purl, PQconninfoOption *options,
printfPQExpBuffer(errorMessage, libpq_gettext( printfPQExpBuffer(errorMessage, libpq_gettext(
"missing \"=\" after \"%s\" in connection info string\n"), "missing \"=\" after \"%s\" in connection info string\n"),
optname); optname);
free(result);
return 3; return 3;
} }
break; break;
@ -3467,6 +3470,7 @@ ldapServiceLookup(const char *purl, PQconninfoOption *options,
printfPQExpBuffer(errorMessage, printfPQExpBuffer(errorMessage,
libpq_gettext("invalid connection option \"%s\"\n"), libpq_gettext("invalid connection option \"%s\"\n"),
optname); optname);
free(result);
return 1; return 1;
} }
optname = NULL; optname = NULL;
@ -3475,6 +3479,8 @@ ldapServiceLookup(const char *purl, PQconninfoOption *options,
oldstate = state; oldstate = state;
} }
free(result);
if (state == 5 || state == 6) if (state == 5 || state == 6)
{ {
printfPQExpBuffer(errorMessage, libpq_gettext( printfPQExpBuffer(errorMessage, libpq_gettext(