path: root/camel/providers/local/camel-mh-summary.c

/*
 *  Copyright (C) 2000 Ximian Inc.
 *
 *  Authors: Not Zed <notzed@lostzed.mmc.com.au>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of version 2 of the GNU General Public
 * License as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public
 * License along with this program; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include <sys/uio.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>

#include <dirent.h>

#include <ctype.h>

#include "camel-mh-summary.h"
#include <camel/camel-mime-message.h>

#include "camel-private.h"

#define d(x) /*(printf("%s(%d): ", __FILE__, __LINE__),(x))*/

#define CAMEL_MH_SUMMARY_VERSION (0x2000)

static int mh_summary_check(CamelLocalSummary *cls, CamelFolderChangeInfo *changeinfo, CamelException *ex);
static int mh_summary_sync(CamelLocalSummary *cls, gboolean expunge, CamelFolderChangeInfo *changeinfo, CamelException *ex);
/*static int mh_summary_add(CamelLocalSummary *cls, CamelMimeMessage *msg, CamelMessageInfo *info, CamelFolderChangeInfo *, CamelException *ex);*/

static char *mh_summary_next_uid_string(CamelFolderSummary *s);

static void camel_mh_summary_class_init (CamelMhSummaryClass *class);
static void camel_mh_summary_init   (CamelMhSummary *gspaper);
static void camel_mh_summary_finalise   (CamelObject *obj);

#define _PRIVATE(x) (((CamelMhSummary *)(x))->priv)

struct _CamelMhSummaryPrivate {
    char *current_uid;
};

static CamelLocalSummaryClass *parent_class;

CamelType
camel_mh_summary_get_type (void)
{
    static CamelType type = CAMEL_INVALID_TYPE;
    
    if (type == CAMEL_INVALID_TYPE) {
        type = camel_type_register(camel_local_summary_get_type (), "CamelMhSummary",
                       sizeof(CamelMhSummary),
                       sizeof(CamelMhSummaryClass),
                       (CamelObjectClassInitFunc)camel_mh_summary_class_init,
                       NULL,
                       (CamelObjectInitFunc)camel_mh_summary_init,
                       (CamelObjectFinalizeFunc)camel_mh_summary_finalise);
    }
    
    return type;
}

static void
camel_mh_summary_class_init (CamelMhSummaryClass *class)
{
    CamelFolderSummaryClass *sklass = (CamelFolderSummaryClass *) class;
    CamelLocalSummaryClass *lklass = (CamelLocalSummaryClass *)class;

    parent_class = (CamelLocalSummaryClass *)camel_type_get_global_classfuncs(camel_local_summary_get_type ());

    /* override methods */
    sklass->next_uid_string = mh_summary_next_uid_string;

    lklass->check = mh_summary_check;
    lklass->sync = mh_summary_sync;
    /*lklass->add = mh_summary_add;*/
}

static void
camel_mh_summary_init (CamelMhSummary *o)
{
    struct _CamelFolderSummary *s = (CamelFolderSummary *) o;

    o->priv = g_malloc0(sizeof(*o->priv));
    /* set unique file version */
    s->version += CAMEL_MH_SUMMARY_VERSION;
}

static void
camel_mh_summary_finalise(CamelObject *obj)
{
    CamelMhSummary *o = (CamelMhSummary *)obj;

    g_free(o->priv);
}

/**
 * camel_mh_summary_new:
 *
 * Create a new CamelMhSummary object.
 * 
 * Return value: A new #CamelMhSummary object.
 **/
CamelMhSummary  *camel_mh_summary_new   (const char *filename, const char *mhdir, CamelIndex *index)
{
    CamelMhSummary *o = (CamelMhSummary *)camel_object_new(camel_mh_summary_get_type ());

    camel_local_summary_construct((CamelLocalSummary *)o, filename, mhdir, index);
    return o;
}

static char *mh_summary_next_uid_string(CamelFolderSummary *s)
{
    CamelMhSummary *mhs = (CamelMhSummary *)s;
    CamelLocalSummary *cls = (CamelLocalSummary *)s;
    int fd = -1;
    guint32 uid;
    char *name;
    char *uidstr;

    /* if we are working to add an existing file, then use current_uid */
    if (mhs->priv->current_uid) {
        uidstr = g_strdup(mhs->priv->current_uid);
        /* tell the summary of this, so we always append numbers to the end */
        camel_folder_summary_set_uid(s, strtoul(uidstr, NULL, 10)+1);
    } else {
        /* else scan for one - and create it too, to make sure */
        do {
            close(fd);
            uid = camel_folder_summary_next_uid(s);
            name = g_strdup_printf("%s/%u", cls->folder_path, uid);
            /* O_EXCL isn't guaranteed, sigh.  Oh well, bad luck, mh has problems anyway */
            fd = open(name, O_WRONLY|O_CREAT|O_EXCL, 0600);
            g_free(name);
        } while (fd == -1 && errno == EEXIST);

        close(fd);

        uidstr = g_strdup_printf("%u", uid);
    }

    return uidstr;
}

