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	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 CouchDB;
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	CouchDB::audit( $msg, $daemon, { daemon => $daemon, message => $msg, @_ } );
33	}
34
35	sub start_stop {
36	my $daemon = shift;
37
38	my $pid = $pids->{$daemon};
39	my $pid_path = "$daemon.pid";
40	$pid_path =~ s{/}{-}g;
41	$pid_path = "$server::conf/$pid_path";
42
43	if ( ! $pid && -e $pid_path ) {
44	my $p = read_file $pid_path;
45	if ( kill 0, $p ) {
46	audit 'adopted', $daemon, pid => $p;
47	return $pids->{$daemon} = $p;
48	} else {
49	audit 'not running', $daemon, pid => $p;
50	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	audit 'stopped', $daemon, pid => $pid;
64	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	audit 'forked', $daemon, pid => $pid;
71	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	audit 'exec', $daemon, 'exec' => $exec;
79	x11::xterm( $daemon => $exec );
80	} else {
81	my $eval = $daemon . '::' . $invoke . '(' . ( @_ ? dump(@_) : '' ) . ')';
82	audit 'eval', $daemon, 'eval' => $eval;
83	eval $eval;
84	audit 'error', $daemon, error => $@ if $@;
85	}
86	exit;
87	} else {
88	audit 'error', $daemon, error => $!;
89	die "fork error $!";
90	}
91	}
92	}
93