Signed-off-by: Jonathon Jongsma <[email protected]>
---
 src/hyperv/hyperv_driver.c | 48 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/src/hyperv/hyperv_driver.c b/src/hyperv/hyperv_driver.c
index 4a4db0e76b..dd1248ced1 100644
--- a/src/hyperv/hyperv_driver.c
+++ b/src/hyperv/hyperv_driver.c
@@ -4235,6 +4235,52 @@ hypervParseInstanceIdFromParentPath(const char *obj_path)
 }
 
 
+static int
+hypervDomainHasCurrentSnapshot(virDomainPtr domain,
+                               unsigned int flags)
+{
+    g_autoptr(Msvm_VirtualSystemSettingData) vssd = NULL;
+    hypervPrivate *priv = domain->conn->privateData;
+    char uuid_string[VIR_UUID_STRING_BUFLEN];
+
+    virCheckFlags(0, -1);
+
+    virUUIDFormat(domain->uuid, uuid_string);
+
+    if (hypervGetMsvmVirtualSystemSettingDataFromUUID(priv, uuid_string, 
&vssd) < 0)
+        return -1;
+
+    return (vssd->data->Parent != NULL);
+}
+
+
+static virDomainSnapshotPtr
+hypervDomainSnapshotCurrent(virDomainPtr domain,
+                            unsigned int flags)
+{
+    hypervPrivate *priv = domain->conn->privateData;
+    char uuid_string[VIR_UUID_STRING_BUFLEN];
+    g_autoptr(Msvm_VirtualSystemSettingData) vssd = NULL;
+    g_autofree char* current_snapshot = NULL;
+
+    virCheckFlags(0, NULL);
+
+    virUUIDFormat(domain->uuid, uuid_string);
+
+    if (hypervGetMsvmVirtualSystemSettingDataFromUUID(priv, uuid_string, 
&vssd) < 0)
+        return NULL;
+
+    current_snapshot = hypervParseInstanceIdFromParentPath(vssd->data->Parent);
+    if (!current_snapshot) {
+        virReportError(VIR_ERR_NO_DOMAIN_SNAPSHOT, "%s",
+                       _("the domain does not have a current snapshot"));
+        return NULL;
+    }
+
+    return virGetDomainSnapshot(domain, current_snapshot);
+}
+
+
 static char *
 hypervDomainSnapshotGetXMLDesc(virDomainSnapshotPtr snapshot,
                                unsigned int flags)
@@ -4345,6 +4391,8 @@ static virHypervisorDriver hypervHypervisorDriver = {
     .domainListAllSnapshots = hypervDomainListAllSnapshots, /* 12.2.0 */
     .domainSnapshotNum = hypervDomainSnapshotNum, /* 12.2.0 */
     .domainSnapshotGetXMLDesc = hypervDomainSnapshotGetXMLDesc, /* 12.2.0 */
+    .domainHasCurrentSnapshot = hypervDomainHasCurrentSnapshot, /* 12.2.0 */
+    .domainSnapshotCurrent = hypervDomainSnapshotCurrent, /* 12.2.0 */
 };
 
 
-- 
2.53.0

Reply via email to