path: root/camel/providers/imap/camel-imap-store.c

/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
/* camel-imap-store.c : class for an imap store */

/*
 *  Authors: Jeffrey Stedfast <fejj@helixcode.com>
 *
 *  Copyright 2000 Helix Code, Inc. (www.helixcode.com)
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  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 Street #330, Boston, MA 02111-1307, USA.
 *
 */


#include <config.h>

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>


#include "camel-imap-store.h"
#include "camel-imap-folder.h"
#include "camel-folder.h"
#include "camel-exception.h"
#include "camel-session.h"
#include "camel-stream.h"
#include "camel-stream-buffer.h"
#include "camel-stream-fs.h"
#include "camel-url.h"

/* Specified in RFC 2060 */
#define IMAP_PORT 143

static CamelServiceClass *service_class = NULL;

static void finalize (GtkObject *object);
static gboolean imap_create (CamelFolder *folder, CamelException *ex);
static gboolean imap_connect (CamelService *service, CamelException *ex);
static gboolean imap_disconnect (CamelService *service, CamelException *ex);
static GList *query_auth_types (CamelService *service, CamelException *ex);
static void free_auth_types (CamelService *service, GList *authtypes);
static CamelFolder *get_folder (CamelStore *store, const char *folder_name, gboolean create,
                CamelException *ex);
static char *get_folder_name (CamelStore *store, const char *folder_name, CamelException *ex);
static int camel_imap_status (char *cmdid, char *respbuf);

static void
camel_imap_store_class_init (CamelImapStoreClass *camel_imap_store_class)
{
    /* virtual method overload */
    GtkObjectClass *object_class =
        GTK_OBJECT_CLASS (camel_imap_store_class);
    CamelServiceClass *camel_service_class =
        CAMEL_SERVICE_CLASS (camel_imap_store_class);
    CamelStoreClass *camel_store_class =
        CAMEL_STORE_CLASS (camel_imap_store_class);
    
    service_class = gtk_type_class (camel_service_get_type ());

    /* virtual method overload */
    object_class->finalize = finalize;

    camel_service_class->connect = imap_connect;
    camel_service_class->disconnect = imap_disconnect;
    camel_service_class->query_auth_types = query_auth_types;
    camel_service_class->free_auth_types = free_auth_types;

    camel_store_class->get_folder = get_folder;
    camel_store_class->get_folder_name = get_folder_name;
}

static void
camel_imap_store_init (gpointer object, gpointer klass)
{
    CamelService *service = CAMEL_SERVICE (object);
    CamelStore *store = CAMEL_STORE (object);

    service->url_flags = (CAMEL_SERVICE_URL_NEED_USER | CAMEL_SERVICE_URL_NEED_HOST);

    store->folders = g_hash_table_new (g_str_hash, g_str_equal);
}

GtkType
camel_imap_store_get_type (void)
{
    static GtkType camel_imap_store_type = 0;
    
    if (!camel_imap_store_type) {
        GtkTypeInfo camel_imap_store_info = 
        {
            "CamelImapStore",
            sizeof (CamelImapStore),
            sizeof (CamelImapStoreClass),
            (GtkClassInitFunc) camel_imap_store_class_init,
            (GtkObjectInitFunc) camel_imap_store_init,
                /* reserved_1 */ NULL,
                /* reserved_2 */ NULL,
            (GtkClassInitFunc) NULL,
        };
        
        camel_imap_store_type = gtk_type_unique (CAMEL_STORE_TYPE, &camel_imap_store_info);
    }
    
    return camel_imap_store_type;
}

static void
finalize (GtkObject *object)
{
    CamelException ex;

    camel_exception_init (&ex);
    imap_disconnect (CAMEL_SERVICE (object), &ex);
    camel_exception_clear (&ex);
}

static CamelServiceAuthType password_authtype = {
    "Password",

    "This option will connect to the IMAP server using a "
    "plaintext password.",

    "",
    TRUE
};

static gboolean
try_connect (CamelService *service, CamelException *ex)
{
    struct hostent *h;
    struct sockaddr_in sin;
    gint fd;

    h = camel_service_gethost (service, ex);
    if (!h)
        return FALSE;

    sin.sin_family = h->h_addrtype;
    sin.sin_port = htons (service->url->port ? service->url->port : IMAP_PORT);
    memcpy (&sin.sin_addr, h->h_addr, sizeof (sin.sin_addr));

    fd = socket (h->h_addrtype, SOCK_STREAM, 0);
    if (fd == -1 || connect (fd, (struct sockaddr *)&sin, sizeof(sin)) == -1) {

        /* We don't want to set a CamelException here */

        if (fd > -1)
            close (fd);

        return FALSE;
    }

    close (fd);
    return TRUE;
}

