From 8101cc3de4a846f11b76be637c3260aaa73e0b3c Mon Sep 17 00:00:00 2001 From: nobu Date: Sat, 25 Jul 2015 02:05:34 +0000 Subject: string.c: fill the terminator * string.c (str_replace_shared_without_enc): fill the terminator of embedded strings in wide char encodings. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@51372 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- ChangeLog | 5 +++++ string.c | 18 +++++++++++++----- test/-ext-/string/test_cstr.rb | 8 ++++++++ 3 files changed, 26 insertions(+), 5 deletions(-) diff --git a/ChangeLog b/ChangeLog index 9424f140f5..3aba54b561 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,8 @@ +Sat Jul 25 11:05:31 2015 Nobuyoshi Nakada + + * string.c (str_replace_shared_without_enc): fill the terminator + of embedded strings in wide char encodings. + Sat Jul 25 06:38:36 2015 Koichi Sasada * vm_core.h: size should be unsigned. diff --git a/string.c b/string.c index 08fa48b791..bf07d97c1f 100644 --- a/string.c +++ b/string.c @@ -962,16 +962,24 @@ rb_str_export_to_enc(VALUE str, rb_encoding *enc) static VALUE str_replace_shared_without_enc(VALUE str2, VALUE str) { - if (RSTRING_LEN(str) <= RSTRING_EMBED_LEN_MAX) { + const int termlen = TERM_LEN(str); + char *ptr; + long len; + + RSTRING_GETMEM(str, ptr, len); + if (len+termlen <= RSTRING_EMBED_LEN_MAX+1) { + char *ptr2 = RSTRING(str2)->as.ary; STR_SET_EMBED(str2); - memcpy(RSTRING_PTR(str2), RSTRING_PTR(str), RSTRING_LEN(str)+1); - STR_SET_EMBED_LEN(str2, RSTRING_LEN(str)); + memcpy(ptr2, RSTRING_PTR(str), len); + STR_SET_EMBED_LEN(str2, len); + TERM_FILL(ptr2+len, termlen); } else { str = rb_str_new_frozen(str); FL_SET(str2, STR_NOEMBED); - RSTRING(str2)->as.heap.len = RSTRING_LEN(str); - RSTRING(str2)->as.heap.ptr = RSTRING_PTR(str); + RSTRING_GETMEM(str, ptr, len); + RSTRING(str2)->as.heap.len = len; + RSTRING(str2)->as.heap.ptr = ptr; STR_SET_SHARED(str2, str); } return str2; diff --git a/test/-ext-/string/test_cstr.rb b/test/-ext-/string/test_cstr.rb index 1e973d0a39..7e24ec6a70 100644 --- a/test/-ext-/string/test_cstr.rb +++ b/test/-ext-/string/test_cstr.rb @@ -86,6 +86,14 @@ class Test_StringCStr < Test::Unit::TestCase } end + def test_wchar_replace + assert_wchars_term_char("abc") {|s| + w = s.dup + s.replace("abcdefghijklmnop") + s.replace(w) + } + end + def test_embedded_from_heap gh821 = "[GH-821]" embedded_string = "abcdefghi" -- cgit v1.2.3