static int camel_mh_summary_add(CamelLocalSummary *cls, const char *name, int forceindex)
{
    CamelMhSummary *mhs = (CamelMhSummary *)cls;
    char *filename = g_strdup_printf("%s/%s", cls->folder_path, name);
    int fd;
    CamelMimeParser *mp;

    d(printf("summarising: %s\n", name));

    fd = open(filename, O_RDONLY);
    if (fd == -1) {
        g_warning ("Cannot summarise/index: %s: %s", filename, strerror (errno));
        g_free(filename);
        return -1;
    }
    mp = camel_mime_parser_new();
    camel_mime_parser_scan_from(mp, FALSE);
    camel_mime_parser_init_with_fd(mp, fd);
    if (cls->index && (forceindex || !camel_index_has_name(cls->index, name))) {
        d(printf("forcing indexing of message content\n"));
        camel_folder_summary_set_index((CamelFolderSummary *)mhs, cls->index);
    } else {
        camel_folder_summary_set_index((CamelFolderSummary *)mhs, NULL);
    }
    mhs->priv->current_uid = (char *)name;
    camel_folder_summary_add_from_parser((CamelFolderSummary *)mhs, mp);
    camel_object_unref((CamelObject *)mp);
    mhs->priv->current_uid = NULL;
    camel_folder_summary_set_index((CamelFolderSummary *)mhs, NULL);
    g_free(filename);
    return 0;
}

static void
remove_summary(char *key, CamelMessageInfo *info, CamelLocalSummary *cls)
{
    d(printf("removing message %s from summary\n", key));
    if (cls->index)
        camel_index_delete_name(cls->index, camel_message_info_uid(info));
    camel_folder_summary_remove((CamelFolderSummary *)cls, info);
    camel_folder_summary_info_free((CamelFolderSummary *)cls, info);
}

static int
sort_uid_cmp(const void *ap, const void *bp)
{
    const CamelMessageInfo
        *a = *((CamelMessageInfo **)ap),
        *b = *((CamelMessageInfo **)bp);
    const char
        *auid = camel_message_info_uid(a),
        *buid = camel_message_info_uid(b);
    int aval = atoi(auid), bval = atoi(buid);

    return (aval < bval) ? -1 : (aval > bval) ? 1 : 0;
}

static int
mh_summary_check(CamelLocalSummary *cls, CamelFolderChangeInfo *changeinfo, CamelException *ex)
{
    DIR *dir;
    struct dirent *d;
    char *p, c;
    CamelMessageInfo *info;
    CamelFolderSummary *s = (CamelFolderSummary *)cls;
    GHashTable *left;
    int i, count;
    int forceindex;

    /* FIXME: Handle changeinfo */

    d(printf("checking summary ...\n"));

    /* scan the directory, check for mail files not in the index, or index entries that
       no longer exist */
    dir = opendir(cls->folder_path);
    if (dir == NULL) {
        camel_exception_setv (ex, CAMEL_EXCEPTION_SYSTEM,
                      _("Cannot open MH directory path: %s: %s"),
                      cls->folder_path, g_strerror (errno));
        return -1;
    }

    /* keeps track of all uid's that have not been processed */
    left = g_hash_table_new(g_str_hash, g_str_equal);
    count = camel_folder_summary_count((CamelFolderSummary *)cls);
    forceindex = count == 0;
    for (i=0;i<count;i++) {
        info = camel_folder_summary_index((CamelFolderSummary *)cls, i);
        if (info) {
            g_hash_table_insert(left, (char *)camel_message_info_uid(info), info);
        }
    }

    while ( (d = readdir(dir)) ) {
        /* FIXME: also run stat to check for regular file */
        p = d->d_name;
        while ( (c = *p++) ) {
            if (!isdigit(c))
                break;
        }
        if (c==0) {
            info = camel_folder_summary_uid((CamelFolderSummary *)cls, d->d_name);
            if (info == NULL || (cls->index && (!camel_index_has_name(cls->index, d->d_name)))) {
                /* need to add this file to the summary */
                if (info != NULL) {
                    g_hash_table_remove(left, camel_message_info_uid(info));
                    camel_folder_summary_remove((CamelFolderSummary *)cls, info);
                    camel_folder_summary_info_free((CamelFolderSummary *)cls, info);
                }
                camel_mh_summary_add(cls, d->d_name, forceindex);
            } else {
                const char *uid = camel_message_info_uid(info);
                CamelMessageInfo *old = g_hash_table_lookup(left, uid);

                if (old) {
                    camel_folder_summary_info_free((CamelFolderSummary *)cls, old);
                    g_hash_table_remove(left, uid);
                }
                camel_folder_summary_info_free((CamelFolderSummary *)cls, info);
            }
        }
    }
    closedir(dir);
    g_hash_table_foreach(left, (GHFunc)remove_summary, cls);
    g_hash_table_destroy(left);

    /* sort the summary based on message number (uid), since the directory order is not useful */
    CAMEL_SUMMARY_LOCK(s, summary_lock);
    qsort(s->messages->pdata, s->messages->len, sizeof(CamelMessageInfo *), sort_uid_cmp);
    CAMEL_SUMMARY_UNLOCK(s, summary_lock);

    return 0;
}

