X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=libbb%2Fxgetcwd.c;h=10febe32d44c8964b6e11e299357ab67a14aaff6;hb=d7171c114028815df4e92f9c26875f1d60c6af2a;hp=ec1d8f7a4231669a6bff3534c311a28444d600b4;hpb=6ca0444420223c224162674902d4f6e4e093962d;p=oweals%2Fbusybox.git

diff --git a/libbb/xgetcwd.c b/libbb/xgetcwd.c
index ec1d8f7a4..10febe32d 100644
--- a/libbb/xgetcwd.c
+++ b/libbb/xgetcwd.c
@@ -5,40 +5,39 @@
  * Written by David MacKenzie <djm@gnu.ai.mit.edu>.
  *
  * Special function for busybox written by Vladimir Oleynik <dzo@simtreas.ru>
-*/
+ *
+ * Licensed under GPLv2, see file LICENSE in this tarball for details.
+ */
 
 #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()
 */
 
-char *
+char* FAST_FUNC
 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. */
+	path_max = 128;	/* 128 + 64 should be enough for 99% of cases */
 
-	if (cwd == NULL)
-		cwd = xmalloc(path_max);
-
-	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;
 }