From eba8393e91f7b6965ac29b170c0356812871bb99 Mon Sep 17 00:00:00 2001
From: Murray Stokely <murray@FreeBSD.org>
Date: Sat, 22 Sep 2001 18:10:56 +0000
Subject: Add a function to free all of sysinstall's internal variables from
 the environment.  This fixes an annoying bug where hitting Ctrl-C and telling
 sysinstall to 'restart' will do no such thing since many of the options are
 still set and so you won't be prompted for them again.

MFC after:	1 week
---
 usr.sbin/sysinstall/sysinstall.h |  1 +
 usr.sbin/sysinstall/system.c     |  5 ++++-
 usr.sbin/sysinstall/variable.c   | 34 ++++++++++++++++++++++++++++++++--
 3 files changed, 37 insertions(+), 3 deletions(-)

(limited to 'usr.sbin/sysinstall')

diff --git a/usr.sbin/sysinstall/sysinstall.h b/usr.sbin/sysinstall/sysinstall.h
index b473ab6eb192..2b1a06dd5c9e 100644
--- a/usr.sbin/sysinstall/sysinstall.h
+++ b/usr.sbin/sysinstall/sysinstall.h
@@ -771,6 +771,7 @@ extern void	variable_unset(char *var);
 extern char	*variable_get_value(char *var, char *prompt, int dirty);
 extern int 	variable_check(char *data);
 extern int	dump_variables(dialogMenuItem *self);
+extern void	free_variables(void);
 
 /* wizard.c */
 extern void	slice_wizard(Disk *d);
diff --git a/usr.sbin/sysinstall/system.c b/usr.sbin/sysinstall/system.c
index 077cf82bc9c7..0a1a040c3fcc 100644
--- a/usr.sbin/sysinstall/system.c
+++ b/usr.sbin/sysinstall/system.c
@@ -56,7 +56,10 @@ intr_reboot(dialogMenuItem *self)
 static int
 intr_restart(dialogMenuItem *self)
 {
-    execl(StartName, StartName, (char *)NULL);
+    int ret;
+    free_variables();
+    ret = execl(StartName, StartName, (char *)NULL);
+    msgDebug("execl failed (%s)\n", strerror(errno));
     /* NOTREACHED */
     return -1;
 }
diff --git a/usr.sbin/sysinstall/variable.c b/usr.sbin/sysinstall/variable.c
index ee5627958853..2def090ac272 100644
--- a/usr.sbin/sysinstall/variable.c
+++ b/usr.sbin/sysinstall/variable.c
@@ -171,8 +171,8 @@ variable_get_value(char *var, char *prompt, int dirty)
     return cp;
 }
 
-/* Check if value passed in data (in the form "variable=value") is equal to value of 
-   variable stored in env */
+/* Check if value passed in data (in the form "variable=value") is
+   equal to value of variable stored in env */
 int
 variable_check(char *data)
 {
@@ -227,3 +227,33 @@ dump_variables(dialogMenuItem *unused)
 
     return DITEM_SUCCESS;
 }
+
+/* Free all of the variables, useful to really start over as when the
+   user selects "restart" from the interrupt menu. */
+void
+free_variables(void)
+{
+    Variable *vp;
+
+    /* Free the variables from our list, if we have one.. */
+    if (!VarHead)
+	return;
+    else if (!VarHead->next) {
+	unsetenv(VarHead->name);
+	safe_free(VarHead->name);
+	safe_free(VarHead->value);
+	free(VarHead);
+	VarHead = NULL;
+    }
+    else {
+	for (vp = VarHead; vp; ) {
+	    Variable *save = vp;
+	    unsetenv(vp->name);
+	    safe_free(vp->name);
+	    safe_free(vp->value);
+	    vp = vp->next;
+	    safe_free(save);
+	}
+	VarHead = NULL;
+    }
+}
-- 
cgit v1.3