static int
mh_summary_sync_message(CamelLocalSummary *cls, CamelMessageInfo *info, CamelException *ex)
{
    CamelMimeParser *mp;
    const char *xev, *buffer;
    int xevoffset;
    int fd, outfd, len, outlen, ret=0;
    char *name, *tmpname, *xevnew;

    name = g_strdup_printf("%s/%s", cls->folder_path, camel_message_info_uid(info));
    fd = open(name, O_RDWR);
    if (fd == -1)
        return -1;

    mp = camel_mime_parser_new();
    camel_mime_parser_init_with_fd(mp, fd);
    if (camel_mime_parser_step(mp, 0, 0) != CAMEL_MIME_PARSER_STATE_EOF) {
        xev = camel_mime_parser_header(mp, "X-Evolution", &xevoffset);
        d(printf("xev = '%s'\n", xev));
        xevnew = camel_local_summary_encode_x_evolution(cls, info);
        if (xev == NULL
            || camel_local_summary_decode_x_evolution(cls, xev, NULL) == -1
            || strlen(xev)-1 != strlen(xevnew)) {

            d(printf("camel local summary_decode_xev = %d\n", camel_local_summary_decode_x_evolution(cls, xev, NULL)));

            /* need to write a new copy/unlink old */
            tmpname = g_strdup_printf("%s/.tmp.%d.%s", cls->folder_path, getpid(), camel_message_info_uid(info));
            d(printf("old xev was %d %s new xev is %d %s\n", strlen(xev), xev, strlen(xevnew), xevnew));
            d(printf("creating new message %s\n", tmpname));
            outfd = open(tmpname, O_CREAT|O_WRONLY|O_TRUNC, 0600);
            if (outfd != -1) {
                outlen = 0;
                len = camel_local_summary_write_headers(outfd, camel_mime_parser_headers_raw(mp), xevnew, NULL, NULL);
                if (len != -1) {
                    while (outlen != -1 && (len = camel_mime_parser_read(mp, &buffer, 10240)) > 0) {
                        d(printf("camel mime parser read, read %d bytes: %.*s\n", len, len, buffer));
                        do {
                            outlen = write(outfd, buffer, len);
                        } while (outlen == -1 && errno == EINTR);
                    }
                }

                d(printf("len = %d outlen = %d, renaming/finishing\n", len, outlen));
                if (close(outfd) == -1
                    || len == -1
                    || outlen == -1
                    || rename(tmpname, name) == -1) {
                    unlink(tmpname);
                    ret = -1;
                }
            } else {
                g_warning("sync can't create tmp file: %s", strerror (errno));
            }
            g_free(tmpname);
        } else {
            d(printf("stamping in updated X-EV at %d\n", (int)xevoffset));
            /* else, we can just update the flags field */
            lseek(fd, xevoffset+strlen("X-Evolution: "), SEEK_SET);
            do {
                len = write(fd, xevnew, strlen(xevnew));
            } while (len == -1 && errno == EINTR);
            if (len == -1)
                ret = -1;
        }

        g_free(xevnew);
    }

    camel_object_unref((CamelObject *)mp);
    g_free(name);
    return ret;
}

/* sync the summary file with the ondisk files */
static int
mh_summary_sync(CamelLocalSummary *cls, gboolean expunge, CamelFolderChangeInfo *changes, CamelException *ex)
{
    int count, i;
    CamelMessageInfo *info;
    char *name;
    const char *uid;

    d(printf("summary_sync(expunge=%s)\n", expunge?"true":"false"));

    /* we could probably get away without this ... but why not use it, esp if we're going to
       be doing any significant io already */
    if (camel_local_summary_check(cls, changes, ex) == -1)
        return -1;

    count = camel_folder_summary_count((CamelFolderSummary *)cls);
    for (i=count-1;i>=0;i--) {
        info = camel_folder_summary_index((CamelFolderSummary *)cls, i);
        g_assert(info);
        if (expunge && (info->flags & CAMEL_MESSAGE_DELETED)) {
            uid = camel_message_info_uid(info);
            name = g_strdup_printf("%s/%s", cls->folder_path, uid);
            d(printf("deleting %s\n", name));
            if (unlink(name) == 0 || errno==ENOENT) {

                /* FIXME: put this in folder_summary::remove()? */
                if (cls->index)
                    camel_index_delete_name(cls->index, (char *)uid);
                
                camel_folder_change_info_remove_uid(changes, uid);
                camel_folder_summary_remove((CamelFolderSummary *)cls, info);
            }
            g_free(name);
        } else if (info->flags & (CAMEL_MESSAGE_FOLDER_NOXEV|CAMEL_MESSAGE_FOLDER_FLAGGED)) {
            if (mh_summary_sync_message(cls, info, ex) != -1) {
                info->flags &= 0xffff;
            } else {
                g_warning("Problem occured when trying to expunge, ignored");
            }
        }
        camel_folder_summary_info_free((CamelFolderSummary *)cls, info);
    }

    return ((CamelLocalSummaryClass *)parent_class)->sync(cls, expunge, changes, ex);
}