From dffb4d82dc83054eade20679f069979d242d9745 Mon Sep 17 00:00:00 2001
From: "Yukihiro \"Matz\" Matsumoto" <matz@ruby-lang.org>
Date: Tue, 6 Dec 2016 11:40:49 +0900
Subject: Add type check for cls before allocation

---
 src/gc.c | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

(limited to 'src/gc.c')

diff --git a/src/gc.c b/src/gc.c
index 69708e40b..470fffc5f 100644
--- a/src/gc.c
+++ b/src/gc.c
@@ -480,7 +480,18 @@ mrb_obj_alloc(mrb_state *mrb, enum mrb_vtype ttype, struct RClass *cls)
   mrb_gc *gc = &mrb->gc;
 
   if (cls) {
-    enum mrb_vtype tt = MRB_INSTANCE_TT(cls);
+    enum mrb_vtype tt;
+
+    switch (cls->tt) {
+    case MRB_TT_CLASS:
+    case MRB_TT_SCLASS:
+    case MRB_TT_MODULE:
+    case MRB_TT_ENV:
+      break;
+    default:
+      mrb_raise(mrb, E_TYPE_ERROR, "allocation failure");
+    }
+    tt = MRB_INSTANCE_TT(cls);
     if (tt != MRB_TT_FALSE &&
         ttype != MRB_TT_SCLASS &&
         ttype != MRB_TT_ICLASS &&
-- 
cgit v1.2.3