kivitendo/scripts/task_server.pl @ de53e9a8
| 149d2f33 | Moritz Bunkus | #!/usr/bin/perl
|
||
| 29458987 | Moritz Bunkus | |||
use List::MoreUtils qw(any);
|
||||
| 149d2f33 | Moritz Bunkus | use strict;
|
||
| 22e5a8bb | Moritz Bunkus | my $exe_dir;
|
||
| 149d2f33 | Moritz Bunkus | BEGIN {
|
||
| ecb3c2e6 | Moritz Bunkus | use FindBin;
|
||
use lib "$FindBin::Bin/..";
|
||||
| 4ab40559 | Moritz Bunkus | use SL::System::Process;
|
||
| 22e5a8bb | Moritz Bunkus | $exe_dir = SL::System::Process::exe_dir;
|
||
| 88e5a86e | Moritz Bunkus | |||
| 4ab40559 | Moritz Bunkus | unshift @INC, "${exe_dir}/modules/override"; # Use our own versions of various modules (e.g. YAML).
|
||
push @INC, "${exe_dir}/modules/fallback"; # Only use our own versions of modules if there's no system version.
|
||||
unshift @INC, $exe_dir;
|
||||
| 88e5a86e | Moritz Bunkus | |||
| 4ab40559 | Moritz Bunkus | chdir($exe_dir) || die "Cannot change directory to ${exe_dir}\n";
|
||
| 149d2f33 | Moritz Bunkus | }
|
||
use CGI qw( -no_xhtml);
|
||||
use Cwd;
|
||||
use Daemon::Generic;
|
||||
use Data::Dumper;
|
||||
use DateTime;
|
||||
| 29458987 | Moritz Bunkus | use Encode qw();
|
||
| 149d2f33 | Moritz Bunkus | use English qw(-no_match_vars);
|
||
| db7dbf26 | Moritz Bunkus | use File::Spec;
|
||
| a63be3ad | Moritz Bunkus | use List::Util qw(first);
|
||
| 7056eb31 | Moritz Bunkus | use POSIX qw(setuid setgid);
|
||
| 149d2f33 | Moritz Bunkus | use SL::Auth;
|
||
use SL::DB::BackgroundJob;
|
||||
use SL::BackgroundJob::ALL;
|
||||
use SL::Form;
|
||||
use SL::Helper::DateTime;
|
||||
| c3f94f18 | Sven Schöling | use SL::InstanceConfiguration;
|
||
| 149d2f33 | Moritz Bunkus | use SL::LXDebug;
|
||
| 67b21d42 | Moritz Bunkus | use SL::LxOfficeConf;
|
||
| 149d2f33 | Moritz Bunkus | use SL::Locale;
|
||
| 29458987 | Moritz Bunkus | use SL::Mailer;
|
||
| db7dbf26 | Moritz Bunkus | use SL::System::TaskServer;
|
||
| 29458987 | Moritz Bunkus | use Template;
|
||
| 149d2f33 | Moritz Bunkus | |||
| e08abae0 | Moritz Bunkus | our %lx_office_conf;
|
||
| 149d2f33 | Moritz Bunkus | |||
| e36a3c06 | Moritz Bunkus | sub debug {
|
||
return if !$lx_office_conf{task_server}->{debug};
|
||||
$::lxdebug->message(0, @_);
|
||||
}
|
||||
| 149d2f33 | Moritz Bunkus | sub lxinit {
|
||
| 57a728bd | Moritz Bunkus | my $login = $lx_office_conf{task_server}->{login};
|
||
my $client = $lx_office_conf{task_server}->{client};
|
||||
| 149d2f33 | Moritz Bunkus | |||
package main;
|
||||
| c3f94f18 | Sven Schöling | $::lxdebug = LXDebug->new;
|
||
$::locale = Locale->new($::lx_office_conf{system}->{language});
|
||||
$::form = Form->new;
|
||||
$::auth = SL::Auth->new;
|
||||
| 57a728bd | Moritz Bunkus | die "No client configured or no client found with the name/ID '$client'" unless $::auth->set_client($client);
|
||
| c3f94f18 | Sven Schöling | $::instance_conf = SL::InstanceConfiguration->new;
|
||
$::request = { cgi => CGI->new({}) };
|
||||
| 149d2f33 | Moritz Bunkus | |||
die 'cannot reach auth db' unless $::auth->session_tables_present;
|
||||
$::auth->restore_session;
|
||||
require "bin/mozilla/common.pl";
|
||||
| 4531a6c7 | Sven Schöling | die "cannot find user $login" unless %::myconfig = $::auth->read_user(login => $login);
|
||
| 149d2f33 | Moritz Bunkus | die "cannot find locale for user $login" unless $::locale = Locale->new('de');
|
||
}
|
||||
| d7fddb8d | Moritz Bunkus | sub per_job_initialization {
|
||
| 4bb9bf41 | Moritz Bunkus | $::locale = Locale->new($::myconfig{countrycode} || $::lx_office_conf{system}->{language});
|
||
| d7fddb8d | Moritz Bunkus | $::form = Form->new;
|
||
$::instance_conf = SL::InstanceConfiguration->new;
|
||||
$::request = SL::Request->new(
|
||||
cgi => CGI->new({}),
|
||||
layout => SL::Layout::None->new,
|
||||
);
|
||||
$::auth->restore_session;
|
||||
$::form->{login} = $lx_office_conf{task_server}->{login};
|
||||
$::instance_conf->init;
|
||||
| 0dc0e521 | Sven Schöling | |||
$::form->{__ERROR_HANDLER} = sub { die @_ };
|
||||
| d7fddb8d | Moritz Bunkus | }
|
||
| 7056eb31 | Moritz Bunkus | sub drop_privileges {
|
||
| e08abae0 | Moritz Bunkus | my $user = $lx_office_conf{task_server}->{run_as};
|
||
| 7056eb31 | Moritz Bunkus | return unless $user;
|
||
my ($uid, $gid);
|
||||
while (my @details = getpwent()) {
|
||||
next unless $details[0] eq $user;
|
||||
($uid, $gid) = @details[2, 3];
|
||||
last;
|
||||
}
|
||||
endpwent();
|
||||
if (!$uid) {
|
||||
print "Error: Cannot drop privileges to ${user}: user does not exist\n";
|
||||
exit 1;
|
||||
}
|
||||
if (!setgid($gid)) {
|
||||
print "Error: Cannot drop group privileges to ${user} (group ID $gid): $!\n";
|
||||
exit 1;
|
||||
}
|
||||
if (!setuid($uid)) {
|
||||
print "Error: Cannot drop user privileges to ${user} (user ID $uid): $!\n";
|
||||
exit 1;
|
||||
}
|
||||
}
|
||||
| 29458987 | Moritz Bunkus | sub notify_on_failure {
|
||
my (%params) = @_;
|
||||
my $cfg = $lx_office_conf{'task_server/notify_on_failure'} || {};
|
||||
return if any { !$cfg->{$_} } qw(send_email_to email_from email_subject email_template);
|
||||
chdir $exe_dir;
|
||||
return debug("Template " . $cfg->{email_template} . " missing!") unless -f $cfg->{email_template};
|
||||
my $email_to = $cfg->{send_email_to};
|
||||
if ($email_to !~ m{\@}) {
|
||||
my %user = $::auth->read_user(login => $email_to);
|
||||
return debug("cannot find user for notification $email_to") unless %user;
|
||||
$email_to = $user{email};
|
||||
return debug("user for notification " . $user{login} . " doesn't have a valid email address") unless $email_to =~ m{\@};
|
||||
}
|
||||
my $template = Template->new({
|
||||
INTERPOLATE => 0,
|
||||
EVAL_PERL => 0,
|
||||
ABSOLUTE => 1,
|
||||
CACHE_SIZE => 0,
|
||||
});
|
||||
return debug("Could not create Template instance") unless $template;
|
||||
$params{client} = $::auth->client;
|
||||
my $body;
|
||||
$template->process($cfg->{email_template}, \%params, \$body);
|
||||
Mailer->new(
|
||||
from => $cfg->{email_from},
|
||||
to => $email_to,
|
||||
subject => $cfg->{email_subject},
|
||||
content_type => 'text/plain',
|
||||
charset => 'utf-8',
|
||||
message => Encode::decode('utf-8', $body),
|
||||
)->send;
|
||||
}
|
||||
| 149d2f33 | Moritz Bunkus | sub gd_preconfig {
|
||
my $self = shift;
|
||||
| 31eb98ad | Moritz Bunkus | SL::LxOfficeConf->read($self->{configfile});
|
||
| 149d2f33 | Moritz Bunkus | |||
| 57a728bd | Moritz Bunkus | die "Missing section [task_server] in config file" unless $lx_office_conf{task_server};
|
||
die "Missing key 'login' in section [task_server] in config file" unless $lx_office_conf{task_server}->{login};
|
||||
die "Missing key 'client' in section [task_server] in config file" unless $lx_office_conf{task_server}->{client};
|
||||
| 149d2f33 | Moritz Bunkus | |||
| 7056eb31 | Moritz Bunkus | drop_privileges();
|
||
| 149d2f33 | Moritz Bunkus | lxinit();
|
||
return ();
|
||||
}
|
||||
sub gd_run {
|
||||
while (1) {
|
||||
my $ok = eval {
|
||||
| e36a3c06 | Moritz Bunkus | debug("Retrieving jobs");
|
||
| 149d2f33 | Moritz Bunkus | |||
my $jobs = SL::DB::Manager::BackgroundJob->get_all_need_to_run;
|
||||
| e36a3c06 | Moritz Bunkus | debug(" Found: " . join(' ', map { $_->package_name } @{ $jobs })) if @{ $jobs };
|
||
| 149d2f33 | Moritz Bunkus | |||
| f6669610 | Moritz Bunkus | foreach my $job (@{ $jobs }) {
|
||
# Provide fresh global variables in case legacy code modifies
|
||||
# them somehow.
|
||||
| d7fddb8d | Moritz Bunkus | per_job_initialization();
|
||
| f6669610 | Moritz Bunkus | |||
| 22e5a8bb | Moritz Bunkus | chdir $exe_dir;
|
||
| 29458987 | Moritz Bunkus | my $history = $job->run;
|
||
notify_on_failure(history => $history) if $history && $history->has_failed;
|
||||
| f6669610 | Moritz Bunkus | }
|
||
| 149d2f33 | Moritz Bunkus | |||
1;
|
||||
};
|
||||
| 29458987 | Moritz Bunkus | if (!$ok) {
|
||
my $error = $EVAL_ERROR;
|
||||
debug("Exception during execution: ${error}");
|
||||
notify_on_failure(exception => $error);
|
||||
}
|
||||
| e36a3c06 | Moritz Bunkus | debug("Sleeping");
|
||
| 149d2f33 | Moritz Bunkus | |||
my $seconds = 60 - (localtime)[0];
|
||||
| 2dffed32 | Moritz Bunkus | if (!eval {
|
||
local $SIG{'ALRM'} = sub {
|
||||
| e36a3c06 | Moritz Bunkus | debug("Got woken up by SIGALRM");
|
||
| 2dffed32 | Moritz Bunkus | die "Alarm!\n"
|
||
};
|
||||
sleep($seconds < 30 ? $seconds + 60 : $seconds);
|
||||
1;
|
||||
}) {
|
||||
die $@ unless $@ eq "Alarm!\n";
|
||||
}
|
||||
| 149d2f33 | Moritz Bunkus | }
|
||
}
|
||||
| 0dc0e521 | Sven Schöling | sub end_of_request {
|
||
$main::lxdebug->show_backtrace();
|
||||
die <<EOF;
|
||||
Job called ::end_of_request()!
|
||||
This usually indicates success but should not be used by background jobs. A
|
||||
backtrace has been logged. Please tell the job author to have a look at it.
|
||||
EOF
|
||||
}
|
||||
| db7dbf26 | Moritz Bunkus | chdir $exe_dir;
|
||
| 149d2f33 | Moritz Bunkus | |||
| db7dbf26 | Moritz Bunkus | mkdir SL::System::TaskServer::PID_BASE() if !-d SL::System::TaskServer::PID_BASE();
|
||
my $file = first { -f } ("${exe_dir}/config/kivitendo.conf", "${exe_dir}/config/lx_office.conf", "${exe_dir}/config/kivitendo.conf.default");
|
||||
die "No configuration file found." unless $file;
|
||||
$file = File::Spec->abs2rel(Cwd::abs_path($file), Cwd::abs_path($exe_dir));
|
||||
| 149d2f33 | Moritz Bunkus | |||
| 67b21d42 | Moritz Bunkus | newdaemon(configfile => $file,
|
||
| b7348417 | Moritz Bunkus | progname => 'kivitendo-background-jobs',
|
||
| db7dbf26 | Moritz Bunkus | pidbase => SL::System::TaskServer::PID_BASE() . '/',
|
||
| 149d2f33 | Moritz Bunkus | );
|
||
1;
|