trunk/Src/wptKeyEditDlgs.cpp

/* wptKeyEditDlgs.cpp - GPG key edit dialogs
 *      Copyright (C) 2002-2006 Timo Schulz
 *
 * This file is part of WinPT.
 *
 * WinPT 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.
 * 
 * WinPT 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 WinPT; 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 <windows.h>
#include <commctrl.h>
#include <time.h>
#include "resource.h"

#include "wptTypes.h"
#include "wptW32API.h"
#include "wptVersion.h"
#include "wptGPG.h"
#include "wptCommonCtl.h"
#include "wptContext.h"
#include "wptDlgs.h"
#include "wptNLS.h"
#include "wptUTF8.h"
#include "wptErrors.h"
#include "wptKeylist.h"
#include "wptKeyManager.h"
#include "wptRegistry.h"
#include "wptKeyEdit.h"

/* All edit key commands. */
enum keyedit_commands {    
    CMD_ADDKEY = 0,
    CMD_ADDUID,
    CMD_ADDPHOTO,
    CMD_ADDREVOKER,
    CMD_DELUID,
    CMD_DELKEY,
    CMD_EXPIRE,
    CMD_SHOWPREF,
    //CMD_SETPREF,
    CMD_PASSWD,
    CMD_PRIMARY,
    CMD_TRUST,
    CMD_REVUID,
    CMD_REVKEY,
    CMD_DISABLE,
    CMD_ENABLE,    
    CMD_SIGN,
    CMD_LSIGN,
    CMD_CHECK
};

struct cmdlist_s {
    const char   *name;
    unsigned int  need_pair:1;
    int       id;
} cmdlist[] = {
    {"ADDKEY", 1, CMD_ADDKEY},
    {"ADDUID", 1, CMD_ADDUID},
    {"ADDPHOTO", 1, CMD_ADDPHOTO},
    {"ADDREVOKER", 1, CMD_ADDREVOKER},
    {"DELUID", 1, CMD_DELUID},
    {"DELKEY", 1, CMD_DELKEY},
    {"EXPIRE", 1, CMD_EXPIRE},
    {"SHOWPREF", 0, CMD_SHOWPREF},
    /*{"SETPREF", 1, CMD_SETPREF},*/
    {"PASSWD", 1, CMD_PASSWD},
    {"PRIMARY", 1, CMD_PRIMARY},
    {"TRUST", 0, CMD_TRUST},
    {"REVUID", 1, CMD_REVUID},
    {"REVKEY", 1, CMD_REVKEY},
    {"DISABLE", 0, CMD_DISABLE},
    {"ENABLE", 0, CMD_ENABLE},
    {"SIGN", 0, CMD_SIGN},
    {"LSIGN", 0, CMD_LSIGN},
    {"CHECK", 0, CMD_CHECK},
    {NULL, 0}  
};


/* Symbolic ids for the subkey columns. */
enum subk_col_t {
    SUBK_COL_DESC       = 0,
    SUBK_COL_KEYID      = 1,    
    SUBK_COL_CREATION   = 2,    
    SUBK_COL_EXPIRES    = 3,
    SUBK_COL_STATUS     = 4,
    SUBK_COL_C_FLAG     = 5,
    SUBK_COL_S_FLAG     = 6,
    SUBK_COL_E_FLAG     = 7,
    SUBK_COL_A_FLAG     = 8
};

/* Symbolic ids for the userid columns. */
enum uid_col_t {
    UID_COL_VALID       = 0,
    UID_COL_NAME        = 1,
    UID_COL_EMAIL       = 2,
    UID_COL_CREATION    = 3
};

/* Key edit callback context. */
struct keyedit_cb_s {
    const char     *keyid;
    const char     *pass;
    listview_ctrl_t lv;
    void           *opaque;
    unsigned int    finished:1;
    unsigned int    is_protected:1;
};
typedef struct keyedit_cb_s *keyedit_cb_t;


/* Key generation callback context. */
struct keygen_cb_s {
    int   bits;
    int   algo;
    u32   expire;
    char *fpr;
};
typedef struct keygen_cb_s *keygen_cb_t;


static subclass_s keyedit_subkey_proc;
static subclass_s keyedit_uid_proc;

int keygen_check_date (SYSTEMTIME *st);
void get_userid_preflist (char **r_prefs, int * r_flags);
char* get_subkey_keyid (const char *keyid);


/* Associate each key with a combo box entry. 
   Skip the key in @k. */
static void
do_init_keylist (HWND dlg, winpt_key_t k)
{
    gpg_keycache_t pub;
    gpgme_key_t key;
    const char *s, *kid;
    char *u;
    int i, n;

    pub = keycache_get_ctx (1);
    if (!pub)
        BUG (0);

    gpg_keycache_rewind (pub);
    while( !gpg_keycache_next_key (pub, 0, &key)) {
        if (key->expired || key->revoked || 
            key->disabled || key->invalid)
            continue;
        s = key->uids->uid;
        kid = key->subkeys->keyid;
        if (!s || !strcmp (kid+8, k->keyid))
            continue;
        u = utf8_to_wincp (s, strlen (s));
        SendDlgItemMessage (dlg, IDC_ADDREV_KEYLIST, CB_ADDSTRING, 
                            0, (WPARAM)(char *)u);
        free (u);
    }
    gpg_keycache_rewind (pub);
    n = SendDlgItemMessage( dlg, IDC_ADDREV_KEYLIST, CB_GETCOUNT, 0, 0 );
    for (i = 0; i < n; i++) {
        gpg_keycache_next_key (pub, 0, &key);
        SendDlgItemMessage (dlg, IDC_ADDREV_KEYLIST, CB_SETITEMDATA, 
                            (WPARAM)(int)i, (LPARAM)key);
    }
    SendDlgItemMessage (dlg, IDC_ADDREV_KEYLIST, CB_SETCURSEL, 0, 0);
}


/* Add a new user-id to the list view @lv. */
static void
do_add_new_userid (listview_ctrl_t lv, 
                   const char * name, const char *email, const char * comment)
{
    char * p;
    size_t n;
    
    n = strlen (name) + strlen (email) + 16;
    if (comment)
        n += strlen (comment);
    p = new char[n+1];
    if (!p)
        BUG( NULL );
    if (comment)
        sprintf (p, "%s (%s)", name, comment);
    else
        sprintf (p, "%s", name);

    listview_add_item (lv, "");
    listview_add_sub_item (lv, 0, 0, _("Ultimate" ));
    listview_add_sub_item (lv, 0, 1, p);
    listview_add_sub_item (lv, 0, 2, email && *email? email : "");
    listview_add_sub_item (lv, 0, 3, get_key_created (time (NULL)));
    free_if_alloc (p);
}


static void
do_add_new_subkey (listview_ctrl_t lv, keygen_cb_t keygen, unsigned int flags)
{
    char info[128], keyid[32];
    const char * expdate, * s;
    int n;
    
    expdate = keygen->expire? get_key_expire_date (keygen->expire) : _("Never");
    _snprintf (info, sizeof info-1, "%d-bit %s",
               keygen->bits, 
               get_key_pubalgo ((gpgme_pubkey_algo_t)keygen->algo));
    _snprintf (keyid, sizeof keyid-1, "0x%s", keygen->fpr+8);
    n = listview_count_items (lv, 0);
    listview_add_item_pos (lv, n);
    listview_add_sub_item (lv, n, 0, info);
    listview_add_sub_item (lv, n, 1, keyid);
    listview_add_sub_item (lv, n, 2, get_key_created (time (NULL)));
    listview_add_sub_item (lv, n, 3, expdate);
    if (flags & KM_FLAG_REVOKED)
        s = _("Revoked");       
    else if (flags & KM_FLAG_EXPIRED)
        s = _("Expired");
    else 
        s = _("OK");
    listview_add_sub_item (lv, n, 4, s);
}


/* Try to find the GPG edit key index which belongs to the user ID
   given by the email address @email, @name is used as a fallback.
   If @r_inf != NULL, the info context will be returned.
   Return value: index of the user ID or -1 on error. */
static int
do_find_userid (const char *keyid, const char *email, 
                const char *name, gpg_uid_info_t *r_inf)
{
    GpgKeyEdit *ke;
    gpgme_error_t err;
    gpg_uid_info_t inf, ui;
    int pos = -1;

    ke = new GpgKeyEdit (keyid);
    if (!ke)
        BUG (NULL);
    err = ke->getUseridInfo (&inf);
    delete ke;
    if (err) {
        log_box (_("user ID"), MB_ERR, 
                _("Could not get key information for: \"%s\":\n%s"), 
                name, gpgme_strerror (err));
        return -1;
    }

    for (ui = inf; ui; ui = ui->next) {
        if (name && email && ui->email && ui->name) {
            if (!strcmp (ui->email, email) && 
                !strncmp (ui->name, name, strlen (name))) {
                pos = ui->index;
                break;
            }
            continue;
        }
        if (ui->email) {
            if (!strcmp (ui->email, email)) {
                pos = ui->index;
                break;
            }
            /* The email address is more unique, use the name just
               as the fallbck when no email address is available. */
            continue;
        }
        if (ui->name && name && !strcmp (ui->name, name)) {
            pos = ui->index;
            break;
        }
    }
    if (r_inf)
        *r_inf = inf;
    else
        gpg_uid_info_release (inf);
    return pos;
}


