Sylpheed

/* passphrase.c - GTK+ based passphrase callback

 *      Copyright (C) 2001 Werner Koch (dd9jn)

 *

 * 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

#if USE_GPGME

#include <string.h>

#include <sys/types.h>

#include <sys/mman.h>

#include <glib.h>

#include <gdk/gdkkeysyms.h>

#ifdef GDK_WINDOWING_X11

#  include <gdk/gdkx.h>  /* GDK_DISPLAY() */

#endif /* GDK_WINDOWING_X11 */

#include <gtk/gtkmain.h>

#include <gtk/gtkwidget.h>

#include <gtk/gtkwindow.h>

#include <gtk/gtkvbox.h>

#include <gtk/gtktable.h>

#include <gtk/gtklabel.h>

#include <gtk/gtkentry.h>

#include <gtk/gtkhbbox.h>

#include <gtk/gtkbutton.h>

#include <gtk/gtkfilesel.h>

#include <gtk/gtksignal.h>

#include "intl.h"

#include "passphrase.h"

#include "prefs_common.h"

#include "manage_window.h"

#include "utils.h"

static int grab_all = 0;

static gboolean pass_ack;

static gchar *last_pass = NULL;

static void passphrase_ok_cb(GtkWidget *widget, gpointer data);

static void passphrase_cancel_cb(GtkWidget *widget, gpointer data);

static gint passphrase_deleted(GtkWidget *widget, GdkEventAny *event,

                               gpointer data);

static gboolean passphrase_key_pressed(GtkWidget *widget, GdkEventKey *event,

                                       gpointer data);

static gchar* passphrase_mbox (const gchar *desc);

static GtkWidget *create_description (const gchar *desc);

void

gpgmegtk_set_passphrase_grab (gint yes)

{

    grab_all = yes;

}

static gchar*

passphrase_mbox (const gchar *desc)

