admin/mruby-mruby
1
0
Fork 0
mirror of https://github.com/mruby/mruby synced 2026-06-08 16:11:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add assertion to make sure new capacity does not overflow.

Browse Source
This commit is contained in:
Yukihiro "Matz" Matsumoto
2016-12-13 10:50:04 +09:00
parent 3a7c369536
commit 3bedd22d55
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/string.c
+5 -2
View File
@@ -118,8 +118,8 @@ mrb_str_buf_new(mrb_state *mrb, size_t capa)
return mrb_obj_value(s);
}
static inline void
resize_capa(mrb_state *mrb, struct RString *s, mrb_int capacity)
static void
resize_capa(mrb_state *mrb, struct RString *s, size_t capacity)
{
if (RSTR_EMBED_P(s)) {
if (RSTRING_EMBED_LEN_MAX < capacity) {
@@ -133,6 +133,9 @@ resize_capa(mrb_state *mrb, struct RString *s, mrb_int capacity)
}
}
else {
#if SIZE_MAX > MRB_INT_MAX
mrb_assert(capacity <= MRB_INT_MAX);
#endif
s->as.heap.ptr = (char *)mrb_realloc(mrb, RSTR_PTR(s), capacity+1);
s->as.heap.aux.capa = capacity;
}