Sylpheed

/*

 * Sylpheed -- a GTK+ based, lightweight, and fast e-mail client

 * Copyright (C) 1999-2005 Hiroyuki Yamamoto

 *

 * 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 Place - Suite 330, Boston, MA 02111-1307, USA.

 */

#ifdef HAVE_CONFIG_H

#  include "config.h"

#endif

#include "defs.h"

#include <glib.h>

#include <glib/gi18n.h>

#include <gtk/gtkmain.h>

#include <gtk/gtksignal.h>

#include <gtk/gtkwindow.h>

#include <stdio.h>

#include <string.h>

#include <sys/types.h>

#if HAVE_SYS_WAIT_H

#  include <sys/wait.h>

#endif

#include "send_message.h"

#include "session.h"

#include "ssl.h"

#include "smtp.h"

#include "news.h"

#include "prefs_common.h"

#include "prefs_account.h"

#include "procheader.h"

#include "account.h"

#include "progressdialog.h"

#include "statusbar.h"

#include "inputdialog.h"

#include "alertpanel.h"

#include "manage_window.h"

#include "socket.h"

#include "utils.h"

#define SMTP_PORT        25

#if USE_SSL

#define SSMTP_PORT        465

#endif

typedef struct _SendProgressDialog        SendProgressDialog;

struct _SendProgressDialog

{

        ProgressDialog *dialog;

        Session *session;

        gboolean cancelled;

};

static gint send_message_local                (const gchar                *command,

                                         FILE                        *fp);

static gint send_message_smtp                (PrefsAccount                *ac_prefs,

                                         GSList                        *to_list,

                                         FILE                        *fp);

static gint send_recv_message                (Session                *session,

                                         const gchar                *msg,

                                         gpointer                 data);

static gint send_send_data_progressive        (Session                *session,

                                         guint                         cur_len,

                                         guint                         total_len,

                                         gpointer                 data);

static gint send_send_data_finished        (Session                *session,

                                         guint                         len,

                                         gpointer                 data);

static SendProgressDialog *send_progress_dialog_create(void);

static void send_progress_dialog_destroy(SendProgressDialog *dialog);

static void send_cancel_button_cb        (GtkWidget        *widget,

                                         gpointer         data);

static void send_put_error                (Session        *session);

gint send_message(const gchar *file, PrefsAccount *ac_prefs, GSList *to_list)

{

        FILE *fp;

        gint val;

        g_return_val_if_fail(file != NULL, -1);

        g_return_val_if_fail(ac_prefs != NULL, -1);

        g_return_val_if_fail(to_list != NULL, -1);

        if ((fp = g_fopen(file, "rb")) == NULL) {

                FILE_OP_ERROR(file, "fopen");

                return -1;

        }

        if (prefs_common.use_extsend && prefs_common.extsend_cmd) {

                val = send_message_local(prefs_common.extsend_cmd, fp);

                fclose(fp);

                return val;

        }

        val = send_message_smtp(ac_prefs, to_list, fp);

        fclose(fp);

        return val;

}

enum

{

        Q_SENDER     = 0,

        Q_SMTPSERVER = 1,

        Q_RECIPIENTS = 2,

        Q_ACCOUNT_ID = 3

};

QueueInfo *send_get_queue_info(const gchar *file)

{

        static HeaderEntry qentry[] = {{"S:",   NULL, FALSE},

                                       {"SSV:", NULL, FALSE},

                                       {"R:",   NULL, FALSE},

                                       {"AID:", NULL, FALSE},

                                       {NULL,   NULL, FALSE}};

        FILE *fp;

        gchar buf[BUFFSIZE];

        gint hnum;

        QueueInfo *qinfo;

        g_return_val_if_fail(file != NULL, NULL);

        if ((fp = g_fopen(file, "rb")) == NULL) {

                FILE_OP_ERROR(file, "fopen");

                return NULL;

        }

        qinfo = g_new0(QueueInfo, 1);

        while ((hnum = procheader_get_one_field(buf, sizeof(buf), fp, qentry))

               != -1) {

                gchar *p;

                p = buf + strlen(qentry[hnum].name);

                switch (hnum) {

                case Q_SENDER:

                        if (!qinfo->from)

                                qinfo->from = g_strdup(p);

                        break;

                case Q_SMTPSERVER:

                        if (!qinfo->server)

                                qinfo->server = g_strdup(p);

                        break;

                case Q_RECIPIENTS:

                        qinfo->to_list =

                                address_list_append(qinfo->to_list, p);

                        break;

                case Q_ACCOUNT_ID:

                        qinfo->ac = account_find_from_id(atoi(p));

                        break;

                default:

                        break;

                }

        }

        qinfo->fp = fp;

        if (((!qinfo->ac || (qinfo->ac && qinfo->ac->protocol != A_NNTP)) &&

             !qinfo->to_list) || !qinfo->from) {

                g_warning(_("Queued message header is broken.\n"));

                send_queue_info_free(qinfo);

                return NULL;

        }

        if (!qinfo->ac) {

                qinfo->ac = account_find_from_smtp_server(qinfo->from,

                                                          qinfo->server);

                if (!qinfo->ac) {

                        g_warning("Account not found. "

                                  "Using current account...\n");

                        qinfo->ac = cur_account;

                }

        }

        return qinfo;

}

