From 45df1c24d269f93a2bc1e7a6fe0ffcecc1193051 Mon Sep 17 00:00:00 2001 From: nobu Date: Tue, 3 Feb 2015 05:04:49 +0000 Subject: dir.c: glob legacy short name * dir.c (glob_helper): obtain real name with FindFirstFile API instead of matchin all entries, on Windows. [ruby-core:67954] [Bug #10819] git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@49478 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- ChangeLog | 6 +++++ dir.c | 75 ++++++++++++++++++++++++++++++++++++++++++++++----- test/ruby/test_dir.rb | 12 +++++++++ 3 files changed, 86 insertions(+), 7 deletions(-) diff --git a/ChangeLog b/ChangeLog index cc7a64b250..f3b76091f4 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,9 @@ +Tue Feb 3 14:04:47 2015 Nobuyoshi Nakada + + * dir.c (glob_helper): obtain real name with FindFirstFile API + instead of matchin all entries, on Windows. + [ruby-core:67954] [Bug #10819] + Tue Feb 3 12:26:35 2015 Katsuhiko Nishimra * lib/mkmf.rb (configuration): set the default cxxflags, which is diff --git a/dir.c b/dir.c index c7ba1b0c04..e983ea2552 100644 --- a/dir.c +++ b/dir.c @@ -81,6 +81,10 @@ char *strchr(char*,char); # define USE_NAME_ON_FS 1 # define RUP32(size) ((size)+3/4) # define SIZEUP32(type) RUP32(sizeof(type)) +#elif defined _WIN32 +# define USE_NAME_ON_FS 1 +#elif defined DOSISH +# define USE_NAME_ON_FS 2 /* by fnmatch */ #else # define USE_NAME_ON_FS 0 #endif @@ -1177,10 +1181,30 @@ has_magic(const char *p, const char *pend, int flags, rb_encoding *enc) return PLAIN; continue; +#ifdef _WIN32 + case '~': + /* possibly legacy 8.3 short name */ + if (p < pend && (c = *p, ISDIGIT(c))) { + while (++p < pend && (c = *p, ISDIGIT(c))); + if (p == pend || c == '.') hasalpha = 1; + } + continue; +#endif + default: if (ISALPHA(c)) { hasalpha = 1; } +#ifdef _WIN32 + else if (!rb_isascii(c)) { + unsigned int code = rb_enc_mbc_to_codepoint(p, pend, enc); + if (ONIGENC_IS_CODE_ALPHA(enc, c)) { + /* Full width alphabets */ + hasalpha = 1; + } + } +#endif + break; } p = Next(p-1, pend, enc); @@ -1373,7 +1397,7 @@ is_case_sensitive(DIR *dirp) } static char * -replace_real_basename(char *path, long base, int norm_p) +replace_real_basename(char *path, long base, rb_encoding *enc, int norm_p) { u_int32_t attrbuf[SIZEUP32(attrreference_t) + RUP32(MAXPATHLEN * 3) + 1]; struct attrlist al = {ATTR_BIT_MAP_COUNT, 0, ATTR_CMN_NAME}; @@ -1408,6 +1432,43 @@ replace_real_basename(char *path, long base, int norm_p) IF_NORMALIZE_UTF8PATH(if (!NIL_P(utf8str)) rb_str_resize(utf8str, 0)); return path; } +#elif defined _WIN32 +VALUE rb_w32_conv_from_wchar(const WCHAR *wstr, rb_encoding *enc); + +static char * +replace_real_basename(char *path, long base, rb_encoding *enc, int norm_p) +{ + char *plainname = path; + volatile VALUE tmp = 0; + WIN32_FIND_DATAW fd; + WCHAR *wplain; + HANDLE h; + long wlen; + if (enc && + enc != rb_usascii_encoding() && + enc != rb_ascii8bit_encoding() && + enc != rb_utf8_encoding()) { + tmp = rb_enc_str_new_cstr(plainname, enc); + tmp = rb_str_encode_ospath(tmp); + plainname = RSTRING_PTR(tmp); + } + wplain = rb_w32_mbstr_to_wstr(CP_UTF8, plainname, -1, &wlen); + if (tmp) rb_str_resize(tmp, 0); + if (!wplain) return path; + h = FindFirstFileW(wplain, &fd); + free(wplain); + if (h == INVALID_HANDLE_VALUE) return path; + FindClose(h); + tmp = rb_w32_conv_from_wchar(fd.cFileName, enc); + wlen = RSTRING_LEN(tmp); + path = GLOB_REALLOC(path, base + wlen + 1); + memcpy(path + base, RSTRING_PTR(tmp), wlen); + path[base + wlen] = 0; + rb_str_resize(tmp, 0); + return path; +} +#elif USE_NAME_ON_FS == 1 +# error not implemented #endif enum answer {UNKNOWN = -1, NO, YES}; @@ -1473,7 +1534,7 @@ glob_helper( plain = 1; break; case ALPHA: -#ifdef HAVE_GETATTRLIST +#if defined HAVE_GETATTRLIST || defined _WIN32 plain = 1; #else magical = 1; @@ -1533,11 +1594,11 @@ glob_helper( if (magical || recursive) { struct dirent *dp; DIR *dirp; -# ifdef DOSISH +# if USE_NAME_ON_FS == 2 char *plainname = 0; # endif IF_NORMALIZE_UTF8PATH(int norm_p); -# ifdef DOSISH +# if USE_NAME_ON_FS == 2 if (cur + 1 == end && (*cur)->type <= ALPHA) { plainname = join_path(path, pathlen, dirsep, (*cur)->str, strlen((*cur)->str)); if (!plainname) return -1; @@ -1633,7 +1694,7 @@ glob_helper( } switch (p->type) { case ALPHA: -# ifdef DOSISH +# if USE_NAME_ON_FS == 2 if (plainname) { *new_end++ = p->next; break; @@ -1703,10 +1764,10 @@ glob_helper( status = -1; break; } -#ifdef HAVE_GETATTRLIST +#if defined HAVE_GETATTRLIST || defined _WIN32 if ((*cur)->type == ALPHA) { long base = pathlen + (dirsep != 0); - buf = replace_real_basename(buf, base, IF_NORMALIZE_UTF8PATH(1)+0); + buf = replace_real_basename(buf, base, enc, IF_NORMALIZE_UTF8PATH(1)+0); } #endif status = glob_helper(buf, 1, UNKNOWN, UNKNOWN, new_beg, diff --git a/test/ruby/test_dir.rb b/test/ruby/test_dir.rb index 6184f05edb..85fdd16dfd 100644 --- a/test/ruby/test_dir.rb +++ b/test/ruby/test_dir.rb @@ -248,6 +248,18 @@ class TestDir < Test::Unit::TestCase assert_equal(roots.map {|n| "/..#{n}"}, Dir.glob("/../*"), bug9648) end + if /mswin|mingw/ =~ RUBY_PLATFORM + def test_glob_legacy_short_name + bug10819 = '[ruby-core:67954] [Bug #10819]' + skip unless /\A\w:/ =~ ENV["ProgramFiles"] + short = "#$&/PROGRA~1" + skip unless File.directory?(short) + entries = Dir.glob("#{short}/Common*") + assert_not_empty(entries, bug10819) + assert_equal(Dir.glob("#{File.expand_path(short)}/Common*"), entries, bug10819) + end + end + def test_home env_home = ENV["HOME"] env_logdir = ENV["LOGDIR"] -- cgit v1.2.3