From a1692f7fdf62dc109886d9910b4dcca635dc2e25 Mon Sep 17 00:00:00 2001 From: nobu Date: Sat, 2 Dec 2017 07:09:16 +0000 Subject: [PATCH] string.c: fix rb_external_str_new_with_enc * string.c (rb_external_str_new_with_enc): do not search non-ascii by NULL pointer. [ruby-core:84055] [Bug #14150] git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@60979 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- ext/-test-/string/new.c | 21 +++++++++++++++++++++ string.c | 4 ++++ test/-ext-/string/test_external_new.rb | 11 +++++++++++ 3 files changed, 36 insertions(+) create mode 100644 ext/-test-/string/new.c create mode 100644 test/-ext-/string/test_external_new.rb diff --git a/ext/-test-/string/new.c b/ext/-test-/string/new.c new file mode 100644 index 0000000000..8a3613e374 --- /dev/null +++ b/ext/-test-/string/new.c @@ -0,0 +1,21 @@ +#include "ruby.h" +#include "ruby/encoding.h" + +VALUE +bug_str_buf_new(VALUE self, VALUE len) +{ + return rb_str_buf_new(NUM2LONG(len)); +} + +VALUE +bug_external_str_new(VALUE self, VALUE len, VALUE enc) +{ + return rb_external_str_new_with_enc(NULL, NUM2LONG(len), rb_to_encoding(enc)); +} + +void +Init_new(VALUE klass) +{ + rb_define_singleton_method(klass, "buf_new", bug_str_buf_new, 1); + rb_define_singleton_method(klass, "external_new", bug_external_str_new, 2); +} diff --git a/string.c b/string.c index 806436e327..56b6f641c6 100644 --- a/string.c +++ b/string.c @@ -1011,6 +1011,10 @@ rb_external_str_new_with_enc(const char *ptr, long len, rb_encoding *eenc) VALUE str; const int eidx = rb_enc_to_index(eenc); + if (!ptr) { + return rb_tainted_str_new_with_enc(ptr, len, eenc); + } + /* ASCII-8BIT case, no conversion */ if ((eidx == rb_ascii8bit_encindex()) || (eidx == rb_usascii_encindex() && search_nonascii(ptr, ptr + len))) { diff --git a/test/-ext-/string/test_external_new.rb b/test/-ext-/string/test_external_new.rb new file mode 100644 index 0000000000..f7c7908db7 --- /dev/null +++ b/test/-ext-/string/test_external_new.rb @@ -0,0 +1,11 @@ +# frozen_string_literal: false +require 'test/unit' +require '-test-/string' + +class Test_StringExternalNew < Test::Unit::TestCase + def test_external_new_with_enc + Encoding.list.each do |enc| + assert_equal(enc, Bug::String.external_new(0, enc).encoding) + end + end +end