diff options
Diffstat (limited to 'grp')
| -rw-r--r-- | grp/compat-initgroups.c | 26 | ||||
| -rw-r--r-- | grp/initgroups.c | 2 |
2 files changed, 9 insertions, 19 deletions
diff --git a/grp/compat-initgroups.c b/grp/compat-initgroups.c index ff10e247f4..3dd50d2306 100644 --- a/grp/compat-initgroups.c +++ b/grp/compat-initgroups.c @@ -14,7 +14,6 @@ compat_call (service_user *nip, const char *user, gid_t group, long int *start, long int *size, gid_t **groupsp, long int limit, int *errnop) { struct group grpbuf; - size_t buflen = __sysconf (_SC_GETGR_R_SIZE_MAX); enum nss_status status; set_function setgrent_fct; get_function getgrent_fct; @@ -35,30 +34,22 @@ compat_call (service_user *nip, const char *user, gid_t group, long int *start, endgrent_fct = __nss_lookup_function (nip, "endgrent"); - char *tmpbuf = __alloca (buflen); - bool use_malloc = false; + struct scratch_buffer tmpbuf; + scratch_buffer_init (&tmpbuf); enum nss_status result = NSS_STATUS_SUCCESS; do { while ((status = DL_CALL_FCT (getgrent_fct, - (&grpbuf, tmpbuf, buflen, errnop)), + (&grpbuf, tmpbuf.data, tmpbuf.length, + errnop)), status == NSS_STATUS_TRYAGAIN) && *errnop == ERANGE) { - if (__libc_use_alloca (buflen * 2)) - tmpbuf = extend_alloca (tmpbuf, buflen, buflen * 2); - else + if (!scratch_buffer_grow (&tmpbuf)) { - buflen *= 2; - char *newbuf = realloc (use_malloc ? tmpbuf : NULL, buflen); - if (newbuf == NULL) - { - result = NSS_STATUS_TRYAGAIN; - goto done; - } - use_malloc = true; - tmpbuf = newbuf; + result = NSS_STATUS_TRYAGAIN; + goto done; } } @@ -116,8 +107,7 @@ compat_call (service_user *nip, const char *user, gid_t group, long int *start, while (status == NSS_STATUS_SUCCESS); done: - if (use_malloc) - free (tmpbuf); + scratch_buffer_free (&tmpbuf); if (endgrent_fct) DL_CALL_FCT (endgrent_fct, ()); diff --git a/grp/initgroups.c b/grp/initgroups.c index 4d083acee5..14a8ab158d 100644 --- a/grp/initgroups.c +++ b/grp/initgroups.c @@ -15,7 +15,6 @@ License along with the GNU C Library; if not, see <http://www.gnu.org/licenses/>. */ -#include <alloca.h> #include <assert.h> #include <errno.h> #include <grp.h> @@ -26,6 +25,7 @@ #include <sys/param.h> #include <sys/types.h> #include <nsswitch.h> +#include <scratch_buffer.h> #include "../nscd/nscd-client.h" #include "../nscd/nscd_proto.h" |