window.c

   1 #include <windows.h>
   2 #include <commctrl.h>
   3 #include <winsock.h>
   4 #include <stdio.h>
   5 #include <stdlib.h>
   6
   7 #define PUTTY_DO_GLOBALS                       /* actually _define_ globals */
   8 #include "putty.h"
   9 #include "win_res.h"
  10
  11 #define IDM_SHOWLOG   0x0010
  12 #define IDM_NEWSESS   0x0020
  13 #define IDM_DUPSESS   0x0030
  14 #define IDM_RECONF    0x0040
  15 #define IDM_CLRSB     0x0050
  16 #define IDM_RESET     0x0060
  17 #define IDM_TEL_AYT   0x0070
  18 #define IDM_TEL_BRK   0x0080
  19 #define IDM_TEL_SYNCH 0x0090
  20 #define IDM_TEL_EC    0x00a0
  21 #define IDM_TEL_EL    0x00b0
  22 #define IDM_TEL_GA    0x00c0
  23 #define IDM_TEL_NOP   0x00d0
  24 #define IDM_TEL_ABORT 0x00e0
  25 #define IDM_TEL_AO    0x00f0
  26 #define IDM_TEL_IP    0x0100
  27 #define IDM_TEL_SUSP  0x0110
  28 #define IDM_TEL_EOR   0x0120
  29 #define IDM_TEL_EOF   0x0130
  30 #define IDM_ABOUT     0x0140
  31 #define IDM_SAVEDSESS 0x0150
  32
  33 #define IDM_SAVED_MIN 0x1000
  34 #define IDM_SAVED_MAX 0x2000
  35
  36 #define WM_IGNORE_SIZE (WM_USER + 2)
  37 #define WM_IGNORE_CLIP (WM_USER + 3)
  38
  39 static int WINAPI WndProc (HWND, UINT, WPARAM, LPARAM);
  40 static int TranslateKey(WPARAM wParam, LPARAM lParam, unsigned char *output);
  41 static void cfgtopalette(void);
  42 static void init_palette(void);
  43 static void init_fonts(void);
  44
  45 static int extra_width, extra_height;
  46
  47 #define FONT_NORMAL 0
  48 #define FONT_BOLD 1
  49 #define FONT_UNDERLINE 2
  50 #define FONT_BOLDUND 3
  51 #define FONT_OEM 4
  52 #define FONT_OEMBOLD 5
  53 #define FONT_OEMBOLDUND 6
  54 #define FONT_OEMUND 7
  55 static HFONT fonts[8];
  56 static enum {
  57     BOLD_COLOURS, BOLD_SHADOW, BOLD_FONT
  58 } bold_mode;
  59 static enum {
  60     UND_LINE, UND_FONT
  61 } und_mode;
  62 static int descent;
  63
  64 #define NCOLOURS 24
  65 static COLORREF colours[NCOLOURS];
  66 static HPALETTE pal;
  67 static LPLOGPALETTE logpal;
  68 static RGBTRIPLE defpal[NCOLOURS];
  69
  70 static HWND hwnd;
  71
  72 static int dbltime, lasttime, lastact;
  73 static Mouse_Button lastbtn;
  74
  75 static char *window_name, *icon_name;
  76
  77 int WINAPI WinMain(HINSTANCE inst, HINSTANCE prev, LPSTR cmdline, int show) {
  78     static char appname[] = "PuTTY";
  79     WORD winsock_ver;
  80     WSADATA wsadata;
  81     WNDCLASS wndclass;
  82     MSG msg;
  83     int guess_width, guess_height;
  84
  85     putty_inst = inst;
  86
  87     winsock_ver = MAKEWORD(1, 1);
  88     if (WSAStartup(winsock_ver, &wsadata)) {
  89         MessageBox(NULL, "Unable to initialise WinSock", "WinSock Error",
  90                    MB_OK | MB_ICONEXCLAMATION);
  91         return 1;
  92     }
  93     if (LOBYTE(wsadata.wVersion) != 1 || HIBYTE(wsadata.wVersion) != 1) {
  94         MessageBox(NULL, "WinSock version is incompatible with 1.1",
  95                    "WinSock Error", MB_OK | MB_ICONEXCLAMATION);
  96         WSACleanup();
  97         return 1;
  98     }
  99     /* WISHLIST: maybe allow config tweaking even if winsock not present? */
 100
 101     InitCommonControls();
 102
 103     /*
 104      * Process the command line.
 105      */
 106     {
 107         char *p;
 108
 109         default_protocol = DEFAULT_PROTOCOL;
 110         default_port = DEFAULT_PORT;
 111
 112         do_defaults(NULL);
 113
 114         p = cmdline;
 115         while (*p && isspace(*p)) p++;
 116
 117         /*
 118          * Process command line options first. Yes, this can be
 119          * done better, and it will be as soon as I have the
 120          * energy...
 121          */
 122         while (*p == '-') {
 123             char *q = p + strcspn(p, " \t");
 124             p++;
 125             if (q == p + 3 &&
 126                 tolower(p[0]) == 's' &&
 127                 tolower(p[1]) == 's' &&
 128                 tolower(p[2]) == 'h') {
 129                 default_protocol = cfg.protocol = PROT_SSH;
 130                 default_port = cfg.port = 22;
 131             }
 132             p = q + strspn(q, " \t");
 133         }
 134
 135         /*
 136          * An initial @ means to activate a saved session.
 137          */
 138         if (*p == '@') {
 139             do_defaults (p+1);
 140             if (!*cfg.host && !do_config()) {
 141                 WSACleanup();
 142                 return 0;
 143             }
 144         } else if (*p == '&') {
 145             /*
 146              * An initial & means we've been given a command line
 147              * containing the hex value of a HANDLE for a file
 148              * mapping object, which we must then extract as a
 149              * config.
 150              */
 151             HANDLE filemap;
 152             Config *cp;
 153             if (sscanf(p+1, "%x", &filemap) == 1 &&
 154                 (cp = MapViewOfFile(filemap, FILE_MAP_READ,
 155                                     0, 0, sizeof(Config))) != NULL) {
 156                 cfg = *cp;
 157                 UnmapViewOfFile(cp);
 158                 CloseHandle(filemap);
 159             } else if (!do_config()) {
 160                 WSACleanup();
 161                 return 0;
 162             }
 163         } else if (*p) {
 164             char *q = p;
 165             while (*p && !isspace(*p)) p++;
 166             if (*p)
 167                 *p++ = '\0';
 168             strncpy (cfg.host, q, sizeof(cfg.host)-1);
 169             cfg.host[sizeof(cfg.host)-1] = '\0';
 170             while (*p && isspace(*p)) p++;
 171             if (*p)
 172                 cfg.port = atoi(p);
 173             else
 174                 cfg.port = -1;
 175         } else {
 176             if (!do_config()) {
 177                 WSACleanup();
 178                 return 0;
 179             }
 180         }
 181     }
 182
 183     back = (cfg.protocol == PROT_SSH ? &ssh_backend :
 184             cfg.protocol == PROT_TELNET ? &telnet_backend :
 185             &raw_backend);
 186
 187     ldisc = (cfg.ldisc_term ? &ldisc_term : &ldisc_simple);
 188
 189     if (!prev) {
 190         wndclass.style         = 0;
 191         wndclass.lpfnWndProc   = WndProc;
 192         wndclass.cbClsExtra    = 0;
 193         wndclass.cbWndExtra    = 0;
 194         wndclass.hInstance     = inst;
 195         wndclass.hIcon         = LoadIcon (inst,
 196                                            MAKEINTRESOURCE(IDI_MAINICON));
 197         wndclass.hCursor       = LoadCursor (NULL, IDC_IBEAM);
 198         wndclass.hbrBackground = GetStockObject (BLACK_BRUSH);
 199         wndclass.lpszMenuName  = NULL;
 200         wndclass.lpszClassName = appname;
 201
 202         RegisterClass (&wndclass);
 203     }
 204
 205     hwnd = NULL;
 206
 207     savelines = cfg.savelines;
 208     term_init();
 209
 210     cfgtopalette();
 211
 212     /*
 213      * Guess some defaults for the window size. This all gets
 214      * updated later, so we don't really care too much. However, we
 215      * do want the font width/height guesses to correspond to a
 216      * large font rather than a small one...
 217      */
 218
 219     font_width = 10;
 220     font_height = 20;
 221     extra_width = 25;
 222     extra_height = 28;
 223     term_size (cfg.height, cfg.width, cfg.savelines);
 224     guess_width = extra_width + font_width * cols;
 225     guess_height = extra_height + font_height * rows;
 226     {
 227         RECT r;
 228         HWND w = GetDesktopWindow();
 229         GetWindowRect (w, &r);
 230         if (guess_width > r.right - r.left)
 231             guess_width = r.right - r.left;
 232         if (guess_height > r.bottom - r.top)
 233             guess_height = r.bottom - r.top;
 234     }
 235
 236     hwnd = CreateWindow (appname, appname,
 237                          WS_OVERLAPPEDWINDOW | WS_VSCROLL,
 238                          CW_USEDEFAULT, CW_USEDEFAULT,
 239                          guess_width, guess_height,
 240                          NULL, NULL, inst, NULL);
 241
 242     /*
 243      * Initialise the fonts, simultaneously correcting the guesses
 244      * for font_{width,height}.
 245      */
 246     bold_mode = cfg.bold_colour ? BOLD_COLOURS : BOLD_FONT;
 247     und_mode = UND_FONT;
 248     init_fonts();
 249
 250     /*
 251      * Correct the guesses for extra_{width,height}.
 252      */
 253     {
 254         RECT cr, wr;
 255         GetWindowRect (hwnd, &wr);
 256         GetClientRect (hwnd, &cr);
 257         extra_width = wr.right - wr.left - cr.right + cr.left;
 258         extra_height = wr.bottom - wr.top - cr.bottom + cr.top;
 259     }
 260
 261     /*
 262      * Resize the window, now we know what size we _really_ want it
 263      * to be.
 264      */
 265     guess_width = extra_width + font_width * cols;
 266     guess_height = extra_height + font_height * rows;
 267     SendMessage (hwnd, WM_IGNORE_SIZE, 0, 0);
 268     SetWindowPos (hwnd, NULL, 0, 0, guess_width, guess_height,
 269                   SWP_NOMOVE | SWP_NOREDRAW | SWP_NOZORDER);
 270
 271     /*
 272      * Initialise the scroll bar.
 273      */
 274     {
 275         SCROLLINFO si;
 276
 277         si.cbSize = sizeof(si);
 278         si.fMask = SIF_RANGE | SIF_PAGE | SIF_POS | SIF_DISABLENOSCROLL;
 279         si.nMin = 0;
 280         si.nMax = rows-1;
 281         si.nPage = rows;
 282         si.nPos = 0;
 283         SetScrollInfo (hwnd, SB_VERT, &si, FALSE);
 284     }
 285
 286     /*
 287      * Start up the telnet connection.
 288      */
 289     {
 290         char *error;
 291         char msg[1024];
 292         char *realhost;
 293
 294         error = back->init (hwnd, cfg.host, cfg.port, &realhost);
 295         if (error) {
 296             sprintf(msg, "Unable to open connection:\n%s", error);
 297             MessageBox(NULL, msg, "PuTTY Error", MB_ICONERROR | MB_OK);
 298             return 0;
 299         }
 300         window_name = icon_name = NULL;
 301         sprintf(msg, "%s - PuTTY", realhost);
 302         set_title (msg);
 303         set_icon (msg);
 304     }
 305
 306     /*
 307      * Set up the input and output buffers.
 308      */
 309     inbuf_reap = inbuf_head = 0;
 310     outbuf_reap = outbuf_head = 0;
 311
 312     /*
 313      * Prepare the mouse handler.
 314      */
 315     lastact = MA_NOTHING;
 316     lastbtn = MB_NOTHING;
 317     dbltime = GetDoubleClickTime();
 318
 319     /*
 320      * Set up the session-control options on the system menu.
 321      */
 322     {
 323         HMENU m = GetSystemMenu (hwnd, FALSE);
 324         HMENU p,s;
 325         int i;
 326
 327         AppendMenu (m, MF_SEPARATOR, 0, 0);
 328         if (cfg.protocol == PROT_TELNET) {
 329             p = CreateMenu();
 330             AppendMenu (p, MF_ENABLED, IDM_TEL_AYT, "Are You There");
 331             AppendMenu (p, MF_ENABLED, IDM_TEL_BRK, "Break");
 332             AppendMenu (p, MF_ENABLED, IDM_TEL_SYNCH, "Synch");
 333             AppendMenu (p, MF_SEPARATOR, 0, 0);
 334             AppendMenu (p, MF_ENABLED, IDM_TEL_EC, "Erase Character");
 335             AppendMenu (p, MF_ENABLED, IDM_TEL_EL, "Erase Line");
 336             AppendMenu (p, MF_ENABLED, IDM_TEL_GA, "Go Ahead");
 337             AppendMenu (p, MF_ENABLED, IDM_TEL_NOP, "No Operation");
 338             AppendMenu (p, MF_SEPARATOR, 0, 0);
 339             AppendMenu (p, MF_ENABLED, IDM_TEL_ABORT, "Abort Process");
 340             AppendMenu (p, MF_ENABLED, IDM_TEL_AO, "Abort Output");
 341             AppendMenu (p, MF_ENABLED, IDM_TEL_IP, "Interrupt Process");
 342             AppendMenu (p, MF_ENABLED, IDM_TEL_SUSP, "Suspend Process");
 343             AppendMenu (p, MF_SEPARATOR, 0, 0);
 344             AppendMenu (p, MF_ENABLED, IDM_TEL_EOR, "End Of Record");
 345             AppendMenu (p, MF_ENABLED, IDM_TEL_EOF, "End Of File");
 346             AppendMenu (m, MF_POPUP | MF_ENABLED, (UINT) p, "Telnet Command");
 347             AppendMenu (m, MF_SEPARATOR, 0, 0);
 348         }
 349         AppendMenu (m, MF_ENABLED, IDM_SHOWLOG, "Event Log");
 350         AppendMenu (m, MF_SEPARATOR, 0, 0);
 351         AppendMenu (m, MF_ENABLED, IDM_NEWSESS, "New Session");
 352         AppendMenu (m, MF_ENABLED, IDM_DUPSESS, "Duplicate Session");
 353         s = CreateMenu();
 354         get_sesslist(TRUE);
 355         for (i = 1 ; i < ((nsessions < 256) ? nsessions : 256) ; i++)
 356           AppendMenu (s, MF_ENABLED, IDM_SAVED_MIN + (16 * i) , sessions[i]);
 357         AppendMenu (m, MF_POPUP | MF_ENABLED, (UINT) s, "Saved Sessions");
 358         AppendMenu (m, MF_ENABLED, IDM_RECONF, "Change Settings");
 359         AppendMenu (m, MF_SEPARATOR, 0, 0);
 360         AppendMenu (m, MF_ENABLED, IDM_CLRSB, "Clear Scrollback");
 361         AppendMenu (m, MF_ENABLED, IDM_RESET, "Reset Terminal");
 362         AppendMenu (m, MF_SEPARATOR, 0, 0);
 363         AppendMenu (m, MF_ENABLED, IDM_ABOUT, "About PuTTY");
 364     }
 365
 366     /*
 367      * Finally show the window!
 368      */
 369     ShowWindow (hwnd, show);
 370
 371     /*
 372      * Set the palette up.
 373      */
 374     pal = NULL;
 375     logpal = NULL;
 376     init_palette();
 377
 378     has_focus = (GetForegroundWindow() == hwnd);
 379     UpdateWindow (hwnd);
 380
 381     while (GetMessage (&msg, NULL, 0, 0)) {
 382         DispatchMessage (&msg);
 383         if (inbuf_reap != inbuf_head)
 384             term_out();
 385         /* In idle moments, do a full screen update */
 386         if (!PeekMessage (&msg, NULL, 0, 0, PM_NOREMOVE))
 387             term_update();
 388     }
 389
 390     /*
 391      * Clean up.
 392      */
 393     {
 394         int i;
 395         for (i=0; i<8; i++)
 396             if (fonts[i])
 397                 DeleteObject(fonts[i]);
 398     }
 399     sfree(logpal);
 400     if (pal)
 401         DeleteObject(pal);
 402     WSACleanup();
 403
 404     if (cfg.protocol == PROT_SSH)
 405         random_save_seed();
 406
 407     return msg.wParam;
 408 }
 409
 410 /*
 411  * Copy the colour palette from the configuration data into defpal.
 412  * This is non-trivial because the colour indices are different.
 413  */
 414 static void cfgtopalette(void) {
 415     int i;
 416     static const int ww[] = {
 417         6, 7, 8, 9, 10, 11, 12, 13,
 418         14, 15, 16, 17, 18, 19, 20, 21,
 419         0, 1, 2, 3, 4, 4, 5, 5
 420     };
 421
 422     for (i=0; i<24; i++) {
 423         int w = ww[i];
 424         defpal[i].rgbtRed = cfg.colours[w][0];
 425         defpal[i].rgbtGreen = cfg.colours[w][1];
 426         defpal[i].rgbtBlue = cfg.colours[w][2];
 427     }
 428 }
 429
 430 /*
 431  * Set up the colour palette.
 432  */
 433 static void init_palette(void) {
 434     int i;
 435     HDC hdc = GetDC (hwnd);
 436     if (hdc) {
 437         if (cfg.try_palette &&
 438             GetDeviceCaps (hdc, RASTERCAPS) & RC_PALETTE) {
 439             logpal = smalloc(sizeof(*logpal)
 440                              - sizeof(logpal->palPalEntry)
 441                              + NCOLOURS * sizeof(PALETTEENTRY));
 442             logpal->palVersion = 0x300;
 443             logpal->palNumEntries = NCOLOURS;
 444             for (i = 0; i < NCOLOURS; i++) {
 445                 logpal->palPalEntry[i].peRed = defpal[i].rgbtRed;
 446                 logpal->palPalEntry[i].peGreen = defpal[i].rgbtGreen;
 447                 logpal->palPalEntry[i].peBlue = defpal[i].rgbtBlue;
 448                 logpal->palPalEntry[i].peFlags = PC_NOCOLLAPSE;
 449             }
 450             pal = CreatePalette (logpal);
 451             if (pal) {
 452                 SelectPalette (hdc, pal, FALSE);
 453                 RealizePalette (hdc);
 454                 SelectPalette (hdc, GetStockObject (DEFAULT_PALETTE),
 455                                FALSE);
 456             }
 457         }
 458         ReleaseDC (hwnd, hdc);
 459     }
 460     if (pal)
 461         for (i=0; i<NCOLOURS; i++)
 462             colours[i] = PALETTERGB(defpal[i].rgbtRed,
 463                                     defpal[i].rgbtGreen,
 464                                     defpal[i].rgbtBlue);
 465     else
 466         for(i=0; i<NCOLOURS; i++)
 467             colours[i] = RGB(defpal[i].rgbtRed,
 468                              defpal[i].rgbtGreen,
 469                              defpal[i].rgbtBlue);
 470 }
 471
 472 /*
 473  * Initialise all the fonts we will need. There may be as many as
 474  * eight or as few as one. We also:
 475  *
 476  * - check the font width and height, correcting our guesses if
 477  *   necessary.
 478  *
 479  * - verify that the bold font is the same width as the ordinary
 480  *   one, and engage shadow bolding if not.
 481  *
 482  * - verify that the underlined font is the same width as the
 483  *   ordinary one (manual underlining by means of line drawing can
 484  *   be done in a pinch).
 485  *
 486  * - verify, in OEM/ANSI combined mode, that the OEM and ANSI base
 487  *   fonts are the same size, and shift to OEM-only mode if not.
 488  */
 489 static void init_fonts(void) {
 490     TEXTMETRIC tm;
 491     int i, j;
 492     int widths[5];
 493     HDC hdc;
 494     int fw_dontcare, fw_bold;
 495
 496     for (i=0; i<8; i++)
 497         fonts[i] = NULL;
 498
 499     if (cfg.fontisbold) {
 500         fw_dontcare = FW_BOLD;
 501         fw_bold = FW_BLACK;
 502    } else {
 503         fw_dontcare = FW_DONTCARE;
 504         fw_bold = FW_BOLD;
 505     }
 506
 507 #define f(i,c,w,u) \
 508     fonts[i] = CreateFont (cfg.fontheight, 0, 0, 0, w, FALSE, u, FALSE, \
 509                            c, OUT_DEFAULT_PRECIS, \
 510                            CLIP_DEFAULT_PRECIS, DEFAULT_QUALITY, \
 511                            FIXED_PITCH | FF_DONTCARE, cfg.font)
 512     if (cfg.vtmode != VT_OEMONLY) {
 513         f(FONT_NORMAL, cfg.fontcharset, fw_dontcare, FALSE);
 514         f(FONT_UNDERLINE, cfg.fontcharset, fw_dontcare, TRUE);
 515     }
 516     if (cfg.vtmode == VT_OEMANSI || cfg.vtmode == VT_OEMONLY) {
 517         f(FONT_OEM, OEM_CHARSET, fw_dontcare, FALSE);
 518         f(FONT_OEMUND, OEM_CHARSET, fw_dontcare, TRUE);
 519     }
 520     if (bold_mode == BOLD_FONT) {
 521         if (cfg.vtmode != VT_OEMONLY) {
 522             f(FONT_BOLD, cfg.fontcharset, fw_bold, FALSE);
 523             f(FONT_BOLDUND, cfg.fontcharset, fw_bold, TRUE);
 524         }
 525         if (cfg.vtmode == VT_OEMANSI || cfg.vtmode == VT_OEMONLY) {
 526             f(FONT_OEMBOLD, OEM_CHARSET, fw_bold, FALSE);
 527             f(FONT_OEMBOLDUND, OEM_CHARSET, fw_bold, TRUE);
 528         }
 529     } else {
 530         fonts[FONT_BOLD] = fonts[FONT_BOLDUND] = NULL;
 531         fonts[FONT_OEMBOLD] = fonts[FONT_OEMBOLDUND] = NULL;
 532     }
 533 #undef f
 534
 535     hdc = GetDC(hwnd);
 536
 537     if (cfg.vtmode == VT_OEMONLY)
 538         j = 4;
 539     else
 540         j = 0;
 541
 542     for (i=0; i<(cfg.vtmode == VT_OEMANSI ? 5 : 4); i++) {
 543         if (fonts[i+j]) {
 544             SelectObject (hdc, fonts[i+j]);
 545             GetTextMetrics(hdc, &tm);
 546             if (i == 0 || i == 4) {
 547                 font_height = tm.tmHeight;
 548                 font_width = tm.tmAveCharWidth;
 549                 descent = tm.tmAscent + 1;
 550                 if (descent >= font_height)
 551                     descent = font_height - 1;
 552             }
 553             widths[i] = tm.tmAveCharWidth;
 554         }
 555     }
 556
 557     ReleaseDC (hwnd, hdc);
 558
 559     if (widths[FONT_UNDERLINE] != widths[FONT_NORMAL] ||
 560         (bold_mode == BOLD_FONT &&
 561          widths[FONT_BOLDUND] != widths[FONT_BOLD])) {
 562         und_mode = UND_LINE;
 563         DeleteObject (fonts[FONT_UNDERLINE]);
 564         if (bold_mode == BOLD_FONT)
 565             DeleteObject (fonts[FONT_BOLDUND]);
 566     }
 567
 568     if (bold_mode == BOLD_FONT &&
 569         widths[FONT_BOLD] != widths[FONT_NORMAL]) {
 570         bold_mode = BOLD_SHADOW;
 571         DeleteObject (fonts[FONT_BOLD]);
 572         if (und_mode == UND_FONT)
 573             DeleteObject (fonts[FONT_BOLDUND]);
 574     }
 575
 576     if (cfg.vtmode == VT_OEMANSI && widths[FONT_OEM] != widths[FONT_NORMAL]) {
 577         MessageBox(NULL, "The OEM and ANSI versions of this font are\n"
 578                    "different sizes. Using OEM-only mode instead",
 579                    "Font Size Mismatch", MB_ICONINFORMATION | MB_OK);
 580         cfg.vtmode = VT_OEMONLY;
 581         for (i=0; i<4; i++)
 582             if (fonts[i])
 583                 DeleteObject (fonts[i]);
 584     }
 585 }
 586
 587 void request_resize (int w, int h) {
 588     int width = extra_width + font_width * w;
 589     int height = extra_height + font_height * h;
 590
 591     SetWindowPos (hwnd, NULL, 0, 0, width, height,
 592                   SWP_NOACTIVATE | SWP_NOCOPYBITS |
 593                   SWP_NOMOVE | SWP_NOZORDER);
 594 }
 595
 596 static void click (Mouse_Button b, int x, int y) {
 597     int thistime = GetMessageTime();
 598
 599     if (lastbtn == b && thistime - lasttime < dbltime) {
 600         lastact = (lastact == MA_CLICK ? MA_2CLK :
 601                    lastact == MA_2CLK ? MA_3CLK :
 602                    lastact == MA_3CLK ? MA_CLICK : MA_NOTHING);
 603     } else {
 604         lastbtn = b;
 605         lastact = MA_CLICK;
 606     }
 607     if (lastact != MA_NOTHING)
 608         term_mouse (b, lastact, x, y);
 609     lasttime = thistime;
 610 }
 611
 612 static int WINAPI WndProc (HWND hwnd, UINT message,
 613                            WPARAM wParam, LPARAM lParam) {
 614     HDC hdc;
 615     static int ignore_size = FALSE;
 616     static int ignore_clip = FALSE;
 617     static int just_reconfigged = FALSE;
 618
 619     switch (message) {
 620       case WM_CREATE:
 621         break;
 622       case WM_CLOSE:
 623         if (!cfg.warn_on_close ||
 624             MessageBox(hwnd, "Are you sure you want to close this session?",
 625                        "PuTTY Exit Confirmation",
 626                        MB_ICONWARNING | MB_OKCANCEL) == IDOK)
 627             DestroyWindow(hwnd);
 628         return 0;
 629       case WM_DESTROY:
 630         PostQuitMessage (0);
 631         return 0;
 632       case WM_SYSCOMMAND:
 633         switch (wParam & ~0xF) {       /* low 4 bits reserved to Windows */
 634           case IDM_SHOWLOG:
 635             showeventlog(hwnd);
 636             break;
 637           case IDM_NEWSESS:
 638           case IDM_DUPSESS:
 639           case IDM_SAVEDSESS:
 640             {
 641                 char b[2048];
 642                 char c[30], *cl;
 643                 int freecl = FALSE;
 644                 STARTUPINFO si;
 645                 PROCESS_INFORMATION pi;
 646                 HANDLE filemap = NULL;
 647
 648                 if (wParam == IDM_DUPSESS) {
 649                     /*
 650                      * Allocate a file-mapping memory chunk for the
 651                      * config structure.
 652                      */
 653                     SECURITY_ATTRIBUTES sa;
 654                     Config *p;
 655
 656                     sa.nLength = sizeof(sa);
 657                     sa.lpSecurityDescriptor = NULL;
 658                     sa.bInheritHandle = TRUE;
 659                     filemap = CreateFileMapping((HANDLE)0xFFFFFFFF,
 660                                                 &sa,
 661                                                 PAGE_READWRITE,
 662                                                 0,
 663                                                 sizeof(Config),
 664                                                 NULL);
 665                     if (filemap) {
 666                         p = (Config *)MapViewOfFile(filemap,
 667                                                     FILE_MAP_WRITE,
 668                                                     0, 0, sizeof(Config));
 669                         if (p) {
 670                             *p = cfg;  /* structure copy */
 671                             UnmapViewOfFile(p);
 672                         }
 673                     }
 674                     sprintf(c, "putty &%08x", filemap);
 675                     cl = c;
 676                 } else if (wParam == IDM_SAVEDSESS) {
 677                     char *session = sessions[(lParam - IDM_SAVED_MIN) / 16];
 678                     cl = malloc(16 + strlen(session)); /* 8, but play safe */
 679                     if (!cl)
 680                         cl = NULL;     /* not a very important failure mode */
 681                     sprintf(cl, "putty @%s", session);
 682                     freecl = TRUE;
 683                 } else
 684                     cl = NULL;
 685
 686                 GetModuleFileName (NULL, b, sizeof(b)-1);
 687                 si.cb = sizeof(si);
 688                 si.lpReserved = NULL;
 689                 si.lpDesktop = NULL;
 690                 si.lpTitle = NULL;
 691                 si.dwFlags = 0;
 692                 si.cbReserved2 = 0;
 693                 si.lpReserved2 = NULL;
 694                 CreateProcess (b, cl, NULL, NULL, TRUE,
 695                                NORMAL_PRIORITY_CLASS, NULL, NULL, &si, &pi);
 696
 697                 if (filemap)
 698                     CloseHandle(filemap);
 699                 if (freecl)
 700                     free(cl);
 701             }
 702             break;
 703           case IDM_RECONF:
 704             if (!do_reconfig(hwnd))
 705                 break;
 706             just_reconfigged = TRUE;
 707             {
 708                 int i;
 709                 for (i=0; i<8; i++)
 710                     if (fonts[i])
 711                         DeleteObject(fonts[i]);
 712             }
 713             bold_mode = cfg.bold_colour ? BOLD_COLOURS : BOLD_FONT;
 714             und_mode = UND_FONT;
 715             init_fonts();
 716             sfree(logpal);
 717             if (pal)
 718                 DeleteObject(pal);
 719             logpal = NULL;
 720             pal = NULL;
 721             cfgtopalette();
 722             init_palette();
 723             term_size(cfg.height, cfg.width, cfg.savelines);
 724             InvalidateRect(hwnd, NULL, TRUE);
 725             SetWindowPos (hwnd, NULL, 0, 0,
 726                           extra_width + font_width * cfg.width,
 727                           extra_height + font_height * cfg.height,
 728                           SWP_NOACTIVATE | SWP_NOCOPYBITS |
 729                           SWP_NOMOVE | SWP_NOZORDER);
 730             if (IsIconic(hwnd)) {
 731                 SetWindowText (hwnd,
 732                                cfg.win_name_always ? window_name : icon_name);
 733             }
 734             break;
 735           case IDM_CLRSB:
 736             term_clrsb();
 737             break;
 738           case IDM_RESET:
 739             term_pwron();
 740             break;
 741           case IDM_TEL_AYT: back->special (TS_AYT); break;
 742           case IDM_TEL_BRK: back->special (TS_BRK); break;
 743           case IDM_TEL_SYNCH: back->special (TS_SYNCH); break;
 744           case IDM_TEL_EC: back->special (TS_EC); break;
 745           case IDM_TEL_EL: back->special (TS_EL); break;
 746           case IDM_TEL_GA: back->special (TS_GA); break;
 747           case IDM_TEL_NOP: back->special (TS_NOP); break;
 748           case IDM_TEL_ABORT: back->special (TS_ABORT); break;
 749           case IDM_TEL_AO: back->special (TS_AO); break;
 750           case IDM_TEL_IP: back->special (TS_IP); break;
 751           case IDM_TEL_SUSP: back->special (TS_SUSP); break;
 752           case IDM_TEL_EOR: back->special (TS_EOR); break;
 753           case IDM_TEL_EOF: back->special (TS_EOF); break;
 754           case IDM_ABOUT:
 755             showabout (hwnd);
 756             break;
 757         default:
 758           if (wParam >= IDM_SAVED_MIN && wParam <= IDM_SAVED_MAX) {
 759             SendMessage(hwnd, WM_SYSCOMMAND, IDM_SAVEDSESS, wParam);
 760           }
 761         }
 762         break;
 763
 764 #define X_POS(l) ((int)(short)LOWORD(l))
 765 #define Y_POS(l) ((int)(short)HIWORD(l))
 766
 767 #define TO_CHR_X(x) (((x)<0 ? (x)-font_width+1 : (x)) / font_width)
 768 #define TO_CHR_Y(y) (((y)<0 ? (y)-font_height+1: (y)) / font_height)
 769
 770       case WM_LBUTTONDOWN:
 771         click (MB_SELECT, TO_CHR_X(X_POS(lParam)),
 772                TO_CHR_Y(Y_POS(lParam)));
 773         SetCapture(hwnd);
 774         return 0;
 775       case WM_LBUTTONUP:
 776         term_mouse (MB_SELECT, MA_RELEASE, TO_CHR_X(X_POS(lParam)),
 777                     TO_CHR_Y(Y_POS(lParam)));
 778         ReleaseCapture();
 779         return 0;
 780       case WM_MBUTTONDOWN:
 781         SetCapture(hwnd);
 782         click (cfg.mouse_is_xterm ? MB_PASTE : MB_EXTEND,
 783                TO_CHR_X(X_POS(lParam)),
 784                TO_CHR_Y(Y_POS(lParam)));
 785         return 0;
 786       case WM_MBUTTONUP:
 787         term_mouse (cfg.mouse_is_xterm ? MB_PASTE : MB_EXTEND,
 788                     MA_RELEASE, TO_CHR_X(X_POS(lParam)),
 789                     TO_CHR_Y(Y_POS(lParam)));
 790         ReleaseCapture();
 791         return 0;
 792       case WM_RBUTTONDOWN:
 793         SetCapture(hwnd);
 794         click (cfg.mouse_is_xterm ? MB_EXTEND : MB_PASTE,
 795                TO_CHR_X(X_POS(lParam)),
 796                TO_CHR_Y(Y_POS(lParam)));
 797         return 0;
 798       case WM_RBUTTONUP:
 799         term_mouse (cfg.mouse_is_xterm ? MB_EXTEND : MB_PASTE,
 800                     MA_RELEASE, TO_CHR_X(X_POS(lParam)),
 801                     TO_CHR_Y(Y_POS(lParam)));
 802         ReleaseCapture();
 803         return 0;
 804       case WM_MOUSEMOVE:
 805         /*
 806          * Add the mouse position and message time to the random
 807          * number noise, if we're using ssh.
 808          */
 809         if (cfg.protocol == PROT_SSH)
 810             noise_ultralight(lParam);
 811
 812         if (wParam & (MK_LBUTTON | MK_MBUTTON | MK_RBUTTON)) {
 813             Mouse_Button b;
 814             if (wParam & MK_LBUTTON)
 815                 b = MB_SELECT;
 816             else if (wParam & MK_MBUTTON)
 817                 b = cfg.mouse_is_xterm ? MB_PASTE : MB_EXTEND;
 818             else
 819                 b = cfg.mouse_is_xterm ? MB_EXTEND : MB_PASTE;
 820             term_mouse (b, MA_DRAG, TO_CHR_X(X_POS(lParam)),
 821                         TO_CHR_Y(Y_POS(lParam)));
 822         }
 823         return 0;
 824       case WM_IGNORE_CLIP:
 825         ignore_clip = wParam;          /* don't panic on DESTROYCLIPBOARD */
 826         break;
 827       case WM_DESTROYCLIPBOARD:
 828         if (!ignore_clip)
 829             term_deselect();
 830         ignore_clip = FALSE;
 831         return 0;
 832       case WM_PAINT:
 833         {
 834             PAINTSTRUCT p;
 835             hdc = BeginPaint (hwnd, &p);
 836             if (pal) {
 837                 SelectPalette (hdc, pal, TRUE);
 838                 RealizePalette (hdc);
 839             }
 840             term_paint (hdc, p.rcPaint.left, p.rcPaint.top,
 841                         p.rcPaint.right, p.rcPaint.bottom);
 842             SelectObject (hdc, GetStockObject(SYSTEM_FONT));
 843             SelectObject (hdc, GetStockObject(WHITE_PEN));
 844             EndPaint (hwnd, &p);
 845         }
 846         return 0;
 847       case WM_NETEVENT:
 848         {
 849             int i = back->msg (wParam, lParam);
 850             if (i < 0) {
 851                 char buf[1024];
 852                 switch (WSABASEERR + (-i) % 10000) {
 853                   case WSAECONNRESET:
 854                     sprintf(buf, "Connection reset by peer");
 855                     break;
 856                   default:
 857                     sprintf(buf, "Unexpected network error %d", -i);
 858                     break;
 859                 }
 860                 MessageBox(hwnd, buf, "PuTTY Fatal Error",
 861                            MB_ICONERROR | MB_OK);
 862                 PostQuitMessage(1);
 863             } else if (i == 0) {
 864                 if (cfg.close_on_exit)
 865                     PostQuitMessage(0);
 866                 else {
 867                     MessageBox(hwnd, "Connection closed by remote host",
 868                                "PuTTY", MB_OK | MB_ICONINFORMATION);
 869                     SetWindowText (hwnd, "PuTTY (inactive)");
 870                 }
 871             }
 872         }
 873         return 0;
 874       case WM_SETFOCUS:
 875         has_focus = TRUE;
 876         term_out();
 877         term_update();
 878         break;
 879       case WM_KILLFOCUS:
 880         has_focus = FALSE;
 881         term_out();
 882         term_update();
 883         break;
 884       case WM_IGNORE_SIZE:
 885         ignore_size = TRUE;            /* don't panic on next WM_SIZE msg */
 886         break;
 887       case WM_ENTERSIZEMOVE:
 888           EnableSizeTip(1);
 889           break;
 890       case WM_EXITSIZEMOVE:
 891           EnableSizeTip(0);
 892           break;
 893       case WM_SIZING:
 894         {
 895             int width, height, w, h, ew, eh;
 896             LPRECT r = (LPRECT)lParam;
 897
 898             width = r->right - r->left - extra_width;
 899             height = r->bottom - r->top - extra_height;
 900             w = (width + font_width/2) / font_width; if (w < 1) w = 1;
 901             h = (height + font_height/2) / font_height; if (h < 1) h = 1;
 902         UpdateSizeTip(hwnd, w, h);
 903             ew = width - w * font_width;
 904             eh = height - h * font_height;
 905             if (ew != 0) {
 906                 if (wParam == WMSZ_LEFT ||
 907                     wParam == WMSZ_BOTTOMLEFT ||
 908                     wParam == WMSZ_TOPLEFT)
 909                     r->left += ew;
 910                 else
 911                     r->right -= ew;
 912             }
 913             if (eh != 0) {
 914                 if (wParam == WMSZ_TOP ||
 915                     wParam == WMSZ_TOPRIGHT ||
 916                     wParam == WMSZ_TOPLEFT)
 917                     r->top += eh;
 918                 else
 919                     r->bottom -= eh;
 920             }
 921             if (ew || eh)
 922                 return 1;
 923             else
 924                 return 0;
 925         }
 926         break;
 927       case WM_SIZE:
 928         if (wParam == SIZE_MINIMIZED) {
 929             SetWindowText (hwnd,
 930                            cfg.win_name_always ? window_name : icon_name);
 931             break;
 932         }
 933         if (wParam == SIZE_RESTORED || wParam == SIZE_MAXIMIZED)
 934             SetWindowText (hwnd, window_name);
 935         if (!ignore_size) {
 936             int width, height, w, h;
 937 #if 0 /* we have fixed this using WM_SIZING now */
 938             int ew, eh;
 939 #endif
 940
 941             width = LOWORD(lParam);
 942             height = HIWORD(lParam);
 943             w = width / font_width; if (w < 1) w = 1;
 944             h = height / font_height; if (h < 1) h = 1;
 945 #if 0 /* we have fixed this using WM_SIZING now */
 946             ew = width - w * font_width;
 947             eh = height - h * font_height;
 948             if (ew != 0 || eh != 0) {
 949                 RECT r;
 950                 GetWindowRect (hwnd, &r);
 951                 SendMessage (hwnd, WM_IGNORE_SIZE, 0, 0);
 952                 SetWindowPos (hwnd, NULL, 0, 0,
 953                               r.right - r.left - ew, r.bottom - r.top - eh,
 954                               SWP_NOACTIVATE | SWP_NOMOVE | SWP_NOZORDER);
 955             }
 956 #endif
 957             if (w != cols || h != rows || just_reconfigged) {
 958                 term_invalidate();
 959                 term_size (h, w, cfg.savelines);
 960                 back->size();
 961                 just_reconfigged = FALSE;
 962             }
 963         }
 964         ignore_size = FALSE;
 965         return 0;
 966       case WM_VSCROLL:
 967         switch (LOWORD(wParam)) {
 968           case SB_BOTTOM: term_scroll(-1, 0); break;
 969           case SB_TOP: term_scroll(+1, 0); break;
 970           case SB_LINEDOWN: term_scroll (0, +1); break;
 971           case SB_LINEUP: term_scroll (0, -1); break;
 972           case SB_PAGEDOWN: term_scroll (0, +rows/2); break;
 973           case SB_PAGEUP: term_scroll (0, -rows/2); break;
 974           case SB_THUMBPOSITION: case SB_THUMBTRACK:
 975             term_scroll (1, HIWORD(wParam)); break;
 976         }
 977         break;
 978      case WM_PALETTECHANGED:
 979         if ((HWND) wParam != hwnd && pal != NULL) {
 980             HDC hdc = get_ctx();
 981             if (hdc) {
 982                 if (RealizePalette (hdc) > 0)
 983                     UpdateColors (hdc);
 984                 free_ctx (hdc);
 985             }
 986         }
 987         break;
 988       case WM_QUERYNEWPALETTE:
 989         if (pal != NULL) {
 990             HDC hdc = get_ctx();
 991             if (hdc) {
 992                 if (RealizePalette (hdc) > 0)
 993                     UpdateColors (hdc);
 994                 free_ctx (hdc);
 995                 return TRUE;
 996             }
 997         }
 998         return FALSE;
 999       case WM_KEYDOWN:
1000       case WM_SYSKEYDOWN:
1001         /*
1002          * Add the scan code and keypress timing to the random
1003          * number noise, if we're using ssh.
1004          */
1005         if (cfg.protocol == PROT_SSH)
1006             noise_ultralight(lParam);
1007
1008         /*
1009          * We don't do TranslateMessage since it disassociates the
1010          * resulting CHAR message from the KEYDOWN that sparked it,
1011          * which we occasionally don't want. Instead, we process
1012          * KEYDOWN, and call the Win32 translator functions so that
1013          * we get the translations under _our_ control.
1014          */
1015         {
1016             unsigned char buf[20];
1017             int len;
1018
1019             len = TranslateKey (wParam, lParam, buf);
1020             if (len == -1)
1021                 return DefWindowProc (hwnd, message, wParam, lParam);
1022             ldisc->send (buf, len);
1023         }
1024         return 0;
1025       case WM_KEYUP:
1026       case WM_SYSKEYUP:
1027         /*
1028          * We handle KEYUP ourselves in order to distinghish left
1029          * and right Alt or Control keys, which Windows won't do
1030          * right if left to itself. See also the special processing
1031          * at the top of TranslateKey.
1032          */
1033         {
1034             BYTE keystate[256];
1035             int ret = GetKeyboardState(keystate);
1036             if (ret && wParam == VK_MENU) {
1037                 if (lParam & 0x1000000) keystate[VK_RMENU] = 0;
1038                 else keystate[VK_LMENU] = 0;
1039                 SetKeyboardState (keystate);
1040             }
1041             if (ret && wParam == VK_CONTROL) {
1042                 if (lParam & 0x1000000) keystate[VK_RCONTROL] = 0;
1043                 else keystate[VK_LCONTROL] = 0;
1044                 SetKeyboardState (keystate);
1045             }
1046         }
1047         /*
1048          * We don't return here, in order to allow Windows to do
1049          * its own KEYUP processing as well.
1050          */
1051         break;
1052       case WM_CHAR:
1053       case WM_SYSCHAR:
1054         /*
1055          * Nevertheless, we are prepared to deal with WM_CHAR
1056          * messages, should they crop up. So if someone wants to
1057          * post the things to us as part of a macro manoeuvre,
1058          * we're ready to cope.
1059          */
1060         {
1061             char c = xlat_kbd2tty((unsigned char)wParam);
1062             ldisc->send (&c, 1);
1063         }
1064         return 0;
1065     }
1066
1067     return DefWindowProc (hwnd, message, wParam, lParam);
1068 }
1069
1070 /*
1071  * Draw a line of text in the window, at given character
1072  * coordinates, in given attributes.
1073  *
1074  * We are allowed to fiddle with the contents of `text'.
1075  */
1076 void do_text (Context ctx, int x, int y, char *text, int len,
1077               unsigned long attr) {
1078     COLORREF fg, bg, t;
1079     int nfg, nbg, nfont;
1080     HDC hdc = ctx;
1081
1082     x *= font_width;
1083     y *= font_height;
1084
1085     if (attr & ATTR_ACTCURS) {
1086         attr &= (bold_mode == BOLD_COLOURS ? 0x200 : 0x300);
1087         attr ^= ATTR_CUR_XOR;
1088     }
1089
1090     nfont = 0;
1091     if (cfg.vtmode == VT_OEMONLY)
1092         nfont |= FONT_OEM;
1093
1094     /*
1095      * Map high-half characters in order to approximate ISO using
1096      * OEM character set. Characters missing are 0xC3 (Atilde) and
1097      * 0xCC (Igrave).
1098      */
1099     if (nfont & FONT_OEM) {
1100         int i;
1101         for (i=0; i<len; i++)
1102             if (text[i] >= '\xA0' && text[i] <= '\xFF') {
1103                 static const char oemhighhalf[] =
1104                     "\x20\xAD\xBD\x9C\xCF\xBE\xDD\xF5" /* A0-A7 */
1105                     "\xF9\xB8\xA6\xAE\xAA\xF0\xA9\xEE" /* A8-AF */
1106                     "\xF8\xF1\xFD\xFC\xEF\xE6\xF4\xFA" /* B0-B7 */
1107                     "\xF7\xFB\xA7\xAF\xAC\xAB\xF3\xA8" /* B8-BF */
1108                     "\xB7\xB5\xB6\x41\x8E\x8F\x92\x80" /* C0-C7 */
1109                     "\xD4\x90\xD2\xD3\x49\xD6\xD7\xD8" /* C8-CF */
1110                     "\xD1\xA5\xE3\xE0\xE2\xE5\x99\x9E" /* D0-D7 */
1111                     "\x9D\xEB\xE9\xEA\x9A\xED\xE8\xE1" /* D8-DF */
1112                     "\x85\xA0\x83\xC6\x84\x86\x91\x87" /* E0-E7 */
1113                     "\x8A\x82\x88\x89\x8D\xA1\x8C\x8B" /* E8-EF */
1114                     "\xD0\xA4\x95\xA2\x93\xE4\x94\xF6" /* F0-F7 */
1115                     "\x9B\x97\xA3\x96\x81\xEC\xE7\x98" /* F8-FF */
1116                     ;
1117                 text[i] = oemhighhalf[(unsigned char)text[i] - 0xA0];
1118             }
1119     }
1120
1121     if (attr & ATTR_GBCHR) {
1122         int i;
1123         /*
1124          * GB mapping: map # to pound, and everything else stays
1125          * normal.
1126          */
1127         for (i=0; i<len; i++)
1128             if (text[i] == '#')
1129                 text[i] = cfg.vtmode == VT_OEMONLY ? '\x9C' : '\xA3';
1130     } else if (attr & ATTR_LINEDRW) {
1131         int i;
1132         static const char poorman[] =
1133             "*#****\xB0\xB1**+++++-----++++|****\xA3\xB7";
1134         static const char oemmap[] =
1135             "*\xB1****\xF8\xF1**\xD9\xBF\xDA\xC0\xC5"
1136             "\xC4\xC4\xC4\xC4\xC4\xC3\xB4\xC1\xC2\xB3****\x9C\xFA";
1137
1138         /*
1139          * Line drawing mapping: map ` thru ~ (0x60 thru 0x7E) to
1140          * VT100 line drawing chars; everything else stays normal.
1141          */
1142         switch (cfg.vtmode) {
1143           case VT_XWINDOWS:
1144             for (i=0; i<len; i++)
1145                 if (text[i] >= '\x60' && text[i] <= '\x7E')
1146                     text[i] += '\x01' - '\x60';
1147             break;
1148           case VT_OEMANSI:
1149           case VT_OEMONLY:
1150             nfont |= FONT_OEM;
1151             for (i=0; i<len; i++)
1152                 if (text[i] >= '\x60' && text[i] <= '\x7E')
1153                     text[i] = oemmap[(unsigned char)text[i] - 0x60];
1154             break;
1155           case VT_POORMAN:
1156             for (i=0; i<len; i++)
1157                 if (text[i] >= '\x60' && text[i] <= '\x7E')
1158                     text[i] = poorman[(unsigned char)text[i] - 0x60];
1159             break;
1160         }
1161     }
1162
1163     nfg = 2 * ((attr & ATTR_FGMASK) >> ATTR_FGSHIFT);
1164     nbg = 2 * ((attr & ATTR_BGMASK) >> ATTR_BGSHIFT);
1165     if (bold_mode == BOLD_FONT && (attr & ATTR_BOLD))
1166         nfont |= FONT_BOLD;
1167     if (und_mode == UND_FONT && (attr & ATTR_UNDER))
1168         nfont |= FONT_UNDERLINE;
1169     if (attr & ATTR_REVERSE) {
1170         t = nfg; nfg = nbg; nbg = t;
1171     }
1172     if (bold_mode == BOLD_COLOURS && (attr & ATTR_BOLD))
1173         nfg++;
1174     fg = colours[nfg];
1175     bg = colours[nbg];
1176     SelectObject (hdc, fonts[nfont]);
1177     SetTextColor (hdc, fg);
1178     SetBkColor (hdc, bg);
1179     SetBkMode (hdc, OPAQUE);
1180     TextOut (hdc, x, y, text, len);
1181     if (bold_mode == BOLD_SHADOW && (attr & ATTR_BOLD)) {
1182         SetBkMode (hdc, TRANSPARENT);
1183         TextOut (hdc, x-1, y, text, len);
1184     }
1185     if (und_mode == UND_LINE && (attr & ATTR_UNDER)) {
1186         HPEN oldpen;
1187         oldpen = SelectObject (hdc, CreatePen(PS_SOLID, 0, fg));
1188         MoveToEx (hdc, x, y+descent, NULL);
1189         LineTo (hdc, x+len*font_width, y+descent);
1190         oldpen = SelectObject (hdc, oldpen);
1191         DeleteObject (oldpen);
1192     }
1193     if (attr & ATTR_PASCURS) {
1194         POINT pts[5];
1195         HPEN oldpen;
1196         pts[0].x = pts[1].x = pts[4].x = x;
1197         pts[2].x = pts[3].x = x+font_width-1;
1198         pts[0].y = pts[3].y = pts[4].y = y;
1199         pts[1].y = pts[2].y = y+font_height-1;
1200         oldpen = SelectObject (hdc, CreatePen(PS_SOLID, 0, colours[23]));
1201         Polyline (hdc, pts, 5);
1202         oldpen = SelectObject (hdc, oldpen);
1203         DeleteObject (oldpen);
1204     }
1205 }
1206
1207 /*
1208  * Translate a WM_(SYS)?KEYDOWN message into a string of ASCII
1209  * codes. Returns number of bytes used.
1210  */
1211 static int TranslateKey(WPARAM wParam, LPARAM lParam, unsigned char *output) {
1212     unsigned char *p = output;
1213     BYTE keystate[256];
1214     int ret, code;
1215     int cancel_alt = FALSE;
1216
1217     /*
1218      * Get hold of the keyboard state, because we'll need it a few
1219      * times shortly.
1220      */
1221     ret = GetKeyboardState(keystate);
1222
1223     /*
1224      * Windows does not always want to distinguish left and right
1225      * Alt or Control keys. Thus we keep track of them ourselves.
1226      * See also the WM_KEYUP handler.
1227      */
1228     if (wParam == VK_MENU) {
1229         if (lParam & 0x1000000) keystate[VK_RMENU] = 0x80;
1230         else keystate[VK_LMENU] = 0x80;
1231         SetKeyboardState (keystate);
1232         return 0;
1233     }
1234     if (wParam == VK_CONTROL) {
1235         if (lParam & 0x1000000) keystate[VK_RCONTROL] = 0x80;
1236         else keystate[VK_LCONTROL] = 0x80;
1237         SetKeyboardState (keystate);
1238         return 0;
1239     }
1240
1241     /*
1242      * Prepend ESC, and cancel ALT, if ALT was pressed at the time
1243      * and it wasn't AltGr.
1244      */
1245     if (lParam & 0x20000000 && (keystate[VK_LMENU] & 0x80)) {
1246         *p++ = 0x1B;
1247         cancel_alt = TRUE;
1248     }
1249
1250     /*
1251      * NetHack keypad mode. This may conflict with Shift-PgUp/PgDn,
1252      * so we do it first.
1253      */
1254     if (cfg.nethack_keypad) {
1255         int shift = keystate[VK_SHIFT] & 0x80;
1256         /*
1257          * NB the shifted versions only work with numlock off.
1258          */
1259         switch ( (lParam >> 16) & 0x1FF ) {
1260           case 0x047: *p++ = shift ? 'Y' : 'y'; return p - output;
1261           case 0x048: *p++ = shift ? 'K' : 'k'; return p - output;
1262           case 0x049: *p++ = shift ? 'U' : 'u'; return p - output;
1263           case 0x04B: *p++ = shift ? 'H' : 'h'; return p - output;
1264           case 0x04C: *p++ = '.'; return p - output;
1265           case 0x04D: *p++ = shift ? 'L' : 'l'; return p - output;
1266           case 0x04F: *p++ = shift ? 'B' : 'b'; return p - output;
1267           case 0x050: *p++ = shift ? 'J' : 'j'; return p - output;
1268           case 0x051: *p++ = shift ? 'N' : 'n'; return p - output;
1269           case 0x053: *p++ = '.'; return p - output;
1270         }
1271     }
1272
1273     /*
1274      * Shift-PgUp, Shift-PgDn, and Alt-F4 all produce window
1275      * events: we'll deal with those now.
1276      */
1277     if (ret && (keystate[VK_SHIFT] & 0x80) && wParam == VK_PRIOR) {
1278         SendMessage (hwnd, WM_VSCROLL, SB_PAGEUP, 0);
1279         return 0;
1280     }
1281     if (ret && (keystate[VK_SHIFT] & 0x80) && wParam == VK_NEXT) {
1282         SendMessage (hwnd, WM_VSCROLL, SB_PAGEDOWN, 0);
1283         return 0;
1284     }
1285     if ((lParam & 0x20000000) && wParam == VK_F4 && cfg.alt_f4) {
1286         return -1;
1287     }
1288     if ((lParam & 0x20000000) && wParam == VK_SPACE && cfg.alt_space) {
1289         SendMessage (hwnd, WM_SYSCOMMAND, SC_KEYMENU, 0);
1290         return -1;
1291     }
1292
1293     /*
1294      * In general, the strategy is to see what the Windows keymap
1295      * translation has to say for itself, and then process function
1296      * keys and suchlike ourselves if that fails. But first we must
1297      * deal with the small number of special cases which the
1298      * Windows keymap translator thinks it can do but gets wrong.
1299      *
1300      * First special case: we might want the Backspace key to send
1301      * 0x7F not 0x08.
1302      */
1303     if (wParam == VK_BACK) {
1304         *p++ = (cfg.bksp_is_delete ? 0x7F : 0x08);
1305         return p - output;
1306     }
1307
1308     /*
1309      * Control-Space should send ^@ (0x00), not Space.
1310      */
1311     if (ret && (keystate[VK_CONTROL] & 0x80) && wParam == VK_SPACE) {
1312         *p++ = 0x00;
1313         return p - output;
1314     }
1315
1316     if (app_keypad_keys) {
1317         /*
1318          * If we're in applications keypad mode, we have to process it
1319          * before char-map translation, because it will pre-empt lots
1320          * of stuff, even if NumLock is off.
1321          */
1322         if (ret) {
1323             /*
1324              * Hack to ensure NumLock doesn't interfere with
1325              * perception of Shift for Keypad Plus. I don't pretend
1326              * to understand this, but it seems to work as is.
1327              * Leave it alone, or die.
1328              */
1329             keystate[VK_NUMLOCK] = 0;
1330             SetKeyboardState (keystate);
1331             GetKeyboardState (keystate);
1332         }
1333         switch ( (lParam >> 16) & 0x1FF ) {
1334           case 0x145: p += sprintf((char *)p, "\x1BOP"); return p - output;
1335           case 0x135: p += sprintf((char *)p, "\x1BOQ"); return p - output;
1336           case 0x037: p += sprintf((char *)p, "\x1BOR"); return p - output;
1337           case 0x047: p += sprintf((char *)p, "\x1BOw"); return p - output;
1338           case 0x048: p += sprintf((char *)p, "\x1BOx"); return p - output;
1339           case 0x049: p += sprintf((char *)p, "\x1BOy"); return p - output;
1340           case 0x04A: p += sprintf((char *)p, "\x1BOS"); return p - output;
1341           case 0x04B: p += sprintf((char *)p, "\x1BOt"); return p - output;
1342           case 0x04C: p += sprintf((char *)p, "\x1BOu"); return p - output;
1343           case 0x04D: p += sprintf((char *)p, "\x1BOv"); return p - output;
1344           case 0x04E: /* keypad + is ^[Ol, but ^[Om with Shift */
1345             p += sprintf((char *)p,
1346                          (ret && (keystate[VK_SHIFT] & 0x80)) ?
1347                          "\x1BOm" : "\x1BOl");
1348             return p - output;
1349           case 0x04F: p += sprintf((char *)p, "\x1BOq"); return p - output;
1350           case 0x050: p += sprintf((char *)p, "\x1BOr"); return p - output;
1351           case 0x051: p += sprintf((char *)p, "\x1BOs"); return p - output;
1352           case 0x052: p += sprintf((char *)p, "\x1BOp"); return p - output;
1353           case 0x053: p += sprintf((char *)p, "\x1BOn"); return p - output;
1354           case 0x11C: p += sprintf((char *)p, "\x1BOM"); return p - output;
1355         }
1356     }
1357
1358     /*
1359      * Before doing Windows charmap translation, remove LeftALT
1360      * from the keymap, since its sole effect should be to prepend
1361      * ESC, which we've already done. Note that removal of LeftALT
1362      * has to happen _after_ the above call to SetKeyboardState, or
1363      * dire things will befall.
1364      */
1365     if (cancel_alt) {
1366         keystate[VK_MENU] = keystate[VK_RMENU];
1367         keystate[VK_LMENU] = 0;
1368     }
1369
1370     /*
1371      * Attempt the Windows char-map translation.
1372      */
1373     if (ret) {
1374         WORD chr;
1375         int r;
1376         BOOL capsOn=keystate[VK_CAPITAL] !=0;
1377
1378         /* helg: clear CAPS LOCK state if caps lock switches to cyrillic */
1379         if(cfg.xlat_capslockcyr)
1380             keystate[VK_CAPITAL] = 0;
1381
1382         r = ToAscii (wParam, (lParam >> 16) & 0xFF,
1383                      keystate, &chr, 0);
1384
1385         if(capsOn)
1386             chr = xlat_latkbd2win((unsigned char)(chr & 0xFF));
1387         if (r == 1) {
1388             *p++ = xlat_kbd2tty((unsigned char)(chr & 0xFF));
1389             return p - output;
1390         }
1391     }
1392
1393     /*
1394      * OK, we haven't had a key code from the keymap translation.
1395      * We'll try our various special cases and function keys, and
1396      * then give up. (There's nothing wrong with giving up:
1397      * Scrollock, Pause/Break, and of course the various buckybit
1398      * keys all produce KEYDOWN events that we really _do_ want to
1399      * ignore.)
1400      */
1401
1402     /*
1403      * Control-2 should return ^@ (0x00), Control-6 should return
1404      * ^^ (0x1E), and Control-Minus should return ^_ (0x1F). Since
1405      * the DOS keyboard handling did it, and we have nothing better
1406      * to do with the key combo in question, we'll also map
1407      * Control-Backquote to ^\ (0x1C).
1408      */
1409     if (ret && (keystate[VK_CONTROL] & 0x80) && wParam == '2') {
1410         *p++ = 0x00;
1411         return p - output;
1412     }
1413     if (ret && (keystate[VK_CONTROL] & 0x80) && wParam == '6') {
1414         *p++ = 0x1E;
1415         return p - output;
1416     }
1417     if (ret && (keystate[VK_CONTROL] & 0x80) && wParam == 0xBD) {
1418         *p++ = 0x1F;
1419         return p - output;
1420     }
1421     if (ret && (keystate[VK_CONTROL] & 0x80) && wParam == 0xDF) {
1422         *p++ = 0x1C;
1423         return p - output;
1424     }
1425
1426     /*
1427      * First, all the keys that do tilde codes. (ESC '[' nn '~',
1428      * for integer decimal nn.)
1429      *
1430      * We also deal with the weird ones here. Linux VCs replace F1
1431      * to F5 by ESC [ [ A to ESC [ [ E. rxvt doesn't do _that_, but
1432      * does replace Home and End (1~ and 4~) by ESC [ H and ESC O w
1433      * respectively.
1434      */
1435     code = 0;
1436     switch (wParam) {
1437       case VK_F1: code = (keystate[VK_SHIFT] & 0x80 ? 23 : 11); break;
1438       case VK_F2: code = (keystate[VK_SHIFT] & 0x80 ? 24 : 12); break;
1439       case VK_F3: code = (keystate[VK_SHIFT] & 0x80 ? 25 : 13); break;
1440       case VK_F4: code = (keystate[VK_SHIFT] & 0x80 ? 26 : 14); break;
1441       case VK_F5: code = (keystate[VK_SHIFT] & 0x80 ? 28 : 15); break;
1442       case VK_F6: code = (keystate[VK_SHIFT] & 0x80 ? 29 : 17); break;
1443       case VK_F7: code = (keystate[VK_SHIFT] & 0x80 ? 31 : 18); break;
1444       case VK_F8: code = (keystate[VK_SHIFT] & 0x80 ? 32 : 19); break;
1445       case VK_F9: code = (keystate[VK_SHIFT] & 0x80 ? 33 : 20); break;
1446       case VK_F10: code = (keystate[VK_SHIFT] & 0x80 ? 34 : 21); break;
1447       case VK_F11: code = 23; break;
1448       case VK_F12: code = 24; break;
1449       case VK_HOME: code = 1; break;
1450       case VK_INSERT: code = 2; break;
1451       case VK_DELETE: code = 3; break;
1452       case VK_END: code = 4; break;
1453       case VK_PRIOR: code = 5; break;
1454       case VK_NEXT: code = 6; break;
1455     }
1456     if (cfg.linux_funkeys && code >= 11 && code <= 15) {
1457         p += sprintf((char *)p, "\x1B[[%c", code + 'A' - 11);
1458         return p - output;
1459     }
1460     if (cfg.rxvt_homeend && (code == 1 || code == 4)) {
1461         p += sprintf((char *)p, code == 1 ? "\x1B[H" : "\x1BOw");
1462         return p - output;
1463     }
1464     if (code) {
1465         p += sprintf((char *)p, "\x1B[%d~", code);
1466         return p - output;
1467     }
1468
1469     /*
1470      * Now the remaining keys (arrows and Keypad 5. Keypad 5 for
1471      * some reason seems to send VK_CLEAR to Windows...).
1472      */
1473     switch (wParam) {
1474       case VK_UP:
1475         p += sprintf((char *)p, app_cursor_keys ? "\x1BOA" : "\x1B[A");
1476         return p - output;
1477       case VK_DOWN:
1478         p += sprintf((char *)p, app_cursor_keys ? "\x1BOB" : "\x1B[B");
1479         return p - output;
1480       case VK_RIGHT:
1481         p += sprintf((char *)p, app_cursor_keys ? "\x1BOC" : "\x1B[C");
1482         return p - output;
1483       case VK_LEFT:
1484         p += sprintf((char *)p, app_cursor_keys ? "\x1BOD" : "\x1B[D");
1485         return p - output;
1486       case VK_CLEAR: p += sprintf((char *)p, "\x1B[G"); return p - output;
1487     }
1488
1489     return 0;
1490 }
1491
1492 void set_title (char *title) {
1493     sfree (window_name);
1494     window_name = smalloc(1+strlen(title));
1495     strcpy (window_name, title);
1496     if (cfg.win_name_always || !IsIconic(hwnd))
1497         SetWindowText (hwnd, title);
1498 }
1499
1500 void set_icon (char *title) {
1501     sfree (icon_name);
1502     icon_name = smalloc(1+strlen(title));
1503     strcpy (icon_name, title);
1504     if (!cfg.win_name_always && IsIconic(hwnd))
1505         SetWindowText (hwnd, title);
1506 }
1507
1508 void set_sbar (int total, int start, int page) {
1509     SCROLLINFO si;
1510     si.cbSize = sizeof(si);
1511     si.fMask = SIF_RANGE | SIF_PAGE | SIF_POS | SIF_DISABLENOSCROLL;
1512     si.nMin = 0;
1513     si.nMax = total - 1;
1514     si.nPage = page;
1515     si.nPos = start;
1516     if (hwnd)
1517         SetScrollInfo (hwnd, SB_VERT, &si, TRUE);
1518 }
1519
1520 Context get_ctx(void) {
1521     HDC hdc;
1522     if (hwnd) {
1523         hdc = GetDC (hwnd);
1524         if (hdc && pal)
1525             SelectPalette (hdc, pal, FALSE);
1526         return hdc;
1527     } else
1528         return NULL;
1529 }
1530
1531 void free_ctx (Context ctx) {
1532     SelectPalette (ctx, GetStockObject (DEFAULT_PALETTE), FALSE);
1533     ReleaseDC (hwnd, ctx);
1534 }
1535
1536 static void real_palette_set (int n, int r, int g, int b) {
1537     if (pal) {
1538         logpal->palPalEntry[n].peRed = r;
1539         logpal->palPalEntry[n].peGreen = g;
1540         logpal->palPalEntry[n].peBlue = b;
1541         logpal->palPalEntry[n].peFlags = PC_NOCOLLAPSE;
1542         colours[n] = PALETTERGB(r, g, b);
1543         SetPaletteEntries (pal, 0, NCOLOURS, logpal->palPalEntry);
1544     } else
1545         colours[n] = RGB(r, g, b);
1546 }
1547
1548 void palette_set (int n, int r, int g, int b) {
1549     static const int first[21] = {
1550         0, 2, 4, 6, 8, 10, 12, 14,
1551         1, 3, 5, 7, 9, 11, 13, 15,
1552         16, 17, 18, 20, 22
1553     };
1554     real_palette_set (first[n], r, g, b);
1555     if (first[n] >= 18)
1556         real_palette_set (first[n]+1, r, g, b);
1557     if (pal) {
1558         HDC hdc = get_ctx();
1559         UnrealizeObject (pal);
1560         RealizePalette (hdc);
1561         free_ctx (hdc);
1562     }
1563 }
1564
1565 void palette_reset (void) {
1566     int i;
1567
1568     for (i = 0; i < NCOLOURS; i++) {
1569         if (pal) {
1570             logpal->palPalEntry[i].peRed = defpal[i].rgbtRed;
1571             logpal->palPalEntry[i].peGreen = defpal[i].rgbtGreen;
1572             logpal->palPalEntry[i].peBlue = defpal[i].rgbtBlue;
1573             logpal->palPalEntry[i].peFlags = 0;
1574             colours[i] = PALETTERGB(defpal[i].rgbtRed,
1575                                     defpal[i].rgbtGreen,
1576                                     defpal[i].rgbtBlue);
1577         } else
1578             colours[i] = RGB(defpal[i].rgbtRed,
1579                              defpal[i].rgbtGreen,
1580                              defpal[i].rgbtBlue);
1581     }
1582
1583     if (pal) {
1584         HDC hdc;
1585         SetPaletteEntries (pal, 0, NCOLOURS, logpal->palPalEntry);
1586         hdc = get_ctx();
1587         RealizePalette (hdc);
1588         free_ctx (hdc);
1589     }
1590 }
1591
1592 void write_clip (void *data, int len) {
1593     HGLOBAL clipdata;
1594     void *lock;
1595
1596     clipdata = GlobalAlloc (GMEM_DDESHARE | GMEM_MOVEABLE, len + 1);
1597     if (!clipdata)
1598         return;
1599     lock = GlobalLock (clipdata);
1600     if (!lock)
1601         return;
1602     memcpy (lock, data, len);
1603     ((unsigned char *) lock) [len] = 0;
1604     GlobalUnlock (clipdata);
1605
1606     SendMessage (hwnd, WM_IGNORE_CLIP, TRUE, 0);
1607     if (OpenClipboard (hwnd)) {
1608         EmptyClipboard();
1609         SetClipboardData (CF_TEXT, clipdata);
1610         CloseClipboard();
1611     } else
1612         GlobalFree (clipdata);
1613     SendMessage (hwnd, WM_IGNORE_CLIP, FALSE, 0);
1614 }
1615
1616 void get_clip (void **p, int *len) {
1617     static HGLOBAL clipdata = NULL;
1618
1619     if (!p) {
1620         if (clipdata)
1621             GlobalUnlock (clipdata);
1622         clipdata = NULL;
1623         return;
1624     } else {
1625         if (OpenClipboard (NULL)) {
1626             clipdata = GetClipboardData (CF_TEXT);
1627             CloseClipboard();
1628             if (clipdata) {
1629                 *p = GlobalLock (clipdata);
1630                 if (*p) {
1631                     *len = strlen(*p);
1632                     return;
1633                 }
1634             }
1635         }
1636     }
1637
1638     *p = NULL;
1639     *len = 0;
1640 }
1641
1642 /*
1643  * Move `lines' lines from position `from' to position `to' in the
1644  * window.
1645  */
1646 void optimised_move (int to, int from, int lines) {
1647     RECT r;
1648     int min, max;
1649
1650     min = (to < from ? to : from);
1651     max = to + from - min;
1652
1653     r.left = 0; r.right = cols * font_width;
1654     r.top = min * font_height; r.bottom = (max+lines) * font_height;
1655     ScrollWindow (hwnd, 0, (to - from) * font_height, &r, &r);
1656 }
1657
1658 /*
1659  * Print a message box and perform a fatal exit.
1660  */
1661 void fatalbox(char *fmt, ...) {
1662     va_list ap;
1663     char stuff[200];
1664
1665     va_start(ap, fmt);
1666     vsprintf(stuff, fmt, ap);
1667     va_end(ap);
1668     MessageBox(hwnd, stuff, "PuTTY Fatal Error", MB_ICONERROR | MB_OK);
1669     exit(1);
1670 }
1671
1672 /*
1673  * Beep.
1674  */
1675 void beep(void) {
1676     MessageBeep(MB_OK);
1677 }