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