From 657bc3ff2d000508bd4b4468e83ce37591655e17 Mon Sep 17 00:00:00 2001 From: Rusty Wagner Date: Thu, 2 Aug 2018 22:46:23 -0400 Subject: Fix crash on flow graphs with no associated basic block --- flowgraphnode.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'flowgraphnode.cpp') diff --git a/flowgraphnode.cpp b/flowgraphnode.cpp index a49cb0d0..eea033a8 100644 --- a/flowgraphnode.cpp +++ b/flowgraphnode.cpp @@ -42,7 +42,10 @@ FlowGraphNode::FlowGraphNode(BNFlowGraphNode* node) Ref FlowGraphNode::GetBasicBlock() const { - return new BasicBlock(BNGetFlowGraphBasicBlock(m_object)); + BNBasicBlock* block = BNGetFlowGraphBasicBlock(m_object); + if (!block) + return nullptr; + return new BasicBlock(block); } -- cgit v1.3.1