static GList *
query_auth_types (CamelService *service, CamelException *ex)
{
    GList *ret = NULL;
    gboolean passwd = TRUE;


    if (service->url) {
        passwd = try_connect (service, ex);
        if (camel_exception_get_id (ex) != CAMEL_EXCEPTION_NONE)
            return NULL;
    }

    if (passwd)
        ret = g_list_append (ret, &password_authtype);

    if (!ret) {
        camel_exception_setv (ex, CAMEL_EXCEPTION_SERVICE_UNAVAILABLE,
                      "Could not connect to IMAP server on "
                      "%s.", service->url->host);
    }                     

    return ret;
}

static void
free_auth_types (CamelService *service, GList *authtypes)
{
    g_list_free (authtypes);
}

static gboolean
imap_connect (CamelService *service, CamelException *ex)
{
    struct hostent *h;
    struct sockaddr_in sin;
    gint fd, status;
    gchar *buf, *msg;
    CamelImapStore *store = CAMEL_IMAP_STORE (service);


    h = camel_service_gethost (service, ex);
    if (!h)
        return FALSE;

    if (!service->url->authmech && !service->url->passwd) {
        gchar *prompt = g_strdup_printf ("Please enter the IMAP password for %s@%s",
                        service->url->user, h->h_name);
        service->url->passwd =
            camel_session_query_authenticator (camel_service_get_session (service),
                               CAMEL_AUTHENTICATOR_ASK, prompt,
                               TRUE, service, "password", ex);
        g_free (prompt);
        if (!service->url->passwd)
            return FALSE;
    }

    sin.sin_family = h->h_addrtype;
    if (service->url->port)
        sin.sin_port = htons(service->url->port);
    else
        sin.sin_port = htons(IMAP_PORT);

    memcpy (&sin.sin_addr, h->h_addr, sizeof (sin.sin_addr));

    fd = socket (h->h_addrtype, SOCK_STREAM, 0);
    if (fd == -1 || connect (fd, (struct sockaddr *)&sin, sizeof(sin)) == -1) {
        camel_exception_setv (ex, CAMEL_EXCEPTION_SERVICE_UNAVAILABLE,
                      "Could not connect to %s (port %s): %s",
                      service->url->host, service->url->port,
                      strerror(errno));
        if (fd > -1)
            close (fd);

        return FALSE;
    }


    store->ostream = camel_stream_fs_new_with_fd (fd);
    store->istream = camel_stream_buffer_new (store->ostream,
                          CAMEL_STREAM_BUFFER_READ);
    store->command = 0;

    /* Read the greeting, if any. */
    buf = camel_stream_buffer_read_line (CAMEL_STREAM_BUFFER (store->istream));
    if (!buf) {
        camel_exception_setv (ex, CAMEL_EXCEPTION_SERVICE_UNAVAILABLE,
                      "Could not read greeting from IMAP "
                      "server: %s",
                      camel_exception_get_description (ex));
        gtk_object_unref (GTK_OBJECT (store->ostream));
        gtk_object_unref (GTK_OBJECT (store->istream));
        return FALSE;
    }
    g_free (buf);

    status = camel_imap_command(store, NULL, &msg, "LOGIN \"%s\" \"%s\"",
                    service->url->user,
                    service->url->passwd);

    if (status != CAMEL_IMAP_OK) {
        camel_exception_setv (ex, CAMEL_EXCEPTION_SERVICE_CANT_AUTHENTICATE,
                      "Unable to authenticate to IMAP "
                      "server. Error sending password:"
                      " %s", msg ? msg : "(Unknown)");
        g_free (msg);
        gtk_object_unref (GTK_OBJECT (store->ostream));
        gtk_object_unref (GTK_OBJECT (store->istream));
        return FALSE;
    } else {
        g_message ("IMAP Service sucessfully authenticated user %s", service->url->user);
    }

    service_class->connect (service, ex);
    return TRUE;
}

static gboolean
imap_disconnect (CamelService *service, CamelException *ex)
{
    CamelImapStore *store = CAMEL_IMAP_STORE (service);

    if (!service->connected)
        return TRUE;

    if (!service_class->disconnect (service, ex))
        return FALSE;

    gtk_object_unref (GTK_OBJECT (store->ostream));
    gtk_object_unref (GTK_OBJECT (store->istream));
    store->ostream = NULL;
    store->istream = NULL;

    return TRUE;
}

const gchar *
camel_imap_store_get_toplevel_dir (CamelImapStore *store)
{
    CamelURL *url = CAMEL_SERVICE (store)->url;

    g_assert(url != NULL);
    return url->path;
}

