path: root/smime/gui/certificate-manager.c

/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
/*
 *  Authors: Chris Toshok <toshok@ximian.com>
 *
 *  Copyright (C) 2003 Ximian, Inc. (www.ximian.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.
 *
 */

#define GLADE_FILE_NAME "smime-ui.glade"

#include <gtk/gtk.h>

#include <libgnome/gnome-i18n.h>

#include <glade/glade.h>
#include "evolution-config-control.h"
#include "certificate-manager.h"

#include "e-cert.h"
#include "e-cert-db.h"

#include "nss.h"
#include <cms.h>
#include <cert.h>
#include <certdb.h>
#include <pkcs11.h>
#include <pk11func.h>

typedef struct {
    GladeXML *gui;

    GtkWidget *yourcerts_treeview;
    GtkTreeStore *yourcerts_treemodel;
    GHashTable *yourcerts_root_hash;
    GtkWidget *view_your_button;
    GtkWidget *backup_your_button;
    GtkWidget *backup_all_your_button;
    GtkWidget *import_your_button;
    GtkWidget *delete_your_button;

    GtkWidget *contactcerts_treeview;
    GtkTreeStore *contactcerts_treemodel;
    GHashTable *contactcerts_root_hash;

    GtkWidget *authoritycerts_treeview;
    GtkTreeStore *authoritycerts_treemodel;
    GHashTable *authoritycerts_root_hash;
    GtkWidget *view_ca_button;
    GtkWidget *edit_ca_button;
    GtkWidget *import_ca_button;
    GtkWidget *delete_ca_button;

} CertificateManagerData;

typedef void (*AddCertCb)(CertificateManagerData *cfm, ECert *cert);

static void unload_certs (CertificateManagerData *cfm, ECertType type);
static void load_certs (CertificateManagerData *cfm, ECertType type, AddCertCb add_cert);

static void add_user_cert (CertificateManagerData *cfm, ECert *cert);
static void add_contact_cert (CertificateManagerData *cfm, ECert *cert);
static void add_ca_cert (CertificateManagerData *cfm, ECert *cert);

static void
handle_selection_changed (GtkTreeSelection *selection,
              int cert_column,
              GtkWidget *view_button,
              GtkWidget *edit_button,
              GtkWidget *delete_button)
{
    GtkTreeIter iter;
    gboolean cert_selected = FALSE;
    GtkTreeModel *model;

    if (gtk_tree_selection_get_selected (selection,
                         &model,
                         &iter)) {
        ECert *cert;

        gtk_tree_model_get (model,
                    &iter,
                    cert_column, &cert,
                    -1);

        if (cert) {
            cert_selected = TRUE;
            g_object_unref (cert);
        }
    }

    if (delete_button)
        gtk_widget_set_sensitive (delete_button, cert_selected);
    if (edit_button)
        gtk_widget_set_sensitive (edit_button, cert_selected);
    if (view_button)
        gtk_widget_set_sensitive (view_button, cert_selected);
}

static void
yourcerts_selection_changed (GtkTreeSelection *selection, CertificateManagerData *cfm)
{
    handle_selection_changed (gtk_tree_view_get_selection (GTK_TREE_VIEW(cfm->yourcerts_treeview)),
                  4,
                  cfm->view_your_button,
                  cfm->backup_your_button, /* yes yes, not really "edit", it's a hack :) */
                  cfm->delete_your_button);
}

