Add missing NULL check in CanBeType

2017-05-02 22:26:21 -04:00
parent b3b02df569
commit 93c563e073
3 changed files with 14 additions and 12 deletions
--- a/type.cpp
+++ b/type.cpp
@@ -855,15 +855,17 @@ PolyType::CanBeType(const Type *t) const {

    const AtomicType *at = CastType<AtomicType>(t);

-    switch (restriction) {
-        case TYPE_INTEGER:
-            return at->IsIntType();
-        case TYPE_FLOATING:
-            return at->IsFloatType();
-        case TYPE_NUMBER:
-            return at->IsIntType() || at->IsFloatType();
-        default:
-            FATAL("Unmatched case for polymorphic restriction");
+    if (at) {
+        switch (restriction) {
+            case TYPE_INTEGER:
+                return at->IsIntType();
+            case TYPE_FLOATING:
+                return at->IsFloatType();
+            case TYPE_NUMBER:
+                return at->IsIntType() || at->IsFloatType();
+            default:
+                FATAL("Unmatched case for polymorphic restriction");
+        }
    }

    // not an atomic type or polymorphic type