/* Return true if @fname is a JPEG file. */
bool
is_jpg_file (const char *fname)
{
    FILE *fp;
    BYTE buf[10];
    int n;

    fp = fopen (fname, "rb");
    if (!fp)
        return false;
    n = fread (buf, 1, 10, fp);
    fclose (fp);
    if (n < 10)
        return false;
    return buf[6] == 'J' && buf[7] == 'F' &&
           buf[8] == 'I' && buf[9] == 'F';
}


/* Dialog box procedure to add a photo. */
BOOL CALLBACK
keyedit_addphoto_dlg_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam)
{
    static winpt_key_t k;        
    GpgKeyEdit *ke;
    gpgme_error_t ec;    
    const char * s;    
    char pwd[128], file[128];
    int id;

    switch( msg ) {
    case WM_INITDIALOG:
        k = (winpt_key_t)lparam;
        if (!k)
            BUG (NULL);
        SetDlgItemText (dlg, IDC_ADDPHOTO_INF, _("Remember that the image is stored within your public key.  If you use a very large picture, your key will become very large as well! Keeping the image close to 240x288 is a good size to use."));
        SetDlgItemText (dlg, IDC_ADDPHOTO_FILEINF, _("Pick an image to use for your photo ID.\nThe image must be a JPEG file."));
        SetDlgItemText (dlg, IDC_ADDPHOTO_PWDINF, _("Passphrase"));
        SetDlgItemText (dlg, IDCANCEL, _("&Cancel"));
        SetWindowText (dlg, _("Add Photo ID"));
        SetForegroundWindow (dlg);
        break;

    case WM_DESTROY:
        break;

    case WM_SYSCOMMAND:
        if (LOWORD (wparam) == SC_CLOSE)
            EndDialog (dlg, TRUE);
        break;

    case WM_COMMAND:
        switch( LOWORD( wparam ) ) {

        case IDC_ADDPHOTO_SELFILE:
            s = get_fileopen_dlg (dlg, _("Select Image File"), 
                                  _("JPEG Files (*.jpg, *.jpeg)\0*.jpg;*.jpeg\0\0"),
                                  NULL);
            if (s && !is_jpg_file (s)) {
                log_box (_("Add Photo"), MB_ERR, 
                         _("'%s' is not a valid JPEG file."), s);
                return FALSE;
            }
            if (s && *s)
                SetDlgItemText (dlg, IDC_ADDPHOTO_FILE, s);
            break;

        case IDOK:
            if (!GetDlgItemText (dlg, IDC_ADDPHOTO_FILE, file, sizeof (file)-1)){
                msg_box( dlg, _("Please enter a file name."), _("Add Photo"), MB_ERR);
                return FALSE;
            }
            if (get_file_size (file) == 0 || get_file_size (file) > 6144 ) {
                id = msg_box (dlg, _("The JPEG is really large.\n"
                                     "Are you sure you want to use it?"), 
                                     _("Add Photo"), MB_YESNO|MB_INFO);
                if (id == IDNO)
                    return TRUE;
            }
            if (k->is_protected &&
                !GetDlgItemText (dlg, IDC_ADDPHOTO_PASS, pwd, sizeof (pwd)-1)) {
                msg_box (dlg, _("Please enter a passphrase."), _("Add Photo"), MB_ERR);
                return FALSE;
            }
            ke = new GpgKeyEdit (k->keyid);
            if (!ke)
                BUG (NULL);
            if (k->is_protected)
                ke->setPassphrase (pwd);
            else
                ke->setNoPassphrase (true);
            ec = ke->addPhotoid (file);
            delete ke;
            wipememory (pwd, sizeof (pwd));
            if (ec) {
                msg_box (dlg, gpgme_strerror (ec), _("Add Photo"), MB_ERR );
                return FALSE;
            }
            else {
                k->update = 1;
                msg_box (dlg, _("Photo successfully added."), 
                         _("GnuPG Status"), MB_OK);
            }
            EndDialog (dlg, TRUE);
            break;

        case IDCANCEL:
            EndDialog (dlg, FALSE);
            break;
        }
        break;
    }
    return FALSE;
}


/* Dialog box procedure to add a designated revoker. */
BOOL CALLBACK
keyedit_addrevoker_dlg_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam)
{
    static winpt_key_t k;
    static gpgme_key_t seckey;
    gpgme_error_t err;
    GpgKeyEdit *ke;
    char uid[128], pwd[128];

    switch (msg) {
    case WM_INITDIALOG:
        k = (winpt_key_t)lparam;
        if (!k)
            BUG (NULL);
        if (get_seckey (k->keyid, &seckey))
            BUG (NULL);
        if (!k->is_protected)
            EnableWindow (GetDlgItem (dlg, IDC_ADDREV_PASS), FALSE);
        do_init_keylist (dlg, k);
        SetDlgItemText (dlg, IDC_ADDREV_INF, 
                        _("Appointing a key as designated revoker cannot be undone."));
        SetDlgItemText (dlg, IDC_ADDREV_KEYINF, _("Public key"));
        SetDlgItemText (dlg, IDC_ADDREV_PWDINF, _("Passphrase"));
        SetDlgItemText (dlg, IDCANCEL, _("&Cancel"));
        SetWindowText (dlg, _("Add Revoker"));
        SetForegroundWindow (dlg);
        break;

    case WM_DESTROY:    
        break;

    case WM_SYSCOMMAND:
        if( LOWORD (wparam) == SC_CLOSE )
            EndDialog( dlg, TRUE );
        break;

    case WM_COMMAND:
        switch( LOWORD( wparam ) ) {
        case IDOK:          
            if( !GetDlgItemText( dlg, IDC_ADDREV_KEYLIST, uid, sizeof uid-1 ) ) {
                msg_box( dlg, _("Please select a user ID."), _("Add Revoker"), MB_ERR );
                return FALSE;
            }
        
            if( k->is_protected ) {
                if( !GetDlgItemText( dlg, IDC_ADDREV_PASS, pwd, sizeof pwd-1 ) ) {
                    msg_box( dlg, _("Please enter the passphrase."), _("Add Revoker"), MB_ERR );
                    return FALSE;
                }
            }
            ke = new GpgKeyEdit (k->keyid);
            if (k->is_protected)
                ke->setPassphrase (pwd);
            else
                ke->setNoPassphrase (true);
            err = ke->addDesignatedRevoker (uid);
            delete ke;
            wipememory (pwd, sizeof (pwd));
            if (err) {
                msg_box (dlg, gpgme_strerror (err), _("Add Revoker"), MB_ERR);
                return TRUE;
            }
            else {
                k->update = 1;
                msg_box (dlg, _("Revoker successfully addded."), 
                         _("GnuPG Status"), MB_OK);
            }
            EndDialog (dlg, TRUE);
            break;

        case IDCANCEL:
            EndDialog (dlg, FALSE);
            break;
        }
        break;
    }
    return FALSE;
}


/* Dialog box procedure to add a new user-ID. */
BOOL CALLBACK
keyedit_adduid_dlg_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam)
{
    static keyedit_cb_t ctx;
    gpgme_error_t err;
    GpgKeyEdit *ke;
    char *utf8_name = NULL;
    char name[128], email[128], comment[128];
    int rc;
    
    switch (msg) {
    case WM_INITDIALOG:
        ctx = (keyedit_cb_t)lparam;
        if( !ctx )
            dlg_fatal_error(dlg, "Could not get dialog param!");
        SetWindowText (dlg, _("Add new User ID"));
        SetDlgItemText (dlg, IDC_ADDUID_INFNAME, _("&Name"));
        SetDlgItemText (dlg, IDC_ADDUID_INFEMAIL, _("&Email"));
        SetDlgItemText (dlg, IDC_ADDUID_INFCOMMENT, _("&Comment"));
        SetDlgItemText (dlg, IDCANCEL, _("&Cancel"));
        SetForegroundWindow (dlg);
        return FALSE;
        
    case WM_SYSCOMMAND:
        if (LOWORD (wparam) == SC_CLOSE)
            EndDialog(dlg, TRUE);
        return FALSE;
        
    case WM_COMMAND:
        switch ( LOWORD( wparam ) )  {
        case IDOK:
            rc = GetDlgItemText( dlg, IDC_ADDUID_NAME, name, sizeof name-1 );
            if (!rc || rc < 5) {
                msg_box( dlg, _("Please enter a name (min. 5 chars.)"), _("UserID"), MB_ERR );
                return FALSE;
            }
            if (strchr (name, '@')) {
                msg_box( dlg, _("Please enter the email address in the email field and not in the name field"), _("UserID"), MB_INFO );
                return FALSE;
            }
                                   
            if( !GetDlgItemText( dlg, IDC_ADDUID_EMAIL, email, sizeof email -1 ) ) {
                msg_box( dlg, _("Please enter an email address."), _("UserID"), MB_ERR );
                return FALSE;
            }
            if (!strchr (email, '@' ) || strchr (email, ' ')) {
                msg_box (dlg, _("Invalid email address."), _("UserID"), MB_ERR);
                return FALSE;
            }
            
            rc = GetDlgItemText (dlg, IDC_ADDUID_COMMENT, 
                                 comment, sizeof comment -1);

            /* XXX: something is wrong with the encoding :-( */
            utf8_name = wincp_to_utf8 (name, strlen (name));

            ke = new GpgKeyEdit (ctx->keyid);
            if (!ke)
                BUG (NULL);
            if (ctx->is_protected)
                ke->setPassphrase (ctx->pass);
            else
                ke->setNoPassphrase (true);
            err = ke->addUserid (utf8_name? utf8_name : name, 
                                 rc > 0? comment : NULL, email);
            if (err)
                msg_box (dlg, gpgme_strerror (err), _("UserID"), MB_ERR);
            else {
                msg_box (dlg, _("user ID successfully added."), _("GnuPG Status"), MB_OK);
                ctx->finished = 1;
            }
            delete ke;
            free (utf8_name);
            if (!err && ctx->lv)
                do_add_new_userid (ctx->lv, name, email, rc?comment : NULL);
            EndDialog (dlg, TRUE);
            return TRUE;
            
        case IDCANCEL:
            EndDialog (dlg, FALSE);
            return FALSE;
        }
        break;
    }
    
    return FALSE;
}


