Coalesce on remote WIP

Signed-off-by: Damien Thenot <[email protected]>

Author: Nambrok

drivers/blktap2.py

@@ -64,6 +64,7 @@
     LINSTOR_AVAILABLE = False
 
 PLUGIN_TAP_PAUSE = "tapdisk-pause"
+PLUGIN_ON_SLAVE = "on-slave"
 
 SOCKPATH = "/var/xapi/xcp-rrdd"
 
@@ -461,6 +462,13 @@ def query(cls, pid, minor):
         total_coalesce = int(m.group(3))
         return (status, coalesced, total_coalesce)
 
+    @classmethod
+    def cancel_commit(cls, pid, minor, wait=True):
+        args = ["cancel", "-p", pid, "-m", minor]
+        if wait:
+            args.append("-w")
+        cls._pread(args)
+
 class TapdiskExists(Exception):
     """Tapdisk already running."""
 
@@ -1658,6 +1666,8 @@ def _activate_locked(self, sr_uuid, vdi_uuid, options):
         if self.tap_wanted():
             if not self._add_tag(vdi_uuid, not options["rdonly"]):
                 return False
+                #TODO: Need to interrupt coalesce on master, the coalesce will check for host_OpaqueRef on the VDI before trying offline coalesce
+                #TODO: The coalesce could happen on another slave in onlinecoalesce, interrupt coalesce on another slave (online coalesce)?
             refresh = True
 
         try:

drivers/cleanup.py

@@ -757,7 +757,7 @@ def getTreeHeight(self):
 
         return maxChildHeight + 1
 
-    def getAllLeaves(self):
+    def getAllLeaves(self) -> List["VDI"]:
         "Get all leaf nodes in the subtree rooted at self"
         if len(self.children) == 0:
             return [self]
@@ -828,6 +828,57 @@ def _clear(self):
     def _clearRef(self):
         self._vdiRef = None
 