static gboolean
imap_create (CamelFolder *folder, CamelException *ex)
{
    gchar *result;
    gint status;

    g_return_val_if_fail (folder != NULL, FALSE);
    
    if (!(folder->full_name || folder->name)) {
        camel_exception_set (ex, CAMEL_EXCEPTION_FOLDER_INVALID,
                     "invalid folder path. Use set_name ?");
        return FALSE;
    }

    if (!strcmp(folder->full_name, "INBOX"))
        return TRUE;
    
    if (camel_folder_get_subfolder(folder->parent_folder, folder->name, FALSE, ex))
        return TRUE;
    
        /* create the directory for the subfolder */
    status = camel_imap_command_extended (CAMEL_IMAP_STORE (folder->parent_store), NULL,
                          &result, "CREATE %s", folder->full_name);

    if (status != CAMEL_IMAP_OK) {
        CamelService *service = CAMEL_SERVICE (folder->parent_store);
        camel_exception_setv (ex, CAMEL_EXCEPTION_SERVICE_UNAVAILABLE,
                      "Could not CREATE %s on IMAP server %s: %s.",
                      folder->full_name, service->url->host,
                      status == CAMEL_IMAP_ERR ? result :
                      "Unknown error");
        g_free (result);
        return FALSE;
    }
    
    g_free(result);

    return TRUE;
}

static CamelFolder *
get_folder (CamelStore *store, const char *folder_name, gboolean create, CamelException *ex)
{
    CamelFolder *new_folder = camel_imap_folder_new (store, ex);

    if (imap_create (new_folder, ex)) {
        return new_folder;
    }
    
    return NULL;
}

static gchar *
get_folder_name (CamelStore *store, const char *folder_name, CamelException *ex)
{
    return g_strdup (folder_name);
}

static int
camel_imap_status (char *cmdid, char *respbuf)
{
    char *retcode;

    if (respbuf) {
        retcode = strstr (respbuf, cmdid);
        if (retcode) {
            retcode += strlen (cmdid) + 1;
            
            if (!strncmp (retcode, "OK", 2))
                return CAMEL_IMAP_OK;
            else if (!strncmp (retcode, "NO", 2))
                return CAMEL_IMAP_ERR;
        }
    }
    
    return CAMEL_IMAP_FAIL;
}

/**
 * camel_imap_command: Send a command to a IMAP server.
 * @store: the IMAP store
 * @ret: a pointer to return the full server response in
 * @fmt: a printf-style format string, followed by arguments
 *
 * This command sends the command specified by @fmt and the following
 * arguments to the connected IMAP store specified by @store. It then
 * reads the server's response and parses out the status code. If
 * the caller passed a non-NULL pointer for @ret, camel_imap_command
 * will set it to point to an buffer containing the rest of the
 * response from the IMAP server. (If @ret was passed but there was
 * no extended response, @ret will be set to NULL.) The caller must
 * free this buffer when it is done with it.
 *
 * Return value: one of CAMEL_IMAP_OK (command executed successfully),
 * CAMEL_IMAP_ERR (command encounted an error), or CAMEL_IMAP_FAIL
 * (a protocol-level error occurred, and Camel is uncertain of the
 * result of the command.)
 **/
gint
camel_imap_command (CamelImapStore *store, CamelFolder *folder, char **ret, char *fmt, ...)
{
    gchar *cmdbuf, *respbuf;
    gchar *cmdid;
    va_list ap;
    gint status;

    if (folder && store->current_folder != folder && strncmp(fmt, "SELECT", 6) &&
        strncmp(fmt, "STATUS", 6) && strncmp(fmt, "CREATE", 5)) {
        /* We need to select the correct mailbox first */
        char *r;
        int s;

        s = camel_imap_command(store, folder, &r, "SELECT %s", folder->full_name);
        if (s != CAMEL_IMAP_OK) {
            *ret = r;
            return s;
        }

        store->current_folder = folder;
    }
    
    /* create the command */
    cmdid = g_strdup_printf("A%.5d", store->command++);
    va_start (ap, fmt);
    cmdbuf = g_strdup_vprintf (fmt, ap);
    va_end (ap);

    fprintf(stderr, "sending : %s %s\r\n", cmdid, cmdbuf);

    if (camel_stream_printf (store->ostream, "%s %s\r\n", cmdid, cmdbuf) == -1) {
        g_free(cmdbuf);
        g_free(cmdid);
        if (*ret)
            *ret = g_strdup(strerror(errno));
        return CAMEL_IMAP_FAIL;
    }
    g_free(cmdbuf);

    /* Read the response */
    respbuf = camel_stream_buffer_read_line (CAMEL_STREAM_BUFFER (store->istream));
    if (respbuf == NULL) {
        if (*ret)
            *ret = g_strdup(strerror(errno));
        return CAMEL_IMAP_FAIL;
    }

    fprintf(stderr, "received: %s\n", respbuf ? respbuf : "(null)");

    status = camel_imap_status (cmdid, respbuf);
    g_free (cmdid);

    if (ret) {
        if (status != CAMEL_IMAP_FAIL) {
            *ret = strchr (respbuf, ' ');
            if (*ret)
                *ret = g_strdup (*ret + 1);
        } else
            *ret = NULL;
    }
    g_free (respbuf);

    return status;
}

