Add a check for unsigned integer wrapping.

This commit is contained in:
Masaki Muranaka
2012-07-07 12:38:07 +09:00
parent 47407768fd
commit a064038d85
+7 -5
View File
@@ -167,14 +167,16 @@ mrb_malloc(mrb_state *mrb, size_t len)
void*
mrb_calloc(mrb_state *mrb, size_t nelem, size_t len)
{
void *p;
void *p = NULL;
size_t size;
size = nelem * len;
p = mrb_realloc(mrb, 0, size);
if (nelem <= SIZE_MAX / len) {
size = nelem * len;
p = mrb_realloc(mrb, 0, size);
if (p && size > 0)
memset(p, 0, size);
if (p && size > 0)
memset(p, 0, size);
}
return p;
}