summaryrefslogtreecommitdiff
path: root/vm_insnhelper.c
diff options
context:
space:
mode:
authormatz <matz@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>2008-08-28 11:29:57 +0000
committermatz <matz@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>2008-08-28 11:29:57 +0000
commitf7f0784270ce51fac9c21b58668b3e455d2ae60e (patch)
treed39193b9d4db25052929ad9250e61f710ec237b0 /vm_insnhelper.c
parenta7fa5122d7c48f2bddeaa74114111b666f986a21 (diff)
* vm_insnhelper.c (vm_yield_setup_args): object with to_ary should
behave like array. a patch from Yusuke Endoh <mame at tsg.ne.jp> in [ruby-dev:35988]. [ruby-dev:35977] git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@18894 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'vm_insnhelper.c')
-rw-r--r--vm_insnhelper.c4
1 files changed, 2 insertions, 2 deletions
diff --git a/vm_insnhelper.c b/vm_insnhelper.c
index 07fe32d2a2..f40d6a2414 100644
--- a/vm_insnhelper.c
+++ b/vm_insnhelper.c
@@ -704,6 +704,7 @@ vm_yield_setup_args(rb_thread_t * const th, const rb_iseq_t *iseq,
int i;
int argc = orig_argc;
const int m = iseq->argc;
+ VALUE ary;
th->mark_stack_len = argc;
@@ -714,8 +715,7 @@ vm_yield_setup_args(rb_thread_t * const th, const rb_iseq_t *iseq,
*/
if (!(iseq->arg_simple & 0x02) &&
(m + iseq->arg_post_len) > 0 &&
- argc == 1 && TYPE(argv[0]) == T_ARRAY) {
- VALUE ary = argv[0];
+ argc == 1 && !NIL_P(ary = rb_check_array_type(argv[0]))) {
th->mark_stack_len = argc = RARRAY_LEN(ary);
CHECK_STACK_OVERFLOW(th->cfp, argc);