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

---
 transform.cpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'transform.cpp')

diff --git a/transform.cpp b/transform.cpp
index d9548689..fb5c568b 100644
--- a/transform.cpp
+++ b/transform.cpp
@@ -39,7 +39,7 @@ Transform::Transform(BNTransformType type, const string& name, const string& lon
 
 BNTransformParameterInfo* Transform::GetParametersCallback(void* ctxt, size_t* count)
 {
-	Transform* xform = (Transform*)ctxt;
+	CallbackRef<Transform> xform(ctxt);
 	vector<TransformParameter> params = xform->GetParameters();
 	*count = params.size();
 	BNTransformParameterInfo* result = new BNTransformParameterInfo[params.size()];
@@ -76,7 +76,7 @@ bool Transform::DecodeCallback(
 	DataBuffer inputBuf(BNDuplicateDataBuffer(input));
 	DataBuffer outputBuf;
 
-	Transform* xform = (Transform*)ctxt;
+	CallbackRef<Transform> xform(ctxt);
 	bool result = xform->Decode(inputBuf, outputBuf, paramMap);
 	BNAssignDataBuffer(output, outputBuf.GetBufferObject());
 	return result;
@@ -93,7 +93,7 @@ bool Transform::EncodeCallback(
 	DataBuffer inputBuf(BNDuplicateDataBuffer(input));
 	DataBuffer outputBuf;
 
-	Transform* xform = (Transform*)ctxt;
+	CallbackRef<Transform> xform(ctxt);
 	bool result = xform->Encode(inputBuf, outputBuf, paramMap);
 	BNAssignDataBuffer(output, outputBuf.GetBufferObject());
 	return result;
-- 
cgit v1.3.1