The existing heuristic we use in virt-install to detect whether
an install has completed successfully is dependant on the type
of installer being used - i.e. if you create a VM with a livecd
or a raw ext3 image, there may be no disk with an MBR at the end.

This patch moves that logic into a new DistroInstaller method.

Signed-off-by: Mark McLoughlin <markmc@redhat.com>

Index: virtinst--devel/virt-install
===================================================================
--- virtinst--devel.orig/virt-install
+++ virtinst--devel/virt-install
@@ -17,7 +17,6 @@
 import os, sys, string
 from optparse import OptionParser, OptionValueError
 import subprocess
-import struct
 import logging
 import libxml2
 import urlgrabber.progress as progress
@@ -582,11 +581,7 @@ def main():
     # the domain is no longer running
     # FIXME: this is just a hacky heuristic, but I'll take what I can get
     try:
-        fd = os.open(guest.disks[0].path, os.O_RDONLY)
-        buf = os.read(fd, 512)
-        os.close(fd)
-        if len(buf) == 512 and \
-               struct.unpack("H", buf[0x1fe: 0x200]) == (0xaa55,):
+        if guest.post_install_check():
             # things installed enough that we should be able to restart
             # the domain
             if continue_inst:
Index: virtinst--devel/virtinst/DistroManager.py
===================================================================
--- virtinst--devel.orig/virtinst/DistroManager.py
+++ virtinst--devel/virtinst/DistroManager.py
@@ -18,6 +18,7 @@ import os
 import gzip
 import re
 import stat
+import struct
 import subprocess
 import urlgrabber.grabber as grabber
 import urlgrabber.progress as progress
@@ -698,3 +699,10 @@ class DistroInstaller(Guest.Installer):
             osblob += "    <bootloader>/usr/bin/pygrub</bootloader>"
 
         return osblob
+
+    def post_install_check(self, guest):
+        # Check for the 0xaa55 signature at the end of the MBR
+        fd = os.open(guest.disks[0].path, os.O_RDONLY)
+        buf = os.read(fd, 512)
+        os.close(fd)
+        return len(buf) == 512 and struct.unpack("H", buf[0x1fe: 0x200]) == (0xaa55,)
Index: virtinst--devel/virtinst/Guest.py
===================================================================
--- virtinst--devel.orig/virtinst/Guest.py
+++ virtinst--devel/virtinst/Guest.py
@@ -727,6 +727,9 @@ class Guest(object):
         # for inactive guest, or get the still running install..
         return self.conn.lookupByName(self.name)
 
+    def post_install_check(self):
+        return self.installer.post_install_check(self)
+
     def connect_console(self, consolecb):
         logging.debug("Restarted guest, looking to see if it is running")
         # sleep in .25 second increments until either a) we get running

--