X-Git-Url: http://erislabs.net/gitweb/?a=blobdiff_plain;f=lib%2Fxgetcwd.c;h=1409bcfba7d684d628e6b1283a2e7303f1d6dd4e;hb=2bf674b2395b6aff54eff7928d24f2995c37c28e;hp=23fbaeab1dd8b992d8ecd4c32afd65f7fe546c5f;hpb=dec5721106b265a81af5e5cdd8ebd9f7955649a3;p=gnulib.git diff --git a/lib/xgetcwd.c b/lib/xgetcwd.c index 23fbaeab1..1409bcfba 100644 --- a/lib/xgetcwd.c +++ b/lib/xgetcwd.c @@ -39,6 +39,8 @@ extern int errno; #if HAVE_GETCWD char *getcwd (); #else +# include "pathmax.h" +# define INITIAL_BUFFER_SIZE (PATH_MAX + 1) char *getwd (); # define getcwd(Buf, Max) getwd (Buf) #endif @@ -51,26 +53,32 @@ char *getwd (); char * xgetcwd () { -#if defined __GLIBC__ && __GLIBC__ >= 2 - return getcwd (NULL, 0); +#if HAVE_GETCWD_NULL + char *cwd = getcwd (NULL, 0); + if (! cwd && errno == ENOMEM) + xalloc_die (); + return cwd; #else - size_t buf_size = 128; /* must be a power of 2 */ - char *buf = NULL; + + /* The initial buffer size for the working directory. A power of 2 + detects arithmetic overflow earlier, but is not required. */ +# ifndef INITIAL_BUFFER_SIZE +# define INITIAL_BUFFER_SIZE 128 +# endif + + size_t buf_size = INITIAL_BUFFER_SIZE; while (1) { - char *cwd; - buf = (char *) xrealloc (buf, buf_size); - - cwd = getcwd (buf, buf_size); - if (cwd != NULL) + char *buf = xmalloc (buf_size); + char *cwd = getcwd (buf, buf_size); + int saved_errno; + if (cwd) return cwd; - if (errno != ERANGE) - { - free (buf); - return NULL; - } - + saved_errno = errno; + free (buf); + if (saved_errno != ERANGE) + return NULL; buf_size *= 2; if (buf_size == 0) xalloc_die ();