2 * PuTTY key generation front end (Windows).
9 #define PUTTY_DO_GLOBALS
20 #define WM_DONEKEY (WM_APP + 1)
22 #define DEFAULT_KEYSIZE 1024
24 static char *cmdline_keyfile = NULL;
27 * Print a modal (Really Bad) message box and perform a fatal exit.
29 void modalfatalbox(char *fmt, ...)
35 stuff = dupvprintf(fmt, ap);
37 MessageBox(NULL, stuff, "PuTTYgen Fatal Error",
38 MB_SYSTEMMODAL | MB_ICONERROR | MB_OK);
43 /* ----------------------------------------------------------------------
44 * Progress report code. This is really horrible :-)
46 #define PROGRESSRANGE 65535
52 unsigned startpoint, total;
53 unsigned param, current, n; /* if exponential */
54 unsigned mult; /* if linear */
56 unsigned total, divisor, range;
60 static void progress_update(void *param, int action, int phase, int iprogress)
62 struct progress *p = (struct progress *) param;
63 unsigned progress = iprogress;
66 if (action < PROGFN_READY && p->nphases < phase)
69 case PROGFN_INITIALISE:
72 case PROGFN_LIN_PHASE:
73 p->phases[phase-1].exponential = 0;
74 p->phases[phase-1].mult = p->phases[phase].total / progress;
76 case PROGFN_EXP_PHASE:
77 p->phases[phase-1].exponential = 1;
78 p->phases[phase-1].param = 0x10000 + progress;
79 p->phases[phase-1].current = p->phases[phase-1].total;
80 p->phases[phase-1].n = 0;
82 case PROGFN_PHASE_EXTENT:
83 p->phases[phase-1].total = progress;
89 for (i = 0; i < p->nphases; i++) {
90 p->phases[i].startpoint = total;
91 total += p->phases[i].total;
94 p->divisor = ((p->total + PROGRESSRANGE - 1) / PROGRESSRANGE);
95 p->range = p->total / p->divisor;
96 SendMessage(p->progbar, PBM_SETRANGE, 0, MAKELPARAM(0, p->range));
100 if (p->phases[phase-1].exponential) {
101 while (p->phases[phase-1].n < progress) {
102 p->phases[phase-1].n++;
103 p->phases[phase-1].current *= p->phases[phase-1].param;
104 p->phases[phase-1].current /= 0x10000;
106 position = (p->phases[phase-1].startpoint +
107 p->phases[phase-1].total - p->phases[phase-1].current);
109 position = (p->phases[phase-1].startpoint +
110 progress * p->phases[phase-1].mult);
112 SendMessage(p->progbar, PBM_SETPOS, position / p->divisor, 0);
119 #define PASSPHRASE_MAXLEN 512
121 struct PassphraseProcStruct {
127 * Dialog-box function for the passphrase box.
129 static int CALLBACK PassphraseProc(HWND hwnd, UINT msg,
130 WPARAM wParam, LPARAM lParam)
132 static char *passphrase = NULL;
133 struct PassphraseProcStruct *p;
137 SetForegroundWindow(hwnd);
138 SetWindowPos(hwnd, HWND_TOP, 0, 0, 0, 0,
139 SWP_NOMOVE | SWP_NOSIZE | SWP_SHOWWINDOW);
144 { /* centre the window */
148 hw = GetDesktopWindow();
149 if (GetWindowRect(hw, &rs) && GetWindowRect(hwnd, &rd))
151 (rs.right + rs.left + rd.left - rd.right) / 2,
152 (rs.bottom + rs.top + rd.top - rd.bottom) / 2,
153 rd.right - rd.left, rd.bottom - rd.top, TRUE);
156 p = (struct PassphraseProcStruct *) lParam;
157 passphrase = p->passphrase;
159 SetDlgItemText(hwnd, 101, p->comment);
161 SetDlgItemText(hwnd, 102, passphrase);
164 switch (LOWORD(wParam)) {
174 case 102: /* edit box */
175 if ((HIWORD(wParam) == EN_CHANGE) && passphrase) {
176 GetDlgItemText(hwnd, 102, passphrase,
177 PASSPHRASE_MAXLEN - 1);
178 passphrase[PASSPHRASE_MAXLEN - 1] = '\0';
191 * Prompt for a key file. Assumes the filename buffer is of size
194 static int prompt_keyfile(HWND hwnd, char *dlgtitle,
195 char *filename, int save, int ppk)
198 memset(&of, 0, sizeof(of));
201 of.lpstrFilter = "PuTTY Private Key Files (*.ppk)\0*.ppk\0"
202 "All Files (*.*)\0*\0\0\0";
203 of.lpstrDefExt = ".ppk";
205 of.lpstrFilter = "All Files (*.*)\0*\0\0\0";
207 of.lpstrCustomFilter = NULL;
209 of.lpstrFile = filename;
211 of.nMaxFile = FILENAME_MAX;
212 of.lpstrFileTitle = NULL;
213 of.lpstrTitle = dlgtitle;
215 return request_file(NULL, &of, FALSE, save);
219 * Dialog-box function for the Licence box.
221 static int CALLBACK LicenceProc(HWND hwnd, UINT msg,
222 WPARAM wParam, LPARAM lParam)
229 { /* centre the window */
233 hw = GetDesktopWindow();
234 if (GetWindowRect(hw, &rs) && GetWindowRect(hwnd, &rd))
236 (rs.right + rs.left + rd.left - rd.right) / 2,
237 (rs.bottom + rs.top + rd.top - rd.bottom) / 2,
238 rd.right - rd.left, rd.bottom - rd.top, TRUE);
243 switch (LOWORD(wParam)) {
258 * Dialog-box function for the About box.
260 static int CALLBACK AboutProc(HWND hwnd, UINT msg,
261 WPARAM wParam, LPARAM lParam)
268 { /* centre the window */
272 hw = GetDesktopWindow();
273 if (GetWindowRect(hw, &rs) && GetWindowRect(hwnd, &rd))
275 (rs.right + rs.left + rd.left - rd.right) / 2,
276 (rs.bottom + rs.top + rd.top - rd.bottom) / 2,
277 rd.right - rd.left, rd.bottom - rd.top, TRUE);
280 SetDlgItemText(hwnd, 100, ver);
283 switch (LOWORD(wParam)) {
289 EnableWindow(hwnd, 0);
290 DialogBox(hinst, MAKEINTRESOURCE(214), hwnd, LicenceProc);
291 EnableWindow(hwnd, 1);
292 SetActiveWindow(hwnd);
304 * Thread to generate a key.
306 struct rsa_key_thread_params {
307 HWND progressbar; /* notify this with progress */
308 HWND dialog; /* notify this on completion */
309 int keysize; /* bits in key */
312 struct dss_key *dsskey;
314 static DWORD WINAPI generate_rsa_key_thread(void *param)
316 struct rsa_key_thread_params *params =
317 (struct rsa_key_thread_params *) param;
318 struct progress prog;
319 prog.progbar = params->progressbar;
321 progress_update(&prog, PROGFN_INITIALISE, 0, 0);
324 dsa_generate(params->dsskey, params->keysize, progress_update, &prog);
326 rsa_generate(params->key, params->keysize, progress_update, &prog);
328 PostMessage(params->dialog, WM_DONEKEY, 0, 0);
334 struct MainDlgState {
335 int collecting_entropy;
336 int generation_thread_exists;
338 int entropy_got, entropy_required, entropy_size;
341 char **commentptr; /* points to key.comment or ssh2key.comment */
342 struct ssh2_userkey ssh2key;
345 struct dss_key dsskey;
346 HMENU filemenu, keymenu, cvtmenu;
349 static void hidemany(HWND hwnd, const int *ids, int hideit)
352 ShowWindow(GetDlgItem(hwnd, *ids++), (hideit ? SW_HIDE : SW_SHOW));
356 static void setupbigedit1(HWND hwnd, int id, int idstatic, struct RSAKey *key)
361 dec1 = bignum_decimal(key->exponent);
362 dec2 = bignum_decimal(key->modulus);
363 buffer = dupprintf("%d %s %s %s", bignum_bitcount(key->modulus),
364 dec1, dec2, key->comment);
365 SetDlgItemText(hwnd, id, buffer);
366 SetDlgItemText(hwnd, idstatic,
367 "&Public key for pasting into authorized_keys file:");
373 static void setupbigedit2(HWND hwnd, int id, int idstatic,
374 struct ssh2_userkey *key)
376 unsigned char *pub_blob;
381 pub_blob = key->alg->public_blob(key->data, &pub_len);
382 buffer = snewn(strlen(key->alg->name) + 4 * ((pub_len + 2) / 3) +
383 strlen(key->comment) + 3, char);
384 strcpy(buffer, key->alg->name);
385 p = buffer + strlen(buffer);
388 while (i < pub_len) {
389 int n = (pub_len - i < 3 ? pub_len - i : 3);
390 base64_encode_atom(pub_blob + i, n, p);
395 strcpy(p, key->comment);
396 SetDlgItemText(hwnd, id, buffer);
397 SetDlgItemText(hwnd, idstatic, "&Public key for pasting into "
398 "OpenSSH authorized_keys file:");
403 static int save_ssh1_pubkey(char *filename, struct RSAKey *key)
408 dec1 = bignum_decimal(key->exponent);
409 dec2 = bignum_decimal(key->modulus);
410 fp = fopen(filename, "wb");
413 fprintf(fp, "%d %s %s %s\n",
414 bignum_bitcount(key->modulus), dec1, dec2, key->comment);
422 * Warn about the obsolescent key file format.
424 void old_keyfile_warning(void)
426 static const char mbtitle[] = "PuTTY Key File Warning";
427 static const char message[] =
428 "You are loading an SSH-2 private key which has an\n"
429 "old version of the file format. This means your key\n"
430 "file is not fully tamperproof. Future versions of\n"
431 "PuTTY may stop supporting this private key format,\n"
432 "so we recommend you convert your key to the new\n"
435 "Once the key is loaded into PuTTYgen, you can perform\n"
436 "this conversion simply by saving it again.";
438 MessageBox(NULL, message, mbtitle, MB_OK);
441 static int save_ssh2_pubkey(char *filename, struct ssh2_userkey *key)
443 unsigned char *pub_blob;
449 pub_blob = key->alg->public_blob(key->data, &pub_len);
451 fp = fopen(filename, "wb");
455 fprintf(fp, "---- BEGIN SSH2 PUBLIC KEY ----\n");
457 fprintf(fp, "Comment: \"");
458 for (p = key->comment; *p; p++) {
459 if (*p == '\\' || *p == '\"')
467 while (i < pub_len) {
469 int n = (pub_len - i < 3 ? pub_len - i : 3);
470 base64_encode_atom(pub_blob + i, n, buf);
474 if (++column >= 16) {
482 fprintf(fp, "---- END SSH2 PUBLIC KEY ----\n");
489 controlidstart = 100,
496 IDC_PKSTATIC, IDC_KEYDISPLAY,
497 IDC_FPSTATIC, IDC_FINGERPRINT,
498 IDC_COMMENTSTATIC, IDC_COMMENTEDIT,
499 IDC_PASSPHRASE1STATIC, IDC_PASSPHRASE1EDIT,
500 IDC_PASSPHRASE2STATIC, IDC_PASSPHRASE2EDIT,
502 IDC_GENSTATIC, IDC_GENERATE,
503 IDC_LOADSTATIC, IDC_LOAD,
504 IDC_SAVESTATIC, IDC_SAVE, IDC_SAVEPUB,
506 IDC_TYPESTATIC, IDC_KEYSSH1, IDC_KEYSSH2RSA, IDC_KEYSSH2DSA,
507 IDC_BITSSTATIC, IDC_BITS,
510 IDC_IMPORT, IDC_EXPORT_OPENSSH, IDC_EXPORT_SSHCOM
513 static const int nokey_ids[] = { IDC_NOKEY, 0 };
514 static const int generating_ids[] = { IDC_GENERATING, IDC_PROGRESS, 0 };
515 static const int gotkey_ids[] = {
516 IDC_PKSTATIC, IDC_KEYDISPLAY,
517 IDC_FPSTATIC, IDC_FINGERPRINT,
518 IDC_COMMENTSTATIC, IDC_COMMENTEDIT,
519 IDC_PASSPHRASE1STATIC, IDC_PASSPHRASE1EDIT,
520 IDC_PASSPHRASE2STATIC, IDC_PASSPHRASE2EDIT, 0
524 * Small UI helper function to switch the state of the main dialog
525 * by enabling and disabling controls and menu items.
527 void ui_set_state(HWND hwnd, struct MainDlgState *state, int status)
533 hidemany(hwnd, nokey_ids, FALSE);
534 hidemany(hwnd, generating_ids, TRUE);
535 hidemany(hwnd, gotkey_ids, TRUE);
536 EnableWindow(GetDlgItem(hwnd, IDC_GENERATE), 1);
537 EnableWindow(GetDlgItem(hwnd, IDC_LOAD), 1);
538 EnableWindow(GetDlgItem(hwnd, IDC_SAVE), 0);
539 EnableWindow(GetDlgItem(hwnd, IDC_SAVEPUB), 0);
540 EnableWindow(GetDlgItem(hwnd, IDC_KEYSSH1), 1);
541 EnableWindow(GetDlgItem(hwnd, IDC_KEYSSH2RSA), 1);
542 EnableWindow(GetDlgItem(hwnd, IDC_KEYSSH2DSA), 1);
543 EnableWindow(GetDlgItem(hwnd, IDC_BITS), 1);
544 EnableMenuItem(state->filemenu, IDC_LOAD, MF_ENABLED|MF_BYCOMMAND);
545 EnableMenuItem(state->filemenu, IDC_SAVE, MF_GRAYED|MF_BYCOMMAND);
546 EnableMenuItem(state->filemenu, IDC_SAVEPUB, MF_GRAYED|MF_BYCOMMAND);
547 EnableMenuItem(state->keymenu, IDC_GENERATE, MF_ENABLED|MF_BYCOMMAND);
548 EnableMenuItem(state->keymenu, IDC_KEYSSH1, MF_ENABLED|MF_BYCOMMAND);
549 EnableMenuItem(state->keymenu, IDC_KEYSSH2RSA, MF_ENABLED|MF_BYCOMMAND);
550 EnableMenuItem(state->keymenu, IDC_KEYSSH2DSA, MF_ENABLED|MF_BYCOMMAND);
551 EnableMenuItem(state->cvtmenu, IDC_IMPORT, MF_ENABLED|MF_BYCOMMAND);
552 EnableMenuItem(state->cvtmenu, IDC_EXPORT_OPENSSH,
553 MF_GRAYED|MF_BYCOMMAND);
554 EnableMenuItem(state->cvtmenu, IDC_EXPORT_SSHCOM,
555 MF_GRAYED|MF_BYCOMMAND);
557 case 1: /* generating key */
558 hidemany(hwnd, nokey_ids, TRUE);
559 hidemany(hwnd, generating_ids, FALSE);
560 hidemany(hwnd, gotkey_ids, TRUE);
561 EnableWindow(GetDlgItem(hwnd, IDC_GENERATE), 0);
562 EnableWindow(GetDlgItem(hwnd, IDC_LOAD), 0);
563 EnableWindow(GetDlgItem(hwnd, IDC_SAVE), 0);
564 EnableWindow(GetDlgItem(hwnd, IDC_SAVEPUB), 0);
565 EnableWindow(GetDlgItem(hwnd, IDC_KEYSSH1), 0);
566 EnableWindow(GetDlgItem(hwnd, IDC_KEYSSH2RSA), 0);
567 EnableWindow(GetDlgItem(hwnd, IDC_KEYSSH2DSA), 0);
568 EnableWindow(GetDlgItem(hwnd, IDC_BITS), 0);
569 EnableMenuItem(state->filemenu, IDC_LOAD, MF_GRAYED|MF_BYCOMMAND);
570 EnableMenuItem(state->filemenu, IDC_SAVE, MF_GRAYED|MF_BYCOMMAND);
571 EnableMenuItem(state->filemenu, IDC_SAVEPUB, MF_GRAYED|MF_BYCOMMAND);
572 EnableMenuItem(state->keymenu, IDC_GENERATE, MF_GRAYED|MF_BYCOMMAND);
573 EnableMenuItem(state->keymenu, IDC_KEYSSH1, MF_GRAYED|MF_BYCOMMAND);
574 EnableMenuItem(state->keymenu, IDC_KEYSSH2RSA, MF_GRAYED|MF_BYCOMMAND);
575 EnableMenuItem(state->keymenu, IDC_KEYSSH2DSA, MF_GRAYED|MF_BYCOMMAND);
576 EnableMenuItem(state->cvtmenu, IDC_IMPORT, MF_GRAYED|MF_BYCOMMAND);
577 EnableMenuItem(state->cvtmenu, IDC_EXPORT_OPENSSH,
578 MF_GRAYED|MF_BYCOMMAND);
579 EnableMenuItem(state->cvtmenu, IDC_EXPORT_SSHCOM,
580 MF_GRAYED|MF_BYCOMMAND);
583 hidemany(hwnd, nokey_ids, TRUE);
584 hidemany(hwnd, generating_ids, TRUE);
585 hidemany(hwnd, gotkey_ids, FALSE);
586 EnableWindow(GetDlgItem(hwnd, IDC_GENERATE), 1);
587 EnableWindow(GetDlgItem(hwnd, IDC_LOAD), 1);
588 EnableWindow(GetDlgItem(hwnd, IDC_SAVE), 1);
589 EnableWindow(GetDlgItem(hwnd, IDC_SAVEPUB), 1);
590 EnableWindow(GetDlgItem(hwnd, IDC_KEYSSH1), 1);
591 EnableWindow(GetDlgItem(hwnd, IDC_KEYSSH2RSA), 1);
592 EnableWindow(GetDlgItem(hwnd, IDC_KEYSSH2DSA), 1);
593 EnableWindow(GetDlgItem(hwnd, IDC_BITS), 1);
594 EnableMenuItem(state->filemenu, IDC_LOAD, MF_ENABLED|MF_BYCOMMAND);
595 EnableMenuItem(state->filemenu, IDC_SAVE, MF_ENABLED|MF_BYCOMMAND);
596 EnableMenuItem(state->filemenu, IDC_SAVEPUB, MF_ENABLED|MF_BYCOMMAND);
597 EnableMenuItem(state->keymenu, IDC_GENERATE, MF_ENABLED|MF_BYCOMMAND);
598 EnableMenuItem(state->keymenu, IDC_KEYSSH1, MF_ENABLED|MF_BYCOMMAND);
599 EnableMenuItem(state->keymenu, IDC_KEYSSH2RSA,MF_ENABLED|MF_BYCOMMAND);
600 EnableMenuItem(state->keymenu, IDC_KEYSSH2DSA,MF_ENABLED|MF_BYCOMMAND);
601 EnableMenuItem(state->cvtmenu, IDC_IMPORT, MF_ENABLED|MF_BYCOMMAND);
603 * Enable export menu items if and only if the key type
604 * supports this kind of export.
606 type = state->ssh2 ? SSH_KEYTYPE_SSH2 : SSH_KEYTYPE_SSH1;
607 #define do_export_menuitem(x,y) \
608 EnableMenuItem(state->cvtmenu, x, MF_BYCOMMAND | \
609 (import_target_type(y)==type?MF_ENABLED:MF_GRAYED))
610 do_export_menuitem(IDC_EXPORT_OPENSSH, SSH_KEYTYPE_OPENSSH);
611 do_export_menuitem(IDC_EXPORT_SSHCOM, SSH_KEYTYPE_SSHCOM);
612 #undef do_export_menuitem
617 void load_key_file(HWND hwnd, struct MainDlgState *state,
618 Filename filename, int was_import_cmd)
620 char passphrase[PASSPHRASE_MAXLEN];
624 const char *errmsg = NULL;
626 struct PassphraseProcStruct pps;
627 struct RSAKey newkey1;
628 struct ssh2_userkey *newkey2 = NULL;
630 type = realtype = key_type(&filename);
631 if (type != SSH_KEYTYPE_SSH1 &&
632 type != SSH_KEYTYPE_SSH2 &&
633 !import_possible(type)) {
634 char *msg = dupprintf("Couldn't load private key (%s)",
635 key_type_to_str(type));
636 message_box(msg, "PuTTYgen Error", MB_OK | MB_ICONERROR,
637 HELPCTXID(errors_cantloadkey));
642 if (type != SSH_KEYTYPE_SSH1 &&
643 type != SSH_KEYTYPE_SSH2) {
645 type = import_target_type(type);
649 if (realtype == SSH_KEYTYPE_SSH1)
650 needs_pass = rsakey_encrypted(&filename, &comment);
651 else if (realtype == SSH_KEYTYPE_SSH2)
653 ssh2_userkey_encrypted(&filename, &comment);
655 needs_pass = import_encrypted(&filename, realtype,
657 pps.passphrase = passphrase;
658 pps.comment = comment;
662 dlgret = DialogBoxParam(hinst,
663 MAKEINTRESOURCE(210),
664 NULL, PassphraseProc,
672 if (type == SSH_KEYTYPE_SSH1) {
673 if (realtype == type)
674 ret = loadrsakey(&filename, &newkey1,
675 passphrase, &errmsg);
677 ret = import_ssh1(&filename, realtype,
678 &newkey1, passphrase, &errmsg);
680 if (realtype == type)
681 newkey2 = ssh2_load_userkey(&filename,
682 passphrase, &errmsg);
684 newkey2 = import_ssh2(&filename, realtype,
685 passphrase, &errmsg);
686 if (newkey2 == SSH2_WRONG_PASSPHRASE)
697 char *msg = dupprintf("Couldn't load private key (%s)", errmsg);
698 message_box(msg, "PuTTYgen Error", MB_OK | MB_ICONERROR,
699 HELPCTXID(errors_cantloadkey));
701 } else if (ret == 1) {
703 * Now update the key controls with all the
707 SetDlgItemText(hwnd, IDC_PASSPHRASE1EDIT,
709 SetDlgItemText(hwnd, IDC_PASSPHRASE2EDIT,
711 if (type == SSH_KEYTYPE_SSH1) {
716 state->commentptr = &state->key.comment;
717 state->key = newkey1;
720 * Set the key fingerprint.
722 savecomment = state->key.comment;
723 state->key.comment = NULL;
724 rsa_fingerprint(buf, sizeof(buf),
726 state->key.comment = savecomment;
728 SetDlgItemText(hwnd, IDC_FINGERPRINT, buf);
730 * Construct a decimal representation
731 * of the key, for pasting into
732 * .ssh/authorized_keys on a Unix box.
734 setupbigedit1(hwnd, IDC_KEYDISPLAY,
735 IDC_PKSTATIC, &state->key);
742 &state->ssh2key.comment;
743 state->ssh2key = *newkey2; /* structure copy */
746 savecomment = state->ssh2key.comment;
747 state->ssh2key.comment = NULL;
750 fingerprint(state->ssh2key.data);
751 state->ssh2key.comment = savecomment;
753 SetDlgItemText(hwnd, IDC_FINGERPRINT, fp);
756 setupbigedit2(hwnd, IDC_KEYDISPLAY,
757 IDC_PKSTATIC, &state->ssh2key);
759 SetDlgItemText(hwnd, IDC_COMMENTEDIT,
763 * Finally, hide the progress bar and show
766 ui_set_state(hwnd, state, 2);
767 state->key_exists = TRUE;
770 * If the user has imported a foreign key
771 * using the Load command, let them know.
772 * If they've used the Import command, be
775 if (realtype != type && !was_import_cmd) {
777 sprintf(msg, "Successfully imported foreign key\n"
779 "To use this key with PuTTY, you need to\n"
780 "use the \"Save private key\" command to\n"
781 "save it in PuTTY's own format.",
782 key_type_to_str(realtype));
783 MessageBox(NULL, msg, "PuTTYgen Notice",
784 MB_OK | MB_ICONINFORMATION);
790 * Dialog-box function for the main PuTTYgen dialog box.
792 static int CALLBACK MainDlgProc(HWND hwnd, UINT msg,
793 WPARAM wParam, LPARAM lParam)
795 static const char generating_msg[] =
796 "Please wait while a key is generated...";
797 static const char entropy_msg[] =
798 "Please generate some randomness by moving the mouse over the blank area.";
799 struct MainDlgState *state;
804 SetWindowLongPtr(hwnd, GWL_EXSTYLE,
805 GetWindowLongPtr(hwnd, GWL_EXSTYLE) |
809 * If we add a Help button, this is where we destroy it
810 * if the help file isn't present.
813 requested_help = FALSE;
814 SendMessage(hwnd, WM_SETICON, (WPARAM) ICON_BIG,
815 (LPARAM) LoadIcon(hinst, MAKEINTRESOURCE(200)));
817 state = snew(struct MainDlgState);
818 state->generation_thread_exists = FALSE;
819 state->collecting_entropy = FALSE;
820 state->entropy = NULL;
821 state->key_exists = FALSE;
822 SetWindowLongPtr(hwnd, GWLP_USERDATA, (LONG_PTR) state);
828 menu1 = CreateMenu();
829 AppendMenu(menu1, MF_ENABLED, IDC_LOAD, "&Load private key");
830 AppendMenu(menu1, MF_ENABLED, IDC_SAVEPUB, "Save p&ublic key");
831 AppendMenu(menu1, MF_ENABLED, IDC_SAVE, "&Save private key");
832 AppendMenu(menu1, MF_SEPARATOR, 0, 0);
833 AppendMenu(menu1, MF_ENABLED, IDC_QUIT, "E&xit");
834 AppendMenu(menu, MF_POPUP | MF_ENABLED, (UINT) menu1, "&File");
835 state->filemenu = menu1;
837 menu1 = CreateMenu();
838 AppendMenu(menu1, MF_ENABLED, IDC_GENERATE, "&Generate key pair");
839 AppendMenu(menu1, MF_SEPARATOR, 0, 0);
840 AppendMenu(menu1, MF_ENABLED, IDC_KEYSSH1, "SSH-&1 key (RSA)");
841 AppendMenu(menu1, MF_ENABLED, IDC_KEYSSH2RSA, "SSH-2 &RSA key");
842 AppendMenu(menu1, MF_ENABLED, IDC_KEYSSH2DSA, "SSH-2 &DSA key");
843 AppendMenu(menu, MF_POPUP | MF_ENABLED, (UINT) menu1, "&Key");
844 state->keymenu = menu1;
846 menu1 = CreateMenu();
847 AppendMenu(menu1, MF_ENABLED, IDC_IMPORT, "&Import key");
848 AppendMenu(menu1, MF_SEPARATOR, 0, 0);
849 AppendMenu(menu1, MF_ENABLED, IDC_EXPORT_OPENSSH,
850 "Export &OpenSSH key");
851 AppendMenu(menu1, MF_ENABLED, IDC_EXPORT_SSHCOM,
852 "Export &ssh.com key");
853 AppendMenu(menu, MF_POPUP | MF_ENABLED, (UINT) menu1,
855 state->cvtmenu = menu1;
857 menu1 = CreateMenu();
858 AppendMenu(menu1, MF_ENABLED, IDC_ABOUT, "&About");
860 AppendMenu(menu1, MF_ENABLED, IDC_GIVEHELP, "&Help");
861 AppendMenu(menu, MF_POPUP | MF_ENABLED, (UINT) menu1, "&Help");
869 { /* centre the window */
873 hw = GetDesktopWindow();
874 if (GetWindowRect(hw, &rs) && GetWindowRect(hwnd, &rd))
876 (rs.right + rs.left + rd.left - rd.right) / 2,
877 (rs.bottom + rs.top + rd.top - rd.bottom) / 2,
878 rd.right - rd.left, rd.bottom - rd.top, TRUE);
882 struct ctlpos cp, cp2;
884 /* Accelerators used: acglops1rbd */
886 ctlposinit(&cp, hwnd, 4, 4, 4);
887 beginbox(&cp, "Key", IDC_BOX_KEY);
889 statictext(&cp2, "No key.", 1, IDC_NOKEY);
891 statictext(&cp2, "", 1, IDC_GENERATING);
892 progressbar(&cp2, IDC_PROGRESS);
894 "&Public key for pasting into authorized_keys file:",
895 IDC_PKSTATIC, IDC_KEYDISPLAY, 5);
896 SendDlgItemMessage(hwnd, IDC_KEYDISPLAY, EM_SETREADONLY, 1, 0);
897 staticedit(&cp, "Key f&ingerprint:", IDC_FPSTATIC,
898 IDC_FINGERPRINT, 75);
899 SendDlgItemMessage(hwnd, IDC_FINGERPRINT, EM_SETREADONLY, 1,
901 staticedit(&cp, "Key &comment:", IDC_COMMENTSTATIC,
902 IDC_COMMENTEDIT, 75);
903 staticpassedit(&cp, "Key p&assphrase:", IDC_PASSPHRASE1STATIC,
904 IDC_PASSPHRASE1EDIT, 75);
905 staticpassedit(&cp, "C&onfirm passphrase:",
906 IDC_PASSPHRASE2STATIC, IDC_PASSPHRASE2EDIT, 75);
908 beginbox(&cp, "Actions", IDC_BOX_ACTIONS);
909 staticbtn(&cp, "Generate a public/private key pair",
910 IDC_GENSTATIC, "&Generate", IDC_GENERATE);
911 staticbtn(&cp, "Load an existing private key file",
912 IDC_LOADSTATIC, "&Load", IDC_LOAD);
913 static2btn(&cp, "Save the generated key", IDC_SAVESTATIC,
914 "Save p&ublic key", IDC_SAVEPUB,
915 "&Save private key", IDC_SAVE);
917 beginbox(&cp, "Parameters", IDC_BOX_PARAMS);
918 radioline(&cp, "Type of key to generate:", IDC_TYPESTATIC, 3,
919 "SSH-&1 (RSA)", IDC_KEYSSH1,
920 "SSH-2 &RSA", IDC_KEYSSH2RSA,
921 "SSH-2 &DSA", IDC_KEYSSH2DSA, NULL);
922 staticedit(&cp, "Number of &bits in a generated key:",
923 IDC_BITSSTATIC, IDC_BITS, 20);
926 CheckRadioButton(hwnd, IDC_KEYSSH1, IDC_KEYSSH2DSA, IDC_KEYSSH2RSA);
927 CheckMenuRadioItem(state->keymenu, IDC_KEYSSH1, IDC_KEYSSH2DSA,
928 IDC_KEYSSH2RSA, MF_BYCOMMAND);
929 SetDlgItemInt(hwnd, IDC_BITS, DEFAULT_KEYSIZE, FALSE);
932 * Initially, hide the progress bar and the key display,
933 * and show the no-key display. Also disable the Save
934 * buttons, because with no key we obviously can't save
937 ui_set_state(hwnd, state, 0);
940 * Load a key file if one was provided on the command line.
943 load_key_file(hwnd, state, filename_from_str(cmdline_keyfile), 0);
947 state = (struct MainDlgState *) GetWindowLongPtr(hwnd, GWLP_USERDATA);
948 if (state->collecting_entropy &&
949 state->entropy && state->entropy_got < state->entropy_required) {
950 state->entropy[state->entropy_got++] = lParam;
951 state->entropy[state->entropy_got++] = GetMessageTime();
952 SendDlgItemMessage(hwnd, IDC_PROGRESS, PBM_SETPOS,
953 state->entropy_got, 0);
954 if (state->entropy_got >= state->entropy_required) {
955 struct rsa_key_thread_params *params;
959 * Seed the entropy pool
961 random_add_heavynoise(state->entropy, state->entropy_size);
962 memset(state->entropy, 0, state->entropy_size);
963 sfree(state->entropy);
964 state->collecting_entropy = FALSE;
966 SetDlgItemText(hwnd, IDC_GENERATING, generating_msg);
967 SendDlgItemMessage(hwnd, IDC_PROGRESS, PBM_SETRANGE, 0,
968 MAKELPARAM(0, PROGRESSRANGE));
969 SendDlgItemMessage(hwnd, IDC_PROGRESS, PBM_SETPOS, 0, 0);
971 params = snew(struct rsa_key_thread_params);
972 params->progressbar = GetDlgItem(hwnd, IDC_PROGRESS);
973 params->dialog = hwnd;
974 params->keysize = state->keysize;
975 params->is_dsa = state->is_dsa;
976 params->key = &state->key;
977 params->dsskey = &state->dsskey;
979 if (!CreateThread(NULL, 0, generate_rsa_key_thread,
980 params, 0, &threadid)) {
981 MessageBox(hwnd, "Out of thread resources",
982 "Key generation error",
983 MB_OK | MB_ICONERROR);
986 state->generation_thread_exists = TRUE;
992 switch (LOWORD(wParam)) {
997 state = (struct MainDlgState *)
998 GetWindowLongPtr(hwnd, GWLP_USERDATA);
999 if (!IsDlgButtonChecked(hwnd, LOWORD(wParam)))
1000 CheckRadioButton(hwnd, IDC_KEYSSH1, IDC_KEYSSH2DSA,
1002 CheckMenuRadioItem(state->keymenu, IDC_KEYSSH1, IDC_KEYSSH2DSA,
1003 LOWORD(wParam), MF_BYCOMMAND);
1007 PostMessage(hwnd, WM_CLOSE, 0, 0);
1009 case IDC_COMMENTEDIT:
1010 if (HIWORD(wParam) == EN_CHANGE) {
1011 state = (struct MainDlgState *)
1012 GetWindowLongPtr(hwnd, GWLP_USERDATA);
1013 if (state->key_exists) {
1014 HWND editctl = GetDlgItem(hwnd, IDC_COMMENTEDIT);
1015 int len = GetWindowTextLength(editctl);
1016 if (*state->commentptr)
1017 sfree(*state->commentptr);
1018 *state->commentptr = snewn(len + 1, char);
1019 GetWindowText(editctl, *state->commentptr, len + 1);
1021 setupbigedit2(hwnd, IDC_KEYDISPLAY, IDC_PKSTATIC,
1024 setupbigedit1(hwnd, IDC_KEYDISPLAY, IDC_PKSTATIC,
1031 EnableWindow(hwnd, 0);
1032 DialogBox(hinst, MAKEINTRESOURCE(213), hwnd, AboutProc);
1033 EnableWindow(hwnd, 1);
1034 SetActiveWindow(hwnd);
1037 if (HIWORD(wParam) == BN_CLICKED ||
1038 HIWORD(wParam) == BN_DOUBLECLICKED) {
1040 WinHelp(hwnd, help_path, HELP_COMMAND,
1041 (DWORD)"JI(`',`puttygen.general')");
1042 requested_help = TRUE;
1047 if (HIWORD(wParam) != BN_CLICKED &&
1048 HIWORD(wParam) != BN_DOUBLECLICKED)
1051 (struct MainDlgState *) GetWindowLongPtr(hwnd, GWLP_USERDATA);
1052 if (!state->generation_thread_exists) {
1054 state->keysize = GetDlgItemInt(hwnd, IDC_BITS, &ok, FALSE);
1056 state->keysize = DEFAULT_KEYSIZE;
1057 /* If we ever introduce a new key type, check it here! */
1058 state->ssh2 = !IsDlgButtonChecked(hwnd, IDC_KEYSSH1);
1059 state->is_dsa = IsDlgButtonChecked(hwnd, IDC_KEYSSH2DSA);
1060 if (state->keysize < 256) {
1061 int ret = MessageBox(hwnd,
1062 "PuTTYgen will not generate a key"
1063 " smaller than 256 bits.\n"
1064 "Key length reset to 256. Continue?",
1066 MB_ICONWARNING | MB_OKCANCEL);
1069 state->keysize = 256;
1070 SetDlgItemInt(hwnd, IDC_BITS, 256, FALSE);
1072 ui_set_state(hwnd, state, 1);
1073 SetDlgItemText(hwnd, IDC_GENERATING, entropy_msg);
1074 state->key_exists = FALSE;
1075 state->collecting_entropy = TRUE;
1078 * My brief statistical tests on mouse movements
1079 * suggest that there are about 2.5 bits of
1080 * randomness in the x position, 2.5 in the y
1081 * position, and 1.7 in the message time, making
1082 * 5.7 bits of unpredictability per mouse movement.
1083 * However, other people have told me it's far less
1084 * than that, so I'm going to be stupidly cautious
1085 * and knock that down to a nice round 2. With this
1086 * method, we require two words per mouse movement,
1087 * so with 2 bits per mouse movement we expect 2
1088 * bits every 2 words.
1090 state->entropy_required = (state->keysize / 2) * 2;
1091 state->entropy_got = 0;
1092 state->entropy_size = (state->entropy_required *
1094 state->entropy = snewn(state->entropy_required, unsigned);
1096 SendDlgItemMessage(hwnd, IDC_PROGRESS, PBM_SETRANGE, 0,
1097 MAKELPARAM(0, state->entropy_required));
1098 SendDlgItemMessage(hwnd, IDC_PROGRESS, PBM_SETPOS, 0, 0);
1102 case IDC_EXPORT_OPENSSH:
1103 case IDC_EXPORT_SSHCOM:
1104 if (HIWORD(wParam) != BN_CLICKED)
1107 (struct MainDlgState *) GetWindowLongPtr(hwnd, GWLP_USERDATA);
1108 if (state->key_exists) {
1109 char filename[FILENAME_MAX];
1110 char passphrase[PASSPHRASE_MAXLEN];
1111 char passphrase2[PASSPHRASE_MAXLEN];
1115 realtype = SSH_KEYTYPE_SSH2;
1117 realtype = SSH_KEYTYPE_SSH1;
1119 if (LOWORD(wParam) == IDC_EXPORT_OPENSSH)
1120 type = SSH_KEYTYPE_OPENSSH;
1121 else if (LOWORD(wParam) == IDC_EXPORT_SSHCOM)
1122 type = SSH_KEYTYPE_SSHCOM;
1126 if (type != realtype &&
1127 import_target_type(type) != realtype) {
1129 sprintf(msg, "Cannot export an SSH-%d key in an SSH-%d"
1130 " format", (state->ssh2 ? 2 : 1),
1131 (state->ssh2 ? 1 : 2));
1132 MessageBox(hwnd, msg,
1133 "PuTTYgen Error", MB_OK | MB_ICONERROR);
1137 GetDlgItemText(hwnd, IDC_PASSPHRASE1EDIT,
1138 passphrase, sizeof(passphrase));
1139 GetDlgItemText(hwnd, IDC_PASSPHRASE2EDIT,
1140 passphrase2, sizeof(passphrase2));
1141 if (strcmp(passphrase, passphrase2)) {
1143 "The two passphrases given do not match.",
1144 "PuTTYgen Error", MB_OK | MB_ICONERROR);
1149 ret = MessageBox(hwnd,
1150 "Are you sure you want to save this key\n"
1151 "without a passphrase to protect it?",
1153 MB_YESNO | MB_ICONWARNING);
1157 if (prompt_keyfile(hwnd, "Save private key as:",
1158 filename, 1, (type == realtype))) {
1160 FILE *fp = fopen(filename, "r");
1164 buffer = dupprintf("Overwrite existing file\n%s?",
1166 ret = MessageBox(hwnd, buffer, "PuTTYgen Warning",
1167 MB_YESNO | MB_ICONWARNING);
1174 Filename fn = filename_from_str(filename);
1175 if (type != realtype)
1176 ret = export_ssh2(&fn, type, &state->ssh2key,
1177 *passphrase ? passphrase : NULL);
1179 ret = ssh2_save_userkey(&fn, &state->ssh2key,
1180 *passphrase ? passphrase :
1183 Filename fn = filename_from_str(filename);
1184 if (type != realtype)
1185 ret = export_ssh1(&fn, type, &state->key,
1186 *passphrase ? passphrase : NULL);
1188 ret = saversakey(&fn, &state->key,
1189 *passphrase ? passphrase : NULL);
1192 MessageBox(hwnd, "Unable to save key file",
1193 "PuTTYgen Error", MB_OK | MB_ICONERROR);
1199 if (HIWORD(wParam) != BN_CLICKED)
1202 (struct MainDlgState *) GetWindowLongPtr(hwnd, GWLP_USERDATA);
1203 if (state->key_exists) {
1204 char filename[FILENAME_MAX];
1205 if (prompt_keyfile(hwnd, "Save public key as:",
1208 FILE *fp = fopen(filename, "r");
1212 buffer = dupprintf("Overwrite existing file\n%s?",
1214 ret = MessageBox(hwnd, buffer, "PuTTYgen Warning",
1215 MB_YESNO | MB_ICONWARNING);
1221 ret = save_ssh2_pubkey(filename, &state->ssh2key);
1223 ret = save_ssh1_pubkey(filename, &state->key);
1226 MessageBox(hwnd, "Unable to save key file",
1227 "PuTTYgen Error", MB_OK | MB_ICONERROR);
1234 if (HIWORD(wParam) != BN_CLICKED)
1237 (struct MainDlgState *) GetWindowLongPtr(hwnd, GWLP_USERDATA);
1238 if (!state->generation_thread_exists) {
1239 char filename[FILENAME_MAX];
1240 if (prompt_keyfile(hwnd, "Load private key:",
1241 filename, 0, LOWORD(wParam)==IDC_LOAD))
1242 load_key_file(hwnd, state, filename_from_str(filename),
1243 LOWORD(wParam) != IDC_LOAD);
1249 state = (struct MainDlgState *) GetWindowLongPtr(hwnd, GWLP_USERDATA);
1250 state->generation_thread_exists = FALSE;
1251 state->key_exists = TRUE;
1252 SendDlgItemMessage(hwnd, IDC_PROGRESS, PBM_SETRANGE, 0,
1253 MAKELPARAM(0, PROGRESSRANGE));
1254 SendDlgItemMessage(hwnd, IDC_PROGRESS, PBM_SETPOS, PROGRESSRANGE, 0);
1256 if (state->is_dsa) {
1257 state->ssh2key.data = &state->dsskey;
1258 state->ssh2key.alg = &ssh_dss;
1260 state->ssh2key.data = &state->key;
1261 state->ssh2key.alg = &ssh_rsa;
1263 state->commentptr = &state->ssh2key.comment;
1265 state->commentptr = &state->key.comment;
1268 * Invent a comment for the key. We'll do this by including
1269 * the date in it. This will be so horrifyingly ugly that
1270 * the user will immediately want to change it, which is
1273 *state->commentptr = snewn(30, char);
1278 strftime(*state->commentptr, 30, "dsa-key-%Y%m%d", &tm);
1280 strftime(*state->commentptr, 30, "rsa-key-%Y%m%d", &tm);
1284 * Now update the key controls with all the key data.
1289 * Blank passphrase, initially. This isn't dangerous,
1290 * because we will warn (Are You Sure?) before allowing
1291 * the user to save an unprotected private key.
1293 SetDlgItemText(hwnd, IDC_PASSPHRASE1EDIT, "");
1294 SetDlgItemText(hwnd, IDC_PASSPHRASE2EDIT, "");
1298 SetDlgItemText(hwnd, IDC_COMMENTEDIT, *state->commentptr);
1300 * Set the key fingerprint.
1302 savecomment = *state->commentptr;
1303 *state->commentptr = NULL;
1306 fp = state->ssh2key.alg->fingerprint(state->ssh2key.data);
1307 SetDlgItemText(hwnd, IDC_FINGERPRINT, fp);
1311 rsa_fingerprint(buf, sizeof(buf), &state->key);
1312 SetDlgItemText(hwnd, IDC_FINGERPRINT, buf);
1314 *state->commentptr = savecomment;
1316 * Construct a decimal representation of the key, for
1317 * pasting into .ssh/authorized_keys or
1318 * .ssh/authorized_keys2 on a Unix box.
1321 setupbigedit2(hwnd, IDC_KEYDISPLAY,
1322 IDC_PKSTATIC, &state->ssh2key);
1324 setupbigedit1(hwnd, IDC_KEYDISPLAY,
1325 IDC_PKSTATIC, &state->key);
1329 * Finally, hide the progress bar and show the key data.
1331 ui_set_state(hwnd, state, 2);
1335 int id = ((LPHELPINFO)lParam)->iCtrlId;
1338 case IDC_GENERATING:
1342 topic = "puttygen.generate"; break;
1344 case IDC_KEYDISPLAY:
1345 topic = "puttygen.pastekey"; break;
1347 case IDC_FINGERPRINT:
1348 topic = "puttygen.fingerprint"; break;
1349 case IDC_COMMENTSTATIC:
1350 case IDC_COMMENTEDIT:
1351 topic = "puttygen.comment"; break;
1352 case IDC_PASSPHRASE1STATIC:
1353 case IDC_PASSPHRASE1EDIT:
1354 case IDC_PASSPHRASE2STATIC:
1355 case IDC_PASSPHRASE2EDIT:
1356 topic = "puttygen.passphrase"; break;
1357 case IDC_LOADSTATIC:
1359 topic = "puttygen.load"; break;
1360 case IDC_SAVESTATIC:
1362 topic = "puttygen.savepriv"; break;
1364 topic = "puttygen.savepub"; break;
1365 case IDC_TYPESTATIC:
1367 case IDC_KEYSSH2RSA:
1368 case IDC_KEYSSH2DSA:
1369 topic = "puttygen.keytype"; break;
1370 case IDC_BITSSTATIC:
1372 topic = "puttygen.bits"; break;
1374 case IDC_EXPORT_OPENSSH:
1375 case IDC_EXPORT_SSHCOM:
1376 topic = "puttygen.conversions"; break;
1379 char *cmd = dupprintf("JI(`',`%s')", topic);
1380 WinHelp(hwnd, help_path, HELP_COMMAND, (DWORD)cmd);
1382 requested_help = TRUE;
1389 state = (struct MainDlgState *) GetWindowLongPtr(hwnd, GWLP_USERDATA);
1391 if (requested_help) {
1392 WinHelp(hwnd, help_path, HELP_QUIT, 0);
1393 requested_help = FALSE;
1401 void cleanup_exit(int code) { exit(code); }
1403 int WINAPI WinMain(HINSTANCE inst, HINSTANCE prev, LPSTR cmdline, int show)
1408 InitCommonControls();
1413 * See if we can find our Help file.
1416 char b[2048], *p, *q, *r;
1418 GetModuleFileName(NULL, b, sizeof(b) - 1);
1420 p = strrchr(b, '\\');
1421 if (p && p >= r) r = p+1;
1422 q = strrchr(b, ':');
1423 if (q && q >= r) r = q+1;
1424 strcpy(r, PUTTY_HELP_FILE);
1425 if ( (fp = fopen(b, "r")) != NULL) {
1426 help_path = dupstr(b);
1432 split_into_argv(cmdline, &argc, &argv, NULL);
1435 if (!strcmp(argv[0], "-pgpfp")) {
1440 * Assume the first argument to be a private key file, and
1441 * attempt to load it.
1443 cmdline_keyfile = argv[0];
1448 return DialogBox(hinst, MAKEINTRESOURCE(201), NULL,
1449 MainDlgProc) != IDOK;