$self->SUPER::close;
}
+# must be called inside a state_dbh transaction with flock held
+sub __cleanup_state {
+ my ($self, $txn_id) = @_;
+ my $user_id = $self->{user_id} // die 'BUG: no {user_id}';
+ $self->{pop3d}->{-state_dbh}->prepare_cached(<<'')->execute($txn_id);
+DELETE FROM deletes WHERE txn_id = ? AND uid_dele = -1
+
+ my $sth = $self->{pop3d}->{-state_dbh}->prepare_cached(<<'');
+SELECT COUNT(*) FROM deletes WHERE user_id = ?
+
+ $sth->execute($user_id);
+ my $nr = $sth->fetchrow_array;
+ if ($nr == 0) {
+ $sth = $self->{pop3d}->{-state_dbh}->prepare_cached(<<'');
+DELETE FROM users WHERE user_id = ?
+
+ $sth->execute($user_id);
+ }
+ $nr;
+}
+
sub cmd_quit {
my ($self) = @_;
if (defined(my $txn_id = $self->{txn_id})) {
if (my $exp = delete $self->{expire}) {
mark_dele($self, $_) for unpack('S*', $exp);
}
+ my $keep = 1;
my $dbh = $self->{pop3d}->{-state_dbh};
my $lk = $self->{pop3d}->lock_for_scope;
- my $sth;
$dbh->begin_work;
- if (defined $self->{txn_max_uid}) {
- $sth = $dbh->prepare_cached(<<'');
+ if (defined(my $max = $self->{txn_max_uid})) {
+ $dbh->prepare_cached(<<'')->execute($max, $txn_id, $max)
UPDATE deletes SET uid_dele = ? WHERE txn_id = ? AND uid_dele < ?
- $sth->execute($self->{txn_max_uid}, $txn_id,
- $self->{txn_max_uid});
+ } else {
+ $keep = $self->__cleanup_state($txn_id);
}
- $sth = $dbh->prepare_cached(<<'');
+ $dbh->prepare_cached(<<'')->execute(time, $user_id) if $keep;
UPDATE users SET last_seen = ? WHERE user_id = ?
- $sth->execute(time, $user_id);
$dbh->commit;
+ # we MUST do txn_id F_UNLCK here inside ->lock_for_scope:
+ $self->{did_quit} = 1;
+ $self->{pop3d}->unlock_mailbox($self);
}
$self->write(\"+OK public-inbox POP3 server signing off\r\n");
$self->close;
my $locked_mb = ('e'x32)."\@$group";
ok($oldc->apop("$locked_mb.0", 'anonymous'), 'APOP to old');
+my $dbh = DBI->connect("dbi:SQLite:dbname=$tmpdir/p3state/db.sqlite3",'','', {
+ AutoCommit => 1,
+ RaiseError => 1,
+ PrintError => 0,
+ sqlite_use_immediate_transaction => 1,
+ sqlite_see_if_its_a_number => 1,
+});
+
{ # locking within the same process
my $x = Net::POP3->new(@old_args);
ok(!$x->apop("$locked_mb.0", 'anonymous'), 'APOP lock failure');
ok(!$np3->apop($mailbox, 'anonymous'), "APOP $mailbox reject");
ok($np3->quit, "QUIT after APOP fail $mailbox");
}
+
+ # we do connect+QUIT bumps to try ensuring non-QUIT disconnects
+ # get processed below:
for my $mailbox ($group, "$group.0") {
my $u = ('f'x32)."\@$mailbox";
+ undef $np3;
+ ok(Net::POP3->new(@np3_args)->quit, 'connect+QUIT bump');
$np3 = Net::POP3->new(@np3_args);
+ my $n0 = $dbh->selectrow_array('SELECT COUNT(*) FROM deletes');
+ my $u0 = $dbh->selectrow_array('SELECT COUNT(*) FROM users');
ok($np3->user($u), "UUID\@$mailbox accept");
ok($np3->pass('anonymous'), 'pass works');
+ my $n1 = $dbh->selectrow_array('SELECT COUNT(*) FROM deletes');
+ is($n1 - $n0, 1, 'deletes bumped while connected');
+ ok($np3->quit, 'client QUIT');
+
+ $n1 = $dbh->selectrow_array('SELECT COUNT(*) FROM deletes');
+ is($n1, $n0, 'deletes row gone on no-op after QUIT');
+ my $u1 = $dbh->selectrow_array('SELECT COUNT(*) FROM users');
+ is($u1, $u0, 'users row gone on no-op after QUIT');
$np3 = Net::POP3->new(@np3_args);
ok($np3->user($u), "UUID\@$mailbox accept");
ok($_ > 0, 'bytes in LIST result') for values %$list;
like($_, qr/\A[a-z0-9]{40,}\z/,
'blob IDs in UIDL result') for values %$uidl;
+ ok($np3->quit, 'QUIT after LIST+UIDL');
+ $n1 = $dbh->selectrow_array('SELECT COUNT(*) FROM deletes');
+ is($n1, $n0, 'deletes row gone on no-op after LIST+UIDL');
+ $n0 = $n1;
+
+ $np3 = Net::POP3->new(@np3_args);
+ ok($np3->user($u), "UUID\@$mailbox accept");
+ ok($np3->pass('anonymous'), 'pass works');
+ undef $np3; # QUIT-less disconnect
+ ok(Net::POP3->new(@np3_args)->quit, 'connect+QUIT bump');
+
+ $u1 = $dbh->selectrow_array('SELECT COUNT(*) FROM users');
+ is($u1, $u0, 'users row gone on QUIT-less disconnect');
+ $n1 = $dbh->selectrow_array('SELECT COUNT(*) FROM deletes');
+ is($n1, $n0, 'deletes row gone on QUIT-less disconnect');
+ $n0 = $n1;
$np3 = Net::POP3->new(@np3_args);
ok(!$np3->apop($u, 'anonumuss'), 'APOP wrong pass reject');
+ $n1 = $dbh->selectrow_array('SELECT COUNT(*) FROM deletes');
+ is($n1, $n0, 'deletes row not bumped w/ wrong pass');
+ undef $np3; # QUIT-less disconnect
+ ok(Net::POP3->new(@np3_args)->quit, 'connect+QUIT bump');
+
+ $n1 = $dbh->selectrow_array('SELECT COUNT(*) FROM deletes');
+ is($n1, $n0, 'deletes row not bumped w/ wrong pass');
$np3 = Net::POP3->new(@np3_args);
ok($np3->apop($u, 'anonymous'), "APOP UUID\@$mailbox");