Fix DownloadProvider reference counting and instance cleanup.

1 files changed, 7 insertions, 0 deletions

diff --git a/downloadprovider.cpp b/downloadprovider.cpp
index 0e3483df..436e2516 100644
--- a/downloadprovider.cpp
+++ b/downloadprovider.cpp
@@ -10,6 +10,7 @@ DownloadInstance::DownloadInstance(DownloadProvider* provider)
 	cb.context = this;
 	cb.destroyInstance = DestroyInstanceCallback;
 	cb.performRequest = PerformRequestCallback;
+	AddRefForRegistration();
 	m_object = BNInitDownloadInstance(provider->GetObject(), &cb);
 }
 
@@ -20,6 +21,12 @@ DownloadInstance::DownloadInstance(BNDownloadInstance* instance)
 }
 
 
+DownloadInstance::~DownloadInstance()
+{
+	BNFreeDownloadInstance(m_object);
+}
+
+
 void DownloadInstance::DestroyInstanceCallback(void* ctxt)
 {
 	DownloadInstance* instance = (DownloadInstance*)ctxt;