static void
initialize_yourcerts_ui (CertificateManagerData *cfm)
{
    GtkCellRenderer *cell = gtk_cell_renderer_text_new ();
    GtkTreeSelection *selection;

    gtk_tree_view_append_column (GTK_TREE_VIEW (cfm->yourcerts_treeview),
                     gtk_tree_view_column_new_with_attributes (_("Certificate Name"),
                                           cell,
                                           "text", 0,
                                           NULL));

    gtk_tree_view_append_column (GTK_TREE_VIEW (cfm->yourcerts_treeview),
                     gtk_tree_view_column_new_with_attributes (_("Purposes"),
                                           cell,
                                           "text", 1,
                                           NULL));

    gtk_tree_view_append_column (GTK_TREE_VIEW (cfm->yourcerts_treeview),
                     gtk_tree_view_column_new_with_attributes (_("Serial Number"),
                                           cell,
                                           "text", 2,
                                           NULL));

    gtk_tree_view_append_column (GTK_TREE_VIEW (cfm->yourcerts_treeview),
                     gtk_tree_view_column_new_with_attributes (_("Expires"),
                                           cell,
                                           "text", 3,
                                           NULL));

    cfm->yourcerts_treemodel = gtk_tree_store_new (5,
                               G_TYPE_STRING,
                               G_TYPE_STRING,
                               G_TYPE_STRING,
                               G_TYPE_STRING,
                               G_TYPE_OBJECT);

    gtk_tree_view_set_model (GTK_TREE_VIEW (cfm->yourcerts_treeview),
                 GTK_TREE_MODEL (cfm->yourcerts_treemodel));

    cfm->yourcerts_root_hash = g_hash_table_new (g_str_hash, g_str_equal);

    selection = gtk_tree_view_get_selection (GTK_TREE_VIEW (cfm->yourcerts_treeview));
    g_signal_connect (selection, "changed", G_CALLBACK (yourcerts_selection_changed), cfm);

    if (cfm->import_your_button) {
        /* g_signal_connect */
    }

    if (cfm->delete_your_button) {
        /* g_signal_connect */
    }

    if (cfm->view_your_button) {
        /* g_signal_connect */
    }

    if (cfm->backup_your_button) {
        /* g_signal_connect */
    }

    if (cfm->backup_all_your_button) {
        /* g_signal_connect */
    }
}

static void
contactcerts_selection_changed (GtkTreeSelection *selection, CertificateManagerData *cfm)
{
#if 0
    handle_selection_changed (gtk_tree_view_get_selection (GTK_TREE_VIEW(cfm->contactcerts_treeview)),
                  1 /* XXX */,
                  NULL,
                  NULL,
                  NULL);
#endif
}

static void
initialize_contactcerts_ui (CertificateManagerData *cfm)
{
    GtkCellRenderer *cell = gtk_cell_renderer_text_new ();
    GtkTreeSelection *selection;

    gtk_tree_view_append_column (GTK_TREE_VIEW (cfm->contactcerts_treeview),
                     gtk_tree_view_column_new_with_attributes (_("Certificate Name"),
                                           cell,
                                           "text", 0,
                                           NULL));

    gtk_tree_view_append_column (GTK_TREE_VIEW (cfm->contactcerts_treeview),
                     gtk_tree_view_column_new_with_attributes (_("E-Mail Address"),
                                           cell,
                                           "text", 1,
                                           NULL));

    gtk_tree_view_append_column (GTK_TREE_VIEW (cfm->contactcerts_treeview),
                     gtk_tree_view_column_new_with_attributes (_("Purposes"),
                                           cell,
                                           "text", 2,
                                           NULL));

    cfm->contactcerts_treemodel = gtk_tree_store_new (3,
                              G_TYPE_STRING,
                              G_TYPE_STRING,
                              G_TYPE_STRING,
                              G_TYPE_OBJECT);
    
    gtk_tree_view_set_model (GTK_TREE_VIEW (cfm->contactcerts_treeview),
                 GTK_TREE_MODEL (cfm->contactcerts_treemodel));

    cfm->contactcerts_root_hash = g_hash_table_new (g_str_hash, g_str_equal);

    selection = gtk_tree_view_get_selection (GTK_TREE_VIEW (cfm->contactcerts_treeview));
    g_signal_connect (selection, "changed", G_CALLBACK (contactcerts_selection_changed), cfm);
}

