25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "lib.h"

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "buffer.h"

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "mmap-util.h"

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "read-full.h"

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "write-full.h"

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "mail-index-private.h"

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "mail-index-sync-private.h"

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "mail-transaction-log.h"

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "mail-cache.h"

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include <stdio.h>

49e513d090753ccbf95560b2f3a21f081a5b6c51Timo Sirainen#include <stddef.h>

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include <time.h>

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include <sys/stat.h>

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainenstatic int mail_index_try_open_only(struct mail_index *index);

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen

473080c7c0d25ddfdf77e7dfa0ba8f73c6c669d5Timo Sirainenstruct mail_index *mail_index_alloc(const char *dir, const char *prefix)

473080c7c0d25ddfdf77e7dfa0ba8f73c6c669d5Timo Sirainen{

473080c7c0d25ddfdf77e7dfa0ba8f73c6c669d5Timo Sirainen struct mail_index *index;

473080c7c0d25ddfdf77e7dfa0ba8f73c6c669d5Timo Sirainen

473080c7c0d25ddfdf77e7dfa0ba8f73c6c669d5Timo Sirainen index = i_new(struct mail_index, 1);

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen index->dir = i_strdup(dir);

356303df200c991580bd24041996a070ad08c05eTimo Sirainen index->prefix = i_strdup(prefix);

473080c7c0d25ddfdf77e7dfa0ba8f73c6c669d5Timo Sirainen index->fd = -1;

49e513d090753ccbf95560b2f3a21f081a5b6c51Timo Sirainen

49e513d090753ccbf95560b2f3a21f081a5b6c51Timo Sirainen index->extension_pool = pool_alloconly_create("extension", 256);

473080c7c0d25ddfdf77e7dfa0ba8f73c6c669d5Timo Sirainen index->extensions = buffer_create_dynamic(index->extension_pool, 64);

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen index->sync_handlers = buffer_create_dynamic(default_pool, 64);

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen index->sync_lost_handlers = buffer_create_dynamic(default_pool, 64);

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen index->expunge_handlers = buffer_create_dynamic(default_pool, 32);

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen

49e513d090753ccbf95560b2f3a21f081a5b6c51Timo Sirainen index->mode = 0600;

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen index->gid = (gid_t)-1;

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen return index;

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen}

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainenvoid mail_index_free(struct mail_index *index)

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen{

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen mail_index_close(index);

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen pool_unref(index->extension_pool);

3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen

66d2db642fe24d555d113ba463e446b038d476efTimo Sirainen buffer_free(index->sync_handlers);

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen buffer_free(index->sync_lost_handlers);

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen buffer_free(index->expunge_handlers);

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen

9f431ccfb6932746db56245c8a3d3415717ef545Timo Sirainen i_free(index->error);

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen i_free(index->dir);

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen i_free(index->prefix);

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen i_free(index);

49e513d090753ccbf95560b2f3a21f081a5b6c51Timo Sirainen}

22535a9e685e29214082878e37a267157044618eTimo Sirainen

66d2db642fe24d555d113ba463e446b038d476efTimo Sirainenvoid mail_index_set_permissions(struct mail_index *index,

22535a9e685e29214082878e37a267157044618eTimo Sirainen mode_t mode, gid_t gid)

22535a9e685e29214082878e37a267157044618eTimo Sirainen{

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen index->mode = mode & 0666;

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen index->gid = gid;

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen}

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen

356303df200c991580bd24041996a070ad08c05eTimo Sirainenuint32_t mail_index_ext_register(struct mail_index *index, const char *name,

49e513d090753ccbf95560b2f3a21f081a5b6c51Timo Sirainen uint32_t default_hdr_size,

66d2db642fe24d555d113ba463e446b038d476efTimo Sirainen uint16_t default_record_size,

22535a9e685e29214082878e37a267157044618eTimo Sirainen uint16_t default_record_align)

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen{

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen const struct mail_index_ext *extensions;

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen struct mail_index_ext ext;

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen size_t ext_count;

356303df200c991580bd24041996a070ad08c05eTimo Sirainen unsigned int i;

49e513d090753ccbf95560b2f3a21f081a5b6c51Timo Sirainen

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen extensions = buffer_get_data(index->extensions, &ext_count);

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen ext_count /= sizeof(*extensions);

356303df200c991580bd24041996a070ad08c05eTimo Sirainen

49e513d090753ccbf95560b2f3a21f081a5b6c51Timo Sirainen i_assert(index->sync_handlers->used /

49e513d090753ccbf95560b2f3a21f081a5b6c51Timo Sirainen sizeof(struct mail_index_sync_handler) == ext_count);

66d2db642fe24d555d113ba463e446b038d476efTimo Sirainen

22535a9e685e29214082878e37a267157044618eTimo Sirainen /* see if it's already there */

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen for (i = 0; i < ext_count; i++) {

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen if (strcmp(extensions[i].name, name) == 0)

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen return i;

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen }

22535a9e685e29214082878e37a267157044618eTimo Sirainen

66d2db642fe24d555d113ba463e446b038d476efTimo Sirainen memset(&ext, 0, sizeof(ext));

22535a9e685e29214082878e37a267157044618eTimo Sirainen ext.name = p_strdup(index->extension_pool, name);

22535a9e685e29214082878e37a267157044618eTimo Sirainen ext.index_idx = ext_count;

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen ext.hdr_size = default_hdr_size;

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen ext.record_size = default_record_size;

9f431ccfb6932746db56245c8a3d3415717ef545Timo Sirainen ext.record_align = default_record_align;

9f431ccfb6932746db56245c8a3d3415717ef545Timo Sirainen

9f431ccfb6932746db56245c8a3d3415717ef545Timo Sirainen buffer_append(index->extensions, &ext, sizeof(ext));

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen buffer_append_zero(index->sync_handlers,

66d2db642fe24d555d113ba463e446b038d476efTimo Sirainen sizeof(struct mail_index_sync_handler));

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen return ext_count;

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen}

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen

