Added way to clean things during a fork for kadmin

The MIT kadmin module currently directly runs the MIT kadmin program.  Some
data needs to be cleaned during the forks for this.  This provides a
callback that can be registered and is called during the fork process,
currently just to mark database handles inactive.  It was added to both the
MIT and Heimdal modules, though it's only a stub in the Heimdal module.
Heimdal is not forking kadmin, but the stub is there in order to allow the
caller to not care which module is being used and just always register the
callbacks.

1 files changed, 7 insertions, 5 deletions

diff --git a/perl/Wallet/Kadmin/MIT.pm b/perl/Wallet/Kadmin/MIT.pm
index 2e9b0b4..c3ad901 100644
--- a/perl/Wallet/Kadmin/MIT.pm
+++ b/perl/Wallet/Kadmin/MIT.pm
@@ -39,6 +39,12 @@ sub error {
     return $self->{error};
 }
 
+# Set a callback to be called for forked kadmin processes.
+sub fork_callback {
+    my ($self, $callback) = @_;
+    $self->{fork_callback} = $callback;
+}
+
 ##############################################################################
 # kadmin Interaction
 ##############################################################################
@@ -73,11 +79,7 @@ sub kadmin {
         $self->error ("cannot fork: $!");
         return;
     } elsif ($pid == 0) {
-        # TODO - How should I handle the db handle?
-        # Don't use die here; it will get trapped as an exception.  Also be
-        # careful about our database handles.  (We still lose if there's some
-        # other database handle open we don't know about.)
-        #$object->{dbh}->{InactiveDestroy} = 1;
+        $self->{fork_callback} ();
         unless (open (STDERR, '>&STDOUT')) {
             warn "wallet: cannot dup stdout: $!\n";
             exit 1;