lib/PXElator/daemons.pm

package daemons;

use warnings;
use strict;

use File::Slurp;
use Data::Dump qw/dump/;

use x11;
use CouchDB;

our $pids;
$pids = {
        httpd => $$,
        kvm => 'not started',
        wireshark => 'not started',
} unless defined $pids; # keep pids on refresh

sub DESTROY {
        warn "pids ",dump( $pids );
        foreach ( values %$pids ) {
                warn "kill $_";
                kill 1,$_ || kill 9, $_;
        }
}

$SIG{CHLD} = 'IGNORE';

sub audit {
        my $msg = shift;
        my $daemon = shift;
        CouchDB::audit( $msg, $daemon, { daemon => $daemon, message => $msg, @_ } );
}

sub start_stop {
        my $daemon = shift;

        my $pid = $pids->{$daemon};
        my $pid_path = "$daemon.pid";
        $pid_path =~ s{/}{-}g;
        $pid_path = "$server::conf/$pid_path";

        if ( ! $pid && -e $pid_path ) {
                my $p = read_file $pid_path;
                if ( kill 0, $p ) {
                        audit 'adopted', $daemon, pid => $p;
                        return $pids->{$daemon} = $p;
                } else {
                        audit 'not running', $daemon, pid => $p;
                        unlink $pid_path;
                }
        }

        $pid ||= 'not started';
        warn "start_stop $daemon $pid\n";

        if ( $pid =~ m{^\d+$} ) {
                my $pstree = `pstree -p $pid`;
                my @pids = $pstree =~ m{\((\d+)\)}g;
                warn "pstree $pstree pids ",dump( @pids );
                kill 1, $_ foreach reverse @pids;
                $pids->{$daemon} = 'stopped';
                audit 'stopped', $daemon, pid => $pid;
                return qq|$daemon pid $pid stopped|;
        } else {
                if ( $pid = fork ) {
                        # parent
                        $pids->{$daemon} = $pid;
                        write_file $pid_path, $pid;
                        audit 'forked', $daemon, pid => $pid;
                        return qq|$daemon pid $pid started|;
                } elsif ( defined $pid ) {
                        # child
                        my $invoke = 'start';
                        $invoke = $1 if $daemon =~ s{/(.+)}{};
                        if ( $daemon =~ m{dhcpd|tftpd|dnsd} ) {
                                my $exec = "perl -I$server::base_dir/lib -I$server::base_dir/lib/PXElator -M$daemon -e ${daemon}::${invoke}";
                                audit 'exec', $daemon, 'exec' => $exec;
                                x11::xterm( $daemon => $exec );
                        } else {
                                my $eval = $daemon . '::' . $invoke . '(' . ( @_ ? dump(@_) : '' ) . ')';
                                audit 'eval', $daemon, 'eval' => $eval;
                                eval $eval;
                                audit 'error', $daemon, error => $@  if $@;
                        }
                        exit;
                } else {
                        audit 'error', $daemon, error => $!;
                        die "fork error $!";
                }
        }
}

1	dpavlin	185	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	dpavlin	208	use CouchDB;
11	dpavlin	185
12			our $pids;
13	dpavlin	225	$pids = {
14			httpd => $$,
15			kvm => 'not started',
16			wireshark => 'not started',
17			} unless defined $pids; # keep pids on refresh
18	dpavlin	185
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	dpavlin	208	sub audit {
30			my $msg = shift;
31			my $daemon = shift;
32			CouchDB::audit( $msg, $daemon, { daemon => $daemon, message => $msg, @_ } );
33			}
34
35	dpavlin	185	sub start_stop {
36			my $daemon = shift;
37
38			my $pid = $pids->{$daemon};
39	dpavlin	225	my $pid_path = "$daemon.pid";
40			$pid_path =~ s{/}{-}g;
41			$pid_path = "$server::conf/$pid_path";
42	dpavlin	185
43			if ( ! $pid && -e $pid_path ) {
44			my $p = read_file $pid_path;
45			if ( kill 0, $p ) {
46	dpavlin	208	audit 'adopted', $daemon, pid => $p;
47	dpavlin	185	return $pids->{$daemon} = $p;
48			} else {
49	dpavlin	208	audit 'not running', $daemon, pid => $p;
50	dpavlin	185	unlink $pid_path;
51			}
52			}
53
54			$pid \|\|= 'not started';
55			warn "start_stop $daemon $pid\n";
56
57			if ( $pid =~ m{^\d+$} ) {
58			my $pstree = `pstree -p $pid`;
59			my @pids = $pstree =~ m{\((\d+)\)}g;
60			warn "pstree $pstree pids ",dump( @pids );
61			kill 1, $_ foreach reverse @pids;
62			$pids->{$daemon} = 'stopped';
63	dpavlin	208	audit 'stopped', $daemon, pid => $pid;
64	dpavlin	185	return qq\|$daemon pid $pid stopped\|;
65			} else {
66			if ( $pid = fork ) {
67			# parent
68			$pids->{$daemon} = $pid;
69			write_file $pid_path, $pid;
70	dpavlin	208	audit 'forked', $daemon, pid => $pid;
71	dpavlin	185	return qq\|$daemon pid $pid started\|;
72			} elsif ( defined $pid ) {
73			# child
74			my $invoke = 'start';
75			$invoke = $1 if $daemon =~ s{/(.+)}{};
76			if ( $daemon =~ m{dhcpd\|tftpd\|dnsd} ) {
77			my $exec = "perl -I$server::base_dir/lib -I$server::base_dir/lib/PXElator -M$daemon -e ${daemon}::${invoke}";
78	dpavlin	208	audit 'exec', $daemon, 'exec' => $exec;
79	dpavlin	185	x11::xterm( $daemon => $exec );
80			} else {
81			my $eval = $daemon . '::' . $invoke . '(' . ( @_ ? dump(@_) : '' ) . ')';
82	dpavlin	208	audit 'eval', $daemon, 'eval' => $eval;
83	dpavlin	185	eval $eval;
84	dpavlin	208	audit 'error', $daemon, error => $@ if $@;
85	dpavlin	185	}
86			exit;
87			} else {
88	dpavlin	208	audit 'error', $daemon, error => $!;
89	dpavlin	185	die "fork error $!";
90			}
91			}
92			}
93