Use SIGTERM to kill iurt and install a handler

1 files changed, 11 insertions, 6 deletions

diff --git a/iurt b/iurt
index 7dd57f8..0a24ded 100755
--- a/iurt
+++ b/iurt
@@ -71,13 +71,10 @@ sub bug_handler {
     #exit(1);
 }
 
+my ($chroot_name, $chroot_tmp, $chroot, $chroot_ref);
+
 $SIG{SEGV} = sub { bug_handler(@_, 1) };
 #$SIG{__DIE__} = \&bug_handler;
-$SIG{TERM} = sub {
-    warn "Got KILLED by SIGTERM at " . strftime("%c", localtime()) . " .\n";
-    exit(1);
-};
-
 my $program_name = 'iurt';
 
 # sessing parameters
@@ -516,6 +513,15 @@ if (!$config->{repository}) {
     die "FATAL $program_name: no repository have been defined (use --repository to specify one on the command line";
 }
 
+$SIG{TERM} = sub {
+    warn "Got KILLED by SIGTERM at " . strftime("%c", localtime()) . " .\n";
+    if (!$run{use_old_chroot} && $chroot_tmp) {
+	print "Cleaning up before dying.\n";
+	clean_chroot($chroot_tmp, \%run, $config);
+    }
+    exit(1);
+};
+
 my $urpmi = Iurt::Urpmi->new(run => \%run, config => $config, urpmi_options => "-v --no-verify-rpm --tune-rpm=all --nolock --auto --no-recommends --ignoresize $config->{urpmi_options}");
 $run{urpmi} = $urpmi;
 
@@ -612,7 +618,6 @@ $run{done} = \%done;
 my $home = $config->{local_home};
 
 my $chroot_base = $config->{chroot_base};
-my ($chroot_name, $chroot_tmp, $chroot, $chroot_ref);
 $chroot_name = "chroot_$run{distro_tag}$debug_tag.$run{my_arch}";
 if (!$run{use_old_chroot}) {
     $chroot_tmp = "$chroot_base/chroot_tmp";