123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158 |
- #!/usr/bin/perl
- use DBI;
- use LockFile::Simple qw(lock trylock unlock);
- use Proc::ProcessTable;
- use Data::Dumper qw(Dumper);
- use IPC::Run 'run';
- use String::Util 'trim';
- use File::Temp;
- use Try::Tiny;
- use sigtrap 'handler' => \&sig_handler, qw(INT TERM KILL QUIT);
- my $t = Proc::ProcessTable->new;
- my $imapsync_running = grep { $_->{cmndline} =~ /^\/usr\/bin\/perl \/usr\/local\/bin\/imapsync\s/ } @{$t->table};
- if ($imapsync_running eq 1)
- {
- print "imapsync is active, exiting...";
- exit;
- }
- $run_dir="/tmp";
- $dsn = 'DBI:mysql:database=__DBNAME__;mysql_socket=/var/run/mysqld/mysqld.sock';
- $lock_file = $run_dir . "/imapsync_busy";
- $lockmgr = LockFile::Simple->make(-autoclean => 1, -max => 1);
- $lockmgr->lock($lock_file) || die "can't lock ${lock_file}";
- $dbh = DBI->connect($dsn, '__DBUSER__', '__DBPASS__', {
- mysql_auto_reconnect => 1,
- mysql_enable_utf8mb4 => 1
- });
- sub sig_handler {
- # Send die to force exception in "run"
- die "sig_handler received signal, preparing to exit...\n";
- };
- open my $file, '<', "/etc/sogo/sieve.creds";
- my $creds = <$file>;
- close $file;
- my ($master_user, $master_pass) = split /:/, $creds;
- my $sth = $dbh->prepare("SELECT id,
- user1,
- user2,
- host1,
- authmech1,
- password1,
- exclude,
- port1,
- enc1,
- delete2duplicates,
- maxage,
- subfolder2,
- delete1,
- delete2,
- automap,
- skipcrossduplicates,
- maxbytespersecond,
- custom_params,
- subscribeall,
- timeout1,
- timeout2
- FROM imapsync
- WHERE active = 1
- AND is_running = 0
- AND (
- UNIX_TIMESTAMP(NOW()) - UNIX_TIMESTAMP(last_run) > mins_interval * 60
- OR
- last_run IS NULL)
- ORDER BY last_run");
- $sth->execute();
- my $row;
- while ($row = $sth->fetchrow_arrayref()) {
- $id = @$row[0];
- $user1 = @$row[1];
- $user2 = @$row[2];
- $host1 = @$row[3];
- $authmech1 = @$row[4];
- $password1 = @$row[5];
- $exclude = @$row[6];
- $port1 = @$row[7];
- $enc1 = @$row[8];
- $delete2duplicates = @$row[9];
- $maxage = @$row[10];
- $subfolder2 = @$row[11];
- $delete1 = @$row[12];
- $delete2 = @$row[13];
- $automap = @$row[14];
- $skipcrossduplicates = @$row[15];
- $maxbytespersecond = @$row[16];
- $custom_params = @$row[17];
- $subscribeall = @$row[18];
- $timeout1 = @$row[19];
- $timeout2 = @$row[20];
- if ($enc1 eq "TLS") { $enc1 = "--tls1"; } elsif ($enc1 eq "SSL") { $enc1 = "--ssl1"; } else { undef $enc1; }
- my $template = $run_dir . '/imapsync.XXXXXXX';
- my $passfile1 = File::Temp->new(TEMPLATE => $template);
- my $passfile2 = File::Temp->new(TEMPLATE => $template);
- print $passfile1 "$password1\n";
- print $passfile2 trim($master_pass) . "\n";
- my $command = "/usr/local/bin/imapsync";
- $command .= " --tmpdir /tmp";
- $command .= " --pidfile /tmp/imapsync.pid";
- $command .= " --nofoldersizes";
- ($timeout1 gt "0" ? () : ($command .= " --timeout1 ${timeout1}"));
- ($timeout2 gt "0" ? () : ($command .= " --timeout2 ${timeout2}"));
- ($exclude eq "" ? () : ($command .= qq` --exclude ${exclude}`));
- ($subfolder2 eq "" ? () : ($command .= qq` --subfolder2 ${subfolder2}`));
- ($maxage eq "0" ? () : ($command .= " --maxage ${maxage}"));
- ($maxbytespersecond eq "0" ? () : ($command .= " --maxbytespersecond ${maxbytespersecond}"));
- ($delete2duplicates ne "1" ? () : ($command .= " --delete2duplicates"));
- ($subscribeall ne "1" ? () : ($command .= " --subscribeall"));
- ($delete1 ne "1" ? () : ($command .= " --delete"));
- ($delete2 ne "1" ? () : ($command .= " --delete2"));
- ($automap ne "1" ? () : ($command .= " --automap"));
- ($skipcrossduplicates ne "1" ? () : ($command .= " --skipcrossduplicates"));
- (!defined($enc1) ? () : ($command .= " ${enc1}"));
- $command .= " --host1 ${host1}";
- $command .= qq` --user1 ${user1}`;
- $command .= " --passfile1 $passfile1->filename";
- $command .= " --port1 ${port1}";
- $command .= " --host2 localhost";
- $command .= " --user2 ${user2}" . '*' . trim($master_user);
- $command .= " --passfile2 $passfile2->filename";
- $command .= " --no-modulesversion";
- ($custom_params eq "" ? () : ($command .= qq` ${custom_params}`));
- try {
- $is_running = $dbh->prepare("UPDATE imapsync SET is_running = 1 WHERE id = ?");
- $is_running->bind_param( 1, ${id} );
- $is_running->execute();
- my $stdout = `${command}`
- $update = $dbh->prepare("UPDATE imapsync SET returned_text = ?, last_run = NOW(), is_running = 0 WHERE id = ?");
- $update->bind_param( 1, ${stdout} );
- $update->bind_param( 2, ${id} );
- $update->execute();
- } catch {
- $update = $dbh->prepare("UPDATE imapsync SET returned_text = 'Could not start or finish imapsync' WHERE id = ?");
- $update->bind_param( 1, ${id} );
- $update->execute();
- } finally {
- $update = $dbh->prepare("UPDATE imapsync SET last_run = NOW(), is_running = 0 WHERE id = ?");
- $update->bind_param( 1, ${id} );
- $update->execute();
- $lockmgr->unlock($lock_file);
- };
- }
- $sth->finish();
- $dbh->disconnect();
- $lockmgr->unlock($lock_file);
|