From 2450e4ba44707a64920ea6c9276930a1210e76cc Mon Sep 17 00:00:00 2001 From: Denis Vlasenko Date: Sat, 29 Sep 2007 19:19:55 +0000 Subject: xrealloc_getcwd_or_warn: smaller cod and less wasted RAM at run time --- libbb/xgetcwd.c | 31 ++++++++++++++----------------- 1 file changed, 14 insertions(+), 17 deletions(-) (limited to 'libbb/xgetcwd.c') diff --git a/libbb/xgetcwd.c b/libbb/xgetcwd.c index ec1d8f7a4..c194e2303 100644 --- a/libbb/xgetcwd.c +++ b/libbb/xgetcwd.c @@ -9,9 +9,6 @@ #include "libbb.h" -/* Amount to increase buffer size by in each try. */ -#define PATH_INCR 32 - /* Return the current directory, newly allocated, arbitrarily long. Return NULL and set errno on error. If argument is not NULL (previous usage allocate memory), call free() @@ -20,25 +17,25 @@ char * xrealloc_getcwd_or_warn(char *cwd) { +#define PATH_INCR 64 + char *ret; unsigned path_max; - path_max = (unsigned) PATH_MAX; - path_max += 2; /* The getcwd docs say to do this. */ - - if (cwd == NULL) - cwd = xmalloc(path_max); + path_max = 128; /* 128 + 64 should be enough for 99% of cases */ - while ((ret = getcwd(cwd, path_max)) == NULL && errno == ERANGE) { + while (1) { path_max += PATH_INCR; cwd = xrealloc(cwd, path_max); + ret = getcwd(cwd, path_max); + if (ret == NULL) { + if (errno == ERANGE) + continue; + free(cwd); + bb_perror_msg("getcwd"); + return NULL; + } + cwd = xrealloc(cwd, strlen(cwd) + 1); + return cwd; } - - if (ret == NULL) { - free(cwd); - bb_perror_msg("getcwd"); - return NULL; - } - - return cwd; } -- cgit v1.2.3