From 1ad9f1f651ef3f0ee26d69007dd27e60f99f1f12 Mon Sep 17 00:00:00 2001 From: Vsevolod Stakhov Date: Thu, 12 Nov 2009 19:50:28 +0300 Subject: [PATCH] * New version of rspamd statfiles that can be used in sync * Add convertion routines for converting 1.0 to 1.2 statfiles * Reserve some place in header for future features --- src/statfile.c | 130 +++++++++++++++++++++++++++++++++++++++++++++++-- src/statfile.h | 27 +++++++++- 2 files changed, 151 insertions(+), 6 deletions(-) diff --git a/src/statfile.c b/src/statfile.c index d8eb7635e..4b8fe59b9 100644 --- a/src/statfile.c +++ b/src/statfile.c @@ -27,6 +27,9 @@ #include "statfile.h" #include "main.h" +#define RSPAMD_STATFILE_VERSION {'1', '2'} +#define BACKUP_SUFFIX ".old" + /* Maximum number of statistics files */ #define STATFILES_MAX 255 static void statfile_pool_set_block_common ( @@ -43,12 +46,80 @@ cmpstatfile (const void *a, const void *b) return g_ascii_strcasecmp (s1->filename, s2->filename); } +/* Convert statfile version 1.0 to statfile version 1.2, saving backup */ +struct stat_file_header_10 { + u_char magic[3]; /**< magic signature ('r' 's' 'd') */ + u_char version[2]; /**< version of statfile */ + u_char padding[3]; /**< padding */ + uint64_t create_time; /**< create time (time_t->uint64_t) */ +}; + +static gboolean +convert_statfile_10 (stat_file_t * file) +{ + char *backup_name; + struct stat st; + struct stat_file_header header = { + .magic = {'r', 's', 'd'}, + .version = RSPAMD_STATFILE_VERSION, + .padding = {0, 0, 0}, + .revision = 0, + .rev_time = 0 + }; + + + /* Format backup name */ + backup_name = g_strdup_printf ("%s.%s", file->filename, BACKUP_SUFFIX); + + msg_info ("convert_statfile_10: convert old statfile %s to version %c.%c, backup in %s", file->filename, + header.version[0], header.version[1], backup_name); + + if (stat (backup_name, &st) != -1) { + msg_info ("convert_statfile_10: replace old %s", backup_name); + unlink (backup_name); + } + + rename (file->filename, backup_name); + g_free (backup_name); + + /* XXX: maybe race condition here */ + unlock_file (file->fd, FALSE); + close (file->fd); + if ((file->fd = open (file->filename, O_RDWR | O_TRUNC | O_CREAT, S_IWUSR | S_IRUSR)) == -1) { + msg_info ("convert_statfile_10: cannot create file %s, error %d, %s", file->filename, errno, strerror (errno)); + return FALSE; + } + lock_file (file->fd, FALSE); + /* Now make new header and copy it to new file */ + if (write (file->fd, &header, sizeof (header)) == -1) { + msg_info ("convert_statfile_10: cannot write to file %s, error %d, %s", file->filename, errno, strerror (errno)); + return FALSE; + } + /* Now write old map to new file */ + if (write (file->fd, ((u_char *)file->map + sizeof (struct stat_file_header_10)), + file->len - sizeof (struct stat_file_header_10)) == -1) { + msg_info ("convert_statfile_10: cannot write to file %s, error %d, %s", file->filename, errno, strerror (errno)); + return FALSE; + } + /* Unmap old memory and map new */ + munmap (file->map, file->len); + file->len = file->len + sizeof (struct stat_file_header) - sizeof (struct stat_file_header_10); + if ((file->map = mmap (NULL, file->len, PROT_READ | PROT_WRITE, MAP_SHARED, file->fd, 0)) == MAP_FAILED) { + msg_info ("convert_statfile_10: cannot mmap file %s, error %d, %s", file->filename, errno, strerror (errno)); + return FALSE; + } + + return TRUE; +} + /* Check whether specified file is statistic file and calculate its len in blocks */ static int statfile_pool_check (stat_file_t * file) { struct stat_file *f; char *c; + static char valid_version[] = RSPAMD_STATFILE_VERSION; + if (!file || !file->map) { return -1; @@ -62,11 +133,22 @@ statfile_pool_check (stat_file_t * file) f = (struct stat_file *)file->map; c = f->header.magic; /* Check magic and version */ - if (*c++ != 'r' || *c++ != 's' || *c++ != 'd' || - /* version */ *c++ != 1 || *c != 0) { + if (*c++ != 'r' || *c++ != 's' || *c++ != 'd') { msg_info ("statfile_pool_check: file %s is invalid stat file", file->filename); return -1; } + /* Now check version and convert old version to new one (that can be used for sync */ + if (*c == 1 && *(c + 1) == 0) { + if (!convert_statfile_10 (file)) { + return -1; + } + f = (struct stat_file *)file->map; + } + else if (memcmp (c, valid_version, sizeof (valid_version)) != 0) { + /* Unknown version */ + msg_info ("statfile_pool_check: file %s has invalid version %c.%c", file->filename, '0' + *c, '0' + *(c + 1)); + return -1; + } /* Check first section and set new offset */ file->cur_section.code = f->section.code; @@ -223,7 +305,7 @@ statfile_pool_open (statfile_pool_t * pool, char *filename, size_t size, gboolea } memory_pool_lock_mutex (pool->lock); - if (!forced && abs (st.st_size - size) > sizeof (struct stat_file_block)) { + if (!forced && abs (st.st_size - size) > sizeof (struct stat_file)) { memory_pool_unlock_mutex (pool->lock); msg_warn ("statfile_pool_open: need to reindex statfile old size: %zd, new size: %zd", st.st_size, size); return statfile_pool_reindex (pool, filename, st.st_size, size); @@ -259,11 +341,15 @@ statfile_pool_open (statfile_pool_t * pool, char *filename, size_t size, gboolea g_strlcpy (new_file->filename, filename, sizeof (new_file->filename)); new_file->len = st.st_size; + /* Aqquire lock for this operation */ + lock_file (new_file->fd, FALSE); if (statfile_pool_check (new_file) == -1) { pool->opened--; memory_pool_unlock_mutex (pool->lock); + unlock_file (new_file->fd, FALSE); return NULL; } + unlock_file (new_file->fd, FALSE); pool->occupied += st.st_size; new_file->open_time = time (NULL); @@ -316,8 +402,10 @@ statfile_pool_create (statfile_pool_t * pool, char *filename, size_t size) { struct stat_file_header header = { .magic = {'r', 's', 'd'}, - .version = {1, 0}, + .version = RSPAMD_STATFILE_VERSION, .padding = {0, 0, 0}, + .revision = 0, + .rev_time = 0 }; struct stat_file_section section = { .code = STATFILE_SECTION_COMMON, @@ -662,3 +750,37 @@ statfile_get_section_by_name (const char *name) return 0; } + +gboolean +statfile_set_revision (statfile_pool_t *pool, stat_file_t *file, uint64_t rev, time_t time) +{ + struct stat_file_header *header; + + if (pool == NULL || file == NULL || file->map == NULL) { + return FALSE; + } + + header = (struct stat_file_header *)file->map; + + header->revision = rev; + header->rev_time = time; + + return FALSE; +} + +gboolean +statfile_get_revision (statfile_pool_t *pool, stat_file_t *file, uint64_t *rev, time_t *time) +{ + struct stat_file_header *header; + + if (pool == NULL || file == NULL || file->map == NULL) { + return FALSE; + } + + header = (struct stat_file_header *)file->map; + + *rev = header->revision; + *time = header->rev_time; + + return FALSE; +} diff --git a/src/statfile.h b/src/statfile.h index 76efe264f..2269d7ced 100644 --- a/src/statfile.h +++ b/src/statfile.h @@ -23,10 +23,13 @@ */ struct stat_file_header { u_char magic[3]; /**< magic signature ('r' 's' 'd') */ - u_char version[2]; /**< version of statfile (1.0) */ + u_char version[2]; /**< version of statfile */ u_char padding[3]; /**< padding */ uint64_t create_time; /**< create time (time_t->uint64_t) */ -} __attribute__((__packed__)); + uint64_t revision; /**< revision number */ + uint64_t rev_time; /**< revision time */ + u_char unused[255]; /**< some bytes that can be used in future */ +}; /** * Section header @@ -206,4 +209,24 @@ gboolean statfile_pool_add_section (statfile_pool_t *pool, stat_file_t *file, ui */ uint32_t statfile_get_section_by_name (const char *name); +/** + * Set statfile revision and revision time + * @param pool statfile pool object + * @param filename name of statfile + * @param revision number of revision + * @param time time of revision + * @return TRUE if revision was set + */ +gboolean statfile_set_revision (statfile_pool_t *pool, stat_file_t *file, uint64_t rev, time_t time); + +/** + * Set statfile revision and revision time + * @param pool statfile pool object + * @param filename name of statfile + * @param revision saved number of revision + * @param time saved time of revision + * @return TRUE if revision was saved in rev and time + */ +gboolean statfile_get_revision (statfile_pool_t *pool, stat_file_t *file, uint64_t *rev, time_t *time); + #endif -- 2.39.5