Undo entry states and reverting

1 files changed, 13 insertions, 4 deletions

diff --git a/filemetadata.cpp b/filemetadata.cpp
index 0a12eb00..4f75129e 100644
--- a/filemetadata.cpp
+++ b/filemetadata.cpp
@@ -270,15 +270,24 @@ bool FileMetadata::CreateSnapshotedView(BinaryView* data, const std::string& vie
 }
 
 
-void FileMetadata::BeginUndoActions()
+std::string FileMetadata::BeginUndoActions()
 {
-	BNBeginUndoActions(m_object);
+	char* id = BNBeginUndoActions(m_object);
+	std::string result = id;
+	BNFreeString(id);
+	return result;
+}
+
+
+void FileMetadata::CommitUndoActions(const std::string& id)
+{
+	BNCommitUndoActions(m_object, id.c_str());
 }
 
 
-void FileMetadata::CommitUndoActions()
+void FileMetadata::RevertUndoActions(const std::string& id)
 {
-	BNCommitUndoActions(m_object);
+	BNRevertUndoActions(m_object, id.c_str());
 }