* win32/win32.c (rb_w32_getcwd): runtime's getcwd() will not success

  if the length of the cwd is longer than MAX_PATH.
  fixed [ruby-list:42335]


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/branches/ruby_1_8@10201 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 30 insertions, 16 deletions

diff --git a/win32/win32.c b/win32/win32.c
index c044efdd22..f06073e001 100644
--- a/win32/win32.c
+++ b/win32/win32.c
@@ -2609,31 +2609,45 @@ gettimeofday(struct timeval *tv, struct timezone *tz)
 char *
 rb_w32_getcwd(char *buffer, int size)
 {
-    int length;
+    char *p = buffer;
     char *bp;
-    int save_errno = errno;
+    int len;
+
+    len = GetCurrentDirectory(0, NULL);
+    if (!len) {
+	errno = map_errno(GetLastError());
+	return NULL;
+    }
 
-#undef getcwd
-    errno = 0;
-    SetLastError(0);
-    if (getcwd(buffer, size) == NULL) {
-	if (!errno)
-	    errno = GetLastError() ? map_errno(GetLastError()) : ERANGE;
-        return NULL;
+    if (p) {
+	if (size < len) {
+	    errno = ERANGE;
+	    return NULL;
+	}
+    }
+    else {
+	p = malloc(len);
+	size = len;
+	if (!p) {
+	    errno = ENOMEM;
+	    return NULL;
+	}
     }
-    length = strlen(buffer);
-    if (length >= size) {
-	errno = ERANGE;
-        return NULL;
+
+    if (!GetCurrentDirectory(size, p)) {
+	errno = map_errno(GetLastError());
+	if (!buffer)
+	    free(p);
+	return NULL;
     }
-    errno = save_errno;
 
-    for (bp = buffer; *bp != '\0'; bp = CharNext(bp)) {
+    for (bp = p; *bp != '\0'; bp = CharNext(bp)) {
 	if (*bp == '\\') {
 	    *bp = '/';
 	}
     }
-    return buffer;
+
+    return p;
 }
 
 int