static void
init_keysize_box (HWND dlg, int ctlid)
{
    const char *sizelist[] = {
        "1024", "1536", "2048", "2560", "3072", "3854", "4096", NULL
    };
    int i;
    for (i=0; sizelist[i] != NULL; i++)
        SendDlgItemMessage (dlg, ctlid, CB_ADDSTRING, 0, (LPARAM)(char*)sizelist[i]);
    SendDlgItemMessage (dlg, ctlid, CB_SETCURSEL, (WPARAM)2, 0);
}

static int
get_keysize_from_box (HWND dlg, int ctlid)
{
    int pos;
    char buf[32];

    pos = SendDlgItemMessage (dlg, ctlid, CB_GETCURSEL, 0, 0);
    if (pos == CB_ERR)
        return -1;
    SendDlgItemMessage (dlg, ctlid, CB_GETLBTEXT, pos, (LPARAM)(char*)buf);
    return atol (buf);
}


/* Create a time_t from a system time @st. */
time_t
w32_mktime (SYSTEMTIME *st)
{
    struct tm tm;

    memset (&tm, 0, sizeof (tm));
    tm.tm_year = st->wYear-1900;
    tm.tm_mday = st->wDay+1;
    tm.tm_mon = st->wMonth-1;
    return mktime (&tm);
}


/* Dialog procedure for adding a new secondary key. */
BOOL CALLBACK
keyedit_addsubkey_dlg_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam)
{
    static keyedit_cb_t ctx;
    static keygen_cb_t keygen;
    GpgKeyEdit *ke;
    gpgme_error_t err;
    SYSTEMTIME st;
    HWND lb;
    int index, size, valid;
    
    switch (msg) {
    case WM_INITDIALOG:
        ctx = (keyedit_cb_t)lparam;
        if (!ctx)
            dlg_fatal_error (dlg, "Could not get dialog param!");
        keygen = (keygen_cb_t)ctx->opaque;

        SetWindowText (dlg, _("Add new Subkey"));
        SetDlgItemText (dlg, IDC_ADDSUBKEY_INFALGO, _("Key type"));
        SetDlgItemText (dlg, IDC_ADDSUBKEY_INFSIZE, _("Size in bits"));
        SetDlgItemText (dlg, IDC_ADDSUBKEY_INFVALID, _("Key expiration"));
        SetDlgItemText (dlg, IDC_ADDSUBKEY_EXPIRE, _("&Never"));
        SetDlgItemText (dlg, IDCANCEL, _("&Cancel"));

        lb = GetDlgItem (dlg, IDC_ADDSUBKEY_ALGO);
        listbox_add_string (lb, "DSA (sign only)");
        listbox_add_string (lb, "ElGamal (encrypt only)");
        listbox_add_string (lb, "RSA (sign only)");
        listbox_add_string (lb, "RSA (encrypt only)");
        CheckDlgButton (dlg, IDC_ADDSUBKEY_EXPIRE, BST_CHECKED);
        EnableWindow (GetDlgItem (dlg, IDC_ADDSUBKEY_EXPDATE), FALSE);
        init_keysize_box (dlg, IDC_ADDSUBKEY_SIZE);
        SetForegroundWindow (dlg);
        return FALSE;
        
    case WM_SYSCOMMAND:
        if (LOWORD (wparam) == SC_CLOSE) {
            EndDialog (dlg, TRUE);
        }
        return FALSE;
        
    case WM_COMMAND:
        if (HIWORD (wparam) == BN_CLICKED && 
            LOWORD (wparam) == IDC_ADDSUBKEY_EXPIRE) {
            if (IsDlgButtonChecked (dlg, IDC_ADDSUBKEY_EXPIRE))
                EnableWindow (GetDlgItem (dlg, IDC_ADDSUBKEY_EXPDATE), FALSE);
            else
                EnableWindow (GetDlgItem (dlg, IDC_ADDSUBKEY_EXPDATE), TRUE);
        }
        if (HIWORD (wparam) == LBN_SELCHANGE && 
            LOWORD (wparam) == IDC_ADDSUBKEY_ALGO) {
            index = SendMessage ((HWND)lparam, LB_GETCURSEL, 0, 0);
            if (index == 0)
                SendDlgItemMessage (dlg, IDC_ADDSUBKEY_SIZE, CB_SETCURSEL, 0, 0);
        }

        switch (LOWORD (wparam)) {
        case IDOK:
            lb = GetDlgItem (dlg, IDC_ADDSUBKEY_ALGO);
            switch (listbox_get_cursel (lb)) {
            case 0: index = 2; break;
            case 1: index = 4; break;
            case 2: index = 5; break;
            case 3: index = 6; break;
            default:
                msg_box (dlg, _("Please select one entry."), _("Add Subkey"), MB_ERR);
                return FALSE;
            }
            size = get_keysize_from_box (dlg, IDC_ADDSUBKEY_SIZE);
            if (index == 2 && size != 1024) {
                msg_box (dlg,_("DSS uses a fixed keysize of 1024. Size changed."), 
                         _("Add Subkey"), MB_INFO);
                size = 1024;
            }

            DateTime_GetSystemtime (GetDlgItem (dlg, IDC_ADDSUBKEY_EXPDATE), &st);
            valid = w32_mktime (&st) - time (NULL);
            valid /= 86400;

            keygen->bits = size;
            switch (index) {
            case 2: keygen->algo = GPGME_PK_DSA; break;
            case 4: keygen->algo = GPGME_PK_ELG_E; break;
            case 5: keygen->algo = GPGME_PK_RSA_S; break;
            case 6: keygen->algo = GPGME_PK_RSA_E; break;
            }
            if (valid > 0)
                keygen->expire = time (NULL) + valid*24*60*60;

            ke = new GpgKeyEdit (ctx->keyid);
            if (!ke)
                BUG (NULL);
            ke->setCallback (keygen_cb, NULL);
            if (ctx->is_protected)
                ke->setPassphrase (ctx->pass);
            else
                ke->setNoPassphrase (true);
            keygen_cb_dlg_create ();

            err = ke->addSubkey ((gpgme_pubkey_algo_t)index, size, valid);
            keygen->fpr = get_subkey_keyid (ctx->keyid);
            keygen_cb_dlg_destroy ();
            keygen_cb (NULL, NULL, 0, 0, 0); /* flush */
            if (err)
                msg_box (dlg, gpgme_strerror (err), _("Add Subkey"), MB_ERR);
            else {
                msg_box (dlg, _("Subkey successfully added."), _("GnuPG Status"), MB_OK);
                if (ctx->lv)
                    do_add_new_subkey (ctx->lv, keygen, 0);
                ctx->finished = 1;
            }
            delete ke;
            EndDialog (dlg, TRUE);
            return TRUE;
            
        case IDCANCEL:
            EndDialog (dlg, FALSE);
            return FALSE;
        }
        break;
    }
    
    return FALSE;
}


BOOL
keyedit_add_userid (winpt_key_t k, HWND dlg, listview_ctrl_t lv)
{
    keyedit_cb_s cb;
    char *pass = NULL;
    int cancel = 0;

    if (!k->key_pair) {
        msg_box( dlg, _("There is no secret key available!"), _("Add user ID"), MB_ERR );
        return FALSE;
    }

    if (k->is_protected) {
        pass = request_passphrase( _("Key Edit"), 1, &cancel );
        if (cancel)
            return FALSE;
    }
           
    memset (&cb, 0, sizeof cb);
    cb.is_protected = k->is_protected;
    cb.pass = pass;
    cb.lv = lv;
    cb.keyid = k->keyid;
    dialog_box_param (glob_hinst, (LPCSTR)IDD_WINPT_KEYEDIT_ADDUID,
                      dlg, keyedit_adduid_dlg_proc,         
                      (LPARAM)&cb, _("Add user ID"),
                      IDS_WINPT_KEYEDIT_ADDUID);

    if (cb.finished)
        k->update = 1;

    sfree_if_alloc (pass);
    return TRUE;
}


