create /tmp/cleanup when necessary

1 files changed, 12 insertions, 1 deletions

diff --git a/rescue.py b/rescue.py
index c4c7d803f..17188ea34 100644
--- a/rescue.py
+++ b/rescue.py
@@ -5,6 +5,7 @@ from translate import _
 import raid
 import os
 from log import log
+import isys
 
 class RescueInterface:
 
@@ -21,7 +22,7 @@ class RescueInterface:
     def __init__(self, screen):
 	self.screen = screen
 
-def runRescue(serial):
+def runRescue(url, serial):
 
     from fstab import NewtFstab
 
@@ -43,6 +44,16 @@ def runRescue(serial):
     screen = SnackScreen()
     intf = RescueInterface(screen)
 
+    # go ahead and set things up for reboot
+    if url[0:6] == "cdrom:" or url[0:4] == "nfs:":
+	f = open("/tmp/cleanup", "w")
+
+	isys.makeDevInode("loop0", "/tmp/loop0")
+	f.write("umount /mnt/runtime\n")
+	f.write("lounsetup /tmp/loop0\n")
+
+	f.close()
+
     parts = upgrade.findExistingRoots(intf, fstab)
 
     if not parts: