Avoid null dereference in Visitor::findContext for visitor without co…

…ntext found by clang-tidy
p4lang · Nov 3, 2023 · e7859c3 · e7859c3
1 parent bceab95
commit e7859c3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ir/visitor.h b/ir/visitor.h
@@ -245,6 +245,7 @@ class Visitor {
     template <class T>
     inline const T *findContext(const Context *&c) const {
         if (!c) c = ctxt;
+        if (!c) return nullptr;
         while ((c = c->parent))
             if (auto *rv = dynamic_cast<const T *>(c->node)) return rv;
         return nullptr;
@@ -257,6 +258,7 @@ class Visitor {
     template <class T>
     inline const T *findOrigCtxt(const Context *&c) const {
         if (!c) c = ctxt;
+        if (!c) return nullptr;
         while ((c = c->parent))
             if (auto *rv = dynamic_cast<const T *>(c->original)) return rv;
         return nullptr;