1f1e81aab38d833d1c9cdc244c91fd762e0080d4Timo Sirainenvoid mail_index_register_expunge_handler(struct mail_index *index,

66d2db642fe24d555d113ba463e446b038d476efTimo Sirainen uint32_t ext_id,

1f1e81aab38d833d1c9cdc244c91fd762e0080d4Timo Sirainen mail_index_expunge_handler_t *cb)

08aea01ef9a9d20703e0fcf8618e6195c0037a44Timo Sirainen{

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen buffer_write(index->expunge_handlers, ext_id * sizeof(cb),

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen &cb, sizeof(cb));

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen}

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen

25757faf029c369a8318349dafe952e2358df1d8Timo Sirainenvoid mail_index_register_sync_handler(struct mail_index *index, uint32_t ext_id,

mail_index_sync_handler_t *cb,

enum mail_index_sync_handler_type type)

{

struct mail_index_sync_handler h;

memset(&h, 0, sizeof(h));

h.callback = cb;

h.type = type;

buffer_write(index->sync_handlers, ext_id * sizeof(h), &h, sizeof(h));

}

void mail_index_register_sync_lost_handler(struct mail_index *index,

mail_index_sync_lost_handler_t *cb)

{

buffer_append(index->sync_lost_handlers, &cb, sizeof(cb));

}

static void mail_index_map_init_extbufs(struct mail_index_map *map,

unsigned int initial_count)

{

size_t ext_size, ext_id_map_size, size;

ext_size = initial_count * sizeof(struct mail_index_ext);

ext_id_map_size = initial_count * sizeof(uint32_t);

if (map->extension_pool == NULL) {

size = ext_size + ext_id_map_size +

(initial_count * 20); /* for names */

map->extension_pool =

pool_alloconly_create("extensions",

nearest_power(size));

} else {

p_clear(map->extension_pool);

}

map->extensions = buffer_create_dynamic(map->extension_pool, ext_size);

map->ext_id_map =

buffer_create_dynamic(map->extension_pool, ext_id_map_size);

}

uint32_t mail_index_map_lookup_ext(struct mail_index_map *map, const char *name)

{

const struct mail_index_ext *extensions;

size_t i, size;

if (map->extensions == NULL)

return (uint32_t)-1;

extensions = buffer_get_data(map->extensions, &size);

size /= sizeof(*extensions);

for (i = 0; i < size; i++) {

if (strcmp(extensions[i].name, name) == 0)

return i;

}

return (uint32_t)-1;

}

mail_index_map_register_ext(struct mail_index *index,

struct mail_index_map *map, const char *name,

uint32_t hdr_offset, uint32_t hdr_size,

uint32_t record_offset, uint32_t record_size,

uint32_t record_align, uint32_t reset_id)

{

struct mail_index_ext *ext;

uint32_t idx, ext_id, empty_id = (uint32_t)-1;

if (map->extensions == NULL) {

mail_index_map_init_extbufs(map, 5);

idx = 0;

} else {

idx = map->extensions->used / sizeof(*ext);

}

ext = buffer_append_space_unsafe(map->extensions, sizeof(*ext));

memset(ext, 0, sizeof(*ext));

ext->name = p_strdup(map->extension_pool, name);

ext->hdr_offset = hdr_offset;

ext->hdr_size = hdr_size;

ext->record_offset = record_offset;

ext->record_size = record_size;

ext->record_align = record_align;

ext->reset_id = reset_id;

ext_id = mail_index_ext_register(index, name, hdr_size,

record_size, record_align);

ext->index_idx = ext_id;

while (map->ext_id_map->used < ext_id * sizeof(uint32_t))

buffer_append(map->ext_id_map, &empty_id, sizeof(empty_id));

buffer_write(map->ext_id_map, ext_id * sizeof(uint32_t),

&idx, sizeof(idx));

return idx;

}

static int size_check(size_t *size_left, size_t size)

{

if (size > *size_left)

return FALSE;

*size_left -= size;

return TRUE;

}

static size_t get_align(size_t name_len)

{

size_t size = sizeof(struct mail_index_ext_header) + name_len;

return MAIL_INDEX_HEADER_SIZE_ALIGN(size) - size;

}

static int mail_index_read_extensions(struct mail_index *index,

struct mail_index_map *map)