char*
get_subkey_keyid (const char *keyid)
{
    gpgme_error_t err;
    gpgme_key_t key;
    gpgme_ctx_t ctx;
    gpgme_subkey_t subk;
    char *kid;

    err = gpgme_new (&ctx);
    if (err)
        return NULL;
    err = gpgme_get_key (ctx, keyid, &key, 0);
    gpgme_release (ctx);
    if (err)
        return NULL;
    subk = get_nth_key (key, count_subkeys (key));
    kid = strdup (subk->keyid);
    gpgme_key_release (key);
    return kid;
}


BOOL
keyedit_add_subkey (winpt_key_t k, HWND dlg, listview_ctrl_t lv)
{
    keyedit_cb_s cb;
    keygen_cb_s keygen;
    char *pass = NULL;
    int cancel = 0;

    if (!k->key_pair) {
        msg_box (dlg, _("There is no secret key available!"), _("Add Subkey"), MB_ERR);
        return FALSE;
    }
    if (k->is_protected) {
        pass = request_passphrase (_("Key Edit"), 1, &cancel);
        if (cancel)
            return FALSE;
    }

    memset (&keygen, 0, sizeof (keygen));
    memset (&cb, 0, sizeof (cb));
    cb.lv = lv;
    cb.keyid = k->keyid;
    cb.is_protected = k->is_protected;
    cb.pass = pass;
    cb.opaque = &keygen;
    dialog_box_param (glob_hinst, (LPCSTR)IDD_WINPT_KEYEDIT_ADDSUBKEY,    
                      dlg, keyedit_addsubkey_dlg_proc,
                      (LPARAM)&cb, _("Add new Subkey"),
                      IDS_WINPT_KEYEDIT_ADDSUBKEY);
    safe_free (keygen.fpr);
    if (cb.finished)
        k->update = 1;

    sfree_if_alloc (pass);    
    return cb.finished? TRUE: FALSE;
}


BOOL
keyedit_set_pref_keyserver (winpt_key_t k, HWND dlg)
{
    GpgKeyEdit *ke;
    gpgme_error_t err;
    struct URL_ctx_s *url;
    char *pass;

    url = (struct URL_ctx_s *)get_keyserver_URL_dlg (dlg);
    if (url->cancel == 1) {
        delete url;
        return FALSE;
    }

    pass = request_passphrase (_("Key Edit"), 1, &url->cancel);
    if (url->cancel) {
        delete url;
        return FALSE;
    }

    ke = new GpgKeyEdit (k->keyid);
    if (!ke)
        BUG (NULL);
    if (k->is_protected)
        ke->setPassphrase (pass);
    else
        ke->setNoPassphrase (true);
    err = ke->setPreferredKeyserver (0 /* XXX */, url->url);
    if (!err)
        msg_box (dlg, _("Preferred keyserver successfully set."), _("Key Edit"), MB_OK);

    sfree_if_alloc (pass);
    delete ke;
    delete url;    
    return err == 0? 0 : WPTERR_GENERAL;
}


/* Add a photo-ID to the key specified in @k. @dlg is the handle of
   the calling dialog. */
BOOL
keyedit_add_photo (winpt_key_t k, HWND dlg)
{
    if (!k->key_pair) {
        msg_box (dlg, _("There is no secret key available!"), _("Add Photo"), MB_ERR);
        return FALSE;
    }
    DialogBoxParam (glob_hinst, (LPCTSTR)IDD_WINPT_KEYEDIT_ADDPHOTO, dlg,
                    keyedit_addphoto_dlg_proc, (LPARAM)k);
    return TRUE;
}


BOOL
keyedit_add_revoker (winpt_key_t k, HWND dlg)
{
    if (!k->key_pair) {
        msg_box (dlg, _("There is no secret key available!"), _("Add Revoker"), MB_ERR);
        return FALSE;
    }
    DialogBoxParam (glob_hinst, (LPCTSTR)IDD_WINPT_KEYEDIT_ADDREV, dlg,
                    keyedit_addrevoker_dlg_proc, (LPARAM)k);
    return TRUE;
}


/* Change ownertrust of the given key @key.
   Return TRUE if the ownertrust was changed. */
BOOL
keyedit_change_ownertrust (winpt_key_t key, HWND dlg)
{
    int rc;

    rc = dialog_box_param (glob_hinst, 
                             (LPCSTR)IDD_WINPT_KEYEDIT_OWNERTRUST,
                             dlg, keyedit_ownertrust_dlg_proc,
                             (LPARAM)key, _("Change Ownertrust"),
                             IDS_WINPT_KEYEDIT_OWNERTRUST);
    if (rc == TRUE) {
        msg_box (dlg, _("Key status changed."), _("Key Edit"), MB_OK);
        key->update = 1;
    }
    return rc;
}


static int
is_idea_protect_algo (const char * keyid)
{
    winpt_key_s k;
    const unsigned char *sym_prefs;
    size_t n;

    memset (&k, 0, sizeof (k));
    if (winpt_get_pubkey (keyid, &k))
        BUG (NULL);
    sym_prefs = k.ext->sym_prefs;
    if (!sym_prefs)
        return 1; /* assume that only v3 keys have no symmetric cipher preferences
                     and thus IDEA is explicit. */
    for (n = 0; sym_prefs[n]; n++)
        ;
    if ((n == 0 || n == 1) && *sym_prefs == 0x01)
        return 1;
    return 0;
} /* is_idea_protect_algo */


BOOL
keyedit_change_passwd (winpt_key_t k, HWND dlg)
{
    gpgme_error_t ec;
    GpgKeyEdit *ke;
    char *old_pass = NULL;
    char *new_pass = NULL;
    int cancel = 0;

    if (!k->key_pair) {
        msg_box (dlg, _("There is no secret key available!"), 
                 _("Key Edit"), MB_ERR);
        return FALSE;
    }

    if( !idea_available && is_idea_protect_algo( k->keyid ) ) {
        msg_box( dlg, _("Cannot change passphrase because the key\n"
                        "is protected with the IDEA encryption algorithm."), 
                        _("Key Edit"), MB_ERR );
        return FALSE;
    }

    if (k->is_protected) {
        old_pass = request_passphrase (_("Current (old) Passphrase"), 1, &cancel);
        if (cancel)
            return FALSE;
    }
    new_pass = request_passphrase2 (_("New Passphrase" ), 1, &cancel);
    if (cancel) {
        free_if_alloc (old_pass);
        return FALSE;
    }

    if (is_8bit_string (new_pass)) {
        msg_box (dlg, _("The passphrase contains 8-bit characters.\n"
                         "It is not suggested to use charset specific characters."),
                         _("Key Edit"), MB_ERR);
        free_if_alloc (old_pass);
        free_if_alloc (new_pass);
        return FALSE;
    }

    ke = new GpgKeyEdit (k->keyid);
    if (!ke)
        BUG (NULL);

    ke->setPassphrase (k->is_protected? old_pass : NULL);
    ec = ke->changePassphrase (new_pass, 0);
    if( ec )
        msg_box (dlg, gpgme_strerror (ec), _("Change Passwd"), MB_ERR);
    else
        msg_box (dlg, _("Passphrase successfully changed."),  _("GnuPG status"), MB_OK);
    sfree_if_alloc (old_pass);
    sfree_if_alloc (new_pass);
    delete ke;
    return TRUE;
}


/* Initialize sub key list from key @k and return
   the new listview control. */