static gint
iter_string_compare (GtkTreeModel *model,
             GtkTreeIter  *a,
             GtkTreeIter  *b,
             gpointer      user_data)
{
    char *string1, *string2;

    gtk_tree_model_get (model, a,
                0, &string1,
                -1);

    gtk_tree_model_get (model, b,
                0, &string2,
                -1);

    return g_utf8_collate (string1, string2);
}

static void
import_ca (GtkWidget *widget, CertificateManagerData *cfm)
{
    GtkWidget *filesel = gtk_file_selection_new (_("Select a cert to import..."));

    if (GTK_RESPONSE_OK == gtk_dialog_run (GTK_DIALOG (filesel))) {
        const char *filename = gtk_file_selection_get_filename (GTK_FILE_SELECTION (filesel));

        if (e_cert_db_import_certs_from_file (e_cert_db_peek (),
                              filename,
                              E_CERT_CA,
                              NULL)) {

            /* there's no telling how many certificates were added during the import,
               so we blow away the CA cert display and regenerate it. */
            unload_certs (cfm, E_CERT_CA);
            load_certs (cfm, E_CERT_CA, add_ca_cert);
        }
    }

    gtk_widget_destroy (filesel);
}

static void
delete_ca (GtkWidget *widget, CertificateManagerData *cfm)
{
    GtkTreeIter iter;

    if (gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW(cfm->authoritycerts_treeview)),
                         NULL,
                         &iter)) {
        ECert *cert;

        gtk_tree_model_get (GTK_TREE_MODEL (cfm->authoritycerts_treemodel),
                    &iter,
                    1, &cert,
                    -1);

        if (cert) {
            printf ("DELETE\n");
            e_cert_db_delete_cert (e_cert_db_peek (), cert);
            gtk_tree_store_remove (cfm->authoritycerts_treemodel,
                           &iter);

            /* we need two unrefs here, one to unref the
               gtk_tree_model_get above, and one to unref
               the initial ref when we created the cert
               and added it to the tree */
            g_object_unref (cert);
            g_object_unref (cert);
        }
    }
                         
}

static void
authoritycerts_selection_changed (GtkTreeSelection *selection, CertificateManagerData *cfm)
{
    handle_selection_changed (gtk_tree_view_get_selection (GTK_TREE_VIEW(cfm->authoritycerts_treeview)),
                  1,
                  cfm->view_ca_button,
                  cfm->edit_ca_button,
                  cfm->delete_ca_button);
}

static GtkTreeStore*
create_authoritycerts_treemodel (void)
{
    return gtk_tree_store_new (2,
                   G_TYPE_STRING,
                   G_TYPE_OBJECT);

}

static void
initialize_authoritycerts_ui (CertificateManagerData *cfm)
{
    GtkCellRenderer *cell = gtk_cell_renderer_text_new ();
    GtkTreeSelection *selection;

    gtk_tree_view_append_column (GTK_TREE_VIEW (cfm->authoritycerts_treeview),
                     gtk_tree_view_column_new_with_attributes (_("Certificate Name"),
                                           cell,
                                           "text", 0,
                                           NULL));

    gtk_tree_sortable_set_sort_func (GTK_TREE_SORTABLE (cfm->authoritycerts_treemodel),
                     0,
                     iter_string_compare, NULL, NULL);

    gtk_tree_sortable_set_sort_column_id (GTK_TREE_SORTABLE (cfm->authoritycerts_treemodel),
                          0,
                          GTK_SORT_ASCENDING);

    selection = gtk_tree_view_get_selection (GTK_TREE_VIEW (cfm->authoritycerts_treeview));
    g_signal_connect (selection, "changed", G_CALLBACK (authoritycerts_selection_changed), cfm);

    if (cfm->import_ca_button)
        g_signal_connect (cfm->import_ca_button, "clicked", G_CALLBACK (import_ca), cfm);

    if (cfm->delete_ca_button)
        g_signal_connect (cfm->delete_ca_button, "clicked", G_CALLBACK (delete_ca), cfm);
}