{

const struct mail_index_ext_header *ext_hdr;

unsigned int i, old_count;

const char *name;

uint32_t ext_id, offset, name_offset;

size_t size_left;

/* extension headers always start from 64bit offsets, so if base header

offset = MAIL_INDEX_HEADER_SIZE_ALIGN(map->hdr.base_header_size);

if (offset >= map->hdr.header_size && map->extension_pool == NULL) {

/* nothing to do, skip allocatations and all */

return 1;

}

old_count = index->extensions->used / sizeof(struct mail_index_ext);

mail_index_map_init_extbufs(map, old_count + 5);

ext_id = (uint32_t)-1;

for (i = 0; i < old_count; i++)

buffer_append(map->ext_id_map, &ext_id, sizeof(ext_id));

while (offset < map->hdr.header_size) {

ext_hdr = CONST_PTR_OFFSET(map->hdr_base, offset);

size_left = map->hdr.header_size - offset;

if (!size_check(&size_left, sizeof(*ext_hdr)) ||

!size_check(&size_left, ext_hdr->name_size) ||

!size_check(&size_left, get_align(ext_hdr->name_size)) ||

!size_check(&size_left, ext_hdr->hdr_size)) {

mail_index_set_error(index, "Corrupted index file %s: "

"Header extension goes outside header",

index->filepath);

return -1;

}

offset += sizeof(*ext_hdr);

name_offset = offset;

offset += ext_hdr->name_size + get_align(ext_hdr->name_size);

t_push();

name = t_strndup(CONST_PTR_OFFSET(map->hdr_base, name_offset),

ext_hdr->name_size);

mail_index_map_register_ext(index, map, name,

offset, ext_hdr->hdr_size,

ext_hdr->record_offset,

ext_hdr->record_size,

ext_hdr->record_align,

ext_hdr->reset_id);

t_pop();

offset += MAIL_INDEX_HEADER_SIZE_ALIGN(ext_hdr->hdr_size);

}

return 1;

}

static int mail_index_check_header(struct mail_index *index,

struct mail_index_map *map)

{

const struct mail_index_header *hdr = &map->hdr;

unsigned char compat_data[sizeof(hdr->compat_data)];

memset(compat_data, 0, sizeof(compat_data));

#ifndef WORDS_BIGENDIAN

compat_data[0] = MAIL_INDEX_COMPAT_LITTLE_ENDIAN;

#endif

compat_data[1] = sizeof(uoff_t);

compat_data[2] = sizeof(time_t);

if (hdr->major_version != MAIL_INDEX_MAJOR_VERSION) {

/* major version change - handle silently(?) */

return -1;

}

if (memcmp(hdr->compat_data, compat_data, sizeof(compat_data)) != 0) {

/* architecture change - handle silently(?) */

return -1;

}

if ((map->hdr.flags & MAIL_INDEX_HDR_FLAG_CORRUPTED) != 0) {

/* we've already complained about it */

return -1;

}

/* following some extra checks that only take a bit of CPU */

if (hdr->uid_validity == 0 && hdr->next_uid != 1) {

mail_index_set_error(index, "Corrupted index file %s: "

"uid_validity = 0, next_uid = %u",

index->filepath, hdr->next_uid);

return -1;

}

if (hdr->keywords_mask_size != sizeof(keywords_mask_t)) {

mail_index_set_error(index, "Corrupted index file %s: "

"keywords_mask_size mismatch: %d != %d",

index->filepath, hdr->keywords_mask_size,

(int)sizeof(keywords_mask_t));

return -1;

}

if (hdr->record_size < sizeof(struct mail_index_record)) {

mail_index_set_error(index, "Corrupted index file %s: "

"record_size too small: %u < %"PRIuSIZE_T,

index->filepath, hdr->record_size,

sizeof(struct mail_index_record));

return -1;

}

if (hdr->next_uid == 0)

return 0;

if (hdr->recent_messages_count > hdr->messages_count ||

hdr->seen_messages_count > hdr->messages_count ||

hdr->deleted_messages_count > hdr->messages_count)

return 0;

if (hdr->first_recent_uid_lowwater > hdr->next_uid ||

hdr->first_unseen_uid_lowwater > hdr->next_uid ||

hdr->first_deleted_uid_lowwater > hdr->next_uid)

return 0;

return mail_index_read_extensions(index, map);

}

static void mail_index_map_clear(struct mail_index *index,

struct mail_index_map *map)

{

if (map->buffer != NULL) {

i_assert(map->mmap_base == NULL);

buffer_free(map->buffer);

map->buffer = NULL;

} else if (map->mmap_base != NULL) {

i_assert(map->buffer == NULL);

if (munmap(map->mmap_base, map->mmap_size) < 0)

mail_index_set_syscall_error(index, "munmap()");

map->mmap_base = NULL;

}

if (map->refcount > 0) {

memset(&map->hdr, 0, sizeof(map->hdr));

map->mmap_size = 0;

map->mmap_used_size = 0;

map->records = NULL;

map->records_count = 0;

}

}

void mail_index_unmap(struct mail_index *index, struct mail_index_map *map)