listview_ctrl_t
subkey_list_init (HWND dlg, winpt_key_t k)
{
    LV_ITEM lvi;
    gpgme_key_t key;
    gpgme_subkey_t sub;
    struct listview_column_s cols[] = {
        {0, 80, (char *)_("Description")},
        {1, 78, (char *)_("Key ID")},
        {2, 66, (char *)_("Creation")},
        {3, 66, (char *)_("Expires")},
        {4, 64, (char *)_("Status")},
        {5, 16,  "C"/*ertify*/},
        {6, 16,  "S"/*ign*/},
        {7, 16,  "E"/*ncrypt*/},
        {8, 16,  "A"/*uth*/},
        {0, 0, 0}
    };
    listview_ctrl_t lv;
    char buf[256], tmp[128];
    const char *t;
    int nkeys = 0, rc = 0, i, bits;    

    if( get_pubkey( k->keyid, &key ) ) {
        msg_box( dlg, _("Could not find key."), _("Key Edit"), MB_ERR );
        return NULL;
    }
    if (!k->ctx)
        k->ctx = key;
    nkeys = count_subkeys (key);
    if( !nkeys ) {
        msg_box (dlg, _("No subkey(s) found."), _("Key Edit"), MB_ERR);
        return NULL;
    }
        
    rc  = listview_new( &lv );
    if( rc )
        BUG( dlg );
        
    lv->ctrl = GetDlgItem( dlg, IDC_KEYEDIT_KEYLIST );
    for( i = 0; cols[i].fieldname != NULL; i++ )     
        listview_add_column( lv, &cols[i] );
        
    for( i = 0; i < nkeys; i++ ) {
        listview_add_item( lv, "" );
        listview_add_sub_item( lv, 0, 1, "" );
        memset( &lvi, 0, sizeof lvi );
        lvi.mask = LVIF_PARAM;  
        lvi.lParam = (LPARAM)key;
        if( ListView_SetItem( lv->ctrl, &lvi ) == FALSE )
            return NULL;
    }
        
    listview_set_ext_style( lv );
    for( i = 0, sub = key->subkeys; i < nkeys; i++, sub = sub->next ) {
        memset( buf, 0, sizeof buf );

        bits = sub->length;
        _snprintf( tmp, sizeof tmp-1, "%d-bit ", bits );
        strcat( buf, tmp );

        _snprintf( tmp, sizeof tmp-1, "%s", get_key_pubalgo (sub->pubkey_algo));
        strcat( buf, tmp );
        
        listview_add_sub_item( lv, i, 0, buf );
        t = sub->keyid;
        if( !t )
            t = "DEADBEEFDEADBEEF";
        _snprintf( tmp, sizeof tmp-1, "0x%s", t+8 );
        listview_add_sub_item( lv, i, 1, tmp );

        t = get_key_created (sub->timestamp);
        if( !t )
            t = "????" "-??" "-??";
        listview_add_sub_item( lv, i, 2, t );

        if( sub->expires ) {
            t = get_key_created (sub->expires);
            listview_add_sub_item( lv, i, 3, t );
        }
        else
            listview_add_sub_item( lv, i, 3, _("Never") );

        if( sub->expired )
            t = _("Expired");
        else if( sub->revoked )
            t = _("Revoked");
        else
            t = _("OK");
        listview_add_sub_item( lv, i, 4, t );

        if (sub->can_certify) t = "*"; else t = "";
        listview_add_sub_item (lv, i, 5, t);
        if (sub->can_sign) t = "*"; else t = "";
        listview_add_sub_item( lv, i, 6, t );
        if (sub->can_encrypt) t = "*"; else t = "";
        listview_add_sub_item( lv, i, 7, t );
        if (sub->can_authenticate) t = "*"; else t = "";
        listview_add_sub_item (lv, i, 8, t);
    }
    return lv;
} /* subkey_list_init */


static listview_ctrl_t
userid_list_init (HWND dlg, winpt_key_t k)
{
    listview_ctrl_t lv = NULL;
    gpgme_key_t key;
    gpgme_key_sig_t ks;
    gpgme_user_id_t u;
    int nuids = 0, rc, j, u_attr;
    struct listview_column_s cols[] = {
        {0,  72, (char *)_("Validity")},
        {1, 150, (char *)_("Name")},
        {2, 110, (char *)_("Email")},
        {3,  76, (char *)_("Creation")},
        {0, 0, 0}
    };    
    const char *attr;

    if (get_pubkey (k->keyid, &key)) {
        msg_box (dlg, _("Could not find key."), _("Key Edit"), MB_ERR);
        return NULL;
    }
    
    nuids = count_userids (key);
    if (!nuids) {
        msg_box (dlg, _("No user ID(s) found."), _("Key Edit"), MB_ERR);
        return NULL;
    }
        
    rc = listview_new (&lv);
    if (rc)
        BUG (dlg);
    lv->ctrl = GetDlgItem( dlg, IDC_KEYEDIT_UIDLIST );
    for( j = 0; cols[j].fieldname != NULL; j++ )
        listview_add_column( lv, &cols[j] );
        
    for( j = 0; j < nuids; j++ ) {           
        listview_add_item( lv, " " );
        listview_add_sub_item( lv, 0, 1, " " );        
    }

    listview_set_ext_style (lv);
    for (j = 0, u=key->uids; j < nuids; u=u->next, j++) {
        if (u->revoked)
            attr = _("Revoked");
        else {
            u_attr = (int)u->validity;
            attr = get_key_trust2 (NULL, u_attr, 0, 0);
        }
        listview_add_sub_item( lv, j, 0, (char *)attr );        

        /* XXX: add comment if available */
        attr = u->name;
        if (attr) {
            char *uid = utf8_to_wincp (attr, strlen (attr));
            if (uid) {
                listview_add_sub_item (lv, j, 1, uid);
                free (uid);
            }
        }
        else
            listview_add_sub_item (lv, j, 1, _("Invalid user ID"));
        attr = u->email;
        if (attr)
            listview_add_sub_item (lv, j, 2, attr);

        ks = get_selfsig (u, k->keyid, 1);
        if (ks)
            listview_add_sub_item (lv, j, 3, get_key_created (ks->timestamp));
    }
    if( !k->key_pair ) {
        CheckDlgButton( dlg, IDC_KEYUID_ADD, BST_INDETERMINATE );
        CheckDlgButton( dlg, IDC_KEYUID_REVOKE, BST_INDETERMINATE );    
    }
    return lv;
} /* userid_list_init */


static void
do_init_cmdlist (HWND dlg, int is_keypair)
{    
    const char *s;
    int i = 0;

    for (i = 0; (s=cmdlist[i].name); i++) {
        if (is_keypair)
            SendDlgItemMessage (dlg, IDC_KEYEDIT_CMD, CB_ADDSTRING, 0,
                                (LPARAM)(char *)s);
        else if (!cmdlist[i].need_pair)
            SendDlgItemMessage (dlg, IDC_KEYEDIT_CMD, CB_ADDSTRING, 0,
                                (LPARAM)(char *)s);
    }
    SendDlgItemMessage (dlg, IDC_KEYEDIT_CMD, CB_SETCURSEL, 0, 0);
}


/* Return 1 if the requested command is RFC2440. */
static int
is_cmd_openpgp (int cmdid)
{
    switch (cmdid) {
    case CMD_ADDKEY:
    case CMD_ADDPHOTO:
    case CMD_ADDREVOKER:
    //case CMD_SETPREF:
        return 1;
    }
    return 0;
}


/* Display a message box with a short description of the commands. */
static void
do_show_help (HWND dlg)
{
    char helptext[2048];

    _snprintf (helptext, sizeof (helptext)-1,
        _(
         "ADDUID   \t\tadd a user ID\r\n"
         "ADDPHOTO  \t\tadd a photo ID\r\n"
         "DELUID    \t\tdelete a user ID\r\n"
         "ADDKEY    \t\tadd a secondard key\r\n"
         "DELKEY    \t\tdelete a secondary key\r\n"
         "ADDREVOKER\t\tadd a revocation key\r\n"
         "EXPIRE    \t\tchange the expire date\r\n"
         "SHOWPREF  \t\tlist preferences (verbose)\r\n"
         "SETPREF   \t\tset preference list\r\n"
         "UPDPREF   \t\tupdated preferences\r\n"
         "PASSWD    \t\tchange the passphrase\r\n"
         "PRIMARY   \t\tflag user ID as primary\r\n"
         "TRUST     \t\tchange the ownertrust\r\n"
         "REVUID    \t\trevoke a user ID\r\n"
         "REVKEY    \t\trevoke a secondary key\r\n"
         "DISABLE   \t\tdisable a key\r\n"
         "ENABLE    \t\tenable a key\r\n"
         "SIGN      \t\tsign a user-id (exportable)\r\n"
         "LSIGN     \t\tsign a user-id (non-exportable)\r\n"));
    msg_box (dlg, helptext, _("Key Edit Help"), MB_OK);
}


static int
do_editkey_delkey (winpt_key_t k, HWND dlg, listview_ctrl_t lv)
{
    gpgme_error_t err;
    GpgKeyEdit *ke;
    int j, id;
    char tmp[64];

    if (!k->key_pair)
        return FALSE; /* XXX: shall we allow to modify non-secret keys?? */

    if( listview_count_items( lv, 0 ) == 1 ) {
        msg_box( dlg, _("Primary key can not be deleted!"), _("Key Edit"), MB_ERR);
        return FALSE;
    }
    if( (j = listview_get_curr_pos( lv )) == -1 ) {
        msg_box( dlg, _("Please select a key."), _("Key Edit"), MB_ERR );
        return FALSE;
    }
    if( j == 0 ) {
        msg_box( dlg, _("Primary subkey can not be deleted!"), _("Key Edit"), MB_ERR );
        return FALSE;
    }
     
    listview_get_item_text( lv, j, 0, tmp, sizeof tmp -1 );
    id = log_box( _("Key Edit"), MB_YESNO|MB_ICONWARNING,
                    _("\"Subkey %s.\"\n\n"
                      "Anything encrypted to the selected subkey will no longer\n"
                      "be able to be decrypted.\n\n"
                      "Do you really want to delete this subkey?"), tmp );
    if( id == IDNO )
        return FALSE;

    ke = new GpgKeyEdit (k->keyid);
    if (!ke)
        BUG (NULL);
    err = ke->delKey (j);
    if (err)
        msg_box (dlg, gpgme_strerror (err), _("Delete Subkey"), MB_ERR);
    else {
        listview_del_item (lv, j);
        k->update = 1;
        status_box (dlg, _("Subkey successfully deleted."), _("GnuPG status"));
    }
    delete ke;
    return err? FALSE : TRUE;
} /* do_editkey_delkey */


