scripts/task_server.pl

   1 #!/usr/bin/perl
   2
   3
   4 use List::MoreUtils qw(any);
   5
   6 use strict;
   7
   8 my $exe_dir;
   9
  10 BEGIN {
  11   use FindBin;
  12   use lib "$FindBin::Bin/..";
  13
  14   use SL::System::Process;
  15   $exe_dir = SL::System::Process::exe_dir;
  16
  17   unshift @INC, "${exe_dir}/modules/override"; # Use our own versions of various modules (e.g. YAML).
  18   push    @INC, "${exe_dir}/modules/fallback"; # Only use our own versions of modules if there's no system version.
  19   unshift @INC, $exe_dir;
  20
  21   chdir($exe_dir) || die "Cannot change directory to ${exe_dir}\n";
  22 }
  23
  24 use CGI qw( -no_xhtml);
  25 use Cwd;
  26 use Daemon::Generic;
  27 use Data::Dumper;
  28 use DateTime;
  29 use Encode qw();
  30 use English qw(-no_match_vars);
  31 use File::Spec;
  32 use List::Util qw(first);
  33 use POSIX qw(setuid setgid);
  34 use SL::Auth;
  35 use SL::DB::BackgroundJob;
  36 use SL::BackgroundJob::ALL;
  37 use SL::Form;
  38 use SL::Helper::DateTime;
  39 use SL::InstanceConfiguration;
  40 use SL::LXDebug;
  41 use SL::LxOfficeConf;
  42 use SL::Locale;
  43 use SL::Mailer;
  44 use SL::System::TaskServer;
  45 use Template;
  46
  47 our %lx_office_conf;
  48
  49 sub debug {
  50   return if !$lx_office_conf{task_server}->{debug};
  51   $::lxdebug->message(0, @_);
  52 }
  53
  54 sub initialize_kivitendo {
  55   chdir $exe_dir;
  56
  57   my $login  = $lx_office_conf{task_server}->{login};
  58   my $client = $lx_office_conf{task_server}->{client};
  59
  60   package main;
  61
  62   Form::disconnect_standard_dbh;
  63   $::lxdebug       = LXDebug->new;
  64   $::locale        = Locale->new($::lx_office_conf{system}->{language});
  65   $::form          = Form->new;
  66   $::auth          = SL::Auth->new;
  67   die "No client configured or no client found with the name/ID '$client'" unless $::auth->set_client($client);
  68   $::instance_conf = SL::InstanceConfiguration->new;
  69   $::request       = SL::Request->new(
  70     cgi            => CGI->new({}),
  71     layout         => SL::Layout::None->new,
  72   );
  73
  74   die 'cannot reach auth db'               unless $::auth->session_tables_present;
  75
  76   $::auth->restore_session;
  77   $::auth->create_or_refresh_session;
  78
  79   die "cannot find user $login"            unless %::myconfig = $::auth->read_user(login => $login);
  80   die "cannot find locale for user $login" unless $::locale   = Locale->new($::myconfig{countrycode} || $::lx_office_conf{system}->{language});
  81
  82   $::form->{__ERROR_HANDLER} = sub { die @_ };
  83 }
  84
  85 sub drop_privileges {
  86   my $user = $lx_office_conf{task_server}->{run_as};
  87   return unless $user;
  88
  89   my ($uid, $gid);
  90   while (my @details = getpwent()) {
  91     next unless $details[0] eq $user;
  92     ($uid, $gid) = @details[2, 3];
  93     last;
  94   }
  95   endpwent();
  96
  97   if (!$uid) {
  98     print "Error: Cannot drop privileges to ${user}: user does not exist\n";
  99     exit 1;
 100   }
 101
 102   if (!setgid($gid)) {
 103     print "Error: Cannot drop group privileges to ${user} (group ID $gid): $!\n";
 104     exit 1;
 105   }
 106
 107   if (!setuid($uid)) {
 108     print "Error: Cannot drop user privileges to ${user} (user ID $uid): $!\n";
 109     exit 1;
 110   }
 111 }
 112
 113 sub notify_on_failure {
 114   my (%params) = @_;
 115
 116   my $cfg = $lx_office_conf{'task_server/notify_on_failure'} || {};
 117
 118   return if any { !$cfg->{$_} } qw(send_email_to email_from email_subject email_template);
 119
 120   chdir $exe_dir;
 121
 122   return debug("Template " . $cfg->{email_template} . " missing!") unless -f $cfg->{email_template};
 123
 124   my $email_to = $cfg->{send_email_to};
 125   if ($email_to !~ m{\@}) {
 126     my %user = $::auth->read_user(login => $email_to);
 127     return debug("cannot find user for notification $email_to") unless %user;
 128
 129     $email_to = $user{email};
 130     return debug("user for notification " . $user{login} . " doesn't have a valid email address") unless $email_to =~ m{\@};
 131   }
 132
 133   my $template  = Template->new({
 134     INTERPOLATE => 0,
 135     EVAL_PERL   => 0,
 136     ABSOLUTE    => 1,
 137     CACHE_SIZE  => 0,
 138   });
 139
 140   return debug("Could not create Template instance") unless $template;
 141
 142   $params{client} = $::auth->client;
 143
 144   my $body;
 145   $template->process($cfg->{email_template}, \%params, \$body);
 146
 147   Mailer->new(
 148     from         => $cfg->{email_from},
 149     to           => $email_to,
 150     subject      => $cfg->{email_subject},
 151     content_type => 'text/plain',
 152     charset      => 'utf-8',
 153     message      => Encode::decode('utf-8', $body),
 154   )->send;
 155 }
 156
 157 sub gd_preconfig {
 158   my $self = shift;
 159
 160   SL::LxOfficeConf->read($self->{configfile});
 161
 162   die "Missing section [task_server] in config file"                 unless $lx_office_conf{task_server};
 163   die "Missing key 'login' in section [task_server] in config file"  unless $lx_office_conf{task_server}->{login};
 164   die "Missing key 'client' in section [task_server] in config file" unless $lx_office_conf{task_server}->{client};
 165
 166   drop_privileges();
 167   initialize_kivitendo();
 168
 169   return ();
 170 }
 171
 172 sub gd_run {
 173   while (1) {
 174     my $ok = eval {
 175       initialize_kivitendo();
 176
 177       debug("Retrieving jobs");
 178
 179       my $jobs = SL::DB::Manager::BackgroundJob->get_all_need_to_run;
 180
 181       debug("  Found: " . join(' ', map { $_->package_name } @{ $jobs })) if @{ $jobs };
 182
 183       foreach my $job (@{ $jobs }) {
 184         # Provide fresh global variables in case legacy code modifies
 185         # them somehow.
 186         initialize_kivitendo();
 187
 188         my $history = $job->run;
 189
 190         notify_on_failure(history => $history) if $history && $history->has_failed;
 191       }
 192
 193       1;
 194     };
 195
 196     if (!$ok) {
 197       my $error = $EVAL_ERROR;
 198       debug("Exception during execution: ${error}");
 199       notify_on_failure(exception => $error);
 200     }
 201
 202     debug("Sleeping");
 203
 204     my $seconds = 60 - (localtime)[0];
 205     if (!eval {
 206       local $SIG{'ALRM'} = sub {
 207         debug("Got woken up by SIGALRM");
 208         die "Alarm!\n"
 209       };
 210       sleep($seconds < 30 ? $seconds + 60 : $seconds);
 211       1;
 212     }) {
 213       die $@ unless $@ eq "Alarm!\n";
 214     }
 215   }
 216 }
 217
 218 sub end_of_request {
 219   $main::lxdebug->show_backtrace();
 220   die <<EOF;
 221 Job called ::end_of_request()!
 222
 223 This usually indicates success but should not be used by background jobs. A
 224 backtrace has been logged. Please tell the job author to have a look at it.
 225 EOF
 226
 227 }
 228
 229 chdir $exe_dir;
 230
 231 mkdir SL::System::TaskServer::PID_BASE() if !-d SL::System::TaskServer::PID_BASE();
 232
 233 my $file = first { -f } ("${exe_dir}/config/kivitendo.conf", "${exe_dir}/config/lx_office.conf", "${exe_dir}/config/kivitendo.conf.default");
 234
 235 die "No configuration file found." unless $file;
 236
 237 $file = File::Spec->abs2rel(Cwd::abs_path($file), Cwd::abs_path($exe_dir));
 238
 239 newdaemon(configfile => $file,
 240           progname   => 'kivitendo-background-jobs',
 241           pidbase    => SL::System::TaskServer::PID_BASE() . '/',
 242           );
 243
 244 1;