gint send_get_queue_contents(QueueInfo *qinfo, const gchar *dest)

{

        FILE *fp;

        glong pos;

        gchar buf[BUFFSIZE];

        g_return_val_if_fail(qinfo != NULL, -1);

        g_return_val_if_fail(qinfo->fp != NULL, -1);

        g_return_val_if_fail(dest != NULL, -1);

        if ((fp = g_fopen(dest, "wb")) == NULL) {

                FILE_OP_ERROR(dest, "fopen");

                return -1;

        }

        pos = ftell(qinfo->fp);

        while (fgets(buf, sizeof(buf), qinfo->fp) != NULL)

                fputs(buf, fp);

        if (fclose(fp) < 0) {

                FILE_OP_ERROR(dest, "fclose");

                g_unlink(dest);

                return -1;

        }

        fseek(qinfo->fp, pos, SEEK_SET);

        return 0;

}

void send_queue_info_free(QueueInfo *qinfo)

{

        if (qinfo == NULL) return;

        slist_free_strings(qinfo->to_list);

        g_slist_free(qinfo->to_list);

        g_free(qinfo->from);

        g_free(qinfo->server);

        if (qinfo->fp)

                fclose(qinfo->fp);

        g_free(qinfo);

}

gint send_message_queue(QueueInfo *qinfo)

{

        gint val = 0;

        glong fpos;

        PrefsAccount *mailac = NULL, *newsac = NULL;

        g_return_val_if_fail(qinfo != NULL, -1);

        fpos = ftell(qinfo->fp);

        if (prefs_common.use_extsend && prefs_common.extsend_cmd) {

                val = send_message_local(prefs_common.extsend_cmd, qinfo->fp);

        } else {

                if (qinfo->ac && qinfo->ac->protocol == A_NNTP) {

                        newsac = qinfo->ac;

                        /* search mail account */

                        mailac = account_find_from_address(qinfo->from);

                        if (!mailac) {

                                if (cur_account &&

                                    cur_account->protocol != A_NNTP)

                                        mailac = cur_account;

                                else {

                                        mailac = account_get_default();

                                        if (mailac->protocol == A_NNTP)

                                                mailac = NULL;

                                }

                        }

                } else

                        mailac = qinfo->ac;

                if (qinfo->to_list) {

                        if (mailac)

                                val = send_message_smtp(mailac, qinfo->to_list,

                                                        qinfo->fp);

                        else {

                                PrefsAccount tmp_ac;

                                g_warning("Account not found.\n");

                                memset(&tmp_ac, 0, sizeof(PrefsAccount));

                                tmp_ac.address = qinfo->from;

                                tmp_ac.smtp_server = qinfo->server;

                                tmp_ac.smtpport = SMTP_PORT;

                                val = send_message_smtp(&tmp_ac, qinfo->to_list,

                                                        qinfo->fp);

                        }

                }

                if (val == 0 && newsac) {

                        fseek(qinfo->fp, fpos, SEEK_SET);

                        val = news_post_stream(FOLDER(newsac->folder),

                                               qinfo->fp);

                        if (val < 0)

                                alertpanel_error(_("Error occurred while posting the message to %s ."),

                                                 newsac->nntp_server);

                }

        }

        fseek(qinfo->fp, fpos, SEEK_SET);

        return val;

}

static gint send_message_local(const gchar *command, FILE *fp)