{

if (--map->refcount > 0)

return;

i_assert(map->refcount == 0);

mail_index_map_clear(index, map);

if (map->extension_pool != NULL)

pool_unref(map->extension_pool);

buffer_free(map->hdr_copy_buf);

i_free(map);

}

static void mail_index_unmap_forced(struct mail_index *index,

struct mail_index_map *map)

{

mail_index_map_clear(index, map);

mail_index_unmap(index, map);

}

static void mail_index_map_copy_hdr(struct mail_index_map *map,

const struct mail_index_header *hdr)

{

if (hdr->base_header_size < sizeof(map->hdr)) {

/* header smaller than ours, make a copy so our newer headers

memset(&map->hdr, 0, sizeof(map->hdr));

memcpy(&map->hdr, hdr, hdr->base_header_size);

} else {

map->hdr = *hdr;

}

}

static int mail_index_mmap(struct mail_index *index, struct mail_index_map *map)

{

const struct mail_index_header *hdr;

unsigned int records_count;

i_assert(!map->write_to_disk);

if (map->buffer != NULL) {

/* we had temporarily used a buffer, eg. for updating index */

buffer_free(map->buffer);

map->buffer = NULL;

}

map->mmap_base = index->readonly ?

mmap_ro_file(index->fd, &map->mmap_size) :

mmap_rw_file(index->fd, &map->mmap_size);

if (map->mmap_base == MAP_FAILED) {

map->mmap_base = NULL;

mail_index_set_syscall_error(index, "mmap()");

return -1;

}

hdr = map->mmap_base;

if (map->mmap_size >

offsetof(struct mail_index_header, major_version) &&

hdr->major_version != MAIL_INDEX_MAJOR_VERSION) {

/* major version change - handle silently */

return 0;

}

if (map->mmap_size < MAIL_INDEX_HEADER_MIN_SIZE) {

mail_index_set_error(index, "Corrupted index file %s: "

"File too small (%"PRIuSIZE_T")",

index->filepath, map->mmap_size);

return 0;

}

map->mmap_used_size = hdr->header_size +

hdr->messages_count * hdr->record_size;

if (map->mmap_used_size > map->mmap_size) {

records_count = (map->mmap_size - hdr->header_size) /

hdr->record_size;

mail_index_set_error(index, "Corrupted index file %s: "

"messages_count too large (%u > %u)",

index->filepath, hdr->messages_count,

records_count);

return 0;

}

mail_index_map_copy_hdr(map, hdr);

map->hdr_base = map->mmap_base;

map->records = PTR_OFFSET(map->mmap_base, map->hdr.header_size);

map->records_count = map->hdr.messages_count;

return 1;

}

static int mail_index_read_header(struct mail_index *index,

struct mail_index_header *hdr, size_t *pos_r)

{

size_t pos;

int ret;

memset(hdr, 0, sizeof(*hdr));

ret = 1;

for (pos = 0; ret > 0 && pos < sizeof(*hdr); ) {

ret = pread(index->fd, PTR_OFFSET(hdr, pos),

sizeof(*hdr) - pos, pos);

if (ret > 0)

pos += ret;

}

*pos_r = pos;

return ret;

}

static int mail_index_read_map(struct mail_index *index,

struct mail_index_map *map, int *retry_r)

{

struct mail_index_header hdr;

void *data = NULL;

ssize_t ret;

size_t pos, records_size;

i_assert(map->mmap_base == NULL);

*retry_r = FALSE;

ret = mail_index_read_header(index, &hdr, &pos);

if (pos > (ssize_t)offsetof(struct mail_index_header, major_version) &&

hdr.major_version != MAIL_INDEX_MAJOR_VERSION) {

/* major version change - handle silently */

return 0;

}

if (ret >= 0 && pos >= MAIL_INDEX_HEADER_MIN_SIZE &&

(ret > 0 || pos >= hdr.base_header_size)) {

if (hdr.base_header_size < MAIL_INDEX_HEADER_MIN_SIZE ||

hdr.header_size < hdr.base_header_size) {

mail_index_set_error(index, "Corrupted index file %s: "

"Corrupted header sizes (base %u, full %u)",

index->filepath, hdr.base_header_size,

hdr.header_size);

return 0;

}

buffer_reset(map->hdr_copy_buf);

buffer_append(map->hdr_copy_buf, &hdr, hdr.base_header_size);

/* @UNSAFE */

data = buffer_append_space_unsafe(map->hdr_copy_buf,

hdr.header_size -

hdr.base_header_size);

ret = pread_full(index->fd, data,

hdr.header_size - hdr.base_header_size,

hdr.base_header_size);

}

if (ret > 0) {

records_size = hdr.messages_count * hdr.record_size;

if (map->buffer == NULL) {

map->buffer = buffer_create_dynamic(default_pool,

records_size);

}

/* @UNSAFE */

buffer_set_used_size(map->buffer, 0);

data = buffer_append_space_unsafe(map->buffer, records_size);

ret = pread_full(index->fd, data, records_size,

hdr.header_size);

}

if (ret < 0) {

if (errno == ESTALE) {

*retry_r = TRUE;

return 0;

}

mail_index_set_syscall_error(index, "pread_full()");

return -1;

}

if (ret == 0) {

mail_index_set_error(index,

"Corrupted index file %s: File too small",

index->filepath);

return 0;

}

map->records = data;

map->records_count = hdr.messages_count;

mail_index_map_copy_hdr(map, &hdr);

map->hdr_base = map->hdr_copy_buf->data;

index->sync_log_file_seq = hdr.log_file_seq;

index->sync_log_file_offset = hdr.log_file_int_offset;

return 1;

}

