[Bug #21513] Raise on converting endless range to set

viralpraxis · viralpraxis · commit 77535572a525 · 2025-07-17T19:47:34.000+04:00
ref: https://bugs.ruby-lang.org/issues/21513 Before this patch, trying to convert endless range (e.g. `(1..)`) to set (using `to_set`) would hang
diff --git a/set.c b/set.c
@@ -494,6 +494,14 @@ set_i_initialize(int argc, VALUE *argv, VALUE set)
     rb_check_arity(argc, 0, 1);
 
     if (argc > 0 && (other = argv[0]) != Qnil) {
+        ID id_size = rb_intern("size");
+        if (rb_respond_to(other, id_size)) {
+            VALUE size = rb_funcall(other, id_size, 0);
+            if (RB_TYPE_P(size, T_FLOAT) && RFLOAT_VALUE(size) == INFINITY) {
+                rb_raise(rb_eArgError, "cannot initialize Set from a enumerable with infinite size");
+            }
+        }
+
         if (RB_TYPE_P(other, T_ARRAY)) {
             long i;
             int block_given = rb_block_given_p();
diff --git a/test/ruby/test_set.rb b/test/ruby/test_set.rb
@@ -81,6 +81,16 @@ def test_s_new
 
     s = Set.new(ary) { |o| o * 2 }
     assert_equal([2,4,6], s.sort)
+
+    assert_raise(ArgumentError) {
+      Set.new((1..))
+    }
+    assert_raise(ArgumentError) {
+      Set.new((1..), &:succ)
+    }
+    assert_raise(ArgumentError) {
+      Set.new(1.upto(Float::INFINITY))
+    }
   end
 
   def test_clone