/* Set the expiration date for the selected key in list view @lv.
   Return value: TRUE on success. */
static int
do_editkey_expire (winpt_key_t k, HWND dlg, listview_ctrl_t lv)
{
    gpgme_error_t err;
    GpgKeyEdit *ke;
    date_s udd = {0};
    char buf[256], * pass = NULL;
    int j, cancel = 0;

    if (!k->key_pair) {
        msg_box (dlg, _("There is no secret key available!"), _("Key Edit"), MB_ERR);
        return FALSE;
    }
    if ((j = listview_get_curr_pos (lv)) == -1) {
        msg_box( dlg, _("Please select a key."), _("Key Edit"), MB_ERR );
        return FALSE;
    }

    /* If a key already expired, it is possible the user wants to
       set a new expiration date.. */   
    listview_get_item_text (lv, j, SUBK_COL_STATUS, buf, sizeof buf -1);
    if (!strcmp (buf, _("Expired"))) {
        cancel = msg_box (dlg, _("Key already expired.\n\n"
                          "Do you want to change the expiration date?"),
                          _("Key Edit"), MB_QUEST_ASK);
        if (cancel == IDNO)
            return FALSE;
        cancel = 0;
    }

    memset (&udd, 0, sizeof udd);
    udd.text = _("Key Expiration Date");
    dialog_box_param (glob_hinst, (LPCSTR)IDD_WINPT_DATE, dlg,    
                      date_dlg_proc, (LPARAM)&udd,
                      _("Key Expiration Date"), IDS_WINPT_DATE);
    if (udd.cancel == 1)
        return FALSE;
    if (!keygen_check_date (&udd.st)) {
        msg_box (dlg, _("The date you have chosen lies in the past."), 
                 _("Key Edit"), MB_ERR);
        return FALSE;
    }
    if( k->is_protected ) {
        pass = request_passphrase (_("Key Edit"), 1, &cancel);
        if (cancel)
            return FALSE;
    }

    ke = new GpgKeyEdit (k->keyid);
    if (!ke)
        BUG (NULL);
    if (k->is_protected)
        ke->setPassphrase (pass);
    else
        ke->setNoPassphrase (true);
    err = ke->setKeyExpireDate (j, w32_mktime (&udd.st), false);
    if (err)
        msg_box (dlg, gpgme_strerror (err), _("Expire Subkey"), MB_ERR);
    else {
        _snprintf (buf, sizeof buf - 1, "%04d-%02d-%02d",       
                   udd.st.wYear, udd.st.wMonth, udd.st.wDay);
        listview_add_sub_item (lv, j, SUBK_COL_EXPIRES, buf);
        k->update = 1;
        msg_box (dlg, _("Subkey expire date successfully set."), 
                 _("GnuPG status"), MB_OK);
    }
    sfree_if_alloc (pass);
    delete ke;
    return TRUE;
}


/* Revoke the selected key in the list view @lv. @k contains 
   control information about the global key.
   Return value: TRUE on success. */
static int
do_editkey_revoke (winpt_key_t k, HWND dlg, listview_ctrl_t lv)
{
    gpgme_error_t err;
    GpgKeyEdit *ke;
    char buf[256];
    char *pass = NULL;
    int j, cancel = 0;

    if (!k->key_pair) {
        msg_box (dlg, _("There is no secret key available!"), _("Key Edit"), MB_ERR);
        return FALSE;
    }

    if ((j = listview_get_curr_pos (lv)) == -1) {
        msg_box( dlg, _("Please select a key."), _("Key Edit"), MB_ERR );
        return FALSE;
    }
    else if (listview_count_items (lv, 0) == 1) {
        msg_box( dlg, _("No subkeys were found, if you want to revoke the\n"
                        "whole key, please use the Key Manager command directly.\n\n"
                        "This command is only available to revoke single subkeys"),
                 _("Key Edit"), MB_INFO );
        return FALSE;
    }
            
    listview_get_item_text (lv, j, SUBK_COL_STATUS, buf, sizeof (buf)-1);
    if (!strcmp (buf, _("Revoked"))) {
        msg_box (dlg, _("Key already revoked."), _("Key Edit"), MB_ERR);
        return FALSE;
    }
    
    if (k->is_protected) {
        pass = request_passphrase (_("Key Edit"), 1, &cancel);
        if (cancel)
            return FALSE;           
    }

    ke = new GpgKeyEdit (k->keyid);
    if (!ke)
        BUG (NULL);
    if (k->is_protected)
        ke->setPassphrase (pass);
    else
        ke->setNoPassphrase (true);
    err = ke->revokeSubkey (j, 0, NULL);
    if (err)
        msg_box( dlg, gpgme_strerror (err), _("Revoke Subkey"), MB_ERR);
    else {
        listview_add_sub_item (lv, j, SUBK_COL_STATUS, _("Revoked"));
        k->update = 1;
        msg_box( dlg, _("Subkey successfully revoked."), _("GnuPG Status"), MB_OK );
    }
    sfree_if_alloc (pass);
    delete ke;
    return TRUE;
}


/* Revoked the selected userid in list view @lv.
   Return value: TRUE on success. */
int
do_editkey_revuid (winpt_key_t k, HWND dlg, listview_ctrl_t lv)
{
    gpgme_error_t err;
    GpgKeyEdit *ke;
    char buf[128], email[128];
    char inf[512];
    char *pass = NULL;
    int cancel = 0, id = 0, j;

    if (!k->key_pair) {
        msg_box (dlg, _("There is no secret key available!"), 
                 _("Revoke user ID"), MB_ERR);
        return FALSE;
    }

    if (listview_count_items (lv, 0) == 1) {
        msg_box (dlg, _("Key has only one user ID."), _("Key Edit"), MB_ERR);   
        return FALSE;
    }

    if( (j = listview_get_curr_pos( lv )) == -1 ) {
        msg_box( dlg, _("Please select a user ID."), _("Key Edit"), MB_ERR );   
        return FALSE;   
    }
            
    listview_get_item_text( lv, j, UID_COL_VALID, buf, sizeof buf - 1);
    if (strstr (buf, _("Revoked"))) {
        msg_box (dlg, _("This user ID has been already revoked."), 
                 _("Key Edit"), MB_INFO);
        return FALSE;
    }
            
    listview_get_item_text (lv, j, UID_COL_NAME, buf, sizeof buf -1);
    _snprintf (inf, sizeof (inf) -1, _("user ID \"%s\".\n\n"
               "Do you really want to revoke this user ID?"), buf);
    if (msg_box (dlg, inf, _("Key Edit"), MB_WARN_ASK) == IDNO)
        return FALSE;
    if (k->is_protected) {
        pass = request_passphrase (_("Key Edit"), 1, &cancel);
        if (cancel)
            return FALSE;           
    }
    listview_get_item_text (lv, j, UID_COL_EMAIL, email, sizeof (email)-1);
    listview_get_item_text (lv, j, UID_COL_NAME, buf, sizeof (buf)-1);
    id = do_find_userid (k->keyid, email, buf, NULL);
    if (id == -1)
        BUG (NULL);

    ke = new GpgKeyEdit (k->keyid);
    if (!ke)
        BUG (NULL);
    if (k->is_protected)
        ke->setPassphrase (pass);
    else
        ke->setNoPassphrase (true);
    err = ke->revokeUserid (id);
    if (err)
        msg_box (dlg, gpgme_strerror (err), _("Revoke User ID"), MB_ERR);
    else {
        listview_add_sub_item (lv, j, 0, _("Revoked"));
        k->update = 1;
        status_box (dlg, _("User ID successfully revoked"), _("GnuPG Status"));
    }
    sfree_if_alloc (pass);
    delete ke;
    return err? FALSE : TRUE;
}


static int
do_editkey_setpref (winpt_key_t k, HWND dlg, listview_ctrl_t lv)
{
    gpgme_error_t rc;
    GpgKeyEdit *ke;
    char buf[256], * pass = NULL, * prefs;
    int j, id, cancel=0, flags=0;

    if ((j = listview_get_curr_pos (lv)) == -1) {
        msg_box (dlg, _("Please select a user ID."), _("Key Edit"), MB_ERR);
        return FALSE;
    }
    listview_get_item_text (lv, j, 2, buf, sizeof buf-1);
    id = do_find_userid (k->keyid, buf, NULL, NULL);
    if (id == -1)
        BUG (dlg);
    if (k->is_protected) {
        pass = request_passphrase (_("Key Edit"), 1, &cancel);
        if (cancel)
            return FALSE;
    }

    ke = new GpgKeyEdit (k->keyid);
    if (!ke)
        BUG (NULL);
    if (k->is_protected)
        ke->setPassphrase (pass);
    else
        ke->setNoPassphrase (true);

    get_userid_preflist (&prefs, &flags);

    rc = ke->setUseridPreferences (id, prefs);
    if (rc)
        msg_box (dlg, _("Could not set user ID preferences"), _("Key Edit"), MB_ERR);

    sfree_if_alloc (pass);
    free_if_alloc (prefs);
    delete ke;
    return 0;
}


