* 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/trunk@10201 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 30 insertions, 19 deletions

diff --git a/win32/win32.c b/win32/win32.c
index 999ff68044..62a65d2346 100644
--- a/win32/win32.c
+++ b/win32/win32.c
@@ -2933,34 +2933,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;
 
-#undef getcwd
-#ifndef __BORLANDC__
-#define getcwd _getcwd
-#endif
-    errno = 0;
-    SetLastError(0);
-    if (getcwd(buffer, size) == NULL) {
-	if (!errno)
-	    errno = GetLastError() ? map_errno(GetLastError()) : ERANGE;
-        return NULL;
+    len = GetCurrentDirectory(0, NULL);
+    if (!len) {
+	errno = map_errno(GetLastError());
+	return NULL;
     }
-    length = strlen(buffer);
-    if (length >= size) {
-	errno = ERANGE;
-        return NULL;
+
+    if (p) {
+	if (size < len) {
+	    errno = ERANGE;
+	    return NULL;
+	}
+    }
+    else {
+	p = malloc(len);
+	size = len;
+	if (!p) {
+	    errno = ENOMEM;
+	    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