From e2479cc43febe9cf26093fa2103a199d2d245711 Mon Sep 17 00:00:00 2001 From: nobu Date: Fri, 22 Dec 2017 01:03:17 +0000 Subject: encoding.c: rb_enc_find_index2 * string.c (str_undump): use rb_enc_find_index2 to find encoding by unterminated string. check the format before encoding name. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@61396 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- string.c | 17 ++++------------- 1 file changed, 4 insertions(+), 13 deletions(-) (limited to 'string.c') diff --git a/string.c b/string.c index e6db6a502c..30d57b5df4 100644 --- a/string.c +++ b/string.c @@ -6263,7 +6263,6 @@ str_undump(VALUE str) } else { const char *encname; - char *buf; int encidx; ptrdiff_t size; @@ -6280,22 +6279,14 @@ str_undump(VALUE str) s = memchr(s, '"', s_end-s); size = s - encname; if (!s) goto invalid_format; - if (size > 100) { - rb_raise(rb_eRuntimeError, "dumped string has unknown encoding name"); - } - buf = ALLOC_N(char, size+1); - memcpy(buf, encname, size); - buf[size] = '\0'; - encidx = rb_enc_find_index(buf); - xfree(buf); + if (s_end - s != 2) goto invalid_format; + if (s[0] != '"' || s[1] != ')') goto invalid_format; + + encidx = rb_enc_find_index2(encname, (long)size); if (encidx < 0) { rb_raise(rb_eRuntimeError, "dumped string has unknown encoding name"); } rb_enc_associate_index(undumped, encidx); - - if (s_end - s != 2 || - s[0] != '"' || - s[1] != ')') goto invalid_format; } break; } -- cgit v1.2.3