ruby-changes:3094
From: ko1@a...
Date: 24 Dec 2007 17:06:30 +0900
Subject: [ruby-changes:3094] nobu - Ruby:r14586 (trunk): * load.c (rb_feature_p): returns loading path name too.
nobu 2007-12-24 17:06:16 +0900 (Mon, 24 Dec 2007) New Revision: 14586 Modified files: trunk/ChangeLog trunk/include/ruby/st.h trunk/load.c trunk/st.c Log: * load.c (rb_feature_p): returns loading path name too. * load.c (search_required): returns path too if feature is being loaded. [ruby-dev:32048] [TODO: refactoring] http://svn.ruby-lang.org/cgi-bin/viewvc.cgi/trunk/load.c?r1=14586&r2=14585 http://svn.ruby-lang.org/cgi-bin/viewvc.cgi/trunk/include/ruby/st.h?r1=14586&r2=14585 http://svn.ruby-lang.org/cgi-bin/viewvc.cgi/trunk/ChangeLog?r1=14586&r2=14585 http://svn.ruby-lang.org/cgi-bin/viewvc.cgi/trunk/st.c?r1=14586&r2=14585 Index: include/ruby/st.h =================================================================== --- include/ruby/st.h (revision 14585) +++ include/ruby/st.h (revision 14586) @@ -77,6 +77,7 @@ int st_delete_safe(st_table *, st_data_t *, st_data_t *, st_data_t); int st_insert(st_table *, st_data_t, st_data_t); int st_lookup(st_table *, st_data_t, st_data_t *); +int st_get_key(st_table *, st_data_t, st_data_t *); int st_foreach(st_table *, int (*)(ANYARGS), st_data_t); int st_reverse_foreach(st_table *, int (*)(ANYARGS), st_data_t); void st_add_direct(st_table *, st_data_t, st_data_t); Index: ChangeLog =================================================================== --- ChangeLog (revision 14585) +++ ChangeLog (revision 14586) @@ -1,4 +1,4 @@ -Mon Dec 24 16:53:48 2007 Nobuyoshi Nakada <nobu@r...> +Mon Dec 24 17:06:14 2007 Nobuyoshi Nakada <nobu@r...> * load.c (rb_feature_p): returns loading path name too. Index: load.c =================================================================== --- load.c (revision 14585) +++ load.c (revision 14586) @@ -99,14 +99,16 @@ } static int -rb_feature_p(const char *feature, const char *ext, int rb, int expanded) +rb_feature_p(const char *feature, const char *ext, int rb, int expanded, const char **fn) { VALUE v, features, p, load_path = 0; const char *f, *e; long i, len, elen, n; st_table *loading_tbl; + st_data_t data; int type; + if (fn) *fn = 0; if (ext) { len = ext - feature; elen = strlen(ext); @@ -143,6 +145,7 @@ } loading_tbl = get_loading_table(); if (loading_tbl) { + f = 0; if (!expanded) { struct loaded_feature_searching fs; fs.name = feature; @@ -151,9 +154,13 @@ fs.load_path = load_path ? load_path : get_load_path(); fs.result = 0; st_foreach(loading_tbl, loaded_feature_path_i, (st_data_t)&fs); - if (fs.result) goto loading; + if ((f = fs.result) != 0) { + if (fn) *fn = f; + goto loading; + } } - if (st_lookup(loading_tbl, (st_data_t)feature, 0)) { + if (st_get_key(loading_tbl, (st_data_t)feature, &data)) { + if (fn) *fn = (const char*)data; loading: if (!ext) return 'u'; return !IS_RBEXT(ext) ? 's' : 'r'; @@ -166,7 +173,8 @@ MEMCPY(buf, feature, char, len); for (i = 0; (e = loadable_ext[i]) != 0; i++) { strncpy(buf + len, e, DLEXT_MAXLEN + 1); - if (st_lookup(loading_tbl, (st_data_t)buf, 0)) { + if (st_get_key(loading_tbl, (st_data_t)buf, &data)) { + if (fn) *fn = (const char*)data; return i ? 's' : 'r'; } } @@ -182,15 +190,15 @@ if (ext && !strchr(ext, '/')) { if (IS_RBEXT(ext)) { - if (rb_feature_p(feature, ext, Qtrue, Qfalse)) return Qtrue; + if (rb_feature_p(feature, ext, Qtrue, Qfalse, 0)) return Qtrue; return Qfalse; } else if (IS_SOEXT(ext) || IS_DLEXT(ext)) { - if (rb_feature_p(feature, ext, Qfalse, Qfalse)) return Qtrue; + if (rb_feature_p(feature, ext, Qfalse, Qfalse, 0)) return Qtrue; return Qfalse; } } - if (rb_feature_p(feature, feature + strlen(feature), Qtrue, Qfalse)) + if (rb_feature_p(feature, feature + strlen(feature), Qtrue, Qfalse, 0)) return Qtrue; return Qfalse; } @@ -337,7 +345,8 @@ st_insert(loading_tbl, (st_data_t)ftptr, data); return (char *)ftptr; } - return RTEST(rb_barrier_wait((VALUE)data)) ? (char *)ftptr : 0; + rb_barrier_wait((VALUE)data); + return 0; } static void @@ -390,32 +399,37 @@ VALUE tmp; char *ext, *ftptr; int type, ft = 0; + const char *loading; *path = 0; ext = strrchr(ftptr = RSTRING_PTR(fname), '.'); if (ext && !strchr(ext, '/')) { if (IS_RBEXT(ext)) { - if (rb_feature_p(ftptr, ext, Qtrue, Qfalse)) + if (rb_feature_p(ftptr, ext, Qtrue, Qfalse, &loading)) { + if (loading) *path = rb_str_new2(loading); return 'r'; + } if ((tmp = rb_find_file(fname)) != 0) { tmp = rb_file_expand_path(tmp, Qnil); ext = strrchr(ftptr = RSTRING_PTR(tmp), '.'); - if (!rb_feature_p(ftptr, ext, Qtrue, Qtrue)) + if (!rb_feature_p(ftptr, ext, Qtrue, Qtrue, 0)) *path = tmp; return 'r'; } return 0; } else if (IS_SOEXT(ext)) { - if (rb_feature_p(ftptr, ext, Qfalse, Qfalse)) + if (rb_feature_p(ftptr, ext, Qfalse, Qfalse, &loading)) { + if (loading) *path = rb_str_new2(loading); return 's'; + } tmp = rb_str_new(RSTRING_PTR(fname), ext - RSTRING_PTR(fname)); #ifdef DLEXT2 OBJ_FREEZE(tmp); if (rb_find_file_ext(&tmp, loadable_ext + 1)) { tmp = rb_file_expand_path(tmp, Qnil); ext = strrchr(ftptr = RSTRING_PTR(tmp), '.'); - if (!rb_feature_p(ftptr, ext, Qfalse, Qtrue)) + if (!rb_feature_p(ftptr, ext, Qfalse, Qtrue, 0)) *path = tmp; return 's'; } @@ -425,25 +439,28 @@ if ((tmp = rb_find_file(tmp)) != 0) { tmp = rb_file_expand_path(tmp, Qnil); ext = strrchr(ftptr = RSTRING_PTR(tmp), '.'); - if (!rb_feature_p(ftptr, ext, Qfalse, Qtrue)) + if (!rb_feature_p(ftptr, ext, Qfalse, Qtrue, 0)) *path = tmp; return 's'; } #endif } else if (IS_DLEXT(ext)) { - if (rb_feature_p(ftptr, ext, Qfalse, Qfalse)) + if (rb_feature_p(ftptr, ext, Qfalse, Qfalse, &loading)) { + if (loading) *path = rb_str_new2(loading); return 's'; + } if ((tmp = rb_find_file(fname)) != 0) { tmp = rb_file_expand_path(tmp, Qnil); ext = strrchr(ftptr = RSTRING_PTR(tmp), '.'); - if (!rb_feature_p(ftptr, ext, Qfalse, Qtrue)) + if (!rb_feature_p(ftptr, ext, Qfalse, Qtrue, 0)) *path = tmp; return 's'; } } } - else if ((ft = rb_feature_p(ftptr, 0, Qfalse, Qfalse)) == 'r') { + else if ((ft = rb_feature_p(ftptr, 0, Qfalse, Qfalse, &loading)) == 'r') { + if (loading) *path = rb_str_new2(loading); return 'r'; } tmp = fname; @@ -451,17 +468,17 @@ tmp = rb_file_expand_path(tmp, Qnil); switch (type) { case 0: - ftptr = RSTRING_PTR(tmp); if (ft) break; - return rb_feature_p(ftptr, 0, Qfalse, Qtrue); + ftptr = RSTRING_PTR(tmp); + return rb_feature_p(ftptr, 0, Qfalse, Qtrue, 0); default: if (ft) break; case 1: ext = strrchr(ftptr = RSTRING_PTR(tmp), '.'); - if (rb_feature_p(ftptr, ext, !--type, Qtrue)) + if (rb_feature_p(ftptr, ext, !--type, Qtrue, &loading) && !loading) break; *path = tmp; } @@ -494,6 +511,8 @@ } volatile saved; char *volatile ftptr = 0; + FilePathValue(fname); + RB_GC_GUARD(fname) = rb_str_new4(fname); PUSH_TAG(); saved.safe = rb_safe_level(); if ((state = EXEC_TAG()) == 0) { @@ -502,8 +521,6 @@ int found; rb_set_safe_level_force(safe); - FilePathValue(fname); - RB_GC_GUARD(fname) = rb_str_new4(fname); found = search_required(fname, &path); if (found) { if (!path || !(ftptr = load_lock(RSTRING_PTR(path)))) { Index: st.c =================================================================== --- st.c (revision 14585) +++ st.c (revision 14586) @@ -302,6 +302,35 @@ } } +int +st_get_key(st_table *table, register st_data_t key, st_data_t *result) +{ + unsigned int hash_val, bin_pos; + register st_table_entry *ptr; + + if (table->entries_packed) { + int i; + for (i = 0; i < table->num_entries; i++) { + if ((st_data_t)table->bins[i*2] == key) { + if (result !=0) *result = (st_data_t)table->bins[i*2]; + return 1; + } + } + return 0; + } + + hash_val = do_hash(key, table); + FIND_ENTRY(table, ptr, hash_val, bin_pos); + + if (ptr == 0) { + return 0; + } + else { + if (result != 0) *result = ptr->key; + return 1; + } +} + #define ADD_DIRECT(table, key, value, hash_val, bin_pos)\ do {\ st_table_entry *entry, *head;\ -- ML: ruby-changes@q... Info: http://www.atdot.net/~ko1/quickml