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, @_ } );
}

mkdir "$server::conf/pid" unless -d "$server::conf/pid";

sub start_stop {
        my ($daemon,$param) = @_;

        $daemon .= '.' . $param->{nr} if defined $param->{nr}; # XXX nr for kvm

        my $pid = $pids->{$daemon};
        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 ) {
                        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+$} && kill 0, $pid ) {
                my $pstree = `pstree -p $pid`;
                my @pids = $pstree =~ m{\((\d+)\)}g;
                warn "pstree $pstree pids ",dump( @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;
                        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{/(.+)}{};
warn "XXX 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}";
                                audit 'exec', $daemon, 'exec' => $exec;
                                x11::xterm( $daemon => $exec );
                        } else {
                                $daemon =~ s{\.\d+$}{} if defined $ENV{nr};
warn "XXXX $daemon\n";
                                my $eval = $daemon . '::' . $invoke . '(' . dump( $param ) . ')';
                                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	321	mkdir "$server::conf/pid" unless -d "$server::conf/pid";
36
37	dpavlin	185	sub start_stop {
38	dpavlin	390	my ($daemon,$param) = @_;
39	dpavlin	185
40	dpavlin	390	$daemon .= '.' . $param->{nr} if defined $param->{nr}; # XXX nr for kvm
41
42	dpavlin	185	my $pid = $pids->{$daemon};
43	dpavlin	321	my $pid_path = $daemon;
44	dpavlin	225	$pid_path =~ s{/}{-}g;
45	dpavlin	321	$pid_path = "$server::conf/pid/$pid_path";
46	dpavlin	185
47	dpavlin	390	warn "# start_stop $daemon ",dump( $param );
48
49	dpavlin	185	if ( ! $pid && -e $pid_path ) {
50			my $p = read_file $pid_path;
51			if ( kill 0, $p ) {
52	dpavlin	208	audit 'adopted', $daemon, pid => $p;
53	dpavlin	185	return $pids->{$daemon} = $p;
54			} else {
55	dpavlin	208	audit 'not running', $daemon, pid => $p;
56	dpavlin	185	unlink $pid_path;
57			}
58			}
59
60			$pid \|\|= 'not started';
61			warn "start_stop $daemon $pid\n";
62
63	dpavlin	245	if ( $pid =~ m{^\d+$} && kill 0, $pid ) {
64	dpavlin	185	my $pstree = `pstree -p $pid`;
65			my @pids = $pstree =~ m{\((\d+)\)}g;
66			warn "pstree $pstree pids ",dump( @pids );
67	dpavlin	227	foreach ( @pids ) {
68			warn "kill $_\n";
69			kill 1, $_;
70			if ( kill 0, $_ ) {
71			warn "kill 9 $_";
72			kill 9, $_;
73			}
74			}
75	dpavlin	185	$pids->{$daemon} = 'stopped';
76	dpavlin	208	audit 'stopped', $daemon, pid => $pid;
77	dpavlin	185	return qq\|$daemon pid $pid stopped\|;
78			} else {
79	dpavlin	390
80			$ENV{$_} = $param->{$_} foreach keys %$param;
81
82	dpavlin	185	if ( $pid = fork ) {
83			# parent
84			$pids->{$daemon} = $pid;
85			write_file $pid_path, $pid;
86	dpavlin	208	audit 'forked', $daemon, pid => $pid;
87	dpavlin	390
88			delete $ENV{$_} foreach keys %$param; # cleanup enviroment
89
90	dpavlin	185	return qq\|$daemon pid $pid started\|;
91			} elsif ( defined $pid ) {
92			# child
93			my $invoke = 'start';
94			$invoke = $1 if $daemon =~ s{/(.+)}{};
95	dpavlin	390	warn "XXX child ",dump( %ENV );
96	dpavlin	231	if ( $daemon =~ m{dhcpd\|tftpd\|dnsd\|syslogd} ) {
97	dpavlin	185	my $exec = "perl -I$server::base_dir/lib -I$server::base_dir/lib/PXElator -M$daemon -e ${daemon}::${invoke}";
98	dpavlin	208	audit 'exec', $daemon, 'exec' => $exec;
99	dpavlin	185	x11::xterm( $daemon => $exec );
100			} else {
101	dpavlin	390	$daemon =~ s{\.\d+$}{} if defined $ENV{nr};
102			warn "XXXX $daemon\n";
103			my $eval = $daemon . '::' . $invoke . '(' . dump( $param ) . ')';
104	dpavlin	208	audit 'eval', $daemon, 'eval' => $eval;
105	dpavlin	185	eval $eval;
106	dpavlin	208	audit 'error', $daemon, error => $@ if $@;
107	dpavlin	185	}
108			exit;
109			} else {
110	dpavlin	208	audit 'error', $daemon, error => $!;
111	dpavlin	185	die "fork error $!";
112			}
113			}
114			}
115