static void
add_user_cert (CertificateManagerData *cfm, ECert *cert)
{
    GtkTreeIter iter;
    GtkTreeIter *parent_iter = NULL;
    const char *organization = e_cert_get_org (cert);

    if (organization) {
        parent_iter = g_hash_table_lookup (cfm->yourcerts_root_hash, organization);
        if (!parent_iter) {
            /* create a new toplevel node */
            gtk_tree_store_append (GTK_TREE_STORE (cfm->yourcerts_treemodel), &iter, NULL);
        
            gtk_tree_store_set (GTK_TREE_STORE (cfm->yourcerts_treemodel), &iter,
                        0, organization, -1);

            /* now copy it off into parent_iter and insert it into
               the hashtable */
            parent_iter = gtk_tree_iter_copy (&iter);
            g_hash_table_insert (cfm->yourcerts_root_hash, g_strdup (organization), parent_iter);
        }
    }

    gtk_tree_store_append (GTK_TREE_STORE (cfm->yourcerts_treemodel), &iter, parent_iter);

    if (e_cert_get_cn (cert))
        gtk_tree_store_set (GTK_TREE_STORE (cfm->yourcerts_treemodel), &iter,
                    0, e_cert_get_cn (cert),
                    4, cert,
                    -1);
    else
        gtk_tree_store_set (GTK_TREE_STORE (cfm->yourcerts_treemodel), &iter,
                    0, e_cert_get_nickname (cert),
                    4, cert,
                    -1);
}

static void
add_contact_cert (CertificateManagerData *cfm, ECert *cert)
{
    /* nothing yet */
}

static void
add_ca_cert (CertificateManagerData *cfm, ECert *cert)
{
    GtkTreeIter iter;
    GtkTreeIter *parent_iter = NULL;
    const char *organization = e_cert_get_org (cert);

    if (organization) {
        parent_iter = g_hash_table_lookup (cfm->authoritycerts_root_hash, organization);
        if (!parent_iter) {
            /* create a new toplevel node */
            gtk_tree_store_append (GTK_TREE_STORE (cfm->authoritycerts_treemodel), &iter, NULL);
        
            gtk_tree_store_set (GTK_TREE_STORE (cfm->authoritycerts_treemodel), &iter,
                        0, organization, -1);

            /* now copy it off into parent_iter and insert it into
               the hashtable */
            parent_iter = gtk_tree_iter_copy (&iter);
            g_hash_table_insert (cfm->authoritycerts_root_hash, g_strdup (organization), parent_iter);
        }
    }


    gtk_tree_store_append (GTK_TREE_STORE (cfm->authoritycerts_treemodel), &iter, parent_iter);

    if (e_cert_get_cn (cert))
        gtk_tree_store_set (GTK_TREE_STORE (cfm->authoritycerts_treemodel), &iter,
                    0, e_cert_get_cn (cert),
                    1, cert,
                    -1);
    else
        gtk_tree_store_set (GTK_TREE_STORE (cfm->authoritycerts_treemodel), &iter,
                    0, e_cert_get_nickname (cert),
                    1, cert,
                    -1);
}

static void
destroy_key (gpointer data)
{
    g_free (data);
}

static void
destroy_value (gpointer data)
{
    gtk_tree_iter_free (data);
}

static void
unload_certs (CertificateManagerData *cfm,
          ECertType type)
{
    switch (type) {
    case E_CERT_USER:
        break;
    case E_CERT_CONTACT:
        break;
    case E_CERT_SITE:
        break;
    case E_CERT_CA:
        cfm->authoritycerts_treemodel = create_authoritycerts_treemodel ();
        gtk_tree_view_set_model (GTK_TREE_VIEW (cfm->authoritycerts_treeview),
                     GTK_TREE_MODEL (cfm->authoritycerts_treemodel));

        if (cfm->authoritycerts_root_hash)
            g_hash_table_destroy (cfm->authoritycerts_root_hash);

        cfm->authoritycerts_root_hash = g_hash_table_new_full (g_str_hash, g_str_equal,
                                       destroy_key, destroy_value);


        break;
    }
}

