enumerator.c: rb_check_funcall

* enumerator.c (enumerator_size): use rb_check_funcall() instead of
  respond_to? and call.


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@42707 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

ChangeLog

@@ -1,4 +1,7 @@
-Tue Aug 27 16:51:21 2013  Nobuyoshi Nakada  <nobu@ruby-lang.org>
+Tue Aug 27 16:56:50 2013  Nobuyoshi Nakada  <nobu@ruby-lang.org>
+
+	* enumerator.c (enumerator_size): use rb_check_funcall() instead of
+	  respond_to? and call.
 
 	* enumerator.c (enumerator_each): ensure that argument array size
 	  does not overflow at appending.

enumerator.c

@@ -1007,19 +1007,19 @@ static VALUE
 enumerator_size(VALUE obj)
 {
     struct enumerator *e = enumerator_ptr(obj);
+    int argc = 0;
+    const VALUE *argv = NULL;
+    VALUE size;
 
     if (e->size_fn) {
 	return (*e->size_fn)(e->obj, e->args, obj);
     }
-    if (rb_respond_to(e->size, id_call)) {
-	if (e->args) {
-	    int argc = (int)RARRAY_LEN(e->args);
-	    VALUE *argv = RARRAY_PTR(e->args);
-	    return rb_funcall2(e->size, id_call, argc, argv);
-	} else {
-	    return rb_funcall(e->size, id_call, 0);
-	}
+    if (e->args) {
+	argc = (int)RARRAY_LEN(e->args);
+	argv = RARRAY_RAWPTR(e->args);
     }
+    size = rb_check_funcall(e->size, id_call, argc, argv);
+    if (size != Qundef) return size;
     return e->size;
 }