Add a hook to allow BinaryView subclasses to run code after snapshot data is applied

Snapshot data is applied when loading from a database, rebasing the
view, etc.

1 files changed, 8 insertions, 0 deletions

diff --git a/binaryview.cpp b/binaryview.cpp
index 85a7cfbe..c1886ab9 100644
--- a/binaryview.cpp
+++ b/binaryview.cpp
@@ -1365,6 +1365,7 @@ BinaryView::BinaryView(const std::string& typeName, FileMetadata* file, BinaryVi
 	view.isRelocatable = IsRelocatableCallback;
 	view.getAddressSize = GetAddressSizeCallback;
 	view.save = SaveCallback;
+	view.onAfterSnapshotDataApplied = OnAfterSnapshotDataAppliedCallback;
 	m_file = file;
 	AddRefForRegistration();
 	m_object = BNCreateCustomBinaryView(
@@ -1388,6 +1389,13 @@ bool BinaryView::InitCallback(void* ctxt)
 }
 
 
+void BinaryView::OnAfterSnapshotDataAppliedCallback(void* ctxt)
+{
+	CallbackRef<BinaryView> view(ctxt);
+	view->OnAfterSnapshotDataApplied();
+}
+
+
 void BinaryView::FreeCallback(void* ctxt)
 {
 	BinaryView* view = (BinaryView*)ctxt;