From 2cbbeba0e091e95bc8456c29f6cab2c36b44a86b Mon Sep 17 00:00:00 2001 From: "Yukihiro \"Matz\" Matsumoto" <matz@ruby-lang.org> Date: Sun, 19 Mar 2017 00:21:53 +0900 Subject: [PATCH] Check call depth when mrb_yield_class() is called; ref #3521 --- src/vm.c | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/vm.c b/src/vm.c index e860cf9fd..3e40cecb0 100644 --- a/src/vm.c +++ b/src/vm.c @@ -668,6 +668,9 @@ mrb_yield_with_class(mrb_state *mrb, mrb_value b, mrb_int argc, const mrb_value if (mrb_nil_p(b)) { mrb_raise(mrb, E_ARGUMENT_ERROR, "no block given"); } + if (mrb->c->ci - mrb->c->cibase > MRB_FUNCALL_DEPTH_MAX) { + mrb_exc_raise(mrb, mrb_obj_value(mrb->stack_err)); + } p = mrb_proc_ptr(b); ci = cipush(mrb); ci->mid = mid; -- 2.26.2