Add examples for UEFI hook.

Document task package usage in the README

3 files changed, 20 insertions, 0 deletions

diff --git a/vmdebootstrap/base.py b/vmdebootstrap/base.py
index e840855..be51ce6 100644
--- a/vmdebootstrap/base.py
+++ b/vmdebootstrap/base.py
@@ -165,3 +165,16 @@ class Base(object):
             # to autologin, serial_command can contain '-a root'
             with open(inittab, 'a') as ftab:
                 ftab.write('\nS0:23:respawn:%s\n' % serial_command)
+
+    def check_swap_size(self):
+        # swap - modifies extent
+        extent = '100%'
+        swap = 256 * 1024 * 1024
+        if self.settings['swap'] > 0:
+            if self.settings['swap'] > swap:
+                swap = self.settings['swap']
+            else:
+                # minimum 256Mb as default qemu ram is 128Mb
+                logging.debug("Setting minimum 256Mb swap space")
+            extent = "%s%%" % int(100 * (self.settings['size'] - swap) / self.settings['size'])
+        return extent
diff --git a/vmdebootstrap/filesystem.py b/vmdebootstrap/filesystem.py
index ccc95a4..796f7d8 100644
--- a/vmdebootstrap/filesystem.py
+++ b/vmdebootstrap/filesystem.py
@@ -291,3 +291,8 @@ class Filesystem(Base):
                     fhosts.write('%s\n' % line)
         except IOError:
             pass
+
+    def make_rootfs_part(self, extent):
+        bootsize = self.settings['esp-size'] / (1024 * 1024) + 1
+        runcmd(['parted', '-s', self.settings['image'],
+                'mkpart', 'primary', str(bootsize), extent])
diff --git a/vmdebootstrap/uefi.py b/vmdebootstrap/uefi.py
index 2a8bc2a..7f77f9f 100644
--- a/vmdebootstrap/uefi.py
+++ b/vmdebootstrap/uefi.py
@@ -105,6 +105,8 @@ class Uefi(Base):
                 umount_wrapper(rootdir)
 
     def partition_esp(self):
+        if not self.settings['use-uefi']:
+            return
         espsize = self.settings['esp-size'] / (1024 * 1024)
         self.message("Using ESP size: %smib %s bytes" % (espsize, self.settings['esp-size']))
         runcmd(['parted', '-s', self.settings['image'],