static void
load_certs (CertificateManagerData *cfm,
        ECertType type,
        AddCertCb add_cert)
{
    CERTCertList *certList;
    CERTCertListNode *node;

    certList = PK11_ListCerts (PK11CertListUnique, NULL);

    printf ("certList = %p\n", certList);

    for (node = CERT_LIST_HEAD(certList);
         !CERT_LIST_END(node, certList);
         node = CERT_LIST_NEXT(node)) {
        ECert *cert = e_cert_new ((CERTCertificate*)node->cert);
        if (e_cert_get_cert_type(cert) == type) {
            printf ("cert (nickname = '%s') matches\n", e_cert_get_nickname (cert));
            add_cert (cfm, cert);
        }
    }

}

static void
populate_ui (CertificateManagerData *cfm)
{
    unload_certs (cfm, E_CERT_USER);
    load_certs (cfm, E_CERT_USER, add_user_cert);

    unload_certs (cfm, E_CERT_CONTACT);
    load_certs (cfm, E_CERT_CONTACT, add_contact_cert);

    unload_certs (cfm, E_CERT_CA);
    load_certs (cfm, E_CERT_CA, add_ca_cert);
}

EvolutionConfigControl*
certificate_manager_config_control_new (void)
{
    CertificateManagerData *cfm_data;
    GtkWidget *control_widget;

    /* We need to peek the db here to make sure it (and NSS) are fully initialized. */
    e_cert_db_peek ();

    cfm_data = g_new0 (CertificateManagerData, 1);
    cfm_data->gui = glade_xml_new (EVOLUTION_GLADEDIR "/" GLADE_FILE_NAME, NULL, NULL);

    cfm_data->yourcerts_treeview = glade_xml_get_widget (cfm_data->gui, "yourcerts-treeview");
    cfm_data->contactcerts_treeview = glade_xml_get_widget (cfm_data->gui, "contactcerts-treeview");
    cfm_data->authoritycerts_treeview = glade_xml_get_widget (cfm_data->gui, "authoritycerts-treeview");

    cfm_data->view_your_button = glade_xml_get_widget (cfm_data->gui, "your-view-button");
    cfm_data->backup_your_button = glade_xml_get_widget (cfm_data->gui, "your-backup-button");
    cfm_data->backup_all_your_button = glade_xml_get_widget (cfm_data->gui, "your-backup-all-button");
    cfm_data->import_your_button = glade_xml_get_widget (cfm_data->gui, "your-import-button");
    cfm_data->delete_your_button = glade_xml_get_widget (cfm_data->gui, "your-delete-button");

    cfm_data->view_ca_button = glade_xml_get_widget (cfm_data->gui, "authority-view-button");
    cfm_data->edit_ca_button = glade_xml_get_widget (cfm_data->gui, "authority-edit-button");
    cfm_data->import_ca_button = glade_xml_get_widget (cfm_data->gui, "authority-import-button");
    cfm_data->delete_ca_button = glade_xml_get_widget (cfm_data->gui, "authority-delete-button");

    initialize_yourcerts_ui(cfm_data);
    initialize_contactcerts_ui(cfm_data);
    initialize_authoritycerts_ui(cfm_data);

    populate_ui (cfm_data);

    control_widget = glade_xml_get_widget (cfm_data->gui, "cert-manager-notebook");
    gtk_widget_ref (control_widget);

    gtk_container_remove (GTK_CONTAINER (control_widget->parent), control_widget);

    return evolution_config_control_new (control_widget);
}