From 309fc943fb48d9ce4f6f7c59f7d66bf0ca69eef1 Mon Sep 17 00:00:00 2001
From: Josh F <josh@vector35.com>
Date: Thu, 6 Oct 2022 18:52:48 -0400
Subject: Add APIs for checking and writing snapshot data

---
 database.cpp | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'database.cpp')

diff --git a/database.cpp b/database.cpp
index ba20e02b..e60e0d92 100644
--- a/database.cpp
+++ b/database.cpp
@@ -390,6 +390,19 @@ int64_t Database::WriteSnapshotData(std::vector<int64_t> parents, Ref<BinaryView
 }
 
 
+bool Database::StoreDataForSnapshot(int64_t id, const Ref<KeyValueStore>& data, const std::function<bool(size_t, size_t)>& progress)
+{
+	ProgressContext pctxt;
+	pctxt.callback = progress;
+	bool result = BNStoreDataForSnapshot(m_object, id, data->GetObject(), &pctxt, ProgressCallback);
+	if (!result)
+	{
+		throw DatabaseException("BNStoreDataForSnapshot");
+	}
+	return result;
+}
+
+
 void Database::TrimSnapshot(int64_t id)
 {
 	if (!BNTrimDatabaseSnapshot(m_object, id))
@@ -518,3 +531,9 @@ void Database::WriteAnalysisCache(Ref<KeyValueStore> val)
 		throw DatabaseException("BNWriteDatabaseAnalysisCache");
 	}
 }
+
+
+bool Database::SnapshotHasData(int64_t id)
+{
+	return BNSnapshotHasData(m_object, id);
+}
-- 
cgit v1.3.1