remove potential crash when an unfinished tree throws an exception

BehaviorTree · May 2, 2021 · 8d3de63 · 8d3de63
1 parent 33c29e5
commit 8d3de63
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/behavior_tree.cpp b/src/behavior_tree.cpp
@@ -56,7 +56,9 @@ void applyRecursiveVisitor(TreeNode* node, const std::function<void(TreeNode*)>&
     }
     else if (auto decorator = dynamic_cast<BT::DecoratorNode*>(node))
     {
-        applyRecursiveVisitor(decorator->child(), visitor);
+        if( decorator->child() ){
+            applyRecursiveVisitor(decorator->child(), visitor);
+        }
     }
 }
 

diff --git a/src/decorator_node.cpp b/src/decorator_node.cpp
@@ -48,6 +48,9 @@ TreeNode* DecoratorNode::child()
 
 void DecoratorNode::haltChild()
 {
+    if( !child_node_ ){
+        return;
+    }
     if (child_node_->status() == NodeStatus::RUNNING)
     {
         child_node_->halt();