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

---
 websocketprovider.cpp | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'websocketprovider.cpp')

diff --git a/websocketprovider.cpp b/websocketprovider.cpp
index 48782715..52b8e721 100644
--- a/websocketprovider.cpp
+++ b/websocketprovider.cpp
@@ -59,28 +59,28 @@ bool WebsocketClient::ConnectCallback(
 		headers[headerKeys[i]] = headerValues[i];
 	}
 
-	WebsocketClient* client = (WebsocketClient*)ctxt;
+	CallbackRef<WebsocketClient> client(ctxt);
 	return client->Connect(host, headers);
 }
 
 
 bool WebsocketClient::DisconnectCallback(void* ctxt)
 {
-	WebsocketClient* client = (WebsocketClient*)ctxt;
+	CallbackRef<WebsocketClient> client(ctxt);
 	return client->Disconnect();
 }
 
 
 void WebsocketClient::ErrorCallback(const char* msg, void* ctxt)
 {
-	WebsocketClient* client = (WebsocketClient*)ctxt;
+	CallbackRef<WebsocketClient> client(ctxt);
 	BNNotifyWebsocketClientError(client->m_object, msg);
 }
 
 
 bool WebsocketClient::WriteCallback(const uint8_t* data, uint64_t len, void* ctxt)
 {
-	WebsocketClient* client = (WebsocketClient*)ctxt;
+	CallbackRef<WebsocketClient> client(ctxt);
 	return client->Write(vector<uint8_t>(data, data + len));
 }
 
-- 
cgit v1.3.1