+    def _call_plug_cancel(self, hostRef):
+        args = {"path": self.path, "vdi_type": self.vdi_type}
+        self.sr.xapi.session.xenapi.host.call_plugin( \
+                    hostRef, XAPI.PLUGIN_ON_SLAVE, "commit_cancel", args)
+
+    def _call_plugin_coalesce(self, hostRef):
+        args = {"path": self.path, "vdi_type": self.vdi_type}
+        self.sr.xapi.session.xenapi.host.call_plugin( \
+                    hostRef, XAPI.PLUGIN_ON_SLAVE, "commit_tapdisk", args)
+
+    def _doCoalesceOnHost(self, hostRef):
+        self.validate()
+        self.parent.validate(True)
+        self.parent._increaseSizeVirt(self.sizeVirt)
+        self.sr._updateSlavesOnResize(self.parent)
+        #TODO: We might need to make the LV RW on the slave directly for coalesce?
+        # Children and parent need to be RW for QCOW2 coalesce, otherwise tapdisk(libqcow) will crash trying to access them
+
+        def abortTest():
+            file = self.sr._gc_running_file(self)
+            try:
+                with open(file, "r") as f:
+                    if not f.read():
+                        #TODO: Need to call commit cancel on the hostRef if we stop
+                        self._call_plug_cancel(hostRef)
+                        return True
+            except OSError as e:
+                if e.errno == errno.ENOENT:
+                    util.SMlog("File {} does not exist".format(file))
+                else:
+                    util.SMlog("IOError: {}".format(e))
+                return True
+            return False
+
+        Util.runAbortable(lambda: self._call_plugin_coalesce(hostRef),
+                          None, self.sr.uuid, abortTest, VDI.POLL_INTERVAL, 0)
+
+        self.parent.validate(True)
+        #self._verifyContents(0)
+        self.parent.updateBlockInfo()
+
+    def _isOpenOnHosts(self) -> Optional[str]:
+        for pbdRecord in self.sr.xapi.getAttachedPBDs():
+            hostRef = pbdRecord["host"]
+            args = {"path": self.path}
+            is_openers = util.strtobool(self.sr.xapi.session.xenapi.host.call_plugin( \
+                    hostRef, XAPI.PLUGIN_ON_SLAVE, "is_openers", args))
+            if is_openers:
+                return hostRef
+        return None
+
     def _doCoalesce(self) -> None:
         """Coalesce self onto parent. Only perform the actual coalescing of
         an image, but not the subsequent relinking. We'll do that as the next step,
@@ -914,7 +965,7 @@ def coalesce(self) -> int:
         return self.cowutil.coalesce(self.path)
 
     @staticmethod
-    def _doCoalesceCowImage(vdi):
+    def _doCoalesceCowImage(vdi: "VDI"):
         try:
             startTime = time.time()
             allocated_size = vdi.getAllocatedSize()
@@ -943,7 +994,21 @@ def _vdi_is_raw(self, vdi_path):
 
     def _coalesceCowImage(self, timeOut):
         Util.log("  Running COW coalesce on %s" % self)
-        abortTest = lambda: IPCFlag(self.sr.uuid).test(FLAG_TYPE_ABORT)
+        def abortTest():
+            if self.cowutil.isCoalesceableOnRemote():
+                file = self.sr._gc_running_file(self)
+                try:
+                    with open(file, "r") as f:
+                        if not f.read():
+                            return True
+                except OSError as e:
+                    if e.errno == errno.ENOENT:
+                        util.SMlog("File {} does not exist".format(file))
+                    else:
+                        util.SMlog("IOError: {}".format(e))
+                    return True
+            return IPCFlag(self.sr.uuid).test(FLAG_TYPE_ABORT)
+
         try:
             util.fistpoint.activate_custom_fn(
                 "cleanup_coalesceVHD_inject_failure",
@@ -2383,7 +2448,17 @@ def cleanupJournals(self, dryRun=False):
     def cleanupCache(self, maxAge=-1) -> int:
         return 0
 
-    def _coalesce(self, vdi):
+    def _hasLeavesAttachedOn(self, vdi: VDI):
+        leaves = vdi.getAllLeaves()
+        leaves_vdi = [leaf.uuid for leaf in leaves]
+        return util.get_hosts_attached_on(self.xapi.session, leaves_vdi)
+
+    def _gc_running_file(self, vdi):
+        run_file = "gc_running_{}".format(vdi.uuid)
+        return os.path.join(NON_PERSISTENT_DIR, str(self.uuid), run_file)
+
+    def _coalesce(self, vdi: VDI):
+        skipRelink = False
         if self.journaler.get(vdi.JRN_RELINK, vdi.uuid):
             # this means we had done the actual coalescing already and just
             # need to finish relinking and/or refreshing the children
@@ -2393,8 +2468,35 @@ def _coalesce(self, vdi):
             # order to decide whether to abort the coalesce. We remove the
             # journal as soon as the COW coalesce step is done, because we
             # don't expect the rest of the process to take long
+
+            #TODO: Create `gc_running` in `/run/nonpersistent/sm/<sr uuid>/`
+            if os.path.exists(self._gc_running_file(vdi)):
+                util.SMlog("gc_running already exist for {}. Ignoring...".format(self.uuid))
+
+            with open(self._gc_running_file(vdi), "w") as f:
+                f.write("1")
+
             self.journaler.create(vdi.JRN_COALESCE, vdi.uuid, "1")
-            vdi._doCoalesce()
+            host_refs = self._hasLeavesAttachedOn(vdi)
+            #TODO: this check of multiple host_refs should be done earlier in `is_coalesceable` to avoid stopping this late every time
+            if len(host_refs) > 1:
+                util.SMlog("Not coalesceable, chain activated more than once")
+                raise Exception("Not coalesceable, chain activated more than once") #TODO: Use correct error
+
+            try:
+                if host_refs and vdi.cowutil.isCoalesceableOnRemote:
+                    #Leaf opened on another host, we need to call online coalesce
+                    vdi._doCoalesceOnHost(host_refs[0])
+                    skipRelink = True
+                else:
+                    vdi._doCoalesce()
+            except:
+                os.unlink(self._gc_running_file(vdi))
+                raise
+            """
+            vdi._doCoalesce will call vdi._coalesceCowImage (after doing other things).
+            It will then call VDI._doCoalesceCowImage in a runAbortable context
+            """
             self.journaler.remove(vdi.JRN_COALESCE, vdi.uuid)
 
             util.fistpoint.activate("LVHDRT_before_create_relink_journal", self.uuid)
@@ -2403,19 +2505,22 @@ def _coalesce(self, vdi):
             # like SM.clone from manipulating the VDIs we'll be relinking and
             # rescan the SR first in case the children changed since the last
             # scan
-            self.journaler.create(vdi.JRN_RELINK, vdi.uuid, "1")
+            if not skipRelink:
+                self.journaler.create(vdi.JRN_RELINK, vdi.uuid, "1")
 
-        self.lock()
-        try:
-            vdi.parent._tagChildrenForRelink()
-            self.scan()
-            vdi._relinkSkip()
-        finally:
-            self.unlock()
-            # Reload the children to leave things consistent
-            vdi.parent._reloadChildren(vdi)
+        if not skipRelink:
+            self.lock()
+            try:
+                vdi.parent._tagChildrenForRelink()
+                self.scan()
+                vdi._relinkSkip()
+            finally:
+                self.unlock()
+                # Reload the children to leave things consistent
+                vdi.parent._reloadChildren(vdi)
+            self.journaler.remove(vdi.JRN_RELINK, vdi.uuid)
 
-        self.journaler.remove(vdi.JRN_RELINK, vdi.uuid)
+        os.unlink(self._gc_running_file(vdi))
         self.deleteVDI(vdi)
 
     class CoalesceTracker:
@@ -3718,8 +3823,7 @@ def _gc_init_file(sr_uuid):
 
 def _create_init_file(sr_uuid):
     util.makedirs(os.path.join(NON_PERSISTENT_DIR, str(sr_uuid)))
-    with open(os.path.join(
-            NON_PERSISTENT_DIR, str(sr_uuid), 'gc_init'), 'w+') as f:
+    with open(os.path.join(_gc_init_file(sr_uuid)), 'w+') as f:
         f.write('1')
 
 
@@ -3748,7 +3852,7 @@ def abortTest():
         Util.log("GC active, quiet period ended")
 
 
-def _gcLoop(sr, dryRun=False, immediate=False):
+def _gcLoop(sr: SR, dryRun=False, immediate=False):
     if not lockGCActive.acquireNoblock():
         Util.log("Another GC instance already active, exiting")
         return

drivers/cowutil.py

@@ -266,6 +266,18 @@ def getKeyHash(self, path: str) -> Optional[str]:
     def setKey(self, path: str, key_hash: str) -> None:
         pass
 
+    @abstractmethod
+    def isCoalesceableOnRemote(self) -> bool:
+        pass
+
+    @abstractmethod
+    def coalesceOnline(self, path: str) -> int:
+        pass
+
+    @abstractmethod
+    def cancelCoalesceOnline(self, path: str) -> None:
+        pass
+
     def getParentChain(self, lvName: str, extractUuidFunction: Callable[[str], str], vgName: str) -> Dict[str, str]:
         """
         Get the chain of all parents of 'path'. Safe to call for raw VDI's as well.

