Merge pull request #1003 from kano4/pr-add-null-check

Add null check
This commit is contained in:
Yukihiro "Matz" Matsumoto
2013-03-13 20:30:32 -07:00
+5 -1
View File
@@ -53,6 +53,7 @@ mrb_alloca(mrb_state *mrb, size_t size)
struct alloca_header *p;
p = (struct alloca_header*) mrb_malloc(mrb, sizeof(struct alloca_header)+size);
if (p == NULL) return NULL;
p->next = mrb->mems;
mrb->mems = p;
return (void*)p->buf;
@@ -61,9 +62,12 @@ mrb_alloca(mrb_state *mrb, size_t size)
static void
mrb_alloca_free(mrb_state *mrb)
{
struct alloca_header *p = mrb->mems;
struct alloca_header *p;
struct alloca_header *tmp;
if (mrb == NULL) return NULL;
p = mrb->mems;
while (p) {
tmp = p;
p = p->next;