static int
do_editkey_primary (winpt_key_t k, HWND dlg, listview_ctrl_t lv)
{
    gpgme_error_t err;
    GpgKeyEdit *ke;
    int j, id, cancel=0;
    char buf[256], * pass = NULL;

    if (listview_count_items (lv, 0) == 1)
        return TRUE;
    if ((j = listview_get_curr_pos (lv)) == -1) {
        msg_box( dlg, _("Please select a user ID."), _("Key Edit"), MB_ERR );
        return FALSE;
    }
    listview_get_item_text (lv, j, 2, buf, sizeof buf-1);
    id = do_find_userid (k->keyid, buf, NULL, NULL);
    if (id == -1)
        BUG (dlg);
    if (k->is_protected) {
        pass = request_passphrase (_("Key Edit"), 1, &cancel);
        if (cancel)
            return FALSE;
    }

    ke = new GpgKeyEdit (k->keyid);
    if (k->is_protected)
        ke->setPassphrase (pass);
    else
        ke->setNoPassphrase (true);
    err = ke->setPrimaryUserid (id);
    if (err)
        msg_box (dlg, gpgme_strerror (err), _("Primary"), MB_ERR);
    else {
        k->update = 1;
        status_box (dlg, _("User ID successfully flagged"), _("GnuPG Status"));
    }

    sfree_if_alloc (pass);
    delete ke;
    return err? FALSE : TRUE;
}


#define CIPHER  11
#define HASH    11
#define COMPR    4

static int
parse_preflist (HWND dlg, const char *list)
{
    char buf[128] = {0};
    char *p, *pbuf = buf;
    const char *ciphers[CIPHER] = {"", "IDEA", "3DES", 
                                   "CAST5", "BLOWFISH", "", "", 
                                   "AES", "AES192", "AES256", "TWOFISH"};
    const char *hash[HASH] = {"", "MD5", "SHA1", "RMD160", "",
                              "", "", "", "SHA256", "SHA384", "SHA512"};
    const char *compress[COMPR] = {"", "ZIP", "ZLIB", "BZIP2"};
    int n=0;

    strncpy (buf, list, 127);
    p = strtok (pbuf, " ");
    while (p != NULL) {
        int algid = atol (p+1);
        n++;
        switch (*p) {
        case 'S':
            SendDlgItemMessage (dlg, IDC_SHOWPREF_CIPHERS, LB_ADDSTRING, 0, 
                                (LPARAM)(const char*)ciphers[algid % CIPHER]);
            break;

        case 'H':
            SendDlgItemMessage (dlg, IDC_SHOWPREF_HASH, LB_ADDSTRING, 0, 
                                (LPARAM)(const char*)hash[algid % HASH]);
            break;

        case 'Z':
            SendDlgItemMessage (dlg, IDC_SHOWPREF_ZIP, LB_ADDSTRING, 0, 
                                (LPARAM)(const char*)compress[algid % COMPR]);
            break;

        default:
            n--;
        }
        p = strtok (NULL, " ");
    }
    return n;
}


/* Dialog box procedure to show the key preferences. */
BOOL CALLBACK
showpref_dlg_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam)
{
    static keyedit_cb_t cb = NULL;
    gpg_uid_info_t inf=NULL;
    char buf[128];
    int pos;

    switch (msg) {
    case WM_INITDIALOG:
        cb = (keyedit_cb_t)lparam;
        if (cb == NULL)
            BUG (dlg);
        pos = listview_get_curr_pos (cb->lv);
        listview_get_item_text (cb->lv, pos, 2, buf, DIM (buf)-1);
        SetDlgItemText (dlg, IDC_SHOWPREF_INFO, buf);
        pos = do_find_userid (((winpt_key_t)cb->opaque)->keyid, 
                              buf, NULL, &inf);
        if (inf) {
            const char *prefs = inf->prefs;
            if (prefs && *prefs) {
                if (parse_preflist (dlg, prefs) <= 0)
                    pos = -1;
            }
            else
                pos = -1;
            gpg_uid_info_release (inf);
            if (pos == -1) {
                msg_box (dlg, _("No preferences available."), _("Key Edit"), MB_ERR);
                EndDialog (dlg, TRUE);
            }
            if (inf->flags.mdc)
                CheckDlgButton (dlg, IDC_SHOWPREF_MDC, BST_CHECKED);
        }
        SetDlgItemText (dlg, IDC_SHOWPREF_MDC, _("MDC feature"));
        SetDlgItemText (dlg, IDC_SHOWPREF_PREFINF, _("Preferences"));
        SetWindowText (dlg, _("Key Preferences"));
        SetForegroundWindow (dlg);
        break;

    case WM_COMMAND:
        switch (LOWORD (wparam)) {
        case IDOK:
            EndDialog (dlg, TRUE);
            break;
        }
        break;
    }
    return FALSE;
}


static int
do_editkey_showpref (winpt_key_t k, HWND dlg, listview_ctrl_t lv)
{
    struct keyedit_cb_s cb;

    if (k->is_v3)
        return TRUE;

    if (listview_get_curr_pos (lv) == -1) {
        msg_box (dlg, _("Please select a user ID."), _("Key Edit"), MB_ERR);
        return FALSE;
    }

    memset (&cb, 0, sizeof (cb));
    cb.lv = lv;
    cb.opaque = k;
    DialogBoxParam (glob_hinst, (LPCTSTR)IDD_WINPT_KEYEDIT_SHOWPREF, dlg,
                    showpref_dlg_proc, (LPARAM)&cb);
    return TRUE;
}


static int
do_editkey_deluid (winpt_key_t k, HWND dlg, listview_ctrl_t lv)
{
    gpgme_error_t err;
    GpgKeyEdit *ke;
    char email[128], name[128];
    char inf[384];
    int j, id = 0;

    if (!k->key_pair)
        return FALSE; /* XXX: see do_editkey_delsubkey */

    if (listview_count_items (lv, 0) == 1) {
        msg_box (dlg, _("Primary user ID can not be deleted!"),
                 _("Key Edit"), MB_ERR);
        return FALSE;
    }
    if ((j = listview_get_curr_pos (lv)) == -1) {
        msg_box (dlg, _("Please select a user ID."), _("Key Edit"), MB_ERR);
        return FALSE;
    }
    
    /* XXX: add a hint that also all signatures will be deleted? */
    listview_get_item_text (lv, j, UID_COL_NAME, name, DIM(name) -1);
    _snprintf (inf, DIM (inf)-1, _("user ID \"%s\".\n\n"
                                   "Do you really want to delete this user ID?"),
                               name);
    if (msg_box (dlg, inf, _("Key Edit"), MB_YESNO|MB_ICONWARNING) == IDNO)
        return FALSE;
    
    listview_get_item_text (lv, j, UID_COL_EMAIL, email, DIM (email)-1);
    listview_get_item_text (lv, j, UID_COL_NAME, name, DIM (name)-1);
    id = do_find_userid (k->keyid, email, name, NULL);
    if (id == -1)
        BUG (dlg);

    ke = new GpgKeyEdit (k->keyid);
    if (!ke)
        BUG (NULL);

    err = ke->delUserid (id);
    if (err)
        msg_box (dlg, gpgme_strerror (err), _("Delete user ID"), MB_ERR);
    else {
        listview_del_item (lv, j);
        k->update = 1;
        status_box (dlg, _("User ID successfully deleted"), _("GnuPG Status"));
    }
    delete ke;
    return err? FALSE : TRUE;
} /* do_editkey_deluid */


static BOOL CALLBACK
subkey_subclass_proc( HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam )
{
    switch( msg ) {
    case WM_KEYUP:
        int virt_key = (int)wparam;
        switch( virt_key ) {        
        case VK_DELETE:
            SendDlgItemMessage( keyedit_subkey_proc.dlg, IDC_KEYEDIT_CMD, 
                                CB_SETCURSEL, CMD_DELKEY, 0 );
            send_cmd_id( keyedit_subkey_proc.dlg, IDOK );
            break;

        case VK_INSERT:
            SendDlgItemMessage( keyedit_subkey_proc.dlg, IDC_KEYEDIT_CMD, 
                                CB_SETCURSEL, CMD_ADDKEY, 0 );
            send_cmd_id( keyedit_subkey_proc.dlg, IDOK );
            break;
        }
    }
    return CallWindowProc( keyedit_subkey_proc.old, dlg, msg, wparam, lparam );
} /* subkey_subclass_proc */


static BOOL CALLBACK
uid_subclass_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam)
{
    switch( msg ) {
    case WM_KEYUP:
        int virt_key = (int)wparam;
        switch (virt_key) {
        case VK_DELETE:
            SendDlgItemMessage (keyedit_uid_proc.dlg, IDC_KEYEDIT_CMD,
                                CB_SETCURSEL, CMD_DELUID, 0);
            send_cmd_id (keyedit_uid_proc.dlg, IDOK);
            break;

        case VK_INSERT:
            SendDlgItemMessage (keyedit_uid_proc.dlg, IDC_KEYEDIT_CMD,
                                CB_SETCURSEL, CMD_ADDUID, 0);
            send_cmd_id (keyedit_uid_proc.dlg, IDOK);
            break;
        }
    }
    return CallWindowProc( keyedit_uid_proc.old, dlg, msg, wparam, lparam );
} /* uid_subclass_proc */


