From 62181e17da2db729eb6d95f6edff9e028ff557b6 Mon Sep 17 00:00:00 2001 From: Jeremy Evans Date: Tue, 26 Sep 2023 15:31:55 -0700 Subject: Make {Queue,SizedQueue}#freeze raise TypeError Fixes [Bug #17146] --- test/ruby/test_thread_queue.rb | 9 +++++++++ thread_sync.c | 17 +++++++++++++++++ 2 files changed, 26 insertions(+) diff --git a/test/ruby/test_thread_queue.rb b/test/ruby/test_thread_queue.rb index fd77853f0e..545bf98888 100644 --- a/test/ruby/test_thread_queue.rb +++ b/test/ruby/test_thread_queue.rb @@ -19,6 +19,15 @@ class TestThreadQueue < Test::Unit::TestCase } end + def test_freeze + assert_raise(TypeError) { + Queue.new.freeze + } + assert_raise(TypeError) { + SizedQueue.new(5).freeze + } + end + def test_queue grind(5, 1000, 15, Queue) end diff --git a/thread_sync.c b/thread_sync.c index 85ebec4d8c..ca463c35f1 100644 --- a/thread_sync.c +++ b/thread_sync.c @@ -1138,6 +1138,21 @@ rb_queue_length(VALUE self) return LONG2NUM(queue_length(self, queue_ptr(self))); } +NORETURN(static VALUE rb_queue_freeze(VALUE self)); +/* + * call-seq: + * freeze + * + * Raises an exception: + * Queue.new.freeze # Raises TypeError (cannot freeze #) + */ +static VALUE +rb_queue_freeze(VALUE self) +{ + rb_raise(rb_eTypeError, "cannot freeze " "%+"PRIsVALUE, self); + UNREACHABLE_RETURN(self); +} + /* * Document-method: Thread::Queue#num_waiting * @@ -1599,6 +1614,7 @@ Init_thread_sync(void) rb_define_method(rb_cQueue, "clear", rb_queue_clear, 0); rb_define_method(rb_cQueue, "length", rb_queue_length, 0); rb_define_method(rb_cQueue, "num_waiting", rb_queue_num_waiting, 0); + rb_define_method(rb_cQueue, "freeze", rb_queue_freeze, 0); rb_define_alias(rb_cQueue, "enq", "push"); rb_define_alias(rb_cQueue, "<<", "push"); @@ -1615,6 +1631,7 @@ Init_thread_sync(void) rb_define_method(rb_cSizedQueue, "clear", rb_szqueue_clear, 0); rb_define_method(rb_cSizedQueue, "length", rb_szqueue_length, 0); rb_define_method(rb_cSizedQueue, "num_waiting", rb_szqueue_num_waiting, 0); + rb_define_method(rb_cSizedQueue, "freeze", rb_queue_freeze, 0); rb_define_alias(rb_cSizedQueue, "size", "length"); /* CVar */ -- cgit v1.2.3