/* Copyright (c) 2002-2018 Dovecot authors, see the included COPYING file */
#include "lib.h"
#include "str.h"
#include "strescape.h"
#include "net.h"
#include "write-full.h"
#include "mail-search-build.h"
#include "index-storage.h"
#include "index-mailbox-size.h"
/*
Saving new mails: After transaction is committed and synced, trigger
vsize updating. Lock vsize updates. Check if the message count +
last-indexed-uid are still valid. If they are, add all the missing new
mails. Unlock.
Fetching vsize: Lock vsize updates. Check if the message count +
last-indexed-uid are still valid. If not, set them to zero. Add all
the missing mails. Unlock.
Expunging mails: Check if syncing would expunge any mails. If so, lock the
vsize updates before locking syncing (to avoid deadlocks). Check if the
message count + last-indexed-uid are still valid. If not, unlock vsize and
do nothing else. Otherwise, for each expunged mail whose UID <=
last-indexed-uid, decrease the message count and the vsize in memory. After
syncing is successfully committed, write the changes to header. Unlock.
Note that the final expunge handling with some mailbox formats is done while
syncing is no longer locked. Because of this we need to have the vsize
locking. The final vsize header update requires committing a transaction,
which internally is the same as a sync lock. So to avoid deadlocks we always
need to lock vsize updates before sync.
*/
#define VSIZE_LOCK_SUFFIX "dovecot-vsize.lock"
#define VSIZE_UPDATE_MAX_LOCK_SECS 10
#define INDEXER_SOCKET_NAME "indexer"
#define INDEXER_HANDSHAKE "VERSION\tindexer\t1\t0\n"
struct mailbox_vsize_update {
struct mailbox *box;
struct mail_index_view *view;
struct mailbox_index_vsize vsize_hdr, orig_vsize_hdr;
struct file_lock *lock;
bool lock_failed;
bool rebuild;
bool written;
bool finish_in_background;
};
static void vsize_header_refresh(struct mailbox_vsize_update *update)
{
const void *data;
size_t size;
if (update->view != NULL)
mail_index_view_close(&update->view);
(void)mail_index_refresh(update->box->index);
update->view = mail_index_view_open(update->box->index);
mail_index_get_header_ext(update->view, update->box->vsize_hdr_ext_id,
&data, &size);
if (size > 0) {
memcpy(&update->orig_vsize_hdr, data,
I_MIN(size, sizeof(update->orig_vsize_hdr)));
}
if (size == sizeof(update->vsize_hdr))
memcpy(&update->vsize_hdr, data, sizeof(update->vsize_hdr));
else {
if (size != 0) {
mailbox_set_critical(update->box,
"vsize-hdr has invalid size: %"PRIuSIZE_T,
size);
}
update->rebuild = TRUE;
i_zero(&update->vsize_hdr);
}
}
static void
index_mailbox_vsize_check_rebuild(struct mailbox_vsize_update *update)
{
uint32_t seq1, seq2;
if (update->vsize_hdr.highest_uid == 0)
return;
if (!mail_index_lookup_seq_range(update->view, 1,
update->vsize_hdr.highest_uid,
&seq1, &seq2))
seq2 = 0;
if (update->vsize_hdr.message_count != seq2) {
if (update->vsize_hdr.message_count < seq2) {
mailbox_set_critical(update->box,
"vsize-hdr has invalid message-count (%u < %u)",
update->vsize_hdr.message_count, seq2);
} else {
/* some messages have been expunged, rescan */
}
i_zero(&update->vsize_hdr);
update->rebuild = TRUE;
}
}
struct mailbox_vsize_update *
index_mailbox_vsize_update_init(struct mailbox *box)
{
struct mailbox_vsize_update *update;
i_assert(box->opened);
update = i_new(struct mailbox_vsize_update, 1);
update->box = box;
vsize_header_refresh(update);
return update;
}
static bool vsize_update_lock_full(struct mailbox_vsize_update *update,
unsigned int lock_secs)
{
struct mailbox *box = update->box;
const char *error;
int ret;
if (update->lock != NULL)
return TRUE;
if (update->lock_failed)
return FALSE;
if (MAIL_INDEX_IS_IN_MEMORY(box->index))
return FALSE;
ret = mailbox_lock_file_create(box, VSIZE_LOCK_SUFFIX, lock_secs,
&update->lock, &error);
if (ret <= 0) {
/* don't log lock timeouts, because we're somewhat expecting
them. Especially when lock_secs is 0. */
if (ret < 0)
mailbox_set_critical(box, "%s", error);
update->lock_failed = TRUE;
return FALSE;
}
update->rebuild = FALSE;
vsize_header_refresh(update);
index_mailbox_vsize_check_rebuild(update);
return TRUE;
}
bool index_mailbox_vsize_update_try_lock(struct mailbox_vsize_update *update)
{
return vsize_update_lock_full(update, 0);
}
bool index_mailbox_vsize_update_wait_lock(struct mailbox_vsize_update *update)
{
return vsize_update_lock_full(update, VSIZE_UPDATE_MAX_LOCK_SECS);
}
bool index_mailbox_vsize_want_updates(struct mailbox_vsize_update *update)
{
return update->vsize_hdr.highest_uid > 0;
}
static void
index_mailbox_vsize_update_write(struct mailbox_vsize_update *update)
{
struct mail_index_transaction *trans;
if (update->written)
return;
update->written = TRUE;
if (update->rebuild == FALSE &&
memcmp(&update->orig_vsize_hdr, &update->vsize_hdr,
sizeof(update->vsize_hdr)) == 0) {
/* no changes */
return;
}
trans = mail_index_transaction_begin(update->view,
MAIL_INDEX_TRANSACTION_FLAG_EXTERNAL);
mail_index_update_header_ext(trans, update->box->vsize_hdr_ext_id,
0, &update->vsize_hdr,
sizeof(update->vsize_hdr));
(void)mail_index_transaction_commit(&trans);
}
static void index_mailbox_vsize_notify_indexer(struct mailbox *box)
{
string_t *str = t_str_new(256);
const char *path;
int fd;
path = t_strconcat(box->storage->user->set->base_dir,
"/"INDEXER_SOCKET_NAME, NULL);
fd = net_connect_unix(path);
if (fd == -1) {
mailbox_set_critical(box,
"Can't start vsize building on background: "
"net_connect_unix(%s) failed: %m", path);
return;
}
str_append(str, INDEXER_HANDSHAKE);
str_append(str, "APPEND\t0\t");
str_append_tabescaped(str, box->storage->user->username);
str_append_c(str, '\t');
str_append_tabescaped(str, box->vname);
str_append_c(str, '\n');
if (write_full(fd, str_data(str), str_len(str)) < 0) {
mailbox_set_critical(box,
"Can't start vsize building on background: "
"write(%s) failed: %m", path);
}
i_close_fd(&fd);
}
void index_mailbox_vsize_update_deinit(struct mailbox_vsize_update **_update)
{
struct mailbox_vsize_update *update = *_update;
*_update = NULL;
if (update->lock != NULL || update->rebuild)
index_mailbox_vsize_update_write(update);
file_lock_free(&update->lock);
if (update->finish_in_background)
index_mailbox_vsize_notify_indexer(update->box);
mail_index_view_close(&update->view);
i_free(update);
}
void index_mailbox_vsize_hdr_expunge(struct mailbox_vsize_update *update,
uint32_t uid, uoff_t vsize)
{
i_assert(update->lock != NULL);
if (uid > update->vsize_hdr.highest_uid)
return;
if (update->vsize_hdr.message_count == 0) {
mailbox_set_critical(update->box,
"vsize-hdr's message_count shrank below 0");
i_zero(&update->vsize_hdr);
return;
}
update->vsize_hdr.message_count--;
if (update->vsize_hdr.vsize < vsize) {
mailbox_set_critical(update->box,
"vsize-hdr's vsize shrank below 0");
i_zero(&update->vsize_hdr);
return;
}
update->vsize_hdr.vsize -= vsize;
}
static int
index_mailbox_vsize_hdr_add_missing(struct mailbox_vsize_update *update,
bool require_result)
{
struct mailbox_index_vsize *vsize_hdr = &update->vsize_hdr;
struct mailbox_transaction_context *trans;
struct mail_search_context *search_ctx;
struct mail_search_args *search_args;
struct mailbox_status status;
struct mail *mail;
unsigned int mails_left;
uint32_t seq1, seq2;
uoff_t vsize;
int ret = 0;
mailbox_get_open_status(update->box, STATUS_UIDNEXT, &status);
if (vsize_hdr->highest_uid + 1 >= status.uidnext) {
/* nothing to do - we should have usually caught this already
before locking */
return 0;
}
/* note that update->view may be more up-to-date than box->view.
we'll just add whatever new mails are in box->view. if we'll notice
that some of the new mails are missing, we'll need to stop there
since that expunge will be applied later on to the vsize header. */
search_args = mail_search_build_init();
if (!mail_index_lookup_seq_range(update->box->view,
vsize_hdr->highest_uid + 1,
status.uidnext-1, &seq1, &seq2)) {
/* nothing existed, but update uidnext */
vsize_hdr->highest_uid = status.uidnext - 1;
mail_search_args_unref(&search_args);
return 0;
}
mail_search_build_add_seqset(search_args, seq1, seq2);
trans = mailbox_transaction_begin(update->box, 0, "vsize update");
search_ctx = mailbox_search_init(trans, search_args, NULL,
MAIL_FETCH_VIRTUAL_SIZE, NULL);
if (!require_result)
mails_left = 0;
else if (update->box->storage->set->mail_vsize_bg_after_count == 0)
mails_left = UINT_MAX;
else
mails_left = update->box->storage->set->mail_vsize_bg_after_count;
while (mailbox_search_next(search_ctx, &mail)) {
if (mails_left == 0) {
/* if there are any more mails whose vsize can't be
looked up from cache, abort and finish on
background. */
mail->lookup_abort = MAIL_LOOKUP_ABORT_NOT_IN_CACHE;
}
ret = mail_get_virtual_size(mail, &vsize);
mail->lookup_abort = MAIL_LOOKUP_ABORT_NEVER;
if (ret < 0 &&
mailbox_get_last_mail_error(update->box) == MAIL_ERROR_LOOKUP_ABORTED) {
/* abort and finish on background */
i_assert(mails_left == 0);
mail_storage_set_error(update->box->storage, MAIL_ERROR_INUSE,
"Finishing vsize calculation on background");
if (require_result)
update->finish_in_background = TRUE;
break;
}
if (mail->mail_stream_opened || mail->mail_metadata_accessed) {
/* slow vsize lookup */
i_assert(mails_left > 0);
mails_left--;
}
if (ret < 0) {
if (mail->expunged)
continue;
ret = -1;
break;
}
vsize_hdr->vsize += vsize;
vsize_hdr->highest_uid = mail->uid;
vsize_hdr->message_count++;
}
if (mailbox_search_deinit(&search_ctx) < 0)
ret = -1;
mail_search_args_unref(&search_args);
if (ret == 0) {
/* success, cache all */
vsize_hdr->highest_uid = status.uidnext - 1;
} else {
/* search failed, cache only up to highest seen uid */
}
(void)mailbox_transaction_commit(&trans);
return ret;
}
int index_mailbox_get_virtual_size(struct mailbox *box,
struct mailbox_metadata *metadata_r)
{
struct mailbox_vsize_update *update;
struct mailbox_status status;
int ret;
mailbox_get_open_status(box, STATUS_MESSAGES | STATUS_UIDNEXT, &status);
update = index_mailbox_vsize_update_init(box);
if (update->vsize_hdr.highest_uid + 1 == status.uidnext &&
update->vsize_hdr.message_count == status.messages) {
/* up to date */
metadata_r->virtual_size = update->vsize_hdr.vsize;
index_mailbox_vsize_update_deinit(&update);
return 0;
}
/* we need to update it - lock it if possible. if not, update it
anyway internally even though we won't be saving the result. */
(void)index_mailbox_vsize_update_wait_lock(update);
ret = index_mailbox_vsize_hdr_add_missing(update, TRUE);
metadata_r->virtual_size = update->vsize_hdr.vsize;
index_mailbox_vsize_update_deinit(&update);
return ret;
}
int index_mailbox_get_physical_size(struct mailbox *box,
struct mailbox_metadata *metadata_r)
{
struct mailbox_transaction_context *trans;
struct mail_search_context *ctx;
struct mail *mail;
struct mail_search_args *search_args;
uoff_t size;
int ret = 0;
/* if physical size = virtual size always for the storage, we can
use the optimized vsize code for this */
if (box->mail_vfuncs->get_physical_size ==
box->mail_vfuncs->get_virtual_size) {
if (index_mailbox_get_virtual_size(box, metadata_r) < 0)
return -1;
metadata_r->physical_size = metadata_r->virtual_size;
return 0;
}
/* do it the slow way (we could implement similar logic as for vsize,
but for now it's not really needed) */
if (mailbox_sync(box, MAILBOX_SYNC_FLAG_FULL_READ) < 0)
return -1;
trans = mailbox_transaction_begin(box, 0, "mailbox physical size");
search_args = mail_search_build_init();
mail_search_build_add_all(search_args);
ctx = mailbox_search_init(trans, search_args, NULL,
MAIL_FETCH_PHYSICAL_SIZE, NULL);
mail_search_args_unref(&search_args);
metadata_r->physical_size = 0;
while (mailbox_search_next(ctx, &mail)) {
if (mail_get_physical_size(mail, &size) == 0)
metadata_r->physical_size += size;
else {
const char *errstr;
enum mail_error error;
errstr = mailbox_get_last_internal_error(box, &error);
if (error != MAIL_ERROR_EXPUNGED) {
i_error("Couldn't get size of mail UID %u in %s: %s",
mail->uid, box->vname, errstr);
ret = -1;
break;
}
}
}
if (mailbox_search_deinit(&ctx) < 0) {
i_error("Listing mails in %s failed: %s",
box->vname, mailbox_get_last_internal_error(box, NULL));
ret = -1;
}
(void)mailbox_transaction_commit(&trans);
return ret;
}
void index_mailbox_vsize_update_appends(struct mailbox *box)
{
struct mailbox_vsize_update *update;
struct mailbox_status status;
update = index_mailbox_vsize_update_init(box);
/* update here only if we don't need to rebuild the whole vsize. */
index_mailbox_vsize_check_rebuild(update);
if (index_mailbox_vsize_want_updates(update)) {
/* Get the UIDNEXT only after checking that vsize updating is
even potentially wanted for this mailbox. We especially
don't want to do this with imapc, because it could trigger
a remote STATUS (UIDNEXT) call. */
mailbox_get_open_status(update->box, STATUS_UIDNEXT, &status);
if (update->vsize_hdr.highest_uid + 1 != status.uidnext &&
index_mailbox_vsize_update_try_lock(update))
(void)index_mailbox_vsize_hdr_add_missing(update, FALSE);
}
index_mailbox_vsize_update_deinit(&update);
}