{

    gchar *the_passphrase = NULL;

    GtkWidget *vbox;

    GtkWidget *table;

    GtkWidget *pass_label;

    GtkWidget *confirm_box;

    GtkWidget *window;

    GtkWidget *pass_entry;

    GtkWidget *ok_button;

    GtkWidget *cancel_button;

    window = gtk_window_new(GTK_WINDOW_TOPLEVEL);

    gtk_window_set_title(GTK_WINDOW(window), _("Passphrase"));

    gtk_widget_set_size_request(window, 450, -1);

    gtk_container_set_border_width(GTK_CONTAINER(window), 4);

    gtk_window_set_position(GTK_WINDOW(window), GTK_WIN_POS_CENTER);

    gtk_window_set_modal(GTK_WINDOW(window), TRUE);

    gtk_window_set_policy(GTK_WINDOW(window), FALSE, FALSE, FALSE);

    g_signal_connect(G_OBJECT(window), "delete_event",

                     G_CALLBACK(passphrase_deleted), NULL);

    g_signal_connect(G_OBJECT(window), "key_press_event",

                     G_CALLBACK(passphrase_key_pressed), NULL);

    MANAGE_WINDOW_SIGNALS_CONNECT(window);

    manage_window_set_transient(GTK_WINDOW(window));

    vbox = gtk_vbox_new(FALSE, 8);

    gtk_container_add(GTK_CONTAINER(window), vbox);

    if (desc) {

        GtkWidget *label;

        label = create_description (desc);

        gtk_box_pack_start (GTK_BOX(vbox), label, TRUE, TRUE, 0);

    }

    table = gtk_table_new(2, 2, FALSE);

    gtk_box_pack_start(GTK_BOX(vbox), table, FALSE, FALSE, 0);

    gtk_container_set_border_width(GTK_CONTAINER(table), 8);

    gtk_table_set_row_spacings(GTK_TABLE(table), 12);

    gtk_table_set_col_spacings(GTK_TABLE(table), 8);

    pass_label = gtk_label_new("");

    gtk_table_attach (GTK_TABLE(table), pass_label, 0, 1, 0, 1,

                      GTK_FILL, GTK_EXPAND|GTK_FILL, 0, 0);

    gtk_misc_set_alignment (GTK_MISC (pass_label), 1, 0.5);

    pass_entry = gtk_entry_new();

    gtk_table_attach (GTK_TABLE(table), pass_entry, 1, 2, 0, 1,

                      GTK_EXPAND|GTK_SHRINK|GTK_FILL, 0, 0, 0);

    gtk_entry_set_visibility (GTK_ENTRY(pass_entry), FALSE);

    gtk_widget_grab_focus (pass_entry);

    confirm_box = gtk_hbutton_box_new ();

    gtk_button_box_set_layout (GTK_BUTTON_BOX(confirm_box), GTK_BUTTONBOX_END);

    gtk_box_set_spacing (GTK_BOX(confirm_box), 5);

    ok_button = gtk_button_new_with_label (_("OK"));

    GTK_WIDGET_SET_FLAGS (ok_button, GTK_CAN_DEFAULT);

    gtk_box_pack_start (GTK_BOX(confirm_box), ok_button, TRUE, TRUE, 0);

    cancel_button = gtk_button_new_with_label (_("Cancel"));

    GTK_WIDGET_SET_FLAGS (cancel_button, GTK_CAN_DEFAULT);

    gtk_box_pack_start(GTK_BOX(confirm_box), cancel_button, TRUE, TRUE, 0);

    gtk_box_pack_end(GTK_BOX(vbox), confirm_box, FALSE, FALSE, 0);

    gtk_widget_grab_default (ok_button);

    g_signal_connect(G_OBJECT(ok_button), "clicked",

                     G_CALLBACK(passphrase_ok_cb), NULL);

    g_signal_connect(G_OBJECT(pass_entry), "activate",

                     G_CALLBACK(passphrase_ok_cb), NULL);

    g_signal_connect(G_OBJECT(cancel_button), "clicked",

                     G_CALLBACK(passphrase_cancel_cb), NULL);

    if (grab_all)

        g_object_set (G_OBJECT(window), "type", GTK_WINDOW_POPUP, NULL);

    gtk_window_set_position (GTK_WINDOW(window), GTK_WIN_POS_CENTER);

    if (grab_all)   

        gtk_window_set_policy (GTK_WINDOW(window), FALSE, FALSE, TRUE);

    gtk_widget_show_all(window);

    /* don't use XIM on entering passphrase */

    gtkut_editable_disable_im(GTK_EDITABLE(pass_entry));

    if (grab_all) {

#ifdef GDK_WINDOWING_X11

        //XGrabServer(GDK_DISPLAY());

#endif /* GDK_WINDOWING_X11 */

        if ( gdk_pointer_grab ( window->window, TRUE, 0,

                                NULL, NULL, GDK_CURRENT_TIME)) {

#ifdef GDK_WINDOWING_X11

            //XUngrabServer ( GDK_DISPLAY() );

#endif /* GDK_WINDOWING_X11 */

            g_warning ("OOPS: Could not grab mouse\n");

            gtk_widget_destroy (window);

            return NULL;

        }

        if ( gdk_keyboard_grab( window->window, FALSE, GDK_CURRENT_TIME )) {

            gdk_pointer_ungrab (GDK_CURRENT_TIME);

#ifdef GDK_WINDOWING_X11

            //XUngrabServer ( GDK_DISPLAY() );

#endif /* GDK_WINDOWING_X11 */

            g_warning ("OOPS: Could not grab keyboard\n");

            gtk_widget_destroy (window);

            return NULL;

        }

    }

    gtk_main();

    if (grab_all) {

#ifdef GDK_WINDOWING_X11

        //XUngrabServer (GDK_DISPLAY());

#endif /* GDK_WINDOWING_X11 */

        gdk_pointer_ungrab (GDK_CURRENT_TIME);

        gdk_keyboard_ungrab (GDK_CURRENT_TIME);

        gdk_flush();

    }

    manage_window_focus_out(window, NULL, NULL);

    if (pass_ack) {

        const gchar *entry_text;

        entry_text = gtk_entry_get_text(GTK_ENTRY(pass_entry));

        if (entry_text) /* Hmmm: Do we really need this? */

            the_passphrase = g_strdup (entry_text);

    }

    gtk_widget_destroy (window);

    return the_passphrase;

}

