Loop umount'ing rootfs until there are no more mounts

This is simpler than trying to count how many successful mounts we made.

Signed-off-by: Ian Campbell <ian.campbell@docker.com>

mount/mount.go

@@ -22,19 +22,3 @@ func MountAll(mounts []Mount, target string) error {
 	}
 	return nil
 }
-
-// UnmountN tries to unmount the given mount point nr times, which is
-// useful for undoing a stack of mounts on the same mount
-// point. Returns the first error encountered, but always attempts the
-// full nr umounts.
-func UnmountN(mount string, flags, nr int) error {
-	var err error
-	for i := 0; i < nr; i++ {
-		if err2 := Unmount(mount, flags); err2 != nil {
-			if err == nil {
-				err = err2
-			}
-		}
-	}
-	return err
-}

mount/mount_linux.go

@@ -15,6 +15,25 @@ func Unmount(mount string, flags int) error {
 	return unix.Unmount(mount, flags)
 }
 
+// UnmountAll repeatedly unmounts the given mount point until there
+// are no mounts remaining (EINVAL is returned by mount), which is
+// useful for undoing a stack of mounts on the same mount point.
+func UnmountAll(mount string, flags int) error {
+	for {
+		if err := Unmount(mount, flags); err != nil {
+			// EINVAL is returned if the target is not a
+			// mount point, indicating that we are
+			// done. It can also indicate a few other
+			// things (such as invalid flags) which we
+			// unfortunately end up squelching here too.
+			if err == unix.EINVAL {
+				return nil
+			}
+			return err
+		}
+	}
+}
+
 // parseMountOptions takes fstab style mount options and parses them for
 // use with a standard mount() syscall
 func parseMountOptions(options []string) (int, string) {

mount/mount_unix.go

@@ -15,3 +15,7 @@ func (m *Mount) Mount(target string) error {
 func Unmount(mount string, flags int) error {
 	return ErrNotImplementOnUnix
 }
+
+func UnmountAll(mount string, flags int) error {
+	return ErrNotImplementOnUnix
+}

mount/mount_windows.go

@@ -13,3 +13,7 @@ func (m *Mount) Mount(target string) error {
 func Unmount(mount string, flags int) error {
 	return ErrNotImplementOnWindows
 }
+
+func UnmountAll(mount string, flags int) error {
+	return ErrNotImplementOnWindows
+}