/**
 * camel_imap_command_extended: Send a command to a IMAP server and get
 * a multi-line response.
 * @store: the IMAP store
 * @ret: a pointer to return the full server response in
 * @fmt: a printf-style format string, followed by arguments
 *
 * This command sends the command specified by @fmt and the following
 * arguments to the connected IMAP store specified by @store. It then
 * reads the server's response and parses out the status code.
 * Camel_imap_command_extended will set it to point to a buffer containing the
 * response from the IMAP server. (If @ret was passed but there was The caller
 * must free this buffer when it is done with it.
 *
 * This command gets the additional data returned by "multi-line" IMAP
 * commands, such as SELECT, LIST, LSUB, and various other commands.
 * The returned data is un-byte-stuffed, and has lines termined by
 * newlines rather than CR/LF pairs.
 *
 * Return value: one of CAMEL_IMAP_OK (command executed successfully),
 * CAMEL_IMAP_ERR (command encounted an error), or CAMEL_IMAP_FAIL
 * (a protocol-level error occurred, and Camel is uncertain of the
 * result of the command.)
 **/

gint
camel_imap_command_extended (CamelImapStore *store, CamelFolder *folder, char **ret, char *fmt, ...)
{
    CamelStreamBuffer *stream = CAMEL_STREAM_BUFFER (store->istream);
    GPtrArray *data;
    gchar *cmdid, *cmdbuf, *respbuf;
    va_list app;
    gint status = CAMEL_IMAP_OK;

    if (folder && store->current_folder != folder && strncmp(fmt, "SELECT", 6) &&
        strncmp(fmt, "STATUS", 6) && strncmp(fmt, "CREATE", 5)) {
        /* We need to select the correct mailbox first */
        char *r;
        int s;

        s = camel_imap_command(store, folder, &r, "SELECT %s", folder->full_name);
        if (s != CAMEL_IMAP_OK) {
            *ret = r;
            return s;
        }

        store->current_folder = folder;
    }
    
    /* Create the command */
    cmdid = g_strdup_printf("A%.5d", store->command++);
    va_start (app, fmt);
    cmdbuf = g_strdup_vprintf (fmt, app);
    va_end (app);

    fprintf(stderr, "sending : %s %s\r\n", cmdid, cmdbuf);

    if (camel_stream_printf (store->ostream, "%s %s\r\n", cmdid, cmdbuf) == -1) {
        g_free(cmdbuf);
        g_free(cmdid);

        *ret = g_strdup(strerror(errno));

        return CAMEL_IMAP_FAIL;
    }
    g_free(cmdbuf);

    data = g_ptr_array_new ();

    while (1) {
        respbuf = camel_stream_buffer_read_line (stream);
        if (!respbuf || !strncmp(respbuf, cmdid, strlen(cmdid)) ) { 
            /* IMAP's last response starts with our command id */
            break;
        }

        fprintf(stderr, "received: %s\n", respbuf);

        g_ptr_array_add (data, respbuf);
    }

    status = camel_imap_status (cmdid, respbuf);
    g_free (cmdid);

    if (status == CAMEL_IMAP_OK) {
        /* Append an empty string to the end of the array
         * so when we g_strjoinv it, we get a "\n" after
         * the last real line.
         */
        g_ptr_array_add (data, "");
        g_ptr_array_add (data, NULL);
        *ret = g_strjoinv ("\n", (gchar **)data->pdata);
    } else {
        if (status != CAMEL_IMAP_FAIL && respbuf)
                *ret = g_strdup (strchr (respbuf, ' ' + 1));
        else
            *ret = NULL;
    }

    g_ptr_array_free (data, TRUE);
    
    return status;
}

void
camel_imap_store_open (CamelImapStore *store, CamelException *ex)
{
    CamelService *service = CAMEL_SERVICE (store);

    if (!camel_service_is_connected (service))
        imap_connect (service, ex);
}

void
camel_imap_store_close (CamelImapStore *store, gboolean expunge, CamelException *ex)
{
    /*
      if (expunge)
      camel_imap_command (store, NULL, "QUIT");
      else
      camel_imap_command (store, NULL, "RSET");
    */
    
    imap_disconnect (CAMEL_SERVICE (store), ex);
}