{

        gchar **argv;

        GPid pid;

        gint child_stdin;

        gchar buf[BUFFSIZE];

        gboolean err = FALSE;

        gint status;

        g_return_val_if_fail(command != NULL, -1);

        g_return_val_if_fail(fp != NULL, -1);

        log_message(_("Sending message using command: %s\n"), command);

        argv = strsplit_with_quote(command, " ", 0);

        if (g_spawn_async_with_pipes(NULL, argv, NULL,

                                     G_SPAWN_DO_NOT_REAP_CHILD, NULL, NULL,

                                     &pid, &child_stdin, NULL, NULL,

                                     NULL) == FALSE) {

                g_snprintf(buf, sizeof(buf),

                           _("Can't execute command: %s"), command);

                log_warning("%s\n", buf);

                alertpanel_error("%s", buf);

                g_strfreev(argv);

                return -1;

        }

        g_strfreev(argv);

        while (fgets(buf, sizeof(buf), fp) != NULL) {

                strretchomp(buf);

                if (buf[0] == '.' && buf[1] == '\0') {

                        if (fd_write_all(child_stdin, ".", 1) < 0) {

                                err = TRUE;

                                break;

                        }

                }

                if (fd_write_all(child_stdin, buf, strlen(buf)) < 0 ||

                    fd_write_all(child_stdin, "\n", 1) < 0) {

                        err = TRUE;

                        break;

                }

        }

        fd_close(child_stdin);

#ifdef G_OS_UNIX

        waitpid(pid, &status, 0);

        if (!WIFEXITED(status) || WEXITSTATUS(status) != 0)

                err = TRUE;

#endif

        g_spawn_close_pid(pid);

        if (err) {

                g_snprintf(buf, sizeof(buf),

                           _("Error occurred while executing command: %s"),

                           command);

                log_warning("%s\n", buf);

                alertpanel_error("%s", buf);

                return -1;

        }

        return 0;

}

static gint send_message_smtp(PrefsAccount *ac_prefs, GSList *to_list, FILE *fp)

