From 9869b65573d5bbaf5af4f20a577b41f95d73b846 Mon Sep 17 00:00:00 2001 From: nagachika Date: Sun, 25 Mar 2018 05:35:17 +0000 Subject: merge revision(s) 62394,62395: [Backport #14469] vm_insnhelper.c: rb_autoloading_value flag * vm_insnhelper.c (vm_get_ev_const): add flag argument of `rb_autoloading_value`. * constant.h (rb_autoloading_value): moved the declaration from vm_core.h for `rb_const_flag_t`. [ruby-core:85516] [Bug #14469] variable.c: flags at autoloading * variable.c (const_tbl_update): flags by deprecate_constant / private_constant set during autoloading should be preserved after required. [ruby-core:85516] [Bug #14469] git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/branches/ruby_2_4@62916 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- variable.c | 49 ++++++++++++++++++++++++++++++++++++++----------- 1 file changed, 38 insertions(+), 11 deletions(-) (limited to 'variable.c') diff --git a/variable.c b/variable.c index f84475ddf9..b1080d44ab 100644 --- a/variable.c +++ b/variable.c @@ -1903,6 +1903,7 @@ struct autoload_state { struct autoload_data_i { VALUE feature; int safe_level; + rb_const_flag_t flag; VALUE value; struct autoload_state *state; /* points to on-stack struct */ }; @@ -1985,6 +1986,7 @@ rb_autoload_str(VALUE mod, ID id, VALUE file) ele->safe_level = rb_safe_level(); ele->value = Qundef; ele->state = 0; + ele->flag = CONST_PUBLIC; st_insert(tbl, (st_data_t)id, (st_data_t)ad); } @@ -2060,7 +2062,7 @@ check_autoload_required(VALUE mod, ID id, const char **loadingpath) } int -rb_autoloading_value(VALUE mod, ID id, VALUE* value) +rb_autoloading_value(VALUE mod, ID id, VALUE* value, rb_const_flag_t *flag) { VALUE load; struct autoload_data_i *ele; @@ -2073,6 +2075,9 @@ rb_autoloading_value(VALUE mod, ID id, VALUE* value) if (value) { *value = ele->value; } + if (flag) { + *flag = ele->flag; + } return 1; } } @@ -2087,13 +2092,14 @@ autoload_defined_p(VALUE mod, ID id) if (!ce || ce->value != Qundef) { return 0; } - return !rb_autoloading_value(mod, id, NULL); + return !rb_autoloading_value(mod, id, NULL, NULL); } struct autoload_const_set_args { VALUE mod; ID id; VALUE value; + rb_const_flag_t flag; }; static void const_tbl_update(struct autoload_const_set_args *); @@ -2140,6 +2146,7 @@ autoload_reset(VALUE arg) args.mod = state->mod; args.id = state->id; args.value = state->ele->value; + args.flag = state->ele->flag; safe_backup = rb_safe_level(); rb_set_safe_level_force(state->ele->safe_level); rb_ensure(autoload_const_set, (VALUE)&args, @@ -2286,6 +2293,7 @@ static VALUE rb_const_search(VALUE klass, ID id, int exclude, int recurse, int visibility) { VALUE value, tmp, av; + rb_const_flag_t flag; int mod_retry = 0; tmp = klass; @@ -2304,7 +2312,7 @@ rb_const_search(VALUE klass, ID id, int exclude, int recurse, int visibility) if (value == Qundef) { if (am == tmp) break; am = tmp; - if (rb_autoloading_value(tmp, id, &av)) return av; + if (rb_autoloading_value(tmp, id, &av, &flag)) return av; rb_autoload_load(tmp, id); continue; } @@ -2556,7 +2564,7 @@ rb_const_defined_0(VALUE klass, ID id, int exclude, int recurse, int visibility) return (int)Qfalse; } if (ce->value == Qundef && !check_autoload_required(tmp, id, 0) && - !rb_autoloading_value(tmp, id, 0)) + !rb_autoloading_value(tmp, id, NULL, NULL)) return (int)Qfalse; return (int)Qtrue; } @@ -2637,6 +2645,7 @@ rb_const_set(VALUE klass, ID id, VALUE val) args.mod = klass; args.id = id; args.value = val; + args.flag = CONST_PUBLIC; const_tbl_update(&args); } /* @@ -2668,6 +2677,21 @@ rb_const_set(VALUE klass, ID id, VALUE val) } } +static struct autoload_data_i * +current_autoload_data(VALUE mod, ID id) +{ + struct autoload_data_i *ele; + VALUE load = autoload_data(mod, id); + if (!load) return 0; + ele = check_autoload_data(load); + if (!ele) return 0; + /* for autoloading thread, keep the defined value to autoloading storage */ + if (ele->state && (ele->state->thread == rb_thread_current())) { + return ele; + } + return 0; +} + static void const_tbl_update(struct autoload_const_set_args *args) { @@ -2676,19 +2700,15 @@ const_tbl_update(struct autoload_const_set_args *args) VALUE val = args->value; ID id = args->id; struct rb_id_table *tbl = RCLASS_CONST_TBL(klass); - rb_const_flag_t visibility = CONST_PUBLIC; + rb_const_flag_t visibility = args->flag; rb_const_entry_t *ce; if (rb_id_table_lookup(tbl, id, &value)) { ce = (rb_const_entry_t *)value; if (ce->value == Qundef) { - VALUE load; - struct autoload_data_i *ele; + struct autoload_data_i *ele = current_autoload_data(klass, id); - load = autoload_data(klass, id); - /* for autoloading thread, keep the defined value to autoloading storage */ - if (load && (ele = check_autoload_data(load)) && ele->state && - (ele->state->thread == rb_thread_current())) { + if (ele) { rb_clear_constant_cache(); ele->value = val; /* autoload_i is non-WB-protected */ @@ -2777,6 +2797,13 @@ set_const_visibility(VALUE mod, int argc, const VALUE *argv, if ((ce = rb_const_lookup(mod, id))) { ce->flag &= ~mask; ce->flag |= flag; + if (ce->value == Qundef) { + struct autoload_data_i *ele = current_autoload_data(mod, id); + if (ele) { + ele->flag &= ~mask; + ele->flag |= flag; + } + } } else { if (i > 0) { -- cgit v1.2.3