src/lib-index/mailbox-log.c

bcb4e51a409d94ae670de96afb8483a4f7855294Stephan Bosch/* Copyright (c) 2009-2018 Dovecot authors, see the included COPYING file */
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen#include "lib.h"
8a68b403f982d0f1b2d628ea826ea5385771a787Timo Sirainen#include "ioloop.h"
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen#include "eacces-error.h"
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen#include "mailbox-log.h"
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen#include <stdio.h>
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen#include <unistd.h>
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen#include <fcntl.h>
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen#include <sys/stat.h>
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
b3254619187a08d0ecbdfaf4d251468158287c40Timo Sirainen/* How often to reopen the log file to make sure that the changes are written
b3254619187a08d0ecbdfaf4d251468158287c40Timo Sirainen   to the latest file. The main problem here is if the value is too high the
b3254619187a08d0ecbdfaf4d251468158287c40Timo Sirainen   changes could be written to a file that was already rotated and deleted.
b3254619187a08d0ecbdfaf4d251468158287c40Timo Sirainen   That wouldn't happen in any real world situations though, since the file
b3254619187a08d0ecbdfaf4d251468158287c40Timo Sirainen   rotation time is probably measured in months or years. Still, each session
b3254619187a08d0ecbdfaf4d251468158287c40Timo Sirainen   rarely writes anything here, so the value can just as well be a pretty small
b3254619187a08d0ecbdfaf4d251468158287c40Timo Sirainen   one without any performance problems. */
b3254619187a08d0ecbdfaf4d251468158287c40Timo Sirainen#define MAILBOX_LOG_REOPEN_SECS (60)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen#define MAILBOX_LOG_ROTATE_SIZE (1024*4)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenstruct mailbox_log {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    char *filepath, *filepath2;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    int fd;
8a68b403f982d0f1b2d628ea826ea5385771a787Timo Sirainen    time_t open_timestamp;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    mode_t mode;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    gid_t gid;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    char *gid_origin;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen};
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenstruct mailbox_log_iter {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    struct mailbox_log *log;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    int fd;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    const char *filepath;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    struct mailbox_log_record buf[128];
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    unsigned int idx, count;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    uoff_t offset;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    bool failed;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen};
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenstatic void mailbox_log_close(struct mailbox_log *log);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenstruct mailbox_log *mailbox_log_alloc(const char *path)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen{
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    struct mailbox_log *log;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    log = i_new(struct mailbox_log, 1);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    log->filepath = i_strdup(path);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    log->filepath2 = i_strconcat(path, ".2", NULL);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    log->mode = 0644;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    log->gid = (gid_t)-1;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    log->fd = -1;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    return log;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen}
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenvoid mailbox_log_free(struct mailbox_log **_log)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen{
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    struct mailbox_log *log = *_log;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    *_log = NULL;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    mailbox_log_close(log);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    i_free(log->gid_origin);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    i_free(log->filepath);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    i_free(log->filepath2);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    i_free(log);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen}
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenstatic void mailbox_log_close(struct mailbox_log *log)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen{
a943ed0f901e312445fd393249b91932797bba79Josef 'Jeff' Sipek    i_close_fd_path(&log->fd, log->filepath);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen}
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenvoid mailbox_log_set_permissions(struct mailbox_log *log, mode_t mode,
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen                 gid_t gid, const char *gid_origin)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen{
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    log->mode = mode;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    log->gid = gid;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    i_free(log->gid_origin);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    log->gid_origin = i_strdup(gid_origin);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen}
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenstatic int mailbox_log_open(struct mailbox_log *log)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen{
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    mode_t old_mode;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
cbd6cad82e1c9ad60869ad71ae6c42e1af8b10a8Timo Sirainen    i_assert(log->fd == -1);
cbd6cad82e1c9ad60869ad71ae6c42e1af8b10a8Timo Sirainen
8a68b403f982d0f1b2d628ea826ea5385771a787Timo Sirainen    log->open_timestamp = ioloop_time;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    log->fd = open(log->filepath, O_RDWR | O_APPEND);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    if (log->fd != -1)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        return 0;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    /* try to create it */
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    old_mode = umask(0666 ^ log->mode);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    log->fd = open(log->filepath, O_RDWR | O_APPEND | O_CREAT, 0666);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    umask(old_mode);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    if (log->fd == -1) {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        if (errno != EACCES)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            i_error("creat(%s) failed: %m", log->filepath);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        else
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            i_error("%s", eacces_error_get("creat", log->filepath));
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        return -1;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    }
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    if (fchown(log->fd, (uid_t)-1, log->gid) < 0) {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        if (errno != EPERM)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            i_error("fchown(%s) failed: %m", log->filepath);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        else {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            i_error("%s", eperm_error_get_chgrp("fchown",
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen                        log->filepath, log->gid,
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen                        log->gid_origin));
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        }
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    }
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    return 0;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen}
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenstatic int mailbox_log_rotate_if_needed(struct mailbox_log *log)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen{
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    struct stat st;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    if (fstat(log->fd, &st) < 0) {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        i_error("fstat(%s) failed: %m", log->filepath);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        return -1;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    }
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    if (st.st_size < MAILBOX_LOG_ROTATE_SIZE)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        return 0;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
5e9713e7b02b2b47c339476725f9fbfd9bf78cffTimo Sirainen    if (rename(log->filepath, log->filepath2) < 0 && errno != ENOENT) {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        i_error("rename(%s, %s) failed: %m",
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            log->filepath, log->filepath2);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        return -1;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    }
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    return 0;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen}
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenvoid mailbox_log_record_set_timestamp(struct mailbox_log_record *rec,
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen                      time_t stamp)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen{
b9e830a81455faf3c0dadfc9dbf0c7dc8aca955cJosef 'Jeff' Sipek    cpu32_to_be_unaligned(stamp, rec->timestamp);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen}
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainentime_t mailbox_log_record_get_timestamp(const struct mailbox_log_record *rec)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen{
b9e830a81455faf3c0dadfc9dbf0c7dc8aca955cJosef 'Jeff' Sipek    return (time_t) be32_to_cpu_unaligned(rec->timestamp);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen}
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenint mailbox_log_append(struct mailbox_log *log,
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen               const struct mailbox_log_record *rec)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen{
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    struct stat st;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    ssize_t ret;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    /* we don't have to be too strict about appending to the latest log
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen       file. the records' ordering doesn't matter and iteration goes
8a68b403f982d0f1b2d628ea826ea5385771a787Timo Sirainen       through both logs anyway. still, if there's a long running session
8a68b403f982d0f1b2d628ea826ea5385771a787Timo Sirainen       it shouldn't keep writing to a rotated log forever. */
b3254619187a08d0ecbdfaf4d251468158287c40Timo Sirainen    if (log->open_timestamp/MAILBOX_LOG_REOPEN_SECS !=
b3254619187a08d0ecbdfaf4d251468158287c40Timo Sirainen        ioloop_time/MAILBOX_LOG_REOPEN_SECS)
8a68b403f982d0f1b2d628ea826ea5385771a787Timo Sirainen        mailbox_log_close(log);
cbd6cad82e1c9ad60869ad71ae6c42e1af8b10a8Timo Sirainen    if (log->fd == -1) {
cbd6cad82e1c9ad60869ad71ae6c42e1af8b10a8Timo Sirainen        if (mailbox_log_open(log) < 0)
cbd6cad82e1c9ad60869ad71ae6c42e1af8b10a8Timo Sirainen            return -1;
abb404575a238f27ea03b6049880f30f1656ddc4Timo Sirainen        i_assert(log->fd != -1);
cbd6cad82e1c9ad60869ad71ae6c42e1af8b10a8Timo Sirainen    }
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    /* We don't bother with locking, atomic appends will protect us.
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen       If they don't (NFS), the worst that can happen is that a few
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen       records get overwritten (because they're all the same size).
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen       This whole log isn't supposed to be super-reliable anyway. */
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    ret = write(log->fd, rec, sizeof(*rec));
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    if (ret < 0) {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        i_error("write(%s) failed: %m", log->filepath);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        return -1;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    } else if (ret != sizeof(*rec)) {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        i_error("write(%s) wrote %d/%u bytes", log->filepath,
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            (int)ret, (unsigned int)sizeof(*rec));
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        if (fstat(log->fd, &st) == 0) {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            if (ftruncate(log->fd, st.st_size - ret) < 0) {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen                i_error("ftruncate(%s) failed: %m",
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen                    log->filepath);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            }
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        }
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        return -1;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    }
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    (void)mailbox_log_rotate_if_needed(log);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    return 0;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen}
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenstatic bool mailbox_log_iter_open_next(struct mailbox_log_iter *iter)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen{
a943ed0f901e312445fd393249b91932797bba79Josef 'Jeff' Sipek    i_close_fd_path(&iter->fd, iter->filepath);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    if (iter->filepath == NULL)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        iter->filepath = iter->log->filepath2;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    else if (iter->filepath == iter->log->filepath2)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        iter->filepath = iter->log->filepath;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    else
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        return FALSE;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    iter->fd = open(iter->filepath, O_RDONLY | O_APPEND);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    if (iter->fd != -1)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        return TRUE;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    else if (errno == ENOENT) {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        if (iter->filepath == iter->log->filepath2)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            return mailbox_log_iter_open_next(iter);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    } else {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        i_error("open(%s) failed: %m", iter->filepath);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        iter->failed = TRUE;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    }
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    return FALSE;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen}
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenstruct mailbox_log_iter *mailbox_log_iter_init(struct mailbox_log *log)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen{
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    struct mailbox_log_iter *iter;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    iter = i_new(struct mailbox_log_iter, 1);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    iter->log = log;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    iter->fd = -1;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    (void)mailbox_log_iter_open_next(iter);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    return iter;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen}
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenconst struct mailbox_log_record *
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenmailbox_log_iter_next(struct mailbox_log_iter *iter)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen{
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    const struct mailbox_log_record *rec;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    uoff_t offset;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    ssize_t ret;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    if (iter->idx == iter->count) {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        if (iter->fd == -1)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            return NULL;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        ret = pread(iter->fd, iter->buf, sizeof(iter->buf),
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen                iter->offset);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        if (ret < 0) {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            i_error("pread(%s) failed: %m", iter->filepath);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            iter->failed = TRUE;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            return NULL;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        }
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        if (ret == 0) {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            if (!mailbox_log_iter_open_next(iter))
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen                return NULL;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            iter->idx = iter->count = 0;
3d7590a656f4f746b2adc6e00fa607eba5d2412dTimo Sirainen            iter->offset = 0;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            return mailbox_log_iter_next(iter);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        }
3d7590a656f4f746b2adc6e00fa607eba5d2412dTimo Sirainen        iter->idx = 0;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        iter->count = ret / sizeof(iter->buf[0]);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        iter->offset += iter->count * sizeof(iter->buf[0]);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    }
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    rec = &iter->buf[iter->idx++];
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    if (rec->type < MAILBOX_LOG_RECORD_DELETE_MAILBOX ||
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        rec->type > MAILBOX_LOG_RECORD_UNSUBSCRIBE) {
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        offset = iter->offset -
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen            (iter->count - iter->idx) * sizeof(iter->buf[0]);
3d7590a656f4f746b2adc6e00fa607eba5d2412dTimo Sirainen        i_error("Corrupted mailbox log %s at offset %"PRIuUOFF_T": "
3d7590a656f4f746b2adc6e00fa607eba5d2412dTimo Sirainen            "type=%d", iter->filepath, offset, rec->type);
46b823ac3bce2c0f9f0fc73911e48d3a77b04fbeTimo Sirainen        i_unlink(iter->filepath);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen        return NULL;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    }
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    return rec;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen}
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainenint mailbox_log_iter_deinit(struct mailbox_log_iter **_iter)
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen{
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    struct mailbox_log_iter *iter = *_iter;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    int ret = iter->failed ? -1 : 0;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    *_iter = NULL;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen
a943ed0f901e312445fd393249b91932797bba79Josef 'Jeff' Sipek    i_close_fd_path(&iter->fd, iter->filepath);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    i_free(iter);
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen    return ret;
fb15bb48cf8be41cd94738c868b64608b4fda1adTimo Sirainen}