static int mail_index_sync_from_transactions(struct mail_index *index,

struct mail_index_map **map,

int sync_to_index)

{

const struct mail_index_header *map_hdr = &(*map)->hdr;

struct mail_index_view *view;

struct mail_transaction_log_view *log_view;

struct mail_index_sync_map_ctx sync_map_ctx;

struct mail_index_header hdr;

const struct mail_transaction_header *thdr;

const void *tdata;

uint32_t prev_seq, max_seq;

uoff_t prev_offset, max_offset;

size_t pos;

int ret, skipped;

if (sync_to_index) {

/* read the real log position where we are supposed to be

ret = mail_index_read_header(index, &hdr, &pos);

if (ret < 0 && errno != ESTALE) {

mail_index_set_syscall_error(index, "pread()");

return -1;

}

if (pos < MAIL_INDEX_HEADER_MIN_SIZE)

return 0;

if (map_hdr->log_file_seq == hdr.log_file_seq &&

map_hdr->log_file_int_offset == hdr.log_file_int_offset) {

/* nothing to do */

return 1;

}

if (map_hdr->log_file_seq > hdr.log_file_seq ||

(map_hdr->log_file_seq == hdr.log_file_seq &&

map_hdr->log_file_int_offset > hdr.log_file_int_offset)) {

/* we went too far, have to re-read the file */

return 0;

}

if (map_hdr->log_file_ext_offset !=

map_hdr->log_file_int_offset ||

hdr.log_file_ext_offset != hdr.log_file_int_offset) {

/* too much trouble to get this right. */

return 0;

}

max_seq = hdr.log_file_seq;

max_offset = hdr.log_file_int_offset;

} else {

/* sync everything there is */

max_seq = (uint32_t)-1;

max_offset = (uoff_t)-1;

}

log_view = mail_transaction_log_view_open(index->log);

if (mail_transaction_log_view_set(log_view,

map_hdr->log_file_seq,

map_hdr->log_file_int_offset,

max_seq, max_offset,

MAIL_TRANSACTION_TYPE_MASK) < 0) {

mail_transaction_log_view_close(log_view);

return 0;

}

index->map = *map;

view = mail_index_view_open(index);

mail_index_sync_map_init(&sync_map_ctx, view,

MAIL_INDEX_SYNC_HANDLER_VIEW);

while ((ret = mail_transaction_log_view_next(log_view, &thdr, &tdata,

&skipped)) > 0) {

if (mail_index_sync_record(&sync_map_ctx, thdr, tdata) < 0) {

ret = -1;

break;

}

}

mail_transaction_log_view_get_prev_pos(log_view, &prev_seq,

&prev_offset);

index->map->hdr.log_file_seq = prev_seq;

index->map->hdr.log_file_int_offset =

index->map->hdr.log_file_ext_offset = prev_offset;

mail_index_sync_map_deinit(&sync_map_ctx);

mail_index_view_close(view);

mail_transaction_log_view_close(log_view);

*map = index->map;

index->map = NULL;

if (sync_to_index) {

/* make sure log file offsets get copied. most of the other

(*map)->hdr = hdr;

}

return ret < 0 ? -1 : 1;

}

static int mail_index_read_map_with_retry(struct mail_index *index,

struct mail_index_map **map,

int sync_to_index)

{

mail_index_sync_lost_handler_t *const *handlers;

size_t size;

unsigned int i;

int ret, retry;

if (index->log_locked) {

/* we're most likely syncing the index and we really don't

sync_to_index = TRUE;

}

if ((*map)->hdr.indexid != 0) {

/* sync this as a view from transaction log. */

index->sync_update = TRUE;

ret = mail_index_sync_from_transactions(index, map,

sync_to_index);

index->sync_update = FALSE;

if (ret != 0)

return ret;

/* transaction log lost/broken, fallback to re-reading it */

}

/* notify all "sync lost" handlers */

handlers = buffer_get_data(index->sync_lost_handlers, &size);

size /= sizeof(*handlers);

for (i = 0; i < size; i++)

(*handlers[i])(index);

for (i = 0; i < MAIL_INDEX_ESTALE_RETRY_COUNT; i++) {

ret = mail_index_read_map(index, *map, &retry);

if (ret != 0 || !retry)

return ret;

/* ESTALE - reopen index file */

if (close(index->fd) < 0)

mail_index_set_syscall_error(index, "close()");

index->fd = -1;

ret = mail_index_try_open_only(index);

if (ret <= 0) {

if (ret == 0) {

/* the file was lost */

errno = ENOENT;

mail_index_set_syscall_error(index, "open()");

}

return -1;

}

}

/* Too many ESTALE retries */

