From c2247a35e97c96eba9cd4caf09b32b038625cc31 Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty.wagner@gmail.com>
Date: Tue, 31 Jan 2023 21:45:23 -0500
Subject: Clean up possible early deletion issues in API callbacks

---
 datarenderer.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'datarenderer.cpp')

diff --git a/datarenderer.cpp b/datarenderer.cpp
index b6dd6ec9..8d1b16b4 100644
--- a/datarenderer.cpp
+++ b/datarenderer.cpp
@@ -39,7 +39,7 @@ bool DataRenderer::IsStructOfTypeName(Type* type, const string& name, vector<pai
 bool DataRenderer::IsValidForDataCallback(
     void* ctxt, BNBinaryView* view, uint64_t addr, BNType* type, BNTypeContext* typeCtx, size_t ctxCount)
 {
-	DataRenderer* renderer = (DataRenderer*)ctxt;
+	CallbackRef<DataRenderer> renderer(ctxt);
 	Ref<BinaryView> viewObj = new BinaryView(BNNewViewReference(view));
 	Ref<Type> typeObj = new Type(BNNewTypeReference(type));
 	vector<pair<Type*, size_t>> context;
@@ -55,7 +55,7 @@ BNDisassemblyTextLine* DataRenderer::GetLinesForDataCallback(void* ctxt, BNBinar
     BNType* type, const BNInstructionTextToken* prefix, size_t prefixCount, size_t width, size_t* count,
     BNTypeContext* typeCtx, size_t ctxCount)
 {
-	DataRenderer* renderer = (DataRenderer*)ctxt;
+	CallbackRef<DataRenderer> renderer(ctxt);
 	Ref<BinaryView> viewObj = new BinaryView(BNNewViewReference(view));
 	Ref<Type> typeObj = new Type(BNNewTypeReference(type));
 	vector<InstructionTextToken> prefixes = InstructionTextToken::ConvertInstructionTextTokenList(prefix, prefixCount);
-- 
cgit v1.3.1