{

        Session *session;

        SMTPSession *smtp_session;

        gushort port;

        SendProgressDialog *dialog;

        gchar buf[BUFFSIZE];

        gint ret = 0;

        g_return_val_if_fail(ac_prefs != NULL, -1);

        g_return_val_if_fail(ac_prefs->address != NULL, -1);

        g_return_val_if_fail(ac_prefs->smtp_server != NULL, -1);

        g_return_val_if_fail(to_list != NULL, -1);

        g_return_val_if_fail(fp != NULL, -1);

        session = smtp_session_new();

        smtp_session = SMTP_SESSION(session);

        smtp_session->hostname =

                ac_prefs->set_domain ? g_strdup(ac_prefs->domain) : NULL;

        if (ac_prefs->use_smtp_auth) {

                smtp_session->forced_auth_type = ac_prefs->smtp_auth_type;

                if (ac_prefs->smtp_userid) {

                        smtp_session->user = g_strdup(ac_prefs->smtp_userid);

                        if (ac_prefs->smtp_passwd)

                                smtp_session->pass =

                                        g_strdup(ac_prefs->smtp_passwd);

                        else if (ac_prefs->tmp_smtp_pass)

                                smtp_session->pass =

                                        g_strdup(ac_prefs->tmp_smtp_pass);

                        else {

                                smtp_session->pass =

                                        input_dialog_query_password

                                                (ac_prefs->smtp_server,

                                                 smtp_session->user);

                                if (!smtp_session->pass)

                                        smtp_session->pass = g_strdup("");

                                ac_prefs->tmp_smtp_pass =

                                        g_strdup(smtp_session->pass);

                        }

                } else {

                        smtp_session->user = g_strdup(ac_prefs->userid);

                        if (ac_prefs->passwd)

                                smtp_session->pass = g_strdup(ac_prefs->passwd);

                        else if (ac_prefs->tmp_pass)

                                smtp_session->pass =

                                        g_strdup(ac_prefs->tmp_pass);

                        else {

                                smtp_session->pass =

                                        input_dialog_query_password

                                                (ac_prefs->smtp_server,

                                                 smtp_session->user);

                                if (!smtp_session->pass)

                                        smtp_session->pass = g_strdup("");

                                ac_prefs->tmp_pass =

                                        g_strdup(smtp_session->pass);

                        }

                }

        } else {

                smtp_session->user = NULL;

                smtp_session->pass = NULL;

        }

        smtp_session->from = g_strdup(ac_prefs->address);

        smtp_session->to_list = to_list;

        smtp_session->cur_to = to_list;

        smtp_session->send_data = get_outgoing_rfc2822_str(fp);

        smtp_session->send_data_len = strlen(smtp_session->send_data);

#if USE_SSL

        port = ac_prefs->set_smtpport ? ac_prefs->smtpport :

                ac_prefs->ssl_smtp == SSL_TUNNEL ? SSMTP_PORT : SMTP_PORT;

        session->ssl_type = ac_prefs->ssl_smtp;

        if (ac_prefs->ssl_smtp != SSL_NONE)

                session->nonblocking = ac_prefs->use_nonblocking_ssl;

#else

        port = ac_prefs->set_smtpport ? ac_prefs->smtpport : SMTP_PORT;

#endif

        dialog = send_progress_dialog_create();

        dialog->session = session;

        progress_dialog_append(dialog->dialog, NULL, ac_prefs->smtp_server,

                               _("Connecting"), NULL);

        g_snprintf(buf, sizeof(buf), _("Connecting to SMTP server: %s ..."),

                   ac_prefs->smtp_server);

        progress_dialog_set_label(dialog->dialog, buf);

        log_message("%s\n", buf);

        session_set_recv_message_notify(session, send_recv_message, dialog);

        session_set_send_data_progressive_notify

                (session, send_send_data_progressive, dialog);

        session_set_send_data_notify(session, send_send_data_finished, dialog);

        if (session_connect(session, ac_prefs->smtp_server, port) < 0) {

                session_destroy(session);

                send_progress_dialog_destroy(dialog);

                return -1;

        }

        debug_print("send_message_smtp(): begin event loop\n");

        while (session_is_connected(session) && dialog->cancelled == FALSE)

                gtk_main_iteration();

        if (SMTP_SESSION(session)->error_val == SM_AUTHFAIL) {

                if (ac_prefs->smtp_userid && ac_prefs->tmp_smtp_pass) {

                        g_free(ac_prefs->tmp_smtp_pass);

                        ac_prefs->tmp_smtp_pass = NULL;

                }

                ret = -1;

        } else if (session->state == SESSION_EOF &&

                   SMTP_SESSION(session)->state == SMTP_QUIT) {

                /* consider EOF right after QUIT successful */

                log_warning("%s\n", _("Connection closed by the remote host."));

                ret = 0;

        } else if (session->state == SESSION_ERROR ||

                   session->state == SESSION_EOF ||

                   session->state == SESSION_TIMEOUT ||

                   SMTP_SESSION(session)->state == SMTP_ERROR ||

                   SMTP_SESSION(session)->error_val != SM_OK)

                ret = -1;

        else if (dialog->cancelled == TRUE)

                ret = -1;

        if (ret == -1) {

                manage_window_focus_in(dialog->dialog->window, NULL, NULL);

                send_put_error(session);

                manage_window_focus_out(dialog->dialog->window, NULL, NULL);

        }

        session_destroy(session);

        send_progress_dialog_destroy(dialog);

        return ret;

}

static gint send_recv_message(Session *session, const gchar *msg, gpointer data)

{

        gchar buf[BUFFSIZE];

        SMTPSession *smtp_session = SMTP_SESSION(session);

        SendProgressDialog *dialog = (SendProgressDialog *)data;

        gchar *state_str = NULL;

        g_return_val_if_fail(dialog != NULL, -1);

        switch (smtp_session->state) {

        case SMTP_READY:

        case SMTP_CONNECTED:

                return 0;

        case SMTP_HELO:

                g_snprintf(buf, sizeof(buf), _("Sending HELO..."));

                state_str = _("Authenticating");

                statusbar_print_all(_("Sending message..."));

                break;

        case SMTP_EHLO:

                g_snprintf(buf, sizeof(buf), _("Sending EHLO..."));

                state_str = _("Authenticating");

                statusbar_print_all(_("Sending message..."));

                break;

        case SMTP_AUTH:

                g_snprintf(buf, sizeof(buf), _("Authenticating..."));

                state_str = _("Authenticating");

                break;

        case SMTP_FROM:

                g_snprintf(buf, sizeof(buf), _("Sending MAIL FROM..."));

                state_str = _("Sending");

                break;

        case SMTP_RCPT:

                g_snprintf(buf, sizeof(buf), _("Sending RCPT TO..."));

                state_str = _("Sending");

                break;

        case SMTP_DATA:

        case SMTP_EOM:

                g_snprintf(buf, sizeof(buf), _("Sending DATA..."));

                state_str = _("Sending");

                break;

        case SMTP_QUIT:

                g_snprintf(buf, sizeof(buf), _("Quitting..."));

                state_str = _("Quitting");

                break;

        case SMTP_ERROR:

                g_warning("send: error: %s\n", msg);

                return 0;

        default:

                return 0;

        }

        progress_dialog_set_label(dialog->dialog, buf);

        progress_dialog_set_row_status(dialog->dialog, 0, state_str);

        return 0;

}