drivers/on_slave.py

@@ -18,6 +18,9 @@
 # A plugin for synchronizing slaves when something changes on the Master
 
 import sys
+import os
+import time
+import errno
 sys.path.append("/opt/xensource/sm/")
 import util
 import lock
@@ -148,7 +151,6 @@ def is_open(session, args):
         util.logException("is_open")
         raise
 
-
 def refresh_lun_size_by_SCSIid(session, args):
     """Refresh the size of LUNs backing the SCSIid on the local node."""
     util.SMlog("on-slave.refresh_lun_size_by_SCSIid(,%s)" % args)
@@ -160,10 +162,78 @@ def refresh_lun_size_by_SCSIid(session, args):
         util.SMlog("on-slave.refresh_lun_size_by_SCSIid with %s failed" % args)
         return "False"
 
+def commit_tapdisk(session, args):
+    path = args["path"]
+    vdi_type = args["vdi_type"]
+    #TODO: Miss activating/changing RW, naming should reflect that it does more than coalesceing
+    from cowutil import getCowUtil
+    cowutil = getCowUtil(vdi_type)
+    try:
+        return str(cowutil.coalesceOnline(path))
+    except:
+        return "0"
+
+def commit_cancel(session, args):
+    path = args["path"]
+    vdi_type = args["vdi_type"]
+    from cowutil import getCowUtil
+    cowutil = getCowUtil(vdi_type)
+    try:
+        cowutil.cancelCoalesceOnline(path)
+    except:
+        return "False"
+    return "True"
+
+def cancel_coalesce_master(session, args):
+    sr_uuid = args["sr_uuid"]
+    vdi_uuid = args["vdi_uuid"]
+
+    # from ipc import IPCFlag
+    # flag = IPCFlag(sr_uuid)
+
+    # runningStr = "gc_running_{}".format(vdi_uuid)
+    # abortStr = "abort_{}".format(vdi_uuid)
+
+    # if not flag.test(runningStr):
+    #     return "True"
+
+    # if not flag.test(abortStr):
+    #     flag.set(abortStr)
+
+    # while flag.test(abortStr) or flag.test(runningStr):
+    #     time.sleep(1)
+
+    # return "True"
+
+    path = "/run/nonpersistent/sm/{}/gc_running_{}".format(sr_uuid, vdi_uuid)
+
+    try:
+        with open(path, "r+") as f:
+            f.truncate(0)
+            f.flush()
+            os.fsync(f.fileno())
+    except IOError as e:
+        if e.errno == errno.ENOENT:
+            return "True"
+        raise
+
+    while os.path.exists(path):
+        time.sleep(1)
+
+    return "True"
+
+def is_openers(session, args):
+    path = args["path"]
+    openers_pid= util.get_openers_pid(path)
+    return str(bool(openers_pid))
 
 if __name__ == "__main__":
     import XenAPIPlugin
     XenAPIPlugin.dispatch({
         "multi": multi,
         "is_open": is_open,
-        "refresh_lun_size_by_SCSIid": refresh_lun_size_by_SCSIid})
+        "refresh_lun_size_by_SCSIid": refresh_lun_size_by_SCSIid,
+        "is_openers": is_openers,
+        "commit_tapdisk": commit_tapdisk,
+        "commit_cancel": commit_cancel,
+        })