trunk/Src/wptMDSumDlg.cpp

/* wptMDSumDlg.cpp
 *      Copyright (C) 2003, 2005, 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 <stdio.h>

#include "resource.h"
#include "wptTypes.h"
#include "wptW32API.h"
#include "wptGPG.h"
#include "wptCommonCtl.h"
#include "wptContext.h"
#include "wptNLS.h"
#include "wptErrors.h"


/* Return a printable digest of the buffer @mdbuf. */
static const char*
printable_digest (byte *mdbuf, size_t n)
{
    static char mdasc[64];
    size_t i;
    for (i = 0; i < n; i++)
        sprintf (mdasc+2*i, "%02X", mdbuf[i]);
    return mdasc;
}


static const char*
id2algo (int mdalgo)
{
    gpgme_hash_algo_t md = (gpgme_hash_algo_t)mdalgo;

    switch (md) {
    case GPGME_MD_MD5:      return "MD5";
    case GPGME_MD_SHA1:     return "SHA1";
    case GPGME_MD_RMD160:   return "RMD160";
    case GPGME_MD_SHA256:   return "SHA256";
    default: return "";
    }
    return "";
}


/* Dialog box procedure to show and calculate file digests. */
BOOL CALLBACK
mdsum_dlg_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam)
{
    static listview_ctrl_t lv;
    static struct md_file_s * md;
    struct listview_column_s cols[] = {
        {0, 264, (char *)_("Digest")},
        {1, 128, (char *)_("Name")},
        {0, 0, 0}
    };
    gpgme_data_t sumlist;
    const char *name;
    char fname[300], mdasc[64];
    byte mdbuf[20];
    int i;
    size_t n;

    switch( msg ) {
    case WM_INITDIALOG:
        md = (md_file_s *)lparam;
        if( !md )
            BUG( NULL );        
        if( listview_new( &lv ) )
            BUG( NULL );
        lv->ctrl = GetDlgItem( dlg, IDC_MDSUM_LIST );
        for( i = 0; i < cols[i].width; i++ )
            listview_add_column( lv, &cols[i] );
        for( i = 0; i < listview_count_items( md->lv, 0 ); i++ ) {
            if( listview_get_item_state( md->lv, i ) ) {
                listview_get_item_text( md->lv, i, 1, fname, sizeof (fname)-1 );
                if( !gpg_md_hash_file( md->mdalgo, fname, mdbuf, &n ) ) {
                    listview_add_item( lv, "" );
                    listview_add_sub_item( lv, 0, 0, printable_digest( mdbuf, n ) );
                    listview_add_sub_item( lv, 0, 1, fname );
                }
            }
        }
        SetDlgItemText (dlg, IDC_MDSUM_COPY, _("&Save..."));
        SetDlgItemText (dlg, IDOK, _("&Close"));
        SetDlgItemText (dlg, IDC_MDSUM_TOCLIP, _("Save to clipboard"));
        SetWindowText (dlg, _("Print Message Digest"));
        SetForegroundWindow (dlg);
        break;

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

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

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

        case IDC_MDSUM_COPY:
            if (gpgme_data_new (&sumlist))
                BUG(0);
            if (md->mdalgo == (int)GPGME_MD_RMD160) {
                const char *s = "# warning rmd160sum is not yet available\r\n";
                gpgme_data_write (sumlist, s, strlen (s));
            }
            for (i = 0; i < listview_count_items (lv, 0); i++) {
                listview_get_item_text (lv, i, 0, mdasc, DIM (mdasc)-1);
                listview_get_item_text (lv, i, 1, fname, DIM (fname)-1);
                
                gpgme_data_write (sumlist, mdasc, strlen (mdasc));
                gpgme_data_write (sumlist, " ", 1);
                gpgme_data_write (sumlist, fname, strlen (fname));
                gpgme_data_write (sumlist, "\r\n", 2);
            }
            if (IsDlgButtonChecked (dlg, IDC_MDSUM_TOCLIP)) {
                gpg_data_release_and_set_clipboard (sumlist, 0);
                break;
            }
            _snprintf (fname, sizeof (fname)-1, "%s_sums.txt", id2algo (md->mdalgo));
            name = get_filesave_dlg (dlg, _("Select file to save checksums"), 
                                     NULL, fname);
            if (name && *name) {
                gpg_data_release_and_set_file (sumlist, name);
                log_box (_("File Manager"), MB_OK, 
                         _("Checksums successfully saved in '%s'"), name);
            }
            else
                gpgme_data_release (sumlist);
            break;
        }
        break;
    }
    return FALSE;
}
1	werner	36	/* wptMDSumDlg.cpp
2	twoaday	165	* Copyright (C) 2003, 2005, 2006 Timo Schulz
3	werner	36	*
4			* This file is part of WinPT.
5			*
6			* WinPT is free software; you can redistribute it and/or modify
7			* it under the terms of the GNU General Public License as published by
8			* the Free Software Foundation; either version 2 of the License, or
9			* (at your option) any later version.
10			*
11			* WinPT is distributed in the hope that it will be useful,
12			* but WITHOUT ANY WARRANTY; without even the implied warranty of
13			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14			* GNU General Public License for more details.
15			*
16			* You should have received a copy of the GNU General Public License
17			* along with WinPT; if not, write to the Free Software Foundation,
18			* Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
19			*/
20			#ifdef HAVE_CONFIG_H
21			#include <config.h>
22			#endif
23
24			#include <windows.h>
25			#include <stdio.h>
26	twoaday	99
27	werner	47	#include "resource.h"
28	werner	36	#include "wptTypes.h"
29			#include "wptW32API.h"
30			#include "wptGPG.h"
31			#include "wptCommonCtl.h"
32			#include "wptContext.h"
33			#include "wptNLS.h"
34			#include "wptErrors.h"
35
36	twoaday	130
37	werner	36	/* Return a printable digest of the buffer @mdbuf. */
38			static const char*
39			printable_digest (byte *mdbuf, size_t n)
40			{
41			static char mdasc[64];
42			size_t i;
43			for (i = 0; i < n; i++)
44			sprintf (mdasc+2*i, "%02X", mdbuf[i]);
45			return mdasc;
46			}
47
48
49	twoaday	130	static const char*
50			id2algo (int mdalgo)
51			{
52			gpgme_hash_algo_t md = (gpgme_hash_algo_t)mdalgo;
53
54			switch (md) {
55			case GPGME_MD_MD5: return "MD5";
56			case GPGME_MD_SHA1: return "SHA1";
57			case GPGME_MD_RMD160: return "RMD160";
58			case GPGME_MD_SHA256: return "SHA256";
59	twoaday	131	default: return "";
60	twoaday	130	}
61			return "";
62			}
63
64
65	werner	36	/* Dialog box procedure to show and calculate file digests. */
66			BOOL CALLBACK
67			mdsum_dlg_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam)
68			{
69			static listview_ctrl_t lv;
70			static struct md_file_s * md;
71			struct listview_column_s cols[] = {
72			{0, 264, (char *)_("Digest")},
73			{1, 128, (char *)_("Name")},
74			{0, 0, 0}
75			};
76			gpgme_data_t sumlist;
77	twoaday	130	const char *name;
78	werner	36	char fname[300], mdasc[64];
79			byte mdbuf[20];
80			int i;
81			size_t n;
82
83			switch( msg ) {
84			case WM_INITDIALOG:
85			md = (md_file_s *)lparam;
86			if( !md )
87			BUG( NULL );
88			if( listview_new( &lv ) )
89			BUG( NULL );
90			lv->ctrl = GetDlgItem( dlg, IDC_MDSUM_LIST );
91			for( i = 0; i < cols[i].width; i++ )
92			listview_add_column( lv, &cols[i] );
93			for( i = 0; i < listview_count_items( md->lv, 0 ); i++ ) {
94			if( listview_get_item_state( md->lv, i ) ) {
95			listview_get_item_text( md->lv, i, 1, fname, sizeof (fname)-1 );
96			if( !gpg_md_hash_file( md->mdalgo, fname, mdbuf, &n ) ) {
97			listview_add_item( lv, "" );
98			listview_add_sub_item( lv, 0, 0, printable_digest( mdbuf, n ) );
99			listview_add_sub_item( lv, 0, 1, fname );
100			}
101			}
102			}
103	twoaday	99	SetDlgItemText (dlg, IDC_MDSUM_COPY, _("&Save..."));
104	twoaday	119	SetDlgItemText (dlg, IDOK, _("&Close"));
105	twoaday	167	SetDlgItemText (dlg, IDC_MDSUM_TOCLIP, _("Save to clipboard"));
106	twoaday	99	SetWindowText (dlg, _("Print Message Digest"));
107			SetForegroundWindow (dlg);
108	werner	36	break;
109
110	twoaday	130	case WM_SYSCOMMAND:
111			if (wparam == SC_CLOSE)
112			EndDialog (dlg, TRUE);
113			return TRUE;
114
115	werner	36	case WM_DESTROY:
116			if( lv ) {
117			listview_release( lv );
118			lv = NULL;
119			}
120			break;
121
122			case WM_COMMAND:
123			switch (LOWORD (wparam)) {
124			case IDOK:
125			EndDialog (dlg, TRUE);
126			break;
127
128			case IDC_MDSUM_COPY:
129			if (gpgme_data_new (&sumlist))
130			BUG(0);
131	twoaday	130	if (md->mdalgo == (int)GPGME_MD_RMD160) {
132	werner	36	const char *s = "# warning rmd160sum is not yet available\r\n";
133			gpgme_data_write (sumlist, s, strlen (s));
134			}
135			for (i = 0; i < listview_count_items (lv, 0); i++) {
136			listview_get_item_text (lv, i, 0, mdasc, DIM (mdasc)-1);
137			listview_get_item_text (lv, i, 1, fname, DIM (fname)-1);
138
139			gpgme_data_write (sumlist, mdasc, strlen (mdasc));
140			gpgme_data_write (sumlist, " ", 1);
141			gpgme_data_write (sumlist, fname, strlen (fname));
142			gpgme_data_write (sumlist, "\r\n", 2);
143			}
144	twoaday	165	if (IsDlgButtonChecked (dlg, IDC_MDSUM_TOCLIP)) {
145			gpg_data_release_and_set_clipboard (sumlist, 0);
146			break;
147			}
148	twoaday	130	_snprintf (fname, sizeof (fname)-1, "%s_sums.txt", id2algo (md->mdalgo));
149			name = get_filesave_dlg (dlg, _("Select file to save checksums"),
150			NULL, fname);
151	werner	36	if (name && *name) {
152			gpg_data_release_and_set_file (sumlist, name);
153	twoaday	130	log_box (_("File Manager"), MB_OK,
154			_("Checksums successfully saved in '%s'"), name);
155	werner	36	}
156			else
157			gpgme_data_release (sumlist);
158			break;
159			}
160			break;
161			}
162			return FALSE;
163			}