static gint send_send_data_progressive(Session *session, guint cur_len,

                                       guint total_len, gpointer data)

{

        gchar buf[BUFFSIZE];

        SendProgressDialog *dialog = (SendProgressDialog *)data;

        g_return_val_if_fail(dialog != NULL, -1);

        if (SMTP_SESSION(session)->state != SMTP_SEND_DATA &&

            SMTP_SESSION(session)->state != SMTP_EOM)

                return 0;

        g_snprintf(buf, sizeof(buf), _("Sending message (%d / %d bytes)"),

                   cur_len, total_len);

        progress_dialog_set_label(dialog->dialog, buf);

        progress_dialog_set_percentage

                (dialog->dialog, (gfloat)cur_len / (gfloat)total_len);

        return 0;

}

static gint send_send_data_finished(Session *session, guint len, gpointer data)

{

        SendProgressDialog *dialog = (SendProgressDialog *)data;

        g_return_val_if_fail(dialog != NULL, -1);

        send_send_data_progressive(session, len, len, dialog);

        return 0;

}

static SendProgressDialog *send_progress_dialog_create(void)

{

        SendProgressDialog *dialog;

        ProgressDialog *progress;

        dialog = g_new0(SendProgressDialog, 1);

        progress = progress_dialog_create();

        gtk_window_set_title(GTK_WINDOW(progress->window),

                             _("Sending message"));

        g_signal_connect(G_OBJECT(progress->cancel_btn), "clicked",

                         G_CALLBACK(send_cancel_button_cb), dialog);

        g_signal_connect(G_OBJECT(progress->window), "delete_event",

                         G_CALLBACK(gtk_true), NULL);

        gtk_window_set_modal(GTK_WINDOW(progress->window), TRUE);

        manage_window_set_transient(GTK_WINDOW(progress->window));

        progress_dialog_set_value(progress, 0.0);

        gtk_widget_show_now(progress->window);

        dialog->dialog = progress;

        return dialog;

}

static void send_progress_dialog_destroy(SendProgressDialog *dialog)

{

        g_return_if_fail(dialog != NULL);

        manage_window_destroy(dialog->dialog->window, NULL);

        progress_dialog_destroy(dialog->dialog);

        g_free(dialog);

}

static void send_cancel_button_cb(GtkWidget *widget, gpointer data)

{

        SendProgressDialog *dialog = (SendProgressDialog *)data;

        dialog->cancelled = TRUE;

}

static void send_put_error(Session *session)

{

        gchar *msg;

        gchar *log_msg = NULL;

        gchar *err_msg = NULL;

        msg = SMTP_SESSION(session)->error_msg;

        switch (SMTP_SESSION(session)->error_val) {

        case SM_ERROR:

        case SM_UNRECOVERABLE:

                log_msg = _("Error occurred while sending the message.");

                if (msg)

                        err_msg = g_strdup_printf

                                (_("Error occurred while sending the message:\n%s"),

                                 msg);

                else

                        err_msg = g_strdup(log_msg);

                break;

        case SM_AUTHFAIL:

                log_msg = _("Authentication failed.");

                if (msg)

                        err_msg = g_strdup_printf

                                (_("Authentication failed:\n%s"), msg);

                else

                        err_msg = g_strdup(log_msg);

                break;

        default:

                switch (session->state) {

                case SESSION_ERROR:

                        log_msg =

                                _("Error occurred while sending the message.");

                        err_msg = g_strdup(log_msg);

                        break;

                case SESSION_EOF:

                        log_msg = _("Connection closed by the remote host.");

                        err_msg = g_strdup(log_msg);

                        break;

                case SESSION_TIMEOUT:

                        log_msg = _("Session timed out.");

                        err_msg = g_strdup(log_msg);

                        break;

                default:

                        break;

                }

                break;

        }

        if (log_msg)

                log_warning("%s\n", log_msg);

        if (err_msg) {

                alertpanel_error("%s", err_msg);

                g_free(err_msg);

        }

}