From fec883266fe54caeec54f20d9346fee5daa17168 Mon Sep 17 00:00:00 2001 From: usa Date: Thu, 1 Jun 2006 07:18:12 +0000 Subject: * 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: http://svn.ruby-lang.org/repos/ruby/trunk@10201 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- ChangeLog | 6 ++++++ win32/win32.c | 49 ++++++++++++++++++++++++++++++------------------- 2 files changed, 36 insertions(+), 19 deletions(-) diff --git a/ChangeLog b/ChangeLog index c8a983f5d..783c5bdfc 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,9 @@ +Thu Jun 1 16:17:26 2006 NAKAMURA Usaku + + * 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] + Thu Jun 1 11:33:32 2006 NAKAMURA Usaku * win32/win32.c (rb_w32_getcwd): set errno if not set. diff --git a/win32/win32.c b/win32/win32.c index 999ff6804..62a65d234 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 -- cgit