mail_index_set_syscall_error(index, "read_map()");

return -1;

}

static int mail_index_map_try_existing(struct mail_index_map *map)

{

const struct mail_index_header *hdr;

size_t used_size;

if (MAIL_INDEX_MAP_IS_IN_MEMORY(map))

return 0;

hdr = map->mmap_base;

/* always check corrupted-flag to avoid errors later */

if ((hdr->flags & MAIL_INDEX_HDR_FLAG_CORRUPTED) != 0)

return -1;

used_size = hdr->header_size + hdr->messages_count * hdr->record_size;

if (map->mmap_size >= used_size && map->hdr_base == hdr) {

map->records_count = hdr->messages_count;

mail_index_map_copy_hdr(map, hdr);

return 1;

}

return 0;

}

int mail_index_map(struct mail_index *index, int force)

{

struct mail_index_map *map;

int ret;

i_assert(index->map == NULL || index->map->refcount > 0);

i_assert(index->lock_type != F_UNLCK);

if (!force && index->map != NULL) {

i_assert(index->hdr != NULL);

ret = mail_index_map_try_existing(index->map);

if (ret != 0)

return ret;

if (index->lock_type == F_WRLCK) {

/* we're syncing, don't break the mapping */

return 1;

}

}

if (index->map != NULL && index->map->refcount > 1) {

/* this map is already used by some views and they may have

if (!index->mmap_disable) {

map = NULL;

} else {

/* create a copy of the mapping instead so we don't

map = mail_index_map_clone(index->map,

index->map->hdr.record_size);

}

index->map->refcount--;

index->map = NULL;

} else {

map = index->map;

}

if (map == NULL) {

map = i_new(struct mail_index_map, 1);

map->refcount = 1;

map->hdr_copy_buf =

buffer_create_dynamic(default_pool, sizeof(map->hdr));

} else if (MAIL_INDEX_MAP_IS_IN_MEMORY(map)) {

i_assert(!map->write_to_disk);

} else if (map->mmap_base != NULL) {

i_assert(map->buffer == NULL);

if (munmap(map->mmap_base, map->mmap_size) < 0)

mail_index_set_syscall_error(index, "munmap()");

map->mmap_base = NULL;

}

index->hdr = NULL;

index->map = NULL;

if (!index->mmap_disable)

ret = mail_index_mmap(index, map);

else

ret = mail_index_read_map_with_retry(index, &map, force);

i_assert(index->map == NULL);

if (ret <= 0) {

mail_index_unmap_forced(index, map);

return ret;

}

ret = mail_index_check_header(index, map);

if (ret < 0) {

mail_index_unmap_forced(index, map);

return 0;

}

if (ret == 0)

index->fsck = TRUE;

index->hdr = &map->hdr;

index->map = map;

i_assert(map->hdr.messages_count == map->records_count);

return 1;

}

struct mail_index_map *

mail_index_map_clone(struct mail_index_map *map, uint32_t new_record_size)

{

struct mail_index_map *mem_map;

struct mail_index_header *hdr;

struct mail_index_ext *extensions;

void *src, *dest;

size_t size, copy_size;

unsigned int i, count;

size = map->records_count * new_record_size;

mem_map = i_new(struct mail_index_map, 1);

mem_map->refcount = 1;

mem_map->buffer = buffer_create_dynamic(default_pool, size);

if (map->hdr.record_size == new_record_size)

buffer_append(mem_map->buffer, map->records, size);

else {

copy_size = I_MIN(map->hdr.record_size, new_record_size);

src = map->records;

for (i = 0; i < map->records_count; i++) {

dest = buffer_append_space_unsafe(mem_map->buffer,

new_record_size);

memcpy(dest, src, copy_size);

src = PTR_OFFSET(src, map->hdr.record_size);

}

}

mem_map->records = buffer_get_modifyable_data(mem_map->buffer, NULL);

mem_map->records_count = map->records_count;

mem_map->hdr_copy_buf =

buffer_create_dynamic(default_pool, map->hdr.header_size);

if (map->hdr.base_header_size < sizeof(*hdr))

buffer_append_zero(mem_map->hdr_copy_buf, sizeof(*hdr));

buffer_write(mem_map->hdr_copy_buf, 0,

&map->hdr, map->hdr.base_header_size);

buffer_append(mem_map->hdr_copy_buf,

CONST_PTR_OFFSET(map->hdr_base,

map->hdr.base_header_size),

map->hdr.header_size - map->hdr.base_header_size);

hdr = buffer_get_modifyable_data(mem_map->hdr_copy_buf, NULL);

if (hdr->base_header_size < sizeof(*hdr))

hdr->base_header_size = sizeof(*hdr);

hdr->record_size = new_record_size;

mem_map->hdr = *hdr;

mem_map->hdr_base = hdr;

/* copy extensions */

if (map->ext_id_map != NULL) {

count = map->ext_id_map->used / sizeof(uint32_t);

mail_index_map_init_extbufs(mem_map, count + 2);

buffer_append_buf(mem_map->extensions, map->extensions,

0, (size_t)-1);

buffer_append_buf(mem_map->ext_id_map, map->ext_id_map,

0, (size_t)-1);

/* fix the name pointers to use our own pool */

extensions = buffer_get_modifyable_data(mem_map->extensions,

NULL);

for (i = 0; i < count; i++) {

extensions[i].name = p_strdup(mem_map->extension_pool,

extensions[i].name);

}

}

return mem_map;

}

int mail_index_map_get_ext_idx(struct mail_index_map *map,

uint32_t ext_id, uint32_t *idx_r)

{

const uint32_t *id_map;

if (map->ext_id_map == NULL ||

map->ext_id_map->used / sizeof(*id_map) <= ext_id)

return 0;

id_map = map->ext_id_map->data;

*idx_r = id_map[ext_id];

return *idx_r != (uint32_t)-1;

}

static int mail_index_try_open_only(struct mail_index *index)

{

int i;

for (i = 0; i < 3; i++) {

index->fd = open(index->filepath, O_RDWR);

if (index->fd == -1 && errno == EACCES) {

index->fd = open(index->filepath, O_RDONLY);

index->readonly = TRUE;

}

if (index->fd != -1 || errno != ESTALE)

break;

/* May happen with some OSes with NFS. Try again, although

}

if (index->fd == -1) {

if (errno != ENOENT)

return mail_index_set_syscall_error(index, "open()");

/* have to create it */

return 0;

}

return 1;

}

static int

mail_index_try_open(struct mail_index *index, unsigned int *lock_id_r)

{

unsigned int lock_id;

int ret;

if (lock_id_r != NULL)

*lock_id_r = 0;

ret = mail_index_try_open_only(index);

if (ret <= 0)

return ret;

if (mail_index_lock_shared(index, FALSE, &lock_id) < 0) {

(void)close(index->fd);

index->fd = -1;

return -1;

}

ret = mail_index_map(index, FALSE);

if (ret == 0) {

/* it's corrupted - recreate it */

mail_index_unlock(index, lock_id);

if (lock_id_r != NULL)

*lock_id_r = 0;

(void)close(index->fd);

index->fd = -1;

} else {

if (lock_id_r != NULL)

*lock_id_r = lock_id;

else

mail_index_unlock(index, lock_id);

}

return ret;

}

int mail_index_write_base_header(struct mail_index *index,

const struct mail_index_header *hdr)

{

size_t hdr_size;

hdr_size = I_MIN(sizeof(*hdr), hdr->base_header_size);

if (!MAIL_INDEX_MAP_IS_IN_MEMORY(index->map)) {

memcpy(index->map->mmap_base, hdr, hdr_size);

if (msync(index->map->mmap_base, hdr_size, MS_SYNC) < 0)

return mail_index_set_syscall_error(index, "msync()");

} else {

if (pwrite_full(index->fd, hdr, hdr_size, 0) < 0) {

mail_index_set_syscall_error(index, "pwrite_full()");

return -1;

}

index->map->hdr = *hdr;

buffer_write(index->map->hdr_copy_buf, 0, hdr, hdr_size);

}

return 0;

}

int mail_index_create_tmp_file(struct mail_index *index, const char **path_r)

{

mode_t old_mask;

const char *path;

int fd;

path = *path_r = t_strconcat(index->filepath, ".tmp", NULL);

old_mask = umask(0);

fd = open(path, O_RDWR|O_CREAT|O_TRUNC, index->mode);

umask(old_mask);

if (fd == -1)

return mail_index_file_set_syscall_error(index, path, "open()");

if (index->gid != (gid_t)-1 && fchown(fd, (uid_t)-1, index->gid) < 0) {

mail_index_file_set_syscall_error(index, path, "fchown()");

return -1;

}

return fd;

}

static int mail_index_create(struct mail_index *index,

struct mail_index_header *hdr)

{

const char *path;

uint32_t seq;

uoff_t offset;

int ret;

i_assert(index->lock_type == F_UNLCK);

/* log file lock protects index creation */

if (mail_transaction_log_sync_lock(index->log, &seq, &offset) < 0)

return -1;

ret = mail_index_try_open(index, NULL);

if (ret != 0) {

mail_transaction_log_sync_unlock(index->log);

return ret < 0 ? -1 : 0;

}

/* create it fully in index.tmp first */

index->fd = mail_index_create_tmp_file(index, &path);

if (index->fd == -1)

ret = -1;

else if (write_full(index->fd, hdr, sizeof(*hdr)) < 0) {

mail_index_file_set_syscall_error(index, path, "write_full()");

ret = -1;

} else {

index->lock_type = F_WRLCK;

ret = mail_index_map(index, FALSE);

index->lock_type = F_UNLCK;

}

if (ret == 0) {

/* it's corrupted even while we just created it,

mail_index_set_error(index,

"Newly created index file is corrupted: %s", path);

ret = -1;

}

if (ret < 0) {

if (unlink(path) < 0 && errno != ENOENT) {

mail_index_file_set_syscall_error(index, path,

"unlink()");

}

} else {

/* make it visible to others */

if (rename(path, index->filepath) < 0) {

mail_index_set_error(index, "rename(%s, %s) failed: %m",

path, index->filepath);

ret = -1;

}

}

mail_transaction_log_sync_unlock(index->log);

return ret;

}

static void mail_index_header_init(struct mail_index_header *hdr)

{

time_t now = time(NULL);

memset(hdr, 0, sizeof(*hdr));

hdr->major_version = MAIL_INDEX_MAJOR_VERSION;

hdr->minor_version = MAIL_INDEX_MINOR_VERSION;

hdr->base_header_size = sizeof(*hdr);

hdr->header_size = sizeof(*hdr);

hdr->record_size = sizeof(struct mail_index_record);

hdr->keywords_mask_size = sizeof(keywords_mask_t);

#ifndef WORDS_BIGENDIAN

hdr->compat_data[0] = MAIL_INDEX_COMPAT_LITTLE_ENDIAN;

#endif

hdr->compat_data[1] = sizeof(uoff_t);

hdr->compat_data[2] = sizeof(time_t);

hdr->indexid = now;

hdr->next_uid = 1;

}

static int mail_index_open_files(struct mail_index *index,

enum mail_index_open_flags flags)

{

struct mail_index_header hdr;

unsigned int lock_id = 0;

int ret;

ret = mail_index_try_open(index, &lock_id);

if (ret > 0)

hdr = *index->hdr;

else if (ret == 0) {

/* doesn't exist, or corrupted */

if ((flags & MAIL_INDEX_OPEN_FLAG_CREATE) == 0)

return 0;

mail_index_header_init(&hdr);

index->hdr = &hdr;

} else if (ret < 0)

return -1;

index->indexid = hdr.indexid;

index->log = mail_transaction_log_open_or_create(index);

if (index->log == NULL)

return -1;

if (index->fd == -1) {

if (lock_id != 0) {

mail_index_unlock(index, lock_id);

lock_id = 0;

}

if (mail_index_create(index, &hdr) < 0)

return -1;

}

if (lock_id == 0) {

if (mail_index_lock_shared(index, FALSE, &lock_id) < 0)

return -1;

}

index->cache = mail_cache_open_or_create(index);

if (index->cache == NULL)

return -1;

mail_index_unlock(index, lock_id);

return 1;

}

int mail_index_open(struct mail_index *index, enum mail_index_open_flags flags,

enum mail_index_lock_method lock_method)

{

int i = 0, ret;

if (index->opened) {

if (index->hdr != NULL &&

(index->hdr->flags & MAIL_INDEX_HDR_FLAG_CORRUPTED) != 0) {

/* corrupted, reopen files */

mail_index_close(index);

} else {

return 0;

}

}

index->filepath = i_strconcat(index->dir, "/", index->prefix, NULL);

do {

index->shared_lock_count = 0;

index->excl_lock_count = 0;

index->lock_type = F_UNLCK;

index->lock_id = 2;

index->readonly = FALSE;

index->nodiskspace = FALSE;

index->index_lock_timeout = FALSE;

index->log_locked = FALSE;

index->mmap_disable =

(flags & MAIL_INDEX_OPEN_FLAG_MMAP_DISABLE) != 0;

index->mmap_no_write =

(flags & MAIL_INDEX_OPEN_FLAG_MMAP_NO_WRITE) != 0;

index->lock_method = lock_method;

/* don't even bother to handle dotlocking without mmap being

i_assert(lock_method != MAIL_INDEX_LOCK_DOTLOCK ||

index->mmap_disable);

ret = mail_index_open_files(index, flags);

if (ret <= 0)

break;

index->opened = TRUE;

if (index->fsck) {

index->fsck = FALSE;

ret = mail_index_fsck(index);

if (ret == 0) {

/* completely broken, reopen */

if (i++ < 3)

continue;

/* too many tries */

ret = -1;

}

}

break;

} while (1);

