#include #include #include #include #include #include #include "ssh.h" #include "putty.h" #include "win_res.h" #define NPANELS 8 #define MAIN_NPANELS 8 #define RECONF_NPANELS 5 static const char *const puttystr = PUTTY_REG_POS "\\Sessions"; static char **events = NULL; static int nevents = 0, negsize = 0; static HWND logbox = NULL, abtbox = NULL; static char hex[16] = "0123456789ABCDEF"; static void mungestr(char *in, char *out) { int candot = 0; while (*in) { if (*in == ' ' || *in == '\\' || *in == '*' || *in == '?' || *in == '%' || *in < ' ' || *in > '~' || (*in == '.' && !candot)) { *out++ = '%'; *out++ = hex[((unsigned char)*in) >> 4]; *out++ = hex[((unsigned char)*in) & 15]; } else *out++ = *in; in++; candot = 1; } *out = '\0'; return; } static void unmungestr(char *in, char *out) { while (*in) { if (*in == '%' && in[1] && in[2]) { int i, j; i = in[1] - '0'; i -= (i > 9 ? 7 : 0); j = in[2] - '0'; j -= (j > 9 ? 7 : 0); *out++ = (i<<4) + j; in += 3; } else *out++ = *in++; } *out = '\0'; return; } static void wpps(HKEY key, LPCTSTR name, LPCTSTR value) { RegSetValueEx(key, name, 0, REG_SZ, value, 1+strlen(value)); } static void wppi(HKEY key, LPCTSTR name, int value) { RegSetValueEx(key, name, 0, REG_DWORD, (CONST BYTE *)&value, sizeof(value)); } static void gpps(HKEY key, LPCTSTR name, LPCTSTR def, LPTSTR val, int len) { DWORD type, size; size = len; if (key == NULL || RegQueryValueEx(key, name, 0, &type, val, &size) != ERROR_SUCCESS || type != REG_SZ) { strncpy(val, def, len); val[len-1] = '\0'; } } static void gppi(HKEY key, LPCTSTR name, int def, int *i) { DWORD type, val, size; size = sizeof(val); if (key == NULL || RegQueryValueEx(key, name, 0, &type, (BYTE *)&val, &size) != ERROR_SUCCESS || size != sizeof(val) || type != REG_DWORD) *i = def; else *i = val; } static HINSTANCE hinst; static int readytogo; static void save_settings (char *section, int do_host) { int i; HKEY subkey1, sesskey; char *p; p = malloc(3*strlen(section)+1); mungestr(section, p); if (RegCreateKey(HKEY_CURRENT_USER, puttystr, &subkey1)!=ERROR_SUCCESS || RegCreateKey(subkey1, p, &sesskey) != ERROR_SUCCESS) { sesskey = NULL; } free(p); RegCloseKey(subkey1); wppi (sesskey, "Present", 1); if (do_host) { wpps (sesskey, "HostName", cfg.host); wppi (sesskey, "PortNumber", cfg.port); p = "raw"; for (i = 0; backends[i].name != NULL; i++) if (backends[i].protocol == cfg.protocol) { p = backends[i].name; break; } wpps (sesskey, "Protocol", p); } wppi (sesskey, "CloseOnExit", !!cfg.close_on_exit); wppi (sesskey, "WarnOnClose", !!cfg.warn_on_close); wpps (sesskey, "TerminalType", cfg.termtype); wpps (sesskey, "TerminalSpeed", cfg.termspeed); { char buf[2*sizeof(cfg.environmt)], *p, *q; p = buf; q = cfg.environmt; while (*q) { while (*q) { int c = *q++; if (c == '=' || c == ',' || c == '\\') *p++ = '\\'; if (c == '\t') c = '='; *p++ = c; } *p++ = ','; q++; } *p = '\0'; wpps (sesskey, "Environment", buf); } wpps (sesskey, "UserName", cfg.username); wppi (sesskey, "NoPTY", cfg.nopty); wpps (sesskey, "Cipher", cfg.cipher == CIPHER_BLOWFISH ? "blowfish" : cfg.cipher == CIPHER_DES ? "des" : "3des"); wppi (sesskey, "AuthTIS", cfg.try_tis_auth); wppi (sesskey, "SSHCompression", cfg.ssh_compression); wppi (sesskey, "RFCEnviron", cfg.rfc_environ); wppi (sesskey, "BackspaceIsDelete", cfg.bksp_is_delete); wppi (sesskey, "RXVTHomeEnd", cfg.rxvt_homeend); wppi (sesskey, "LinuxFunctionKeys", cfg.linux_funkeys); wppi (sesskey, "ApplicationCursorKeys", cfg.app_cursor); wppi (sesskey, "ApplicationKeypad", cfg.app_keypad); wppi (sesskey, "NetHackKeypad", cfg.nethack_keypad); wppi (sesskey, "AltF4", cfg.alt_f4); wppi (sesskey, "AltSpace", cfg.alt_space); wppi (sesskey, "LdiscTerm", cfg.ldisc_term); wppi (sesskey, "BlinkCur", cfg.blink_cur); wppi (sesskey, "Beep", cfg.beep); wppi (sesskey, "ScrollbackLines", cfg.savelines); wppi (sesskey, "DECOriginMode", cfg.dec_om); wppi (sesskey, "AutoWrapMode", cfg.wrap_mode); wppi (sesskey, "LFImpliesCR", cfg.lfhascr); wppi (sesskey, "WinNameAlways", cfg.win_name_always); wppi (sesskey, "TermWidth", cfg.width); wppi (sesskey, "TermHeight", cfg.height); wpps (sesskey, "Font", cfg.font); wppi (sesskey, "FontIsBold", cfg.fontisbold); wppi (sesskey, "FontCharSet", cfg.fontcharset); wppi (sesskey, "FontHeight", cfg.fontheight); wppi (sesskey, "FontVTMode", cfg.vtmode); wppi (sesskey, "TryPalette", cfg.try_palette); wppi (sesskey, "BoldAsColour", cfg.bold_colour); for (i=0; i<22; i++) { char buf[20], buf2[30]; sprintf(buf, "Colour%d", i); sprintf(buf2, "%d,%d,%d", cfg.colours[i][0], cfg.colours[i][1], cfg.colours[i][2]); wpps (sesskey, buf, buf2); } wppi (sesskey, "MouseIsXterm", cfg.mouse_is_xterm); for (i=0; i<256; i+=32) { char buf[20], buf2[256]; int j; sprintf(buf, "Wordness%d", i); *buf2 = '\0'; for (j=i; j 0) { if (!*p) goto disaster; while (*p) p++; p++; i--; } q = p; if (!*p) goto disaster; while (*p) p++; p++; while (*p) { while (*p) *q++ = *p++; *q++ = *p++; } *q = '\0'; disaster:; } break; } break; } return GeneralPanelProc (hwnd, msg, wParam, lParam); } static int CALLBACK SshProc (HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) { switch (msg) { case WM_INITDIALOG: SetDlgItemText (hwnd, IDC3_TTEDIT, cfg.termtype); SetDlgItemText (hwnd, IDC3_LOGEDIT, cfg.username); CheckDlgButton (hwnd, IDC3_NOPTY, cfg.nopty); CheckRadioButton (hwnd, IDC3_CIPHER3DES, IDC3_CIPHERDES, cfg.cipher == CIPHER_BLOWFISH ? IDC3_CIPHERBLOWF : cfg.cipher == CIPHER_DES ? IDC3_CIPHERDES : IDC3_CIPHER3DES); CheckDlgButton (hwnd, IDC3_AUTHTIS, cfg.try_tis_auth); CheckDlgButton (hwnd, IDC3_COMPRESS, cfg.ssh_compression); break; case WM_COMMAND: switch (LOWORD(wParam)) { case IDC3_TTEDIT: if (HIWORD(wParam) == EN_CHANGE) GetDlgItemText (hwnd, IDC3_TTEDIT, cfg.termtype, sizeof(cfg.termtype)-1); break; case IDC3_LOGEDIT: if (HIWORD(wParam) == EN_CHANGE) GetDlgItemText (hwnd, IDC3_LOGEDIT, cfg.username, sizeof(cfg.username)-1); break; case IDC3_NOPTY: if (HIWORD(wParam) == BN_CLICKED || HIWORD(wParam) == BN_DOUBLECLICKED) cfg.nopty = IsDlgButtonChecked (hwnd, IDC3_NOPTY); break; case IDC3_CIPHER3DES: case IDC3_CIPHERBLOWF: case IDC3_CIPHERDES: if (HIWORD(wParam) == BN_CLICKED || HIWORD(wParam) == BN_DOUBLECLICKED) { if (IsDlgButtonChecked (hwnd, IDC3_CIPHER3DES)) cfg.cipher = CIPHER_3DES; else if (IsDlgButtonChecked (hwnd, IDC3_CIPHERBLOWF)) cfg.cipher = CIPHER_BLOWFISH; else if (IsDlgButtonChecked (hwnd, IDC3_CIPHERDES)) cfg.cipher = CIPHER_DES; } break; case IDC3_AUTHTIS: if (HIWORD(wParam) == BN_CLICKED || HIWORD(wParam) == BN_DOUBLECLICKED) cfg.try_tis_auth = IsDlgButtonChecked (hwnd, IDC3_AUTHTIS); break; case IDC3_COMPRESS: if (HIWORD(wParam) == BN_CLICKED || HIWORD(wParam) == BN_DOUBLECLICKED) cfg.ssh_compression = IsDlgButtonChecked (hwnd, IDC3_COMPRESS); break; } break; } return GeneralPanelProc (hwnd, msg, wParam, lParam); } static int CALLBACK SelectionProc (HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) { int i; switch (msg) { case WM_INITDIALOG: CheckRadioButton (hwnd, IDC4_MBWINDOWS, IDC4_MBXTERM, cfg.mouse_is_xterm ? IDC4_MBXTERM : IDC4_MBWINDOWS); { static int tabs[4] = {25, 61, 96, 128}; SendDlgItemMessage (hwnd, IDC4_CCLIST, LB_SETTABSTOPS, 4, (LPARAM) tabs); } for (i=0; i<256; i++) { char str[100]; sprintf(str, "%d\t(0x%02X)\t%c\t%d", i, i, (i>=0x21 && i != 0x7F) ? i : ' ', cfg.wordness[i]); SendDlgItemMessage (hwnd, IDC4_CCLIST, LB_ADDSTRING, 0, (LPARAM) str); } break; case WM_COMMAND: switch (LOWORD(wParam)) { case IDC4_MBWINDOWS: case IDC4_MBXTERM: cfg.mouse_is_xterm = IsDlgButtonChecked (hwnd, IDC4_MBXTERM); break; case IDC4_CCSET: { BOOL ok; int i; int n = GetDlgItemInt (hwnd, IDC4_CCEDIT, &ok, FALSE); if (!ok) MessageBeep (0); else { for (i=0; i<256; i++) if (SendDlgItemMessage (hwnd, IDC4_CCLIST, LB_GETSEL, i, 0)) { char str[100]; cfg.wordness[i] = n; SendDlgItemMessage (hwnd, IDC4_CCLIST, LB_DELETESTRING, i, 0); sprintf(str, "%d\t(0x%02X)\t%c\t%d", i, i, (i>=0x21 && i != 0x7F) ? i : ' ', cfg.wordness[i]); SendDlgItemMessage (hwnd, IDC4_CCLIST, LB_INSERTSTRING, i, (LPARAM)str); } } } break; } break; } return GeneralPanelProc (hwnd, msg, wParam, lParam); } static int CALLBACK ColourProc (HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) { static const char *const colours[] = { "Default Foreground", "Default Bold Foreground", "Default Background", "Default Bold Background", "Cursor Text", "Cursor Colour", "ANSI Black", "ANSI Black Bold", "ANSI Red", "ANSI Red Bold", "ANSI Green", "ANSI Green Bold", "ANSI Yellow", "ANSI Yellow Bold", "ANSI Blue", "ANSI Blue Bold", "ANSI Magenta", "ANSI Magenta Bold", "ANSI Cyan", "ANSI Cyan Bold", "ANSI White", "ANSI White Bold" }; static const int permanent[] = { TRUE, FALSE, TRUE, FALSE, TRUE, TRUE, TRUE, FALSE, TRUE, FALSE, TRUE, FALSE, TRUE, FALSE, TRUE, FALSE, TRUE, FALSE, TRUE, FALSE, TRUE, FALSE }; switch (msg) { case WM_INITDIALOG: CheckDlgButton (hwnd, IDC5_BOLDCOLOUR, cfg.bold_colour); CheckDlgButton (hwnd, IDC5_PALETTE, cfg.try_palette); { int i; for (i=0; i<22; i++) if (cfg.bold_colour || permanent[i]) SendDlgItemMessage (hwnd, IDC5_LIST, LB_ADDSTRING, 0, (LPARAM) colours[i]); } SendDlgItemMessage (hwnd, IDC5_LIST, LB_SETCURSEL, 0, 0); SetDlgItemInt (hwnd, IDC5_RVALUE, cfg.colours[0][0], FALSE); SetDlgItemInt (hwnd, IDC5_GVALUE, cfg.colours[0][1], FALSE); SetDlgItemInt (hwnd, IDC5_BVALUE, cfg.colours[0][2], FALSE); break; case WM_COMMAND: switch (LOWORD(wParam)) { case IDC5_BOLDCOLOUR: if (HIWORD(wParam) == BN_CLICKED || HIWORD(wParam) == BN_DOUBLECLICKED) { int n, i; cfg.bold_colour = IsDlgButtonChecked (hwnd, IDC5_BOLDCOLOUR); n = SendDlgItemMessage (hwnd, IDC5_LIST, LB_GETCOUNT, 0, 0); if (cfg.bold_colour && n!=22) { for (i=0; i<22; i++) if (!permanent[i]) SendDlgItemMessage (hwnd, IDC5_LIST, LB_INSERTSTRING, i, (LPARAM) colours[i]); } else if (!cfg.bold_colour && n!=12) { for (i=22; i-- ;) if (!permanent[i]) SendDlgItemMessage (hwnd, IDC5_LIST, LB_DELETESTRING, i, 0); } } break; case IDC5_PALETTE: if (HIWORD(wParam) == BN_CLICKED || HIWORD(wParam) == BN_DOUBLECLICKED) cfg.try_palette = IsDlgButtonChecked (hwnd, IDC5_PALETTE); break; case IDC5_LIST: if (HIWORD(wParam) == LBN_DBLCLK || HIWORD(wParam) == LBN_SELCHANGE) { int i = SendDlgItemMessage (hwnd, IDC5_LIST, LB_GETCURSEL, 0, 0); if (!cfg.bold_colour) i = (i < 3 ? i*2 : i == 3 ? 5 : i*2-2); SetDlgItemInt (hwnd, IDC5_RVALUE, cfg.colours[i][0], FALSE); SetDlgItemInt (hwnd, IDC5_GVALUE, cfg.colours[i][1], FALSE); SetDlgItemInt (hwnd, IDC5_BVALUE, cfg.colours[i][2], FALSE); } break; case IDC5_CHANGE: if (HIWORD(wParam) == BN_CLICKED || HIWORD(wParam) == BN_DOUBLECLICKED) { static CHOOSECOLOR cc; static DWORD custom[16] = {0}; /* zero initialisers */ int i = SendDlgItemMessage (hwnd, IDC5_LIST, LB_GETCURSEL, 0, 0); if (!cfg.bold_colour) i = (i < 3 ? i*2 : i == 3 ? 5 : i*2-2); cc.lStructSize = sizeof(cc); cc.hwndOwner = hwnd; cc.hInstance = (HWND)hinst; cc.lpCustColors = custom; cc.rgbResult = RGB (cfg.colours[i][0], cfg.colours[i][1], cfg.colours[i][2]); cc.Flags = CC_FULLOPEN | CC_RGBINIT; if (ChooseColor(&cc)) { cfg.colours[i][0] = (unsigned char) (cc.rgbResult & 0xFF); cfg.colours[i][1] = (unsigned char) (cc.rgbResult >> 8) & 0xFF; cfg.colours[i][2] = (unsigned char) (cc.rgbResult >> 16) & 0xFF; SetDlgItemInt (hwnd, IDC5_RVALUE, cfg.colours[i][0], FALSE); SetDlgItemInt (hwnd, IDC5_GVALUE, cfg.colours[i][1], FALSE); SetDlgItemInt (hwnd, IDC5_BVALUE, cfg.colours[i][2], FALSE); } } break; } break; } return GeneralPanelProc (hwnd, msg, wParam, lParam); } static int CALLBACK LanguageProc (HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) { switch (msg) { case WM_INITDIALOG: CheckRadioButton (hwnd, IDC6_NOXLAT, IDC6_88592WIN1250, cfg.xlat_88592w1250 ? IDC6_88592WIN1250 : cfg.xlat_enablekoiwin ? IDC6_KOI8WIN1251 : IDC6_NOXLAT); CheckDlgButton (hwnd, IDC6_CAPSLOCKCYR, cfg.xlat_capslockcyr); case WM_COMMAND: switch (LOWORD(wParam)) { case IDC6_NOXLAT: case IDC6_KOI8WIN1251: case IDC6_88592WIN1250: cfg.xlat_enablekoiwin = IsDlgButtonChecked (hwnd, IDC6_KOI8WIN1251); cfg.xlat_88592w1250 = IsDlgButtonChecked (hwnd, IDC6_88592WIN1250); break; case IDC6_CAPSLOCKCYR: if (HIWORD(wParam) == BN_CLICKED || HIWORD(wParam) == BN_DOUBLECLICKED) { cfg.xlat_capslockcyr = IsDlgButtonChecked (hwnd, IDC6_CAPSLOCKCYR); } break; } } return GeneralPanelProc (hwnd, msg, wParam, lParam); } static DLGPROC panelproc[NPANELS] = { ConnectionProc, KeyboardProc, TerminalProc, TelnetProc, SshProc, SelectionProc, ColourProc, LanguageProc }; static char *panelids[NPANELS] = { MAKEINTRESOURCE(IDD_PANEL0), MAKEINTRESOURCE(IDD_PANEL1), MAKEINTRESOURCE(IDD_PANEL2), MAKEINTRESOURCE(IDD_PANEL3), MAKEINTRESOURCE(IDD_PANEL35), MAKEINTRESOURCE(IDD_PANEL4), MAKEINTRESOURCE(IDD_PANEL5), MAKEINTRESOURCE(IDD_PANEL6) }; static char *names[NPANELS] = { "Connection", "Keyboard", "Terminal", "Telnet", "SSH", "Selection", "Colours", "Language" }; static int mainp[MAIN_NPANELS] = { 0, 1, 2, 3, 4, 5, 6, 7}; static int reconfp[RECONF_NPANELS] = { 1, 2, 5, 6, 7}; static int GenericMainDlgProc (HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam, int npanels, int *panelnums, HWND *page) { HWND hw; switch (msg) { case WM_INITDIALOG: { /* centre the window */ RECT rs, rd; hw = GetDesktopWindow(); if (GetWindowRect (hw, &rs) && GetWindowRect (hwnd, &rd)) MoveWindow (hwnd, (rs.right + rs.left + rd.left - rd.right)/2, (rs.bottom + rs.top + rd.top - rd.bottom)/2, rd.right-rd.left, rd.bottom-rd.top, TRUE); } *page = NULL; { /* initialise the tab control */ TC_ITEMHEADER tab; int i; hw = GetDlgItem (hwnd, IDC_TAB); for (i=0; icode == TCN_SELCHANGE) { int i = TabCtrl_GetCurSel(((LPNMHDR)lParam)->hwndFrom); if (*page) DestroyWindow (*page); /* *page = CreateDialogIndirect (hinst, panels[panelnums[i]].temp, hwnd, panelproc[panelnums[i]]);*/ *page = CreateDialog (hinst, panelids[panelnums[i]], hwnd, panelproc[panelnums[i]]); SetWindowLong (*page, GWL_EXSTYLE, GetWindowLong (*page, GWL_EXSTYLE) | WS_EX_CONTROLPARENT); SetFocus (((LPNMHDR)lParam)->hwndFrom); /* ensure focus stays */ return 0; } break; /* case WM_CTLCOLORDLG: */ /* return (int) GetStockObject (LTGRAY_BRUSH); */ case WM_COMMAND: switch (LOWORD(wParam)) { case IDOK: if (*cfg.host) EndDialog (hwnd, 1); else MessageBeep (0); return 0; case IDCANCEL: EndDialog (hwnd, 0); return 0; } return 0; case WM_CLOSE: EndDialog (hwnd, 0); return 0; } return 0; } static int CALLBACK MainDlgProc (HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) { #if 0 HWND hw; int i; #endif static HWND page = NULL; if (msg == WM_COMMAND && LOWORD(wParam) == IDOK) { #if 0 /* * If the Connection panel is active and the Session List * box is selected, we treat a press of Open to have an * implicit press of Load preceding it. */ hw = GetDlgItem (hwnd, IDC_TAB); i = TabCtrl_GetCurSel(hw); if (panelproc[mainp[i]] == ConnectionProc && page && implicit_load_ok) { SendMessage (page, WM_COMMAND, MAKELONG(IDC0_SESSLOAD, BN_CLICKED), 0); } #endif } if (msg == WM_COMMAND && LOWORD(wParam) == IDC_ABOUT) { EnableWindow(hwnd, 0); DialogBox(hinst, MAKEINTRESOURCE(IDD_ABOUTBOX), GetParent(hwnd), AboutProc); EnableWindow(hwnd, 1); } return GenericMainDlgProc (hwnd, msg, wParam, lParam, MAIN_NPANELS, mainp, &page); } static int CALLBACK ReconfDlgProc (HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) { static HWND page; return GenericMainDlgProc (hwnd, msg, wParam, lParam, RECONF_NPANELS, reconfp, &page); } void get_sesslist(int allocate) { static char *buffer; int buflen, bufsize, i, ret; char otherbuf[2048]; char *p; HKEY subkey1; if (allocate) { if (RegCreateKey(HKEY_CURRENT_USER, puttystr, &subkey1) != ERROR_SUCCESS) return; buflen = bufsize = 0; buffer = NULL; i = 0; do { ret = RegEnumKey(subkey1, i++, otherbuf, sizeof(otherbuf)); if (ret == ERROR_SUCCESS) { bufsize = buflen + 2048; buffer = srealloc(buffer, bufsize); unmungestr(otherbuf, buffer+buflen); buflen += strlen(buffer+buflen)+1; } } while (ret == ERROR_SUCCESS); buffer = srealloc(buffer, buflen+1); buffer[buflen] = '\0'; p = buffer; nsessions = 1; /* "Default Settings" counts as one */ while (*p) { if (strcmp(p, "Default Settings")) nsessions++; while (*p) p++; p++; } sessions = smalloc(nsessions * sizeof(char *)); sessions[0] = "Default Settings"; p = buffer; i = 1; while (*p) { if (strcmp(p, "Default Settings")) sessions[i++] = p; while (*p) p++; p++; } } else { sfree (buffer); sfree (sessions); } } int do_config (void) { int ret; get_sesslist(TRUE); ret = DialogBox (hinst, MAKEINTRESOURCE(IDD_MAINBOX), NULL, MainDlgProc); get_sesslist(FALSE); return ret; } int do_reconfig (HWND hwnd) { Config backup_cfg; int ret; backup_cfg = cfg; /* structure copy */ ret = DialogBox (hinst, MAKEINTRESOURCE(IDD_RECONF), hwnd, ReconfDlgProc); if (!ret) cfg = backup_cfg; /* structure copy */ return ret; } void do_defaults (char *session) { if (session) load_settings (session, TRUE); else load_settings ("Default Settings", FALSE); } void logevent (char *string) { if (IS_SCP && (scp_flags & SCP_VERBOSE) != 0) fprintf(stderr, "%s\n", string); if (nevents >= negsize) { negsize += 64; events = srealloc (events, negsize * sizeof(*events)); } events[nevents] = smalloc(1+strlen(string)); strcpy (events[nevents], string); nevents++; if (logbox) SendDlgItemMessage (logbox, IDN_LIST, LB_ADDSTRING, 0, (LPARAM)string); } void showeventlog (HWND hwnd) { if (!logbox) { logbox = CreateDialog (hinst, MAKEINTRESOURCE(IDD_LOGBOX), hwnd, LogProc); ShowWindow (logbox, SW_SHOWNORMAL); } } void showabout (HWND hwnd) { if (!abtbox) { abtbox = CreateDialog (hinst, MAKEINTRESOURCE(IDD_ABOUTBOX), hwnd, AboutProc); ShowWindow (abtbox, SW_SHOWNORMAL); } } void verify_ssh_host_key(char *host, char *keystr) { char *otherstr, *mungedhost; int len; HKEY rkey; len = 1 + strlen(keystr); /* * Now read a saved key in from the registry and see what it * says. */ otherstr = malloc(len); mungedhost = malloc(3*strlen(host)+1); if (!otherstr || !mungedhost) fatalbox("Out of memory"); mungestr(host, mungedhost); if (RegCreateKey(HKEY_CURRENT_USER, PUTTY_REG_POS "\\SshHostKeys", &rkey) != ERROR_SUCCESS) { if (MessageBox(NULL, "PuTTY was unable to open the host key cache\n" "in the registry. There is thus no way to tell\n" "if the remote host is what you think it is.\n" "Connect anyway?", "PuTTY Problem", MB_ICONWARNING | MB_YESNO) == IDNO) exit(0); } else { DWORD readlen = len; DWORD type; int ret; ret = RegQueryValueEx(rkey, mungedhost, NULL, &type, otherstr, &readlen); if (ret == ERROR_MORE_DATA || (ret == ERROR_SUCCESS && type == REG_SZ && strcmp(otherstr, keystr))) { if (MessageBox(NULL, "This host's host key is different from the\n" "one cached in the registry! Someone may be\n" "impersonating this host for malicious reasons;\n" "alternatively, the host key may have changed\n" "due to sloppy system administration.\n" "Replace key in registry and connect?", "PuTTY: Security Warning", MB_ICONWARNING | MB_YESNO) == IDNO) exit(0); RegSetValueEx(rkey, mungedhost, 0, REG_SZ, keystr, strlen(keystr)+1); } else if (ret != ERROR_SUCCESS || type != REG_SZ) { if (MessageBox(NULL, "This host's host key is not cached in the\n" "registry. Do you want to add it to the cache\n" "and carry on connecting?", "PuTTY: New Host", MB_ICONWARNING | MB_YESNO) == IDNO) exit(0); RegSetValueEx(rkey, mungedhost, 0, REG_SZ, keystr, strlen(keystr)+1); } RegCloseKey(rkey); } }