Skip to content

Commit

Permalink
Fix unexpected node bug for shareable_constant_value: literal
Browse files Browse the repository at this point in the history
[Bug #20339]
[Bug #20341]

`const_decl_path` changes the value of `NODE **dest`, LHS of an assignment,
with `NODE_LIT` created by `const_decl_path`. `shareable_literal_constant` calls
`const_decl_path` via `ensure_shareable_node` multiple times if RHS of an assignment
is array or hash. This means `NODE **dest` argument of `const_decl_path` can be `NODE_LIT`
from the second time then causes `[BUG] unexpected node: NODE_LIT` in
`rb_node_const_decl_val`.
This commit change to not update `NODE **dest` in `const_decl_path` to
fix the issue.
  • Loading branch information
yui-knk committed Mar 21, 2024
1 parent 6650b3a commit 8ba4d7d
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 10 deletions.
20 changes: 10 additions & 10 deletions parse.y
Expand Up @@ -14068,13 +14068,13 @@ mark_lvar_used(struct parser_params *p, NODE *rhs)
}

static NODE *
const_decl_path(struct parser_params *p, NODE **dest)
const_decl_path(struct parser_params *p, NODE *dest)
{
NODE *n = *dest;
if (!nd_type_p(n, NODE_CALL)) {
const YYLTYPE *loc = &n->nd_loc;
VALUE path = rb_node_const_decl_val(n);
*dest = n = NEW_LIT(path, loc);
NODE *n = dest;
if (!nd_type_p(dest, NODE_CALL)) {
const YYLTYPE *loc = &dest->nd_loc;
VALUE path = rb_node_const_decl_val(dest);
n = NEW_LIT(path, loc);
RB_OBJ_WRITTEN(p->ast, Qnil, RNODE_LIT(n)->nd_lit);
}
return n;
Expand All @@ -14096,7 +14096,7 @@ make_shareable_node(struct parser_params *p, NODE *value, bool copy, const YYLTY
}

static NODE *
ensure_shareable_node(struct parser_params *p, NODE **dest, NODE *value, const YYLTYPE *loc)
ensure_shareable_node(struct parser_params *p, NODE *dest, NODE *value, const YYLTYPE *loc)
{
NODE *fcore = NEW_LIT(rb_mRubyVMFrozenCore, loc);
NODE *args = NEW_LIST(value, loc);
Expand Down Expand Up @@ -14147,7 +14147,7 @@ shareable_literal_value(struct parser_params *p, NODE *node)

static NODE *
shareable_literal_constant(struct parser_params *p, enum shareability shareable,
NODE **dest, NODE *value, const YYLTYPE *loc, size_t level)
NODE *dest, NODE *value, const YYLTYPE *loc, size_t level)
{
# define shareable_literal_constant_next(n) \
shareable_literal_constant(p, shareable, dest, (n), &(n)->nd_loc, level+1)
Expand Down Expand Up @@ -14297,7 +14297,7 @@ shareable_constant_value(struct parser_params *p, enum shareability shareable,

case shareable_literal:
{
NODE *lit = shareable_literal_constant(p, shareable, &lhs, value, loc, 0);
NODE *lit = shareable_literal_constant(p, shareable, lhs, value, loc, 0);
if (lit) return lit;
return value;
}
Expand All @@ -14306,7 +14306,7 @@ shareable_constant_value(struct parser_params *p, enum shareability shareable,
case shareable_copy:
case shareable_everything:
{
NODE *lit = shareable_literal_constant(p, shareable, &lhs, value, loc, 0);
NODE *lit = shareable_literal_constant(p, shareable, lhs, value, loc, 0);
if (lit) return lit;
return make_shareable_node(p, value, shareable == shareable_copy, loc);
}
Expand Down
18 changes: 18 additions & 0 deletions test/ruby/test_parse.rb
Expand Up @@ -1526,6 +1526,24 @@ def test_shareable_constant_value_simple
assert_not_ractor_shareable(obj)
assert_equal obj, a
assert !obj.equal?(a)
bug_20339 = '[ruby-core:117186] [Bug #20339]'
bug_20341 = '[ruby-core:117197] [Bug #20341]'
a, b = eval_separately(<<~'end;')
# shareable_constant_value: literal
foo = 1
bar = 2
A = { foo => bar }
B = [foo, bar]
[A, B]
end;
assert_ractor_shareable(a)
assert_ractor_shareable(b)
assert_equal([1], a.keys, bug_20339)
assert_equal([2], a.values, bug_20339)
assert_equal(1, b[0], bug_20341)
assert_equal(2, b[1], bug_20341)
end
def test_shareable_constant_value_nested
Expand Down

0 comments on commit 8ba4d7d

Please sign in to comment.