From fe6cf485e6e1d23766a81246e825b61eb534c176 Mon Sep 17 00:00:00 2001
From: nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Sat, 28 Feb 2015 09:09:18 +0000
Subject: enum.c: Fixnum only

* enum.c (limit_by_enum_size, enum_size_over_p): check only
  against Fixnum size.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@49782 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 enum.c                            | 10 +++++-----
 test/ruby/test_lazy_enumerator.rb | 12 ++++++++++++
 2 files changed, 17 insertions(+), 5 deletions(-)

diff --git a/enum.c b/enum.c
index 2e78d08a9a..6c68240f0e 100644
--- a/enum.c
+++ b/enum.c
@@ -319,17 +319,17 @@ limit_by_enum_size(VALUE obj, long n)
 {
     unsigned long limit;
     VALUE size = rb_check_funcall(obj, id_size, 0, 0);
-    if (size == Qundef) return n;
-    limit = NUM2ULONG(size);
-    return ((unsigned long)n > limit) ? limit : n;
+    if (!FIXNUM_P(size)) return n;
+    limit = FIX2ULONG(size);
+    return ((unsigned long)n > limit) ? (long)limit : n;
 }
 
 static int
 enum_size_over_p(VALUE obj, long n)
 {
     VALUE size = rb_check_funcall(obj, id_size, 0, 0);
-    if (size == Qundef) return 0;
-    return ((unsigned long)n > NUM2ULONG(size));
+    if (!FIXNUM_P(size)) return 0;
+    return ((unsigned long)n > FIX2ULONG(size));
 }
 
 /*
diff --git a/test/ruby/test_lazy_enumerator.rb b/test/ruby/test_lazy_enumerator.rb
index 75f24580ca..952a773d58 100644
--- a/test/ruby/test_lazy_enumerator.rb
+++ b/test/ruby/test_lazy_enumerator.rb
@@ -483,6 +483,18 @@ EOS
     assert_equal Enumerator::Lazy, [].lazy.slice_when{}.class, bug7507
   end
 
+  def test_each_cons_limit
+    n = 1 << 120
+    assert_equal([1, 2], (1..n).lazy.each_cons(2).first)
+    assert_equal([[1, 2], [2, 3]], (1..n).lazy.each_cons(2).first(2))
+  end
+
+  def test_each_slice_limit
+    n = 1 << 120
+    assert_equal([1, 2], (1..n).lazy.each_slice(2).first)
+    assert_equal([[1, 2], [3, 4]], (1..n).lazy.each_slice(2).first(2))
+  end
+
   def test_no_warnings
     le = (1..3).lazy
     assert_warning("") {le.zip([4,5,6]).force}
-- 
cgit v1.2.3