src/msgcache.c

   1 /*
   2  * Sylpheed -- a GTK+ based, lightweight, and fast e-mail client
   3  * Copyright (C) 1999-2005 Hiroyuki Yamamoto & The Sylpheed Claws Team
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  18  */
  19
  20 #include "defs.h"
  21
  22 #include <glib.h>
  23 #include <glib/gi18n.h>
  24
  25 #include <time.h>
  26
  27 #include <endian.h>
  28
  29 #include "msgcache.h"
  30 #include "utils.h"
  31 #include "procmsg.h"
  32 #include "codeconv.h"
  33
  34 #ifdef __BIG_ENDIAN__
  35 #define bswap_32(x) \
  36      ((((x) & 0xff000000) >> 24) | (((x) & 0x00ff0000) >>  8) | \
  37       (((x) & 0x0000ff00) <<  8) | (((x) & 0x000000ff) << 24))
  38 #else
  39 #define bswap_32(x) (x)
  40 #endif
  41
  42 static gboolean swapping = TRUE;
  43
  44 typedef enum
  45 {
  46         DATA_READ,
  47         DATA_WRITE,
  48         DATA_APPEND
  49 } DataOpenMode;
  50
  51 struct _MsgCache {
  52         GHashTable      *msgnum_table;
  53         GHashTable      *msgid_table;
  54         guint            memusage;
  55         time_t           last_access;
  56 };
  57
  58 typedef struct _StringConverter StringConverter;
  59 struct _StringConverter {
  60         gchar *(*convert) (StringConverter *converter, gchar *srcstr);
  61         void   (*free)    (StringConverter *converter);
  62 };
  63
  64 typedef struct _StrdupConverter StrdupConverter;
  65 struct _StrdupConverter {
  66         StringConverter converter;
  67 };
  68
  69 typedef struct _CharsetConverter CharsetConverter;
  70 struct _CharsetConverter {
  71         StringConverter converter;
  72
  73         gchar *srccharset;
  74         gchar *dstcharset;
  75 };
  76
  77 MsgCache *msgcache_new(void)
  78 {
  79         MsgCache *cache;
  80
  81         cache = g_new0(MsgCache, 1),
  82         cache->msgnum_table = g_hash_table_new(g_int_hash, g_int_equal);
  83         cache->msgid_table = g_hash_table_new(g_str_hash, g_str_equal);
  84         cache->last_access = time(NULL);
  85
  86         return cache;
  87 }
  88
  89 static gboolean msgcache_msginfo_free_func(gpointer num, gpointer msginfo, gpointer user_data)
  90 {
  91         procmsg_msginfo_free((MsgInfo *)msginfo);
  92         return TRUE;
  93 }
  94
  95 void msgcache_destroy(MsgCache *cache)
  96 {
  97         g_return_if_fail(cache != NULL);
  98
  99         g_hash_table_foreach_remove(cache->msgnum_table, msgcache_msginfo_free_func, NULL);
 100         g_hash_table_destroy(cache->msgid_table);
 101         g_hash_table_destroy(cache->msgnum_table);
 102         g_free(cache);
 103 }
 104
 105 void msgcache_add_msg(MsgCache *cache, MsgInfo *msginfo)
 106 {
 107         MsgInfo *newmsginfo;
 108
 109         g_return_if_fail(cache != NULL);
 110         g_return_if_fail(msginfo != NULL);
 111
 112         newmsginfo = procmsg_msginfo_new_ref(msginfo);
 113         g_hash_table_insert(cache->msgnum_table, &newmsginfo->msgnum, newmsginfo);
 114         if(newmsginfo->msgid != NULL)
 115                 g_hash_table_insert(cache->msgid_table, newmsginfo->msgid, newmsginfo);
 116         cache->memusage += procmsg_msginfo_memusage(msginfo);
 117         cache->last_access = time(NULL);
 118
 119         debug_print("Cache size: %d messages, %d bytes\n", g_hash_table_size(cache->msgnum_table), cache->memusage);
 120 }
 121
 122 void msgcache_remove_msg(MsgCache *cache, guint msgnum)
 123 {
 124         MsgInfo *msginfo;
 125
 126         g_return_if_fail(cache != NULL);
 127         g_return_if_fail(msgnum > 0);
 128
 129         msginfo = (MsgInfo *) g_hash_table_lookup(cache->msgnum_table, &msgnum);
 130         if(!msginfo)
 131                 return;
 132
 133         cache->memusage -= procmsg_msginfo_memusage(msginfo);
 134         if(msginfo->msgid)
 135                 g_hash_table_remove(cache->msgid_table, msginfo->msgid);
 136         g_hash_table_remove(cache->msgnum_table, &msginfo->msgnum);
 137         procmsg_msginfo_free(msginfo);
 138         cache->last_access = time(NULL);
 139
 140         debug_print("Cache size: %d messages, %d byte\n", g_hash_table_size(cache->msgnum_table), cache->memusage);
 141 }
 142
 143 void msgcache_update_msg(MsgCache *cache, MsgInfo *msginfo)
 144 {
 145         MsgInfo *oldmsginfo, *newmsginfo;
 146
 147         g_return_if_fail(cache != NULL);
 148         g_return_if_fail(msginfo != NULL);
 149
 150         oldmsginfo = g_hash_table_lookup(cache->msgnum_table, &msginfo->msgnum);
 151         if(oldmsginfo && oldmsginfo->msgid)
 152                 g_hash_table_remove(cache->msgid_table, oldmsginfo->msgid);
 153         if (oldmsginfo) {
 154                 g_hash_table_remove(cache->msgnum_table, &oldmsginfo->msgnum);
 155                 cache->memusage -= procmsg_msginfo_memusage(oldmsginfo);
 156                 procmsg_msginfo_free(oldmsginfo);
 157         }
 158
 159         newmsginfo = procmsg_msginfo_new_ref(msginfo);
 160         g_hash_table_insert(cache->msgnum_table, &newmsginfo->msgnum, newmsginfo);
 161         if(newmsginfo->msgid)
 162                 g_hash_table_insert(cache->msgid_table, newmsginfo->msgid, newmsginfo);
 163         cache->memusage += procmsg_msginfo_memusage(newmsginfo);
 164         cache->last_access = time(NULL);
 165
 166         debug_print("Cache size: %d messages, %d byte\n", g_hash_table_size(cache->msgnum_table), cache->memusage);
 167
 168         return;
 169 }
 170
 171 MsgInfo *msgcache_get_msg(MsgCache *cache, guint num)
 172 {
 173         MsgInfo *msginfo;
 174
 175         g_return_val_if_fail(cache != NULL, NULL);
 176
 177         msginfo = g_hash_table_lookup(cache->msgnum_table, &num);
 178         if(!msginfo)
 179                 return NULL;
 180         cache->last_access = time(NULL);
 181
 182         return procmsg_msginfo_new_ref(msginfo);
 183 }
 184
 185 MsgInfo *msgcache_get_msg_by_id(MsgCache *cache, const gchar *msgid)
 186 {
 187         MsgInfo *msginfo;
 188
 189         g_return_val_if_fail(cache != NULL, NULL);
 190         g_return_val_if_fail(msgid != NULL, NULL);
 191
 192         msginfo = g_hash_table_lookup(cache->msgid_table, msgid);
 193         if(!msginfo)
 194                 return NULL;
 195         cache->last_access = time(NULL);
 196
 197         return procmsg_msginfo_new_ref(msginfo);
 198 }
 199
 200 static void msgcache_get_msg_list_func(gpointer key, gpointer value, gpointer user_data)
 201 {
 202         MsgInfoList **listptr = user_data;
 203         MsgInfo *msginfo = value;
 204
 205         *listptr = g_slist_prepend(*listptr, procmsg_msginfo_new_ref(msginfo));
 206 }
 207
 208 MsgInfoList *msgcache_get_msg_list(MsgCache *cache)
 209 {
 210         MsgInfoList *msg_list = NULL;
 211
 212         g_return_val_if_fail(cache != NULL, NULL);
 213
 214         g_hash_table_foreach((GHashTable *)cache->msgnum_table, msgcache_get_msg_list_func, (gpointer)&msg_list);
 215         cache->last_access = time(NULL);
 216
 217         msg_list = g_slist_reverse(msg_list);
 218
 219         return msg_list;
 220 }
 221
 222 time_t msgcache_get_last_access_time(MsgCache *cache)
 223 {
 224         g_return_val_if_fail(cache != NULL, 0);
 225
 226         return cache->last_access;
 227 }
 228
 229 gint msgcache_get_memory_usage(MsgCache *cache)
 230 {
 231         g_return_val_if_fail(cache != NULL, 0);
 232
 233         return cache->memusage;
 234 }
 235
 236 /*
 237  *  Cache saving functions
 238  */
 239
 240 #define READ_CACHE_DATA(data, fp, total_len) \
 241 { \
 242         if ((tmp_len = msgcache_read_cache_data_str(fp, &data, conv)) < 0) { \
 243                 procmsg_msginfo_free(msginfo); \
 244                 error = TRUE; \
 245                 break; \
 246         } \
 247         total_len += tmp_len; \
 248 }
 249
 250 #define READ_CACHE_DATA_INT(n, fp) \
 251 { \
 252         guint32 idata; \
 253         size_t ni; \
 254  \
 255         if ((ni = fread(&idata, sizeof(idata), 1, fp)) != 1) { \
 256                 g_warning("read_int: Cache data corrupted, read %d of %d at " \
 257                           "offset %ld\n", ni, sizeof(idata), ftell(fp)); \
 258                 procmsg_msginfo_free(msginfo); \
 259                 error = TRUE; \
 260                 break; \
 261         } else \
 262                 n = swapping ? bswap_32(idata) : (idata);\
 263 }
 264
 265 #define WRITE_CACHE_DATA_INT(n, fp)             \
 266 {                                               \
 267         guint32 idata;                          \
 268                                                 \
 269         idata = (guint32)bswap_32(n);                   \
 270         fwrite(&idata, sizeof(idata), 1, fp);   \
 271 }
 272
 273 #define WRITE_CACHE_DATA(data, fp) \
 274 { \
 275         size_t len; \
 276         if (data == NULL) \
 277                 len = 0; \
 278         else \
 279                 len = strlen(data); \
 280         WRITE_CACHE_DATA_INT(len, fp); \
 281         if (len > 0) { \
 282                 fwrite(data, len, 1, fp); \
 283         } \
 284 }
 285
 286 static FILE *msgcache_open_data_file(const gchar *file, guint version,
 287                                      DataOpenMode mode,
 288                                      gchar *buf, size_t buf_size)
 289 {
 290         FILE *fp;
 291         gint32 data_ver;
 292
 293         g_return_val_if_fail(file != NULL, NULL);
 294
 295         if (mode == DATA_WRITE) {
 296                 if ((fp = fopen(file, "wb")) == NULL) {
 297                         FILE_OP_ERROR(file, "fopen");
 298                         return NULL;
 299                 }
 300                 if (change_file_mode_rw(fp, file) < 0)
 301                         FILE_OP_ERROR(file, "chmod");
 302
 303                 WRITE_CACHE_DATA_INT(version, fp);
 304                 return fp;
 305         }
 306
 307         /* check version */
 308         if ((fp = fopen(file, "rb")) == NULL)
 309                 debug_print("Mark/Cache file '%s' not found\n", file);
 310         else {
 311                 if (buf && buf_size > 0)
 312                         setvbuf(fp, buf, _IOFBF, buf_size);
 313                 if (fread(&data_ver, sizeof(data_ver), 1, fp) != 1 ||
 314                          version != bswap_32(data_ver)) {
 315                         g_message("%s: Mark/Cache version is different (%u != %u).\n",
 316                                   file, bswap_32(data_ver), version);
 317                         fclose(fp);
 318                         fp = NULL;
 319                 }
 320                 data_ver = bswap_32(data_ver);
 321         }
 322
 323         if (mode == DATA_READ)
 324                 return fp;
 325
 326         if (fp) {
 327                 /* reopen with append mode */
 328                 fclose(fp);
 329                 if ((fp = fopen(file, "ab")) == NULL)
 330                         FILE_OP_ERROR(file, "fopen");
 331         } else {
 332                 /* open with overwrite mode if mark file doesn't exist or
 333                    version is different */
 334                 fp = msgcache_open_data_file(file, version, DATA_WRITE, buf,
 335                                             buf_size);
 336         }
 337
 338         return fp;
 339 }
 340
 341 static gint msgcache_read_cache_data_str(FILE *fp, gchar **str,
 342                                          StringConverter *conv)
 343 {
 344         gchar *tmpstr = NULL;
 345         size_t ni;
 346         guint32 len;
 347
 348         *str = NULL;
 349         if (!swapping) {
 350                 if ((ni = fread(&len, sizeof(len), 1, fp) != 1) ||
 351                     len > G_MAXINT) {
 352                         g_warning("read_data_str: Cache data (len) corrupted, read %d "
 353                                   "of %d bytes at offset %ld\n", ni, sizeof(len),
 354                                   ftell(fp));
 355                         return -1;
 356                 }
 357         } else {
 358                 if ((ni = fread(&len, sizeof(len), 1, fp) != 1) ||
 359                     bswap_32(len) > G_MAXINT) {
 360                         g_warning("read_data_str: Cache data (len) corrupted, read %d "
 361                                   "of %d bytes at offset %ld\n", ni, sizeof(len),
 362                                   ftell(fp));
 363                         return -1;
 364                 }
 365                 len = bswap_32(len);
 366         }
 367
 368         if (len == 0)
 369                 return 0;
 370
 371         tmpstr = g_malloc(len + 1);
 372
 373         if ((ni = fread(tmpstr, 1, len, fp)) != len) {
 374                 g_warning("read_data_str: Cache data corrupted, read %d of %d "
 375                           "bytes at offset %ld\n",
 376                           ni, len, ftell(fp));
 377                 g_free(tmpstr);
 378                 return -1;
 379         }
 380         tmpstr[len] = 0;
 381
 382         if (conv != NULL) {
 383                 *str = conv->convert(conv, tmpstr);
 384                 g_free(tmpstr);
 385         } else
 386                 *str = tmpstr;
 387
 388         return len;
 389 }
 390
 391 gchar *strconv_strdup_convert(StringConverter *conv, gchar *srcstr)
 392 {
 393         return g_strdup(srcstr);
 394 }
 395
 396 gchar *strconv_charset_convert(StringConverter *conv, gchar *srcstr)
 397 {
 398         CharsetConverter *charsetconv = (CharsetConverter *) conv;
 399
 400         return conv_codeset_strdup(srcstr, charsetconv->srccharset, charsetconv->dstcharset);
 401 }
 402
 403 void strconv_charset_free(StringConverter *conv)
 404 {
 405         CharsetConverter *charsetconv = (CharsetConverter *) conv;
 406
 407         g_free(charsetconv->srccharset);
 408         g_free(charsetconv->dstcharset);
 409 }
 410
 411 MsgCache *msgcache_read_cache(FolderItem *item, const gchar *cache_file)
 412 {
 413         MsgCache *cache;
 414         FILE *fp;
 415         MsgInfo *msginfo;
 416         MsgTmpFlags tmp_flags = 0;
 417         gchar file_buf[BUFFSIZE];
 418         guint32 num;
 419         guint refnum;
 420         gboolean error = FALSE;
 421         StringConverter *conv = NULL;
 422         gchar *srccharset = NULL;
 423         const gchar *dstcharset = NULL;
 424         gchar *ref = NULL;
 425         guint memusage = 0;
 426         guint tmp_len = 0;
 427 #if 0
 428         struct timeval start;
 429         struct timeval end;
 430         struct timeval diff;
 431         gettimeofday(&start, NULL);
 432 #endif
 433         g_return_val_if_fail(cache_file != NULL, NULL);
 434         g_return_val_if_fail(item != NULL, NULL);
 435
 436         swapping = TRUE;
 437
 438         /* In case we can't open the mark file with MARK_VERSION, check if we can open it with the
 439          * swapped MARK_VERSION. As msgcache_open_data_file swaps it too, if this succeeds,
 440          * it means it's the old version (not little-endian) on a big-endian machine. The code has
 441          * no effect on x86 as their file doesn't change. */
 442
 443         if ((fp = msgcache_open_data_file
 444                 (cache_file, CACHE_VERSION, DATA_READ, file_buf, sizeof(file_buf))) == NULL) {
 445                 if ((fp = msgcache_open_data_file
 446                 (cache_file, bswap_32(CACHE_VERSION), DATA_READ, file_buf, sizeof(file_buf))) == NULL)
 447                         return NULL;
 448                 else
 449                         swapping = FALSE;
 450         }
 451
 452         debug_print("\tReading %sswapped message cache from %s...\n", swapping?"":"un", cache_file);
 453
 454         if (item->stype == F_QUEUE) {
 455                 tmp_flags |= MSG_QUEUED;
 456         } else if (item->stype == F_DRAFT) {
 457                 tmp_flags |= MSG_DRAFT;
 458         }
 459
 460         if (msgcache_read_cache_data_str(fp, &srccharset, NULL) < 0)
 461                 return NULL;
 462         dstcharset = CS_UTF_8;
 463         if (srccharset == NULL || dstcharset == NULL) {
 464                 conv = NULL;
 465         } else if (strcmp(srccharset, dstcharset) == 0) {
 466                 StrdupConverter *strdupconv;
 467
 468                 debug_print("using StrdupConverter\n");
 469
 470                 strdupconv = g_new0(StrdupConverter, 1);
 471                 strdupconv->converter.convert = strconv_strdup_convert;
 472                 strdupconv->converter.free = NULL;
 473
 474                 conv = (StringConverter *) strdupconv;
 475         } else {
 476                 CharsetConverter *charsetconv;
 477
 478                 debug_print("using CharsetConverter\n");
 479
 480                 charsetconv = g_new0(CharsetConverter, 1);
 481                 charsetconv->converter.convert = strconv_charset_convert;
 482                 charsetconv->converter.free = strconv_charset_free;
 483                 charsetconv->srccharset = g_strdup(srccharset);
 484                 charsetconv->dstcharset = g_strdup(dstcharset);
 485
 486                 conv = (StringConverter *) charsetconv;
 487         }
 488         g_free(srccharset);
 489
 490         cache = msgcache_new();
 491         g_hash_table_freeze(cache->msgnum_table);
 492         g_hash_table_freeze(cache->msgid_table);
 493
 494         while (fread(&num, sizeof(num), 1, fp) == 1) {
 495                 if (swapping)
 496                         num = bswap_32(num);
 497
 498                 msginfo = procmsg_msginfo_new();
 499                 msginfo->msgnum = num;
 500                 memusage += sizeof(MsgInfo);
 501
 502                 READ_CACHE_DATA_INT(msginfo->size, fp);
 503                 READ_CACHE_DATA_INT(msginfo->mtime, fp);
 504                 READ_CACHE_DATA_INT(msginfo->date_t, fp);
 505                 READ_CACHE_DATA_INT(msginfo->flags.tmp_flags, fp);
 506
 507                 READ_CACHE_DATA(msginfo->fromname, fp, memusage);
 508
 509                 READ_CACHE_DATA(msginfo->date, fp, memusage);
 510                 READ_CACHE_DATA(msginfo->from, fp, memusage);
 511                 READ_CACHE_DATA(msginfo->to, fp, memusage);
 512                 READ_CACHE_DATA(msginfo->cc, fp, memusage);
 513                 READ_CACHE_DATA(msginfo->newsgroups, fp, memusage);
 514                 READ_CACHE_DATA(msginfo->subject, fp, memusage);
 515                 READ_CACHE_DATA(msginfo->msgid, fp, memusage);
 516                 READ_CACHE_DATA(msginfo->inreplyto, fp, memusage);
 517                 READ_CACHE_DATA(msginfo->xref, fp, memusage);
 518
 519                 READ_CACHE_DATA_INT(msginfo->planned_download, fp);
 520                 READ_CACHE_DATA_INT(msginfo->total_size, fp);
 521                 READ_CACHE_DATA_INT(refnum, fp);
 522
 523                 for (; refnum != 0; refnum--) {
 524                         ref = NULL;
 525
 526                         READ_CACHE_DATA(ref, fp, memusage);
 527
 528                         if (ref && strlen(ref))
 529                                 msginfo->references =
 530                                         g_slist_prepend(msginfo->references, ref);
 531                 }
 532                 if (msginfo->references)
 533                         msginfo->references =
 534                                 g_slist_reverse(msginfo->references);
 535
 536                 msginfo->folder = item;
 537                 msginfo->flags.tmp_flags |= tmp_flags;
 538
 539                 g_hash_table_insert(cache->msgnum_table, &msginfo->msgnum, msginfo);
 540                 if(msginfo->msgid)
 541                         g_hash_table_insert(cache->msgid_table, msginfo->msgid, msginfo);
 542         }
 543         fclose(fp);
 544         g_hash_table_thaw(cache->msgnum_table);
 545         g_hash_table_thaw(cache->msgid_table);
 546
 547         if (conv != NULL) {
 548                 if (conv->free != NULL)
 549                         conv->free(conv);
 550                 g_free(conv);
 551         }
 552
 553         if(error) {
 554                 msgcache_destroy(cache);
 555                 return NULL;
 556         }
 557
 558         cache->last_access = time(NULL);
 559         cache->memusage = memusage;
 560
 561         debug_print("done. (%d items read)\n", g_hash_table_size(cache->msgnum_table));
 562         debug_print("Cache size: %d messages, %d byte\n", g_hash_table_size(cache->msgnum_table), cache->memusage);
 563 #if 0
 564         gettimeofday(&end, NULL);
 565         timersub(&end, &start, &diff);
 566         printf("spent %d seconds %d usages %d;%d\n", diff.tv_sec, diff.tv_usec, cache->memusage, memusage);
 567 #endif
 568         return cache;
 569 }
 570
 571 void msgcache_read_mark(MsgCache *cache, const gchar *mark_file)
 572 {
 573         FILE *fp;
 574         MsgInfo *msginfo;
 575         MsgPermFlags perm_flags;
 576         guint32 num;
 577
 578         swapping = TRUE;
 579
 580         /* In case we can't open the mark file with MARK_VERSION, check if we can open it with the
 581          * swapped MARK_VERSION. As msgcache_open_data_file swaps it too, if this succeeds,
 582          * it means it's the old version (not little-endian) on a big-endian machine. The code has
 583          * no effect on x86 as their file doesn't change. */
 584
 585         if ((fp = msgcache_open_data_file(mark_file, MARK_VERSION, DATA_READ, NULL, 0)) == NULL) {
 586                 /* see if it isn't swapped ? */
 587                 if ((fp = msgcache_open_data_file(mark_file, bswap_32(MARK_VERSION), DATA_READ, NULL, 0)) == NULL)
 588                         return;
 589                 else
 590                         swapping = FALSE; /* yay */
 591         }
 592         debug_print("reading %sswapped mark file.\n", swapping?"":"un");
 593         while (fread(&num, sizeof(num), 1, fp) == 1) {
 594                 if (swapping)
 595                         num = bswap_32(num);
 596                 if (fread(&perm_flags, sizeof(perm_flags), 1, fp) != 1) break;
 597                 if (swapping)
 598                         perm_flags = bswap_32(perm_flags);
 599                 msginfo = g_hash_table_lookup(cache->msgnum_table, &num);
 600                 if(msginfo) {
 601                         msginfo->flags.perm_flags = perm_flags;
 602                 }
 603         }
 604         fclose(fp);
 605 }
 606
 607 void msgcache_write_cache(MsgInfo *msginfo, FILE *fp)
 608 {
 609         MsgTmpFlags flags = msginfo->flags.tmp_flags & MSG_CACHED_FLAG_MASK;
 610         GSList *cur;
 611
 612         WRITE_CACHE_DATA_INT(msginfo->msgnum, fp);
 613         WRITE_CACHE_DATA_INT(msginfo->size, fp);
 614         WRITE_CACHE_DATA_INT(msginfo->mtime, fp);
 615         WRITE_CACHE_DATA_INT(msginfo->date_t, fp);
 616         WRITE_CACHE_DATA_INT(flags, fp);
 617
 618         WRITE_CACHE_DATA(msginfo->fromname, fp);
 619
 620         WRITE_CACHE_DATA(msginfo->date, fp);
 621         WRITE_CACHE_DATA(msginfo->from, fp);
 622         WRITE_CACHE_DATA(msginfo->to, fp);
 623         WRITE_CACHE_DATA(msginfo->cc, fp);
 624         WRITE_CACHE_DATA(msginfo->newsgroups, fp);
 625         WRITE_CACHE_DATA(msginfo->subject, fp);
 626         WRITE_CACHE_DATA(msginfo->msgid, fp);
 627         WRITE_CACHE_DATA(msginfo->inreplyto, fp);
 628         WRITE_CACHE_DATA(msginfo->xref, fp);
 629         WRITE_CACHE_DATA_INT(msginfo->planned_download, fp);
 630         WRITE_CACHE_DATA_INT(msginfo->total_size, fp);
 631
 632         WRITE_CACHE_DATA_INT(g_slist_length(msginfo->references), fp);
 633
 634         for (cur = msginfo->references; cur != NULL; cur = cur->next) {
 635                 WRITE_CACHE_DATA((gchar *)cur->data, fp);
 636         }
 637 }
 638
 639 static void msgcache_write_flags(MsgInfo *msginfo, FILE *fp)
 640 {
 641         MsgPermFlags flags = msginfo->flags.perm_flags;
 642
 643         WRITE_CACHE_DATA_INT(msginfo->msgnum, fp);
 644         WRITE_CACHE_DATA_INT(flags, fp);
 645 }
 646
 647 struct write_fps
 648 {
 649         FILE *cache_fp;
 650         FILE *mark_fp;
 651 };
 652
 653 static void msgcache_write_func(gpointer key, gpointer value, gpointer user_data)
 654 {
 655         MsgInfo *msginfo;
 656         struct write_fps *write_fps;
 657
 658         msginfo = (MsgInfo *)value;
 659         write_fps = user_data;
 660
 661         msgcache_write_cache(msginfo, write_fps->cache_fp);
 662         msgcache_write_flags(msginfo, write_fps->mark_fp);
 663 }
 664
 665 gint msgcache_write(const gchar *cache_file, const gchar *mark_file, MsgCache *cache)
 666 {
 667         struct write_fps write_fps;
 668
 669         g_return_val_if_fail(cache_file != NULL, -1);
 670         g_return_val_if_fail(mark_file != NULL, -1);
 671         g_return_val_if_fail(cache != NULL, -1);
 672
 673         write_fps.cache_fp = msgcache_open_data_file(cache_file, CACHE_VERSION,
 674                 DATA_WRITE, NULL, 0);
 675         if (write_fps.cache_fp == NULL)
 676                 return -1;
 677
 678         WRITE_CACHE_DATA(CS_UTF_8, write_fps.cache_fp);
 679
 680         write_fps.mark_fp = msgcache_open_data_file(mark_file, MARK_VERSION,
 681                 DATA_WRITE, NULL, 0);
 682         if (write_fps.mark_fp == NULL) {
 683                 fclose(write_fps.cache_fp);
 684                 return -1;
 685         }
 686
 687         debug_print("\tWriting message cache to %s and %s...\n", cache_file, mark_file);
 688
 689         if (change_file_mode_rw(write_fps.cache_fp, cache_file) < 0)
 690                 FILE_OP_ERROR(cache_file, "chmod");
 691
 692         g_hash_table_foreach(cache->msgnum_table, msgcache_write_func, (gpointer)&write_fps);
 693
 694         fclose(write_fps.cache_fp);
 695         fclose(write_fps.mark_fp);
 696
 697         cache->last_access = time(NULL);
 698
 699         debug_print("done.\n");
 700         return 0;
 701 }
 702