snapshotctl: Initialize snapshot pointer when reverting snapshots

Test: snapshotctl revert-snapshots
Bug: 368487904
Change-Id: I9743bf412d4bfcf508aeb046ec329568da4e2be1
Signed-off-by: Akilesh Kailash <akailash@google.com>
This commit is contained in:
Akilesh Kailash 2024-10-08 20:53:06 -07:00
parent 5684f3e059
commit e69b2d4277

View file

@ -105,7 +105,7 @@ class MapSnapshots {
bool FinishSnapshotWrites();
bool UnmapCowImagePath(std::string& name);
bool DeleteSnapshots();
bool CleanupSnapshot() { return sm_->PrepareDeviceToBootWithoutSnapshot(); }
bool CleanupSnapshot();
bool BeginUpdate();
bool ApplyUpdate();
@ -495,6 +495,11 @@ bool MapSnapshots::UnmapCowImagePath(std::string& name) {
return sm_->UnmapCowImage(name);
}
bool MapSnapshots::CleanupSnapshot() {
sm_ = SnapshotManager::New();
return sm_->PrepareDeviceToBootWithoutSnapshot();
}
bool MapSnapshots::DeleteSnapshots() {
sm_ = SnapshotManager::New();
lock_ = sm_->LockExclusive();