fast-import.c: don't try to commit marks file if write failed

diff --git a/fast-import.c b/fast-import.c
index 3609c24288dedb5b5e1b46d77ebfce6c305a434e..4cf50929765f56f339cd3479312c80c747a78eba 100644 (file)
--- a/fast-import.c
+++ b/fast-import.c
@@ -1545,19 +1545,27 @@ static void dump_marks(void)
                return;
        }
 
-       dump_marks_helper(f, 0, marks);
-       if (ferror(f) || fclose(f))
-               failure |= error("Unable to write marks file %s: %s",
-                       mark_file, strerror(errno));
        /*
-        * Since the lock file was fdopen()'ed and then fclose()'ed above,
-        * assign -1 to the lock file descriptor so that commit_lock_file()
+        * Since the lock file was fdopen()'ed, it should not be close()'ed.
+        * Assign -1 to the lock file descriptor so that commit_lock_file()
         * won't try to close() it.
         */
        mark_lock.fd = -1;
-       if (commit_lock_file(&mark_lock))
-               failure |= error("Unable to write commit file %s: %s",
+
+       dump_marks_helper(f, 0, marks);
+       if (ferror(f) || fclose(f)) {
+               rollback_lock_file(&mark_lock);
+               failure |= error("Unable to write marks file %s: %s",
                        mark_file, strerror(errno));
+               return;
+       }
+
+       if (commit_lock_file(&mark_lock)) {
+               rollback_lock_file(&mark_lock);
+               failure |= error("Unable to commit marks file %s: %s",
+                       mark_file, strerror(errno));
+               return;
+       }
 }
 
 static int read_next_command(void)