git.rot13.org Git - pxelator/blob - lib/PXElator/daemons.pm

   1 package daemons;
   2
   3 use warnings;
   4 use strict;
   5
   6 use File::Slurp;
   7 use Data::Dump qw/dump/;
   8
   9 use x11;
  10 use store;
  11
  12 our $pids;
  13 $pids = {
  14         httpd => $$,
  15         kvm => 'not started',
  16         wireshark => 'not started',
  17 } unless defined $pids; # keep pids on refresh
  18
  19 sub DESTROY {
  20         warn "pids ",dump( $pids );
  21         foreach ( values %$pids ) {
  22                 warn "kill $_";
  23                 kill 1,$_ || kill 9, $_;
  24         }
  25 }
  26
  27 $SIG{CHLD} = 'IGNORE';
  28
  29 sub audit {
  30         my $msg = shift;
  31         my $daemon = shift;
  32         store::audit( $msg, $daemon, { daemon => $daemon, message => $msg, @_ } );
  33 }
  34
  35 mkdir "$server::conf/pid" unless -d "$server::conf/pid";
  36
  37 sub start_stop {
  38         my ($daemon,$param) = @_;
  39
  40         if ( defined $param->{nr} ) { # XXX nr for kvm
  41                 $daemon .= '.' . $param->{nr};
  42         } else {
  43                 $param->{nr} = $1 if $daemon =~ /\.(\d+)/;
  44         }
  45
  46         my $pid = $pids->{$daemon};
  47         my $pid_path = $daemon;
  48         $pid_path =~ s{/}{-}g;
  49         $pid_path = "$server::conf/pid/$pid_path";
  50
  51         warn "# start_stop $daemon ",dump( $param );
  52
  53         if ( ! $pid && -e $pid_path ) {
  54                 my $p = read_file $pid_path;
  55                 if ( kill 0, $p ) {
  56                         audit 'adopted', $daemon, pid => $p;
  57                         return $pids->{$daemon} = $p;
  58                 } else {
  59                         audit 'not running', $daemon, pid => $p;
  60                         unlink $pid_path;
  61                 }
  62         }
  63
  64         $pid ||= 'not started';
  65         warn "start_stop $daemon $pid\n";
  66
  67         if ( $pid =~ m{^\d+$} && kill 0, $pid ) {
  68                 my $pstree = `pstree -p $pid`;
  69                 my @pids = $pstree =~ m{\((\d+)\)}g;
  70                 warn "pstree $pstree pids ",dump( @pids );
  71                 foreach ( @pids ) {
  72                         warn "kill $_\n";
  73                         kill 1, $_;
  74                         if ( kill 0, $_ ) {
  75                                 warn "kill 9 $_";
  76                                 kill 9, $_;
  77                         }
  78                 }
  79                 $pids->{$daemon} = 'stopped';
  80                 audit 'stopped', $daemon, pid => $pid;
  81                 return qq|$daemon pid $pid stopped|;
  82         } else {
  83
  84                 $ENV{$_} = $param->{$_} foreach keys %$param;
  85
  86                 if ( $pid = fork ) {
  87                         # parent
  88                         $pids->{$daemon} = $pid;
  89                         write_file $pid_path, $pid;
  90                         audit 'forked', $daemon, pid => $pid;
  91
  92                         delete $ENV{$_} foreach keys %$param; # cleanup enviroment
  93
  94                         return qq|$daemon pid $pid started|;
  95                 } elsif ( defined $pid ) {
  96                         # child
  97                         my $invoke = 'start';
  98                         $invoke = $1 if $daemon =~ s{/(.+)}{};
  99                         #warn "## ENV child ",dump( %ENV );
 100                         if ( $daemon =~ m{dhcpd|tftpd|dnsd|syslogd} ) {
 101                                 my $exec = "perl -I$server::base_dir/lib -I$server::base_dir/lib/PXElator -M$daemon -e ${daemon}::${invoke}";
 102                                 audit 'exec', $daemon, 'exec' => $exec;
 103                                 x11::xterm( $daemon => $exec );
 104                         } else {
 105                                 $daemon =~ s{\.\d+$}{} if defined $ENV{nr};
 106                                 my $eval = $daemon . '::' . $invoke . '(' . dump( $param ) . ')';
 107                                 audit 'eval', $daemon, 'eval' => $eval;
 108                                 eval $eval;
 109                                 audit 'error', $daemon, error => $@  if $@;
 110                         }
 111                         exit;
 112                 } else {
 113                         audit 'error', $daemon, error => $!;
 114                         die "fork error $!";
 115                 }
 116         }
 117 }
 118