From 6ea26aafd5a2d64987bd1607f9212f6ce494765c Mon Sep 17 00:00:00 2001 From: "Yukihiro \"Matz\" Matsumoto" Date: Sat, 18 Mar 2017 17:37:24 +0900 Subject: Save/restore GC arena index to avoid arena overflow error. --- src/error.c | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/error.c b/src/error.c index 01c0e376c..6d488372e 100644 --- a/src/error.c +++ b/src/error.c @@ -329,6 +329,7 @@ mrb_vformat(mrb_state *mrb, const char *format, va_list ap) const char *b = p; ptrdiff_t size; mrb_value ary = mrb_ary_new_capa(mrb, 4); + int ai = mrb_gc_arena_save(mrb); while (*p) { const char c = *p++; @@ -352,14 +353,18 @@ mrb_vformat(mrb_state *mrb, const char *format, va_list ap) break; } } + mrb_gc_arena_restore(mrb, ai); } if (b == format) { return mrb_str_new_cstr(mrb, format); } else { size = p - b; - mrb_ary_push(mrb, ary, mrb_str_new(mrb, b, size)); - return mrb_ary_join(mrb, ary, mrb_str_new(mrb, NULL, 0)); + if (size > 0) { + mrb_ary_push(mrb, ary, mrb_str_new(mrb, b, size)); + mrb_gc_arena_restore(mrb, ai); + } + return mrb_ary_join(mrb, ary, mrb_nil_value()); } } -- cgit v1.2.3