From 96818620121454d42b3d20c6f3440dcc82edf6a8 Mon Sep 17 00:00:00 2001
From: Benjamin Drung <benjamin.drung@profitbricks.com>
Date: Wed, 13 Dec 2017 18:40:37 +0100
Subject: [PATCH] Support live-{top,premount,bottom} hooks

initramfs-tools support hooks for top, premount, bottom when booting
with boot=local or boot=nfs. Add similar support to live-boot when
booting with boot=live.

Closes: #884355
Signed-off-by: Benjamin Drung <benjamin.drung@profitbricks.com>
---
 backend/initramfs-tools/live.script | 56 +++++++++++++++++++++++++++++++++----
 1 file changed, 51 insertions(+), 5 deletions(-)

diff --git a/backend/initramfs-tools/live.script b/backend/initramfs-tools/live.script
index ff2915a..5bec741 100755
--- a/backend/initramfs-tools/live.script
+++ b/backend/initramfs-tools/live.script
@@ -1,13 +1,59 @@
-#!/bin/sh
-
-#set -e
+# Live system filesystem mounting			-*- shell-script -*-
 
 . /bin/live-boot
 
-. /scripts/functions
+live_top()
+{
+	if [ "${live_top_used}" != "yes" ]; then
+		[ "$quiet" != "y" ] && log_begin_msg "Running /scripts/live-top"
+		run_scripts /scripts/live-top
+		[ "$quiet" != "y" ] && log_end_msg
+	fi
+	live_top_used=yes
+}
+
+live_premount()
+{
+	if [ "${live_premount_used}" != "yes" ]; then
+		[ "$quiet" != "y" ] && log_begin_msg "Running /scripts/live-premount"
+		run_scripts /scripts/live-premount
+		[ "$quiet" != "y" ] && log_end_msg
+	fi
+	live_premount_used=yes
+}
+
+live_bottom()
+{
+	if [ "${live_premount_used}" = "yes" ] || [ "${live_top_used}" = "yes" ]; then
+		[ "$quiet" != "y" ] && log_begin_msg "Running /scripts/live-bottom"
+		run_scripts /scripts/live-bottom
+		[ "$quiet" != "y" ] && log_end_msg
+	fi
+	live_premount_used=no
+	live_top_used=no
+}
+
 
-mountroot ()
+mountroot()
 {
 	# initramfs-tools entry point for live-boot is mountroot(); function
 	Live
 }
+
+mount_top()
+{
+	# Note, also called directly in case it's overridden.
+	live_top
+}
+
+mount_premount()
+{
+	# Note, also called directly in case it's overridden.
+	live_premount
+}
+
+mount_bottom()
+{
+	# Note, also called directly in case it's overridden.
+	live_bottom
+}
-- 
2.1.4