extract nr from kvm.nr

[pxelator] / lib / PXElator / daemons.pm

diff --git a/lib/PXElator/daemons.pm b/lib/PXElator/daemons.pm
index 5bc6889..ee17802 100644 (file)
--- a/lib/PXElator/daemons.pm
+++ b/lib/PXElator/daemons.pm
@@ -7,9 +7,14 @@ use File::Slurp;
 use Data::Dump qw/dump/;
 
 use x11;
+use store;
 
 our $pids;
-$pids = { httpd => $$ } unless defined $pids; # keep pids on refresh
+$pids = {
+       httpd => $$,
+       kvm => 'not started',
+       wireshark => 'not started',
+} unless defined $pids; # keep pids on refresh
 
 sub DESTROY {
        warn "pids ",dump( $pids );
@@ -21,19 +26,37 @@ sub DESTROY {
 
 $SIG{CHLD} = 'IGNORE';
 
-sub start_stop {
+sub audit {
+       my $msg = shift;
        my $daemon = shift;
+       store::audit( $msg, $daemon, { daemon => $daemon, message => $msg, @_ } );
+}
+
+mkdir "$server::conf/pid" unless -d "$server::conf/pid";
+
+sub start_stop {
+       my ($daemon,$param) = @_;
+
+       if ( defined $param->{nr} ) { # XXX nr for kvm
+               $daemon .= '.' . $param->{nr};
+       } else {
+               $param->{nr} = $1 if $daemon =~ /\.(\d+)/;
+       }
 
        my $pid = $pids->{$daemon};
-       my $pid_path = "$server::conf/$daemon.pid";
+       my $pid_path = $daemon;
+       $pid_path =~ s{/}{-}g;
+       $pid_path = "$server::conf/pid/$pid_path";
+
+       warn "# start_stop $daemon ",dump( $param );
 
        if ( ! $pid && -e $pid_path ) {
                my $p = read_file $pid_path;
                if ( kill 0, $p ) {
-                       warn "adopted $daemon $p\n";
+                       audit 'adopted', $daemon, pid => $p;
                        return $pids->{$daemon} = $p;
                } else {
-                       warn "old $daemon $p not running";
+                       audit 'not running', $daemon, pid => $p;
                        unlink $pid_path;
                }
        }
@@ -41,36 +64,53 @@ sub start_stop {
        $pid ||= 'not started';
        warn "start_stop $daemon $pid\n";
 
-       if ( $pid =~ m{^\d+$} ) {
+       if ( $pid =~ m{^\d+$} && kill 0, $pid ) {
                my $pstree = `pstree -p $pid`;
                my @pids = $pstree =~ m{\((\d+)\)}g;
                warn "pstree $pstree pids ",dump( @pids );
-               kill 1, $_ foreach reverse @pids;
+               foreach ( @pids ) {
+                       warn "kill $_\n";
+                       kill 1, $_;
+                       if ( kill 0, $_ ) {
+                               warn "kill 9 $_";
+                               kill 9, $_;
+                       }
+               }
                $pids->{$daemon} = 'stopped';
+               audit 'stopped', $daemon, pid => $pid;
                return qq|$daemon pid $pid stopped|;
        } else {
+
+               $ENV{$_} = $param->{$_} foreach keys %$param;
+
                if ( $pid = fork ) {
                        # parent
                        $pids->{$daemon} = $pid;
                        write_file $pid_path, $pid;
-                       warn "forked $daemon $pid\n";
+                       audit 'forked', $daemon, pid => $pid;
+
+                       delete $ENV{$_} foreach keys %$param; # cleanup enviroment
+
                        return qq|$daemon pid $pid started|;
                } elsif ( defined $pid ) {
                        # child
                        my $invoke = 'start';
                        $invoke = $1 if $daemon =~ s{/(.+)}{};
-                       if ( $daemon =~ m{dhcpd|tftpd|dnsd} ) {
+                       #warn "## ENV child ",dump( %ENV );
+                       if ( $daemon =~ m{dhcpd|tftpd|dnsd|syslogd} ) {
                                my $exec = "perl -I$server::base_dir/lib -I$server::base_dir/lib/PXElator -M$daemon -e ${daemon}::${invoke}";
-                               warn "exec $exec";
+                               audit 'exec', $daemon, 'exec' => $exec;
                                x11::xterm( $daemon => $exec );
                        } else {
-                               my $eval = $daemon . '::' . $invoke . '(' . ( @_ ? dump(@_) : '' ) . ')';
-                               warn "eval $eval";
+                               $daemon =~ s{\.\d+$}{} if defined $ENV{nr};
+                               my $eval = $daemon . '::' . $invoke . '(' . dump( $param ) . ')';
+                               audit 'eval', $daemon, 'eval' => $eval;
                                eval $eval;
-                               warn "can't start $daemon: $@" if $@;
+                               audit 'error', $daemon, error => $@  if $@;
                        }
                        exit;
                } else {
+                       audit 'error', $daemon, error => $!;
                        die "fork error $!";
                }
        }