if (ret <= 0)

mail_index_close(index);

return ret;

}

void mail_index_close(struct mail_index *index)

{

if (index->log != NULL) {

mail_transaction_log_close(index->log);

index->log = NULL;

}

if (index->map != NULL) {

mail_index_unmap(index, index->map);

index->map = NULL;

}

if (index->cache != NULL) {

mail_cache_free(index->cache);

index->cache = NULL;

}

if (index->fd != -1) {

if (close(index->fd) < 0)

mail_index_set_syscall_error(index, "close()");

index->fd = -1;

}

i_free(index->copy_lock_path);

index->copy_lock_path = NULL;

i_free(index->filepath);

index->filepath = NULL;

index->indexid = 0;

index->opened = FALSE;

}

int mail_index_reopen(struct mail_index *index, int fd)

{

struct mail_index_map *old_map;

unsigned int old_shared_locks, old_lock_id, lock_id = 0;

int ret, old_fd, old_lock_type;

old_map = index->map;

old_fd = index->fd;

old_map->refcount++;

/* new file, new locks. the old fd can keep it's locks, they don't

old_lock_type = index->lock_type;

old_lock_id = index->lock_id;

old_shared_locks = index->shared_lock_count;

if (index->lock_type == F_RDLCK)

index->lock_type = F_UNLCK;

index->lock_id += 2;

index->shared_lock_count = 0;

if (fd != -1) {

index->fd = fd;

ret = 0;

} else {