package Mail::SpamAssassin::Locker::UnixNFSSafe;
use strict;
use bytes;
use Mail::SpamAssassin;
use Mail::SpamAssassin::Locker;
use Mail::SpamAssassin::Util;
use File::Spec;
use Time::Local;
use Fcntl qw(:DEFAULT :flock);
use vars qw{
@ISA
};
@ISA = qw(Mail::SpamAssassin::Locker);
sub new {
my $class = shift;
my $self = $class->SUPER::new(@_);
$self;
}
use constant LOCK_MAX_AGE => 600;
sub safe_lock {
my ($self, $path, $max_retries) = @_;
my $is_locked = 0;
my @stat;
$max_retries ||= 30;
my $lock_file = "$path.lock";
my $hname = Mail::SpamAssassin::Util::fq_hostname();
my $lock_tmp = Mail::SpamAssassin::Util::untaint_file_path
($path.".lock.".$hname.".".$$);
$self->{lock_tmp} = $lock_tmp;
my $umask = umask 077;
if (!open(LTMP, ">$lock_tmp")) {
umask $umask; die "lock: $$ cannot create tmp lockfile $lock_tmp for $lock_file: $!\n";
}
umask $umask;
autoflush LTMP 1;
dbg("lock: $$ created $lock_tmp");
for (my $retries = 0; $retries < $max_retries; $retries++) {
if ($retries > 0) { $self->jittery_one_second_sleep(); }
print LTMP "$hname.$$\n";
dbg("lock: $$ trying to get lock on $path with $retries retries");
if (link($lock_tmp, $lock_file)) {
dbg("lock: $$ link to $lock_file: link ok");
$is_locked = 1;
last;
}
@stat = lstat($lock_tmp);
if ($stat[3] > 1) {
dbg("lock: $$ link to $lock_file: stat ok");
$is_locked = 1;
last;
}
my $now = ($ @stat = lstat($lock_file);
my $lock_age = ($ if (!defined($lock_age) || ($now - $lock_age) > LOCK_MAX_AGE) {
dbg("lock: $$ breaking stale $lock_file: age=" .
(defined $lock_age ? $lock_age : "undef") . " now=$now");
unlink ($lock_file) || warn "lock: $$ unlink of lock file $lock_file failed: $!\n";
}
}
close(LTMP);
unlink ($lock_tmp) || warn "lock: $$ unlink of temp lock $lock_tmp failed: $!\n";
if ($is_locked) {
@stat = lstat($lock_file);
my $lock_ctime = ($
$self->{lock_ctimes} ||= { };
$self->{lock_ctimes}->{$path} = $lock_ctime;
}
return $is_locked;
}
sub safe_unlock {
my ($self, $path) = @_;
my $lock_file = "$path.lock";
my $lock_tmp = $self->{lock_tmp};
if (!$lock_tmp) {
dbg("unlock: $$ $path.lock never locked");
return;
}
my @stat_ourtmp;
sysopen(LTMP, $lock_tmp, O_CREAT|O_WRONLY|O_EXCL, 0700);
autoflush LTMP 1;
print LTMP "\n";
if (!(@stat_ourtmp = stat(LTMP)) || (scalar(@stat_ourtmp) < 11)) {
warn "unlock: $$ failed to create lock tmpfile $lock_tmp";
close LTMP; unlink $lock_tmp;
return;
}
my $ourtmp_ctime = $stat_ourtmp[10]; if (!defined $ourtmp_ctime) {
die "stat failed on $lock_tmp";
}
close LTMP; unlink $lock_tmp;
my $lock_ctime = $self->{lock_ctimes}->{$path};
if (!defined $lock_ctime) {
warn "unlock: $$ no ctime recorded for $lock_file";
return;
}
my @stat_lock = lstat ($lock_file);
my $now_ctime = $stat_lock[10];
if (defined $now_ctime && $now_ctime == $lock_ctime)
{
unlink ($lock_file) || warn "unlock: $$ unlink failed: $lock_file\n";
dbg("unlock: $$ unlink $lock_file");
if ($ourtmp_ctime >= $lock_ctime + LOCK_MAX_AGE) {
dbg("unlock: $$ lock expired on $lock_file expired safely; sleeping");
my $i; for ($i = 0; $i < 5; $i++) {
$self->jittery_one_second_sleep();
}
}
return;
}
if ($ourtmp_ctime < $lock_ctime + LOCK_MAX_AGE) {
warn "unlock: $$ lock on $lock_file was stolen";
} else {
warn "unlock: $$ lock on $lock_file was lost due to expiry";
}
}
sub refresh_lock {
my($self, $path) = @_;
return unless $path;
my $lock_file = "$path.lock";
utime time, time, $lock_file;
my @stat = lstat($lock_file);
my $lock_ctime = ($ $self->{lock_ctimes}->{$path} = $lock_ctime;
dbg("refresh: $$ refresh $path.lock");
}
sub dbg { Mail::SpamAssassin::dbg (@_); }
1;