static void 

passphrase_ok_cb(GtkWidget *widget, gpointer data)

{

    pass_ack = TRUE;

    gtk_main_quit();

}

static void 

passphrase_cancel_cb(GtkWidget *widget, gpointer data)

{

    pass_ack = FALSE;

    gtk_main_quit();

}

static gint

passphrase_deleted(GtkWidget *widget, GdkEventAny *event, gpointer data)

{

    passphrase_cancel_cb(NULL, NULL);

    return TRUE;

}

static gboolean

passphrase_key_pressed(GtkWidget *widget, GdkEventKey *event, gpointer data)

{

    if (event && event->keyval == GDK_Escape)

        passphrase_cancel_cb(NULL, NULL);

    return FALSE;

}

static gint 

linelen (const gchar *s)

{

    gint i;

    for (i = 0; *s && *s != '\n'; s++, i++)

        ;

    return i;

}

static GtkWidget *

create_description (const gchar *desc)

{

    const gchar *cmd = NULL, *uid = NULL, *info = NULL;

    gchar *buf;

    GtkWidget *label;

    cmd = desc;

    uid = strchr (cmd, '\n');

    if (uid) {

        info = strchr (++uid, '\n');

        if (info )

            info++;

    }

    if (!uid)

        uid = _("[no user id]");

    if (!info)

        info = "";

    buf = g_strdup_printf (_("%sPlease enter the passphrase for:\n\n"

                           "  %.*s  \n"

                           "(%.*s)\n"),

                           !strncmp (cmd, "TRY_AGAIN", 9 ) ?

                           _("Bad passphrase! Try again...\n\n") : "",

                           linelen (uid), uid, linelen (info), info);

    label = gtk_label_new (buf);

    g_free (buf);

    return label;

}

static int free_passphrase(gpointer _unused)

{

    if (last_pass != NULL) {

        munlock(last_pass, strlen(last_pass));

        g_free(last_pass);

        last_pass = NULL;

        debug_print("%% passphrase removed");

    }

    return FALSE;

}

const char*

gpgmegtk_passphrase_cb (void *opaque, const char *desc, void **r_hd)

{

    struct passphrase_cb_info_s *info = opaque;

    GpgmeCtx ctx = info ? info->c : NULL;

    const char *pass;

    if (!desc) {

        /* FIXME: cleanup by looking at *r_hd */

        return NULL;

    }

    if (prefs_common.store_passphrase && last_pass != NULL &&

        strncmp(desc, "TRY_AGAIN", 9) != 0)

        return g_strdup(last_pass);

    gpgmegtk_set_passphrase_grab (prefs_common.passphrase_grab);

    debug_print ("%% requesting passphrase for `%s': ", desc);

    pass = passphrase_mbox (desc);

    gpgmegtk_free_passphrase();

    if (!pass) {

        debug_print ("%% cancel passphrase entry");

        gpgme_cancel (ctx);

    }

    else {

        if (prefs_common.store_passphrase) {

            last_pass = g_strdup(pass);

            if (mlock(last_pass, strlen(last_pass)) == -1)

                debug_print("%% locking passphrase failed");

            if (prefs_common.store_passphrase_timeout > 0) {

                gtk_timeout_add(prefs_common.store_passphrase_timeout*60*1000,

                                free_passphrase, NULL);

            }

        }

        debug_print ("%% sending passphrase");

    }

    return pass;

}

void gpgmegtk_free_passphrase()

{

    (void)free_passphrase(NULL); // could be inline

}

#endif /* USE_GPGME */