From patchwork Fri Sep  5 19:36:19 2014
Content-Type: text/plain; charset="utf-8"
MIME-Version: 1.0
Content-Transfer-Encoding: 7bit
Subject: [1,of,3] dirstate: add begin/endparentchange to dirstate
From: Durham Goode <durham@fb.com>
X-Patchwork-Id: 5705
Message-Id: <8c6d502c92c8ed8dfa79.1409945779@dev2000.prn2.facebook.com>
To: <mercurial-devel@selenic.com>
Date: Fri, 5 Sep 2014 12:36:19 -0700

# HG changeset patch
# User Durham Goode <durham@fb.com>
# Date 1409942069 25200
#      Fri Sep 05 11:34:29 2014 -0700
# Node ID 8c6d502c92c8ed8dfa79b8fac4c8a2cc9e6c4153
# Parent  188b8aa2120b03eead618ba150319074f4e3b42b
dirstate: add begin/endparentchange to dirstate

It's possible for the dirstate to become incoherent (issue4353) if there is an
exception in the middle of the dirstate parent and entries being written (like
if the user ctrl+c's). This change adds begin/endparentchange which a future
patch will require to be set before changing the dirstate parent.  This will
allow us to prevent writing the dirstate in the event of an exception while
changing the parent.

diff --git a/mercurial/dirstate.py b/mercurial/dirstate.py
--- a/mercurial/dirstate.py
+++ b/mercurial/dirstate.py
@@ -44,6 +44,17 @@
         self._lastnormaltime = 0
         self._ui = ui
         self._filecache = {}
+        self._parentwriters = 0
+
+    def beginparentchange(self):
+        self._parentwriters += 1
+
+    def endparentchange(self):
+        if self._parentwriters > 0:
+            self._parentwriters -= 1
+
+    def pendingparentchange(self):
+        return self._parentwriters > 0
 
     @propertycache
     def _map(self):
@@ -300,6 +311,7 @@
                 delattr(self, a)
         self._lastnormaltime = 0
         self._dirty = False
+        self._parentwriters = 0
 
     def copy(self, source, dest):
         """Mark dest as a copy of source. Unmark dest if source is None."""
diff --git a/mercurial/localrepo.py b/mercurial/localrepo.py
--- a/mercurial/localrepo.py
+++ b/mercurial/localrepo.py
@@ -1108,7 +1108,11 @@
             return l
 
         def unlock():
-            self.dirstate.write()
+            if self.dirstate.pendingparentchange():
+                self.dirstate.invalidate()
+            else:
+                self.dirstate.write()
+
             self._filecache['dirstate'].refresh()
 
         l = self._lock(self.vfs, "wlock", wait, unlock,