diff options
| -rw-r--r-- | ext/-test-/string/capacity.c | 17 | ||||
| -rw-r--r-- | test/-ext-/string/test_capacity.rb | 18 |
2 files changed, 35 insertions, 0 deletions
diff --git a/ext/-test-/string/capacity.c b/ext/-test-/string/capacity.c new file mode 100644 index 0000000000..ba7fb85e9e --- /dev/null +++ b/ext/-test-/string/capacity.c @@ -0,0 +1,17 @@ +#include "ruby.h" +#include "internal.h" + +static VALUE +bug_str_capacity(VALUE klass, VALUE str) +{ + return + STR_EMBED_P(str) ? INT2FIX(RSTRING_EMBED_LEN_MAX) : \ + STR_SHARED_P(str) ? INT2FIX(0) : \ + LONG2FIX(RSTRING(str)->as.heap.aux.capa); +} + +void +Init_capacity(VALUE klass) +{ + rb_define_singleton_method(klass, "capacity", bug_str_capacity, 1); +} diff --git a/test/-ext-/string/test_capacity.rb b/test/-ext-/string/test_capacity.rb new file mode 100644 index 0000000000..b4e4fb7c02 --- /dev/null +++ b/test/-ext-/string/test_capacity.rb @@ -0,0 +1,18 @@ +# frozen_string_literal: true +require 'test/unit' +require '-test-/string' + +class Test_StringCapacity < Test::Unit::TestCase + def test_capacity_embeded + size = RbConfig::SIZEOF['void*'] * 3 - 1 + assert_equal size, Bug::String.capacity('foo') + end + + def test_capacity_shared + assert_equal 0, Bug::String.capacity(:abcdefghijklmnopqrstuvwxyz.to_s) + end + + def test_capacity_normal + assert_equal 128, Bug::String.capacity('1'*128) + end +end |