static void
do_editkey_enable_disable (winpt_key_t k, HWND dlg, listview_ctrl_t lv, int enable)
{
    km_enable_disable_key (lv, dlg, 0, enable);
    k->update = 1;
}


/* Return default secret key. */
static gpgme_key_t
get_default_key (void)
{
    gpgme_key_t def_sk;
    char *keyid = get_gnupg_default_key ();

    get_seckey (keyid, &def_sk);
    free_if_alloc (keyid);
    return def_sk;
}


static void
do_editkey_check (winpt_key_t k, HWND dlg)
{
    if (!k->ctx)
        get_pubkey (k->keyid, &k->ctx);
    if (!k->uid && k->ctx)
        k->uid = k->ctx->uids->uid;
    DialogBoxParam (glob_hinst, (LPCTSTR)IDD_WINPT_KEYSIG_TREE, dlg,
                    sigtree_dlg_proc, (LPARAM)k);
}


static int
do_editkey_sign_userid (winpt_key_t k, HWND dlg, listview_ctrl_t lv, int mode)
{
    gpgme_error_t err;
    GpgKeyEdit *ke;
    char *pass = NULL;
    char email[64], name[128];
    int uid_index;
    int cancel = 0;

    uid_index = listview_get_curr_pos (lv);
    if (uid_index == -1) {
        msg_box (dlg, _("Please select a user ID."), _("Key Edit"), MB_ERR);
        return FALSE;
    }
    if (mode == CMD_SIGN) {
        cancel = msg_box (dlg, _("Do you really want to make this sig exportable?"),
                          _("Key Edit"), MB_QUEST_ASK);
        if (cancel == IDNO)
            return FALSE;
    }

    listview_get_item_text (lv, uid_index, UID_COL_EMAIL, email, sizeof (email)-1);
    listview_get_item_text (lv, uid_index, UID_COL_NAME, name, sizeof (name)-1);
    uid_index = do_find_userid (k->keyid, email, name, NULL);
    if (k->is_protected) {
        pass = request_passphrase (_("Key Edit"), 1, &cancel);
        if (cancel)
            return FALSE;
    }
    ke = new GpgKeyEdit (k->keyid);
    if (k->is_protected)
        ke->setPassphrase (pass);
    else
        ke->setNoPassphrase (true);
    ke->setLocalUser (get_default_key ());
    err = ke->signUserid (uid_index, 
                          mode == CMD_SIGN? GPG_EDITKEY_SIGN : GPG_EDITKEY_LSIGN,
                          0, NULL);
    if (!err) {
        msg_box (dlg, _("Key successfully signed."), _("Key Edit"), MB_OK);
        k->update = 1;
    }
    else
        msg_box (dlg, gpgme_strerror (err), _("Key Edit"), MB_ERR);
    delete ke;
    sfree_if_alloc (pass);
    return !err? TRUE : FALSE;
}


static int
lookup_cmd (HWND dlg)
{
    char buf[64];
    int i;

    i = SendDlgItemMessage (dlg, IDC_KEYEDIT_CMD, CB_GETCURSEL, 0, 0);
    if (i == LB_ERR)
        return LB_ERR;
    GetDlgItemText (dlg, IDC_KEYEDIT_CMD, buf, sizeof (buf)-1);
    for (i=0; cmdlist[i].name != NULL; i++) {
        if (!strcmp (buf, cmdlist[i].name))
            return cmdlist[i].id;
    }
    return LB_ERR;
}

/* Dialog box procedure for the edit key dialog. */
BOOL CALLBACK
keyedit_main_dlg_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam)
{
    static winpt_key_t k;
    static listview_ctrl_t lvsub = NULL, lvuid = NULL;
    int cmd;
    HWND item;

    switch( msg ) {
    case WM_INITDIALOG:
        k = (winpt_key_t)lparam;
        if (!k)
            BUG (NULL);
        do_init_cmdlist (dlg, k->key_pair);
        lvsub = subkey_list_init (dlg, k);
        if( !lvsub )
            BUG( NULL );
        lvuid = userid_list_init (dlg, k);
        if( !lvuid )
            BUG( NULL );
        item = GetDlgItem( dlg, IDC_KEYEDIT_KEYLIST );
        keyedit_subkey_proc.dlg = dlg;
        keyedit_subkey_proc.current = (WNDPROC)subkey_subclass_proc;
        keyedit_subkey_proc.old = (WNDPROC)GetWindowLong( item, GWL_WNDPROC );
        if( keyedit_subkey_proc.old ) {
            if( !SetWindowLong( item, GWL_WNDPROC, (LONG)keyedit_subkey_proc.current ) ) {
                msg_box( dlg, _("Could not set subkey window procedure."), _("Key Edit"), MB_ERR );
                BUG( NULL );
            }
        }
        item = GetDlgItem( dlg, IDC_KEYEDIT_UIDLIST );
        keyedit_uid_proc.dlg = dlg;
        keyedit_uid_proc.current = (WNDPROC)uid_subclass_proc;
        keyedit_uid_proc.old = (WNDPROC)GetWindowLong( item, GWL_WNDPROC );
        if( keyedit_uid_proc.old ) {
            if( !SetWindowLong( item, GWL_WNDPROC, (LONG)keyedit_uid_proc.current ) ) {
                msg_box( dlg, _("Could not set user ID window procedure."), _("Key Edit"), MB_ERR );
                BUG( NULL );
            }
        }
        if (k->ctx->revoked) {
            EnableWindow (GetDlgItem (dlg, IDC_KEYEDIT_CMD), FALSE);
            EnableWindow (GetDlgItem (dlg, IDOK), FALSE);
        }
        SetDlgItemText (dlg, IDC_KEYEDIT_CMDINF, _("Command>"));
        SetDlgItemText (dlg, IDCANCEL, _("&Close"));
        SetDlgItemText (dlg, IDC_KEYEDIT_HELP, _("&Help"));
        SetWindowText (dlg, _("Key Edit"));
        SetForegroundWindow (dlg);
        center_window (dlg, NULL);
        return TRUE;

    case WM_DESTROY:
        if( lvsub ) {
            listview_release( lvsub );
            lvsub = NULL;
        }
        if( lvuid ) {
            listview_release( lvuid );
            lvuid = NULL;
        }
        break;

    case WM_NOTIFY:
        NMHDR * notify;
        notify = (NMHDR *)lparam;
        if (notify && notify->code == NM_DBLCLK &&
            notify->idFrom == IDC_KEYEDIT_UIDLIST)
            do_editkey_showpref (k, dlg, lvuid);
        break;

    case WM_COMMAND:
        switch( LOWORD( wparam ) ) {
        case IDOK:
            cmd = lookup_cmd (dlg);
            if (cmd == LB_ERR) {
                msg_box( dlg, _("Please select a command."), _("Key Edit"), MB_INFO );
                return FALSE;
            }
            if (k->is_v3 && is_cmd_openpgp (cmd)) {
                msg_box (dlg, _("This command cannot be used with PGP 2 (v3) keys.\n"),
                                _("Key Edit"), MB_ERR);
                return FALSE;
            }
            switch (cmd) {
            case CMD_SHOWPREF: do_editkey_showpref (k, dlg, lvuid); break;
            case CMD_DELKEY: do_editkey_delkey (k, dlg, lvsub); break;
            case CMD_ADDKEY: keyedit_add_subkey (k, dlg, lvsub); break;
            case CMD_EXPIRE: do_editkey_expire (k, dlg, lvsub); break;
            case CMD_REVKEY: do_editkey_revoke (k, dlg, lvsub); break;
            /*case CMD_SETPREF:do_editkey_setpref( k, dlg, lvuid ); break;*/
            case CMD_ADDUID: keyedit_add_userid( k, dlg, lvuid ); break;
            case CMD_ADDREVOKER: keyedit_add_revoker( k, dlg ); break;
            case CMD_ADDPHOTO: keyedit_add_photo( k, dlg ); break;
            case CMD_REVUID: do_editkey_revuid( k, dlg, lvuid ); break;
            case CMD_DELUID: do_editkey_deluid( k, dlg, lvuid ); break;
            case CMD_PASSWD: keyedit_change_passwd( k, dlg ); break;
            case CMD_PRIMARY: do_editkey_primary( k, dlg, lvuid ); break;
            case CMD_ENABLE: do_editkey_enable_disable (k, dlg, lvsub, 1); break;
            case CMD_DISABLE: do_editkey_enable_disable (k, dlg, lvsub, 0); break;
            case CMD_CHECK: do_editkey_check (k, dlg); break;
            case CMD_TRUST: keyedit_change_ownertrust (k, dlg); break;
            case CMD_SIGN:
            case CMD_LSIGN: do_editkey_sign_userid (k, dlg, 
                                                    lvuid, cmd);
                            break;
            }
            break;

        case IDCANCEL:
            EndDialog (dlg, FALSE);
            break;

        case IDC_KEYEDIT_HELP:
            do_show_help (dlg);
            break;
        }
        break;
    }
    return FALSE;
}
