]> asedeno.scripts.mit.edu Git - PuTTY.git/blob - window.c
Fix potential segfault on malloc failure
[PuTTY.git] / 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                     else {
682                         sprintf(cl, "putty @%s", session);
683                         freecl = TRUE;
684                     }
685                 } else
686                     cl = NULL;
687
688                 GetModuleFileName (NULL, b, sizeof(b)-1);
689                 si.cb = sizeof(si);
690                 si.lpReserved = NULL;
691                 si.lpDesktop = NULL;
692                 si.lpTitle = NULL;
693                 si.dwFlags = 0;
694                 si.cbReserved2 = 0;
695                 si.lpReserved2 = NULL;
696                 CreateProcess (b, cl, NULL, NULL, TRUE,
697                                NORMAL_PRIORITY_CLASS, NULL, NULL, &si, &pi);
698
699                 if (filemap)
700                     CloseHandle(filemap);
701                 if (freecl)
702                     free(cl);
703             }
704             break;
705           case IDM_RECONF:
706             if (!do_reconfig(hwnd))
707                 break;
708             just_reconfigged = TRUE;
709             {
710                 int i;
711                 for (i=0; i<8; i++)
712                     if (fonts[i])
713                         DeleteObject(fonts[i]);
714             }
715             bold_mode = cfg.bold_colour ? BOLD_COLOURS : BOLD_FONT;
716             und_mode = UND_FONT;
717             init_fonts();
718             sfree(logpal);
719             ldisc = (cfg.ldisc_term ? &ldisc_term : &ldisc_simple);
720             if (pal)
721                 DeleteObject(pal);
722             logpal = NULL;
723             pal = NULL;
724             cfgtopalette();
725             init_palette();
726             term_size(cfg.height, cfg.width, cfg.savelines);
727             InvalidateRect(hwnd, NULL, TRUE);
728             SetWindowPos (hwnd, NULL, 0, 0,
729                           extra_width + font_width * cfg.width,
730                           extra_height + font_height * cfg.height,
731                           SWP_NOACTIVATE | SWP_NOCOPYBITS |
732                           SWP_NOMOVE | SWP_NOZORDER);
733             if (IsIconic(hwnd)) {
734                 SetWindowText (hwnd,
735                                cfg.win_name_always ? window_name : icon_name);
736             }
737             break;
738           case IDM_CLRSB:
739             term_clrsb();
740             break;
741           case IDM_RESET:
742             term_pwron();
743             break;
744           case IDM_TEL_AYT: back->special (TS_AYT); break;
745           case IDM_TEL_BRK: back->special (TS_BRK); break;
746           case IDM_TEL_SYNCH: back->special (TS_SYNCH); break;
747           case IDM_TEL_EC: back->special (TS_EC); break;
748           case IDM_TEL_EL: back->special (TS_EL); break;
749           case IDM_TEL_GA: back->special (TS_GA); break;
750           case IDM_TEL_NOP: back->special (TS_NOP); break;
751           case IDM_TEL_ABORT: back->special (TS_ABORT); break;
752           case IDM_TEL_AO: back->special (TS_AO); break;
753           case IDM_TEL_IP: back->special (TS_IP); break;
754           case IDM_TEL_SUSP: back->special (TS_SUSP); break;
755           case IDM_TEL_EOR: back->special (TS_EOR); break;
756           case IDM_TEL_EOF: back->special (TS_EOF); break;
757           case IDM_ABOUT:
758             showabout (hwnd);
759             break;
760         default:
761           if (wParam >= IDM_SAVED_MIN && wParam <= IDM_SAVED_MAX) {
762             SendMessage(hwnd, WM_SYSCOMMAND, IDM_SAVEDSESS, wParam);
763           }
764         }
765         break;
766
767 #define X_POS(l) ((int)(short)LOWORD(l))
768 #define Y_POS(l) ((int)(short)HIWORD(l))
769
770 #define TO_CHR_X(x) (((x)<0 ? (x)-font_width+1 : (x)) / font_width)
771 #define TO_CHR_Y(y) (((y)<0 ? (y)-font_height+1: (y)) / font_height)
772
773       case WM_LBUTTONDOWN:
774         click (MB_SELECT, TO_CHR_X(X_POS(lParam)),
775                TO_CHR_Y(Y_POS(lParam)));
776         SetCapture(hwnd);
777         return 0;
778       case WM_LBUTTONUP:
779         term_mouse (MB_SELECT, MA_RELEASE, TO_CHR_X(X_POS(lParam)),
780                     TO_CHR_Y(Y_POS(lParam)));
781         ReleaseCapture();
782         return 0;
783       case WM_MBUTTONDOWN:
784         SetCapture(hwnd);
785         click (cfg.mouse_is_xterm ? MB_PASTE : MB_EXTEND,
786                TO_CHR_X(X_POS(lParam)),
787                TO_CHR_Y(Y_POS(lParam)));
788         return 0;
789       case WM_MBUTTONUP:
790         term_mouse (cfg.mouse_is_xterm ? MB_PASTE : MB_EXTEND,
791                     MA_RELEASE, TO_CHR_X(X_POS(lParam)),
792                     TO_CHR_Y(Y_POS(lParam)));
793         ReleaseCapture();
794         return 0;
795       case WM_RBUTTONDOWN:
796         SetCapture(hwnd);
797         click (cfg.mouse_is_xterm ? MB_EXTEND : MB_PASTE,
798                TO_CHR_X(X_POS(lParam)),
799                TO_CHR_Y(Y_POS(lParam)));
800         return 0;
801       case WM_RBUTTONUP:
802         term_mouse (cfg.mouse_is_xterm ? MB_EXTEND : MB_PASTE,
803                     MA_RELEASE, TO_CHR_X(X_POS(lParam)),
804                     TO_CHR_Y(Y_POS(lParam)));
805         ReleaseCapture();
806         return 0;
807       case WM_MOUSEMOVE:
808         /*
809          * Add the mouse position and message time to the random
810          * number noise, if we're using ssh.
811          */
812         if (cfg.protocol == PROT_SSH)
813             noise_ultralight(lParam);
814
815         if (wParam & (MK_LBUTTON | MK_MBUTTON | MK_RBUTTON)) {
816             Mouse_Button b;
817             if (wParam & MK_LBUTTON)
818                 b = MB_SELECT;
819             else if (wParam & MK_MBUTTON)
820                 b = cfg.mouse_is_xterm ? MB_PASTE : MB_EXTEND;
821             else
822                 b = cfg.mouse_is_xterm ? MB_EXTEND : MB_PASTE;
823             term_mouse (b, MA_DRAG, TO_CHR_X(X_POS(lParam)),
824                         TO_CHR_Y(Y_POS(lParam)));
825         }
826         return 0;
827       case WM_IGNORE_CLIP:
828         ignore_clip = wParam;          /* don't panic on DESTROYCLIPBOARD */
829         break;
830       case WM_DESTROYCLIPBOARD:
831         if (!ignore_clip)
832             term_deselect();
833         ignore_clip = FALSE;
834         return 0;
835       case WM_PAINT:
836         {
837             PAINTSTRUCT p;
838             hdc = BeginPaint (hwnd, &p);
839             if (pal) {
840                 SelectPalette (hdc, pal, TRUE);
841                 RealizePalette (hdc);
842             }
843             term_paint (hdc, p.rcPaint.left, p.rcPaint.top,
844                         p.rcPaint.right, p.rcPaint.bottom);
845             SelectObject (hdc, GetStockObject(SYSTEM_FONT));
846             SelectObject (hdc, GetStockObject(WHITE_PEN));
847             EndPaint (hwnd, &p);
848         }
849         return 0;
850       case WM_NETEVENT:
851         {
852             int i = back->msg (wParam, lParam);
853             if (i < 0) {
854                 char buf[1024];
855                 switch (WSABASEERR + (-i) % 10000) {
856                   case WSAECONNRESET:
857                     sprintf(buf, "Connection reset by peer");
858                     break;
859                   default:
860                     sprintf(buf, "Unexpected network error %d", -i);
861                     break;
862                 }
863                 MessageBox(hwnd, buf, "PuTTY Fatal Error",
864                            MB_ICONERROR | MB_OK);
865                 PostQuitMessage(1);
866             } else if (i == 0) {
867                 if (cfg.close_on_exit)
868                     PostQuitMessage(0);
869                 else {
870                     MessageBox(hwnd, "Connection closed by remote host",
871                                "PuTTY", MB_OK | MB_ICONINFORMATION);
872                     SetWindowText (hwnd, "PuTTY (inactive)");
873                 }
874             }
875         }
876         return 0;
877       case WM_SETFOCUS:
878         has_focus = TRUE;
879         term_out();
880         term_update();
881         break;
882       case WM_KILLFOCUS:
883         has_focus = FALSE;
884         term_out();
885         term_update();
886         break;
887       case WM_IGNORE_SIZE:
888         ignore_size = TRUE;            /* don't panic on next WM_SIZE msg */
889         break;
890       case WM_ENTERSIZEMOVE:
891           EnableSizeTip(1);
892           break;
893       case WM_EXITSIZEMOVE:
894           EnableSizeTip(0);
895           break;
896       case WM_SIZING:
897         {
898             int width, height, w, h, ew, eh;
899             LPRECT r = (LPRECT)lParam;
900
901             width = r->right - r->left - extra_width;
902             height = r->bottom - r->top - extra_height;
903             w = (width + font_width/2) / font_width; if (w < 1) w = 1;
904             h = (height + font_height/2) / font_height; if (h < 1) h = 1;
905         UpdateSizeTip(hwnd, w, h);
906             ew = width - w * font_width;
907             eh = height - h * font_height;
908             if (ew != 0) {
909                 if (wParam == WMSZ_LEFT ||
910                     wParam == WMSZ_BOTTOMLEFT ||
911                     wParam == WMSZ_TOPLEFT)
912                     r->left += ew;
913                 else
914                     r->right -= ew;
915             }
916             if (eh != 0) {
917                 if (wParam == WMSZ_TOP ||
918                     wParam == WMSZ_TOPRIGHT ||
919                     wParam == WMSZ_TOPLEFT)
920                     r->top += eh;
921                 else
922                     r->bottom -= eh;
923             }
924             if (ew || eh)
925                 return 1;
926             else
927                 return 0;
928         }
929         break;
930       case WM_SIZE:
931         if (wParam == SIZE_MINIMIZED) {
932             SetWindowText (hwnd,
933                            cfg.win_name_always ? window_name : icon_name);
934             break;
935         }
936         if (wParam == SIZE_RESTORED || wParam == SIZE_MAXIMIZED)
937             SetWindowText (hwnd, window_name);
938         if (!ignore_size) {
939             int width, height, w, h;
940 #if 0 /* we have fixed this using WM_SIZING now */
941             int ew, eh;
942 #endif
943
944             width = LOWORD(lParam);
945             height = HIWORD(lParam);
946             w = width / font_width; if (w < 1) w = 1;
947             h = height / font_height; if (h < 1) h = 1;
948 #if 0 /* we have fixed this using WM_SIZING now */
949             ew = width - w * font_width;
950             eh = height - h * font_height;
951             if (ew != 0 || eh != 0) {
952                 RECT r;
953                 GetWindowRect (hwnd, &r);
954                 SendMessage (hwnd, WM_IGNORE_SIZE, 0, 0);
955                 SetWindowPos (hwnd, NULL, 0, 0,
956                               r.right - r.left - ew, r.bottom - r.top - eh,
957                               SWP_NOACTIVATE | SWP_NOMOVE | SWP_NOZORDER);
958             }
959 #endif
960             if (w != cols || h != rows || just_reconfigged) {
961                 term_invalidate();
962                 term_size (h, w, cfg.savelines);
963                 back->size();
964                 just_reconfigged = FALSE;
965             }
966         }
967         ignore_size = FALSE;
968         return 0;
969       case WM_VSCROLL:
970         switch (LOWORD(wParam)) {
971           case SB_BOTTOM: term_scroll(-1, 0); break;
972           case SB_TOP: term_scroll(+1, 0); break;
973           case SB_LINEDOWN: term_scroll (0, +1); break;
974           case SB_LINEUP: term_scroll (0, -1); break;
975           case SB_PAGEDOWN: term_scroll (0, +rows/2); break;
976           case SB_PAGEUP: term_scroll (0, -rows/2); break;
977           case SB_THUMBPOSITION: case SB_THUMBTRACK:
978             term_scroll (1, HIWORD(wParam)); break;
979         }
980         break; 
981      case WM_PALETTECHANGED:
982         if ((HWND) wParam != hwnd && pal != NULL) {
983             HDC hdc = get_ctx();
984             if (hdc) {
985                 if (RealizePalette (hdc) > 0)
986                     UpdateColors (hdc);
987                 free_ctx (hdc);
988             }
989         }
990         break;
991       case WM_QUERYNEWPALETTE:
992         if (pal != NULL) {
993             HDC hdc = get_ctx();
994             if (hdc) {
995                 if (RealizePalette (hdc) > 0)
996                     UpdateColors (hdc);
997                 free_ctx (hdc);
998                 return TRUE;
999             }
1000         }
1001         return FALSE;
1002       case WM_KEYDOWN:
1003       case WM_SYSKEYDOWN:
1004         /*
1005          * Add the scan code and keypress timing to the random
1006          * number noise, if we're using ssh.
1007          */
1008         if (cfg.protocol == PROT_SSH)
1009             noise_ultralight(lParam);
1010
1011         /*
1012          * We don't do TranslateMessage since it disassociates the
1013          * resulting CHAR message from the KEYDOWN that sparked it,
1014          * which we occasionally don't want. Instead, we process
1015          * KEYDOWN, and call the Win32 translator functions so that
1016          * we get the translations under _our_ control.
1017          */
1018         {
1019             unsigned char buf[20];
1020             int len;
1021
1022             len = TranslateKey (wParam, lParam, buf);
1023             if (len == -1)
1024                 return DefWindowProc (hwnd, message, wParam, lParam);
1025             ldisc->send (buf, len);
1026         }
1027         return 0;
1028       case WM_KEYUP:
1029       case WM_SYSKEYUP:
1030         /*
1031          * We handle KEYUP ourselves in order to distinghish left
1032          * and right Alt or Control keys, which Windows won't do
1033          * right if left to itself. See also the special processing
1034          * at the top of TranslateKey.
1035          */
1036         {
1037             BYTE keystate[256];
1038             int ret = GetKeyboardState(keystate);
1039             if (ret && wParam == VK_MENU) {
1040                 if (lParam & 0x1000000) keystate[VK_RMENU] = 0;
1041                 else keystate[VK_LMENU] = 0;
1042                 SetKeyboardState (keystate);
1043             }
1044             if (ret && wParam == VK_CONTROL) {
1045                 if (lParam & 0x1000000) keystate[VK_RCONTROL] = 0;
1046                 else keystate[VK_LCONTROL] = 0;
1047                 SetKeyboardState (keystate);
1048             }
1049         }
1050         /*
1051          * We don't return here, in order to allow Windows to do
1052          * its own KEYUP processing as well.
1053          */
1054         break;
1055       case WM_CHAR:
1056       case WM_SYSCHAR:
1057         /*
1058          * Nevertheless, we are prepared to deal with WM_CHAR
1059          * messages, should they crop up. So if someone wants to
1060          * post the things to us as part of a macro manoeuvre,
1061          * we're ready to cope.
1062          */
1063         {
1064             char c = xlat_kbd2tty((unsigned char)wParam);
1065             ldisc->send (&c, 1);
1066         }
1067         return 0;
1068     }
1069
1070     return DefWindowProc (hwnd, message, wParam, lParam);
1071 }
1072
1073 /*
1074  * Draw a line of text in the window, at given character
1075  * coordinates, in given attributes.
1076  *
1077  * We are allowed to fiddle with the contents of `text'.
1078  */
1079 void do_text (Context ctx, int x, int y, char *text, int len,
1080               unsigned long attr) {
1081     COLORREF fg, bg, t;
1082     int nfg, nbg, nfont;
1083     HDC hdc = ctx;
1084
1085     x *= font_width;
1086     y *= font_height;
1087
1088     if (attr & ATTR_ACTCURS) {
1089         attr &= (bold_mode == BOLD_COLOURS ? 0x200 : 0x300);
1090         attr ^= ATTR_CUR_XOR;
1091     }
1092
1093     nfont = 0;
1094     if (cfg.vtmode == VT_OEMONLY)
1095         nfont |= FONT_OEM;
1096
1097     /*
1098      * Map high-half characters in order to approximate ISO using
1099      * OEM character set. Characters missing are 0xC3 (Atilde) and
1100      * 0xCC (Igrave).
1101      */
1102     if (nfont & FONT_OEM) {
1103         int i;
1104         for (i=0; i<len; i++)
1105             if (text[i] >= '\xA0' && text[i] <= '\xFF') {
1106                 static const char oemhighhalf[] =
1107                     "\x20\xAD\xBD\x9C\xCF\xBE\xDD\xF5" /* A0-A7 */
1108                     "\xF9\xB8\xA6\xAE\xAA\xF0\xA9\xEE" /* A8-AF */
1109                     "\xF8\xF1\xFD\xFC\xEF\xE6\xF4\xFA" /* B0-B7 */
1110                     "\xF7\xFB\xA7\xAF\xAC\xAB\xF3\xA8" /* B8-BF */
1111                     "\xB7\xB5\xB6\x41\x8E\x8F\x92\x80" /* C0-C7 */
1112                     "\xD4\x90\xD2\xD3\x49\xD6\xD7\xD8" /* C8-CF */
1113                     "\xD1\xA5\xE3\xE0\xE2\xE5\x99\x9E" /* D0-D7 */
1114                     "\x9D\xEB\xE9\xEA\x9A\xED\xE8\xE1" /* D8-DF */
1115                     "\x85\xA0\x83\xC6\x84\x86\x91\x87" /* E0-E7 */
1116                     "\x8A\x82\x88\x89\x8D\xA1\x8C\x8B" /* E8-EF */
1117                     "\xD0\xA4\x95\xA2\x93\xE4\x94\xF6" /* F0-F7 */
1118                     "\x9B\x97\xA3\x96\x81\xEC\xE7\x98" /* F8-FF */
1119                     ;
1120                 text[i] = oemhighhalf[(unsigned char)text[i] - 0xA0];
1121             }
1122     }
1123
1124     if (attr & ATTR_GBCHR) {
1125         int i;
1126         /*
1127          * GB mapping: map # to pound, and everything else stays
1128          * normal.
1129          */
1130         for (i=0; i<len; i++)
1131             if (text[i] == '#')
1132                 text[i] = cfg.vtmode == VT_OEMONLY ? '\x9C' : '\xA3';
1133     } else if (attr & ATTR_LINEDRW) {
1134         int i;
1135         static const char poorman[] =
1136             "*#****\xB0\xB1**+++++-----++++|****\xA3\xB7";
1137         static const char oemmap[] =
1138             "*\xB1****\xF8\xF1**\xD9\xBF\xDA\xC0\xC5"
1139             "\xC4\xC4\xC4\xC4\xC4\xC3\xB4\xC1\xC2\xB3****\x9C\xFA";
1140
1141         /*
1142          * Line drawing mapping: map ` thru ~ (0x60 thru 0x7E) to
1143          * VT100 line drawing chars; everything else stays normal.
1144          */
1145         switch (cfg.vtmode) {
1146           case VT_XWINDOWS:
1147             for (i=0; i<len; i++)
1148                 if (text[i] >= '\x60' && text[i] <= '\x7E')
1149                     text[i] += '\x01' - '\x60';
1150             break;
1151           case VT_OEMANSI:
1152           case VT_OEMONLY:
1153             nfont |= FONT_OEM;
1154             for (i=0; i<len; i++)
1155                 if (text[i] >= '\x60' && text[i] <= '\x7E')
1156                     text[i] = oemmap[(unsigned char)text[i] - 0x60];
1157             break;
1158           case VT_POORMAN:
1159             for (i=0; i<len; i++)
1160                 if (text[i] >= '\x60' && text[i] <= '\x7E')
1161                     text[i] = poorman[(unsigned char)text[i] - 0x60];
1162             break;
1163         }
1164     }
1165
1166     nfg = 2 * ((attr & ATTR_FGMASK) >> ATTR_FGSHIFT);
1167     nbg = 2 * ((attr & ATTR_BGMASK) >> ATTR_BGSHIFT);
1168     if (bold_mode == BOLD_FONT && (attr & ATTR_BOLD))
1169         nfont |= FONT_BOLD;
1170     if (und_mode == UND_FONT && (attr & ATTR_UNDER))
1171         nfont |= FONT_UNDERLINE;
1172     if (attr & ATTR_REVERSE) {
1173         t = nfg; nfg = nbg; nbg = t;
1174     }
1175     if (bold_mode == BOLD_COLOURS && (attr & ATTR_BOLD))
1176         nfg++;
1177     fg = colours[nfg];
1178     bg = colours[nbg];
1179     SelectObject (hdc, fonts[nfont]);
1180     SetTextColor (hdc, fg);
1181     SetBkColor (hdc, bg);
1182     SetBkMode (hdc, OPAQUE);
1183     TextOut (hdc, x, y, text, len);
1184     if (bold_mode == BOLD_SHADOW && (attr & ATTR_BOLD)) {
1185         SetBkMode (hdc, TRANSPARENT);
1186         TextOut (hdc, x-1, y, text, len);
1187     }
1188     if (und_mode == UND_LINE && (attr & ATTR_UNDER)) {
1189         HPEN oldpen;
1190         oldpen = SelectObject (hdc, CreatePen(PS_SOLID, 0, fg));
1191         MoveToEx (hdc, x, y+descent, NULL);
1192         LineTo (hdc, x+len*font_width, y+descent);
1193         oldpen = SelectObject (hdc, oldpen);
1194         DeleteObject (oldpen);
1195     }
1196     if (attr & ATTR_PASCURS) {
1197         POINT pts[5];
1198         HPEN oldpen;
1199         pts[0].x = pts[1].x = pts[4].x = x;
1200         pts[2].x = pts[3].x = x+font_width-1;
1201         pts[0].y = pts[3].y = pts[4].y = y;
1202         pts[1].y = pts[2].y = y+font_height-1;
1203         oldpen = SelectObject (hdc, CreatePen(PS_SOLID, 0, colours[23]));
1204         Polyline (hdc, pts, 5);
1205         oldpen = SelectObject (hdc, oldpen);
1206         DeleteObject (oldpen);
1207     }
1208 }
1209
1210 /*
1211  * Translate a WM_(SYS)?KEYDOWN message into a string of ASCII
1212  * codes. Returns number of bytes used.
1213  */
1214 static int TranslateKey(WPARAM wParam, LPARAM lParam, unsigned char *output) {
1215     unsigned char *p = output;
1216     BYTE keystate[256];
1217     int ret, code;
1218     int cancel_alt = FALSE;
1219
1220     /*
1221      * Get hold of the keyboard state, because we'll need it a few
1222      * times shortly.
1223      */
1224     ret = GetKeyboardState(keystate);
1225
1226     /* 
1227      * Windows does not always want to distinguish left and right
1228      * Alt or Control keys. Thus we keep track of them ourselves.
1229      * See also the WM_KEYUP handler.
1230      */
1231     if (wParam == VK_MENU) {
1232         if (lParam & 0x1000000) keystate[VK_RMENU] = 0x80;
1233         else keystate[VK_LMENU] = 0x80;
1234         SetKeyboardState (keystate);
1235         return 0;
1236     }
1237     if (wParam == VK_CONTROL) {
1238         if (lParam & 0x1000000) keystate[VK_RCONTROL] = 0x80;
1239         else keystate[VK_LCONTROL] = 0x80;
1240         SetKeyboardState (keystate);
1241         return 0;
1242     }
1243
1244     /*
1245      * Prepend ESC, and cancel ALT, if ALT was pressed at the time
1246      * and it wasn't AltGr.
1247      */
1248     if (lParam & 0x20000000 && (keystate[VK_LMENU] & 0x80)) {
1249         *p++ = 0x1B;
1250         cancel_alt = TRUE;
1251     }
1252
1253     /*
1254      * NetHack keypad mode. This may conflict with Shift-PgUp/PgDn,
1255      * so we do it first.
1256      */
1257     if (cfg.nethack_keypad) {
1258         int shift = keystate[VK_SHIFT] & 0x80;
1259         /*
1260          * NB the shifted versions only work with numlock off.
1261          */
1262         switch ( (lParam >> 16) & 0x1FF ) {
1263           case 0x047: *p++ = shift ? 'Y' : 'y'; return p - output;
1264           case 0x048: *p++ = shift ? 'K' : 'k'; return p - output;
1265           case 0x049: *p++ = shift ? 'U' : 'u'; return p - output;
1266           case 0x04B: *p++ = shift ? 'H' : 'h'; return p - output;
1267           case 0x04C: *p++ = '.'; return p - output;
1268           case 0x04D: *p++ = shift ? 'L' : 'l'; return p - output;
1269           case 0x04F: *p++ = shift ? 'B' : 'b'; return p - output;
1270           case 0x050: *p++ = shift ? 'J' : 'j'; return p - output;
1271           case 0x051: *p++ = shift ? 'N' : 'n'; return p - output;
1272           case 0x053: *p++ = '.'; return p - output;
1273         }
1274     }
1275
1276     /*
1277      * Shift-PgUp, Shift-PgDn, and Alt-F4 all produce window
1278      * events: we'll deal with those now.
1279      */
1280     if (ret && (keystate[VK_SHIFT] & 0x80) && wParam == VK_PRIOR) {
1281         SendMessage (hwnd, WM_VSCROLL, SB_PAGEUP, 0);
1282         return 0;
1283     }
1284     if (ret && (keystate[VK_SHIFT] & 0x80) && wParam == VK_NEXT) {
1285         SendMessage (hwnd, WM_VSCROLL, SB_PAGEDOWN, 0);
1286         return 0;
1287     }
1288     if ((lParam & 0x20000000) && wParam == VK_F4 && cfg.alt_f4) {
1289         return -1;
1290     }
1291     if ((lParam & 0x20000000) && wParam == VK_SPACE && cfg.alt_space) {
1292         SendMessage (hwnd, WM_SYSCOMMAND, SC_KEYMENU, 0);
1293         return -1;
1294     }
1295
1296     /*
1297      * In general, the strategy is to see what the Windows keymap
1298      * translation has to say for itself, and then process function
1299      * keys and suchlike ourselves if that fails. But first we must
1300      * deal with the small number of special cases which the
1301      * Windows keymap translator thinks it can do but gets wrong.
1302      *
1303      * First special case: we might want the Backspace key to send
1304      * 0x7F not 0x08.
1305      */
1306     if (wParam == VK_BACK) {
1307         *p++ = (cfg.bksp_is_delete ? 0x7F : 0x08);
1308         return p - output;
1309     }
1310
1311     /*
1312      * Control-Space should send ^@ (0x00), not Space.
1313      */
1314     if (ret && (keystate[VK_CONTROL] & 0x80) && wParam == VK_SPACE) {
1315         *p++ = 0x00;
1316         return p - output;
1317     }
1318
1319     if (app_keypad_keys) {
1320         /*
1321          * If we're in applications keypad mode, we have to process it
1322          * before char-map translation, because it will pre-empt lots
1323          * of stuff, even if NumLock is off.
1324          */
1325         if (ret) {
1326             /*
1327              * Hack to ensure NumLock doesn't interfere with
1328              * perception of Shift for Keypad Plus. I don't pretend
1329              * to understand this, but it seems to work as is.
1330              * Leave it alone, or die.
1331              */
1332             keystate[VK_NUMLOCK] = 0;
1333             SetKeyboardState (keystate);
1334             GetKeyboardState (keystate);
1335         }
1336         switch ( (lParam >> 16) & 0x1FF ) {
1337           case 0x145: p += sprintf((char *)p, "\x1BOP"); return p - output;
1338           case 0x135: p += sprintf((char *)p, "\x1BOQ"); return p - output;
1339           case 0x037: p += sprintf((char *)p, "\x1BOR"); return p - output;
1340           case 0x047: p += sprintf((char *)p, "\x1BOw"); return p - output;
1341           case 0x048: p += sprintf((char *)p, "\x1BOx"); return p - output;
1342           case 0x049: p += sprintf((char *)p, "\x1BOy"); return p - output;
1343           case 0x04A: p += sprintf((char *)p, "\x1BOS"); return p - output;
1344           case 0x04B: p += sprintf((char *)p, "\x1BOt"); return p - output;
1345           case 0x04C: p += sprintf((char *)p, "\x1BOu"); return p - output;
1346           case 0x04D: p += sprintf((char *)p, "\x1BOv"); return p - output;
1347           case 0x04E: /* keypad + is ^[Ol, but ^[Om with Shift */
1348             p += sprintf((char *)p,
1349                          (ret && (keystate[VK_SHIFT] & 0x80)) ?
1350                          "\x1BOm" : "\x1BOl");
1351             return p - output;
1352           case 0x04F: p += sprintf((char *)p, "\x1BOq"); return p - output;
1353           case 0x050: p += sprintf((char *)p, "\x1BOr"); return p - output;
1354           case 0x051: p += sprintf((char *)p, "\x1BOs"); return p - output;
1355           case 0x052: p += sprintf((char *)p, "\x1BOp"); return p - output;
1356           case 0x053: p += sprintf((char *)p, "\x1BOn"); return p - output;
1357           case 0x11C: p += sprintf((char *)p, "\x1BOM"); return p - output;
1358         }
1359     }
1360
1361     /*
1362      * Before doing Windows charmap translation, remove LeftALT
1363      * from the keymap, since its sole effect should be to prepend
1364      * ESC, which we've already done. Note that removal of LeftALT
1365      * has to happen _after_ the above call to SetKeyboardState, or
1366      * dire things will befall.
1367      */
1368     if (cancel_alt) {
1369         keystate[VK_MENU] = keystate[VK_RMENU];
1370         keystate[VK_LMENU] = 0;
1371     }
1372
1373     /*
1374      * Attempt the Windows char-map translation.
1375      */
1376     if (ret) {
1377         WORD chr;
1378         int r;
1379         BOOL capsOn=keystate[VK_CAPITAL] !=0;
1380
1381         /* helg: clear CAPS LOCK state if caps lock switches to cyrillic */
1382         if(cfg.xlat_capslockcyr)
1383             keystate[VK_CAPITAL] = 0;
1384
1385         r = ToAscii (wParam, (lParam >> 16) & 0xFF,
1386                      keystate, &chr, 0);
1387
1388         if(capsOn)
1389             chr = xlat_latkbd2win((unsigned char)(chr & 0xFF));
1390         if (r == 1) {
1391             *p++ = xlat_kbd2tty((unsigned char)(chr & 0xFF));
1392             return p - output;
1393         }
1394     }
1395
1396     /*
1397      * OK, we haven't had a key code from the keymap translation.
1398      * We'll try our various special cases and function keys, and
1399      * then give up. (There's nothing wrong with giving up:
1400      * Scrollock, Pause/Break, and of course the various buckybit
1401      * keys all produce KEYDOWN events that we really _do_ want to
1402      * ignore.)
1403      */
1404
1405     /*
1406      * Control-2 should return ^@ (0x00), Control-6 should return
1407      * ^^ (0x1E), and Control-Minus should return ^_ (0x1F). Since
1408      * the DOS keyboard handling did it, and we have nothing better
1409      * to do with the key combo in question, we'll also map
1410      * Control-Backquote to ^\ (0x1C).
1411      */
1412     if (ret && (keystate[VK_CONTROL] & 0x80) && wParam == '2') {
1413         *p++ = 0x00;
1414         return p - output;
1415     }
1416     if (ret && (keystate[VK_CONTROL] & 0x80) && wParam == '6') {
1417         *p++ = 0x1E;
1418         return p - output;
1419     }
1420     if (ret && (keystate[VK_CONTROL] & 0x80) && wParam == 0xBD) {
1421         *p++ = 0x1F;
1422         return p - output;
1423     }
1424     if (ret && (keystate[VK_CONTROL] & 0x80) && wParam == 0xDF) {
1425         *p++ = 0x1C;
1426         return p - output;
1427     }
1428
1429     /*
1430      * First, all the keys that do tilde codes. (ESC '[' nn '~',
1431      * for integer decimal nn.)
1432      *
1433      * We also deal with the weird ones here. Linux VCs replace F1
1434      * to F5 by ESC [ [ A to ESC [ [ E. rxvt doesn't do _that_, but
1435      * does replace Home and End (1~ and 4~) by ESC [ H and ESC O w
1436      * respectively.
1437      */
1438     code = 0;
1439     switch (wParam) {
1440       case VK_F1: code = (keystate[VK_SHIFT] & 0x80 ? 23 : 11); break;
1441       case VK_F2: code = (keystate[VK_SHIFT] & 0x80 ? 24 : 12); break;
1442       case VK_F3: code = (keystate[VK_SHIFT] & 0x80 ? 25 : 13); break;
1443       case VK_F4: code = (keystate[VK_SHIFT] & 0x80 ? 26 : 14); break;
1444       case VK_F5: code = (keystate[VK_SHIFT] & 0x80 ? 28 : 15); break;
1445       case VK_F6: code = (keystate[VK_SHIFT] & 0x80 ? 29 : 17); break;
1446       case VK_F7: code = (keystate[VK_SHIFT] & 0x80 ? 31 : 18); break;
1447       case VK_F8: code = (keystate[VK_SHIFT] & 0x80 ? 32 : 19); break;
1448       case VK_F9: code = (keystate[VK_SHIFT] & 0x80 ? 33 : 20); break;
1449       case VK_F10: code = (keystate[VK_SHIFT] & 0x80 ? 34 : 21); break;
1450       case VK_F11: code = 23; break;
1451       case VK_F12: code = 24; break;
1452       case VK_HOME: code = 1; break;
1453       case VK_INSERT: code = 2; break;
1454       case VK_DELETE: code = 3; break;
1455       case VK_END: code = 4; break;
1456       case VK_PRIOR: code = 5; break;
1457       case VK_NEXT: code = 6; break;
1458     }
1459     if (cfg.linux_funkeys && code >= 11 && code <= 15) {
1460         p += sprintf((char *)p, "\x1B[[%c", code + 'A' - 11);
1461         return p - output;
1462     }
1463     if (cfg.rxvt_homeend && (code == 1 || code == 4)) {
1464         p += sprintf((char *)p, code == 1 ? "\x1B[H" : "\x1BOw");
1465         return p - output;
1466     }
1467     if (code) {
1468         p += sprintf((char *)p, "\x1B[%d~", code);
1469         return p - output;
1470     }
1471
1472     /*
1473      * Now the remaining keys (arrows and Keypad 5. Keypad 5 for
1474      * some reason seems to send VK_CLEAR to Windows...).
1475      */
1476     switch (wParam) {
1477       case VK_UP:
1478         p += sprintf((char *)p, app_cursor_keys ? "\x1BOA" : "\x1B[A");
1479         return p - output;
1480       case VK_DOWN:
1481         p += sprintf((char *)p, app_cursor_keys ? "\x1BOB" : "\x1B[B");
1482         return p - output;
1483       case VK_RIGHT:
1484         p += sprintf((char *)p, app_cursor_keys ? "\x1BOC" : "\x1B[C");
1485         return p - output;
1486       case VK_LEFT:
1487         p += sprintf((char *)p, app_cursor_keys ? "\x1BOD" : "\x1B[D");
1488         return p - output;
1489       case VK_CLEAR: p += sprintf((char *)p, "\x1B[G"); return p - output;
1490     }
1491
1492     return 0;
1493 }
1494
1495 void set_title (char *title) {
1496     sfree (window_name);
1497     window_name = smalloc(1+strlen(title));
1498     strcpy (window_name, title);
1499     if (cfg.win_name_always || !IsIconic(hwnd))
1500         SetWindowText (hwnd, title);
1501 }
1502
1503 void set_icon (char *title) {
1504     sfree (icon_name);
1505     icon_name = smalloc(1+strlen(title));
1506     strcpy (icon_name, title);
1507     if (!cfg.win_name_always && IsIconic(hwnd))
1508         SetWindowText (hwnd, title);
1509 }
1510
1511 void set_sbar (int total, int start, int page) {
1512     SCROLLINFO si;
1513     si.cbSize = sizeof(si);
1514     si.fMask = SIF_RANGE | SIF_PAGE | SIF_POS | SIF_DISABLENOSCROLL;
1515     si.nMin = 0;
1516     si.nMax = total - 1;
1517     si.nPage = page;
1518     si.nPos = start;
1519     if (hwnd)
1520         SetScrollInfo (hwnd, SB_VERT, &si, TRUE);
1521 }
1522
1523 Context get_ctx(void) {
1524     HDC hdc;
1525     if (hwnd) {
1526         hdc = GetDC (hwnd);
1527         if (hdc && pal)
1528             SelectPalette (hdc, pal, FALSE);
1529         return hdc;
1530     } else
1531         return NULL;
1532 }
1533
1534 void free_ctx (Context ctx) {
1535     SelectPalette (ctx, GetStockObject (DEFAULT_PALETTE), FALSE);
1536     ReleaseDC (hwnd, ctx);
1537 }
1538
1539 static void real_palette_set (int n, int r, int g, int b) {
1540     if (pal) {
1541         logpal->palPalEntry[n].peRed = r;
1542         logpal->palPalEntry[n].peGreen = g;
1543         logpal->palPalEntry[n].peBlue = b;
1544         logpal->palPalEntry[n].peFlags = PC_NOCOLLAPSE;
1545         colours[n] = PALETTERGB(r, g, b);
1546         SetPaletteEntries (pal, 0, NCOLOURS, logpal->palPalEntry);
1547     } else
1548         colours[n] = RGB(r, g, b);
1549 }
1550
1551 void palette_set (int n, int r, int g, int b) {
1552     static const int first[21] = {
1553         0, 2, 4, 6, 8, 10, 12, 14,
1554         1, 3, 5, 7, 9, 11, 13, 15,
1555         16, 17, 18, 20, 22
1556     };
1557     real_palette_set (first[n], r, g, b);
1558     if (first[n] >= 18)
1559         real_palette_set (first[n]+1, r, g, b);
1560     if (pal) {
1561         HDC hdc = get_ctx();
1562         UnrealizeObject (pal);
1563         RealizePalette (hdc);
1564         free_ctx (hdc);
1565     }
1566 }
1567
1568 void palette_reset (void) {
1569     int i;
1570
1571     for (i = 0; i < NCOLOURS; i++) {
1572         if (pal) {
1573             logpal->palPalEntry[i].peRed = defpal[i].rgbtRed;
1574             logpal->palPalEntry[i].peGreen = defpal[i].rgbtGreen;
1575             logpal->palPalEntry[i].peBlue = defpal[i].rgbtBlue;
1576             logpal->palPalEntry[i].peFlags = 0;
1577             colours[i] = PALETTERGB(defpal[i].rgbtRed,
1578                                     defpal[i].rgbtGreen,
1579                                     defpal[i].rgbtBlue);
1580         } else
1581             colours[i] = RGB(defpal[i].rgbtRed,
1582                              defpal[i].rgbtGreen,
1583                              defpal[i].rgbtBlue);
1584     }
1585
1586     if (pal) {
1587         HDC hdc;
1588         SetPaletteEntries (pal, 0, NCOLOURS, logpal->palPalEntry);
1589         hdc = get_ctx();
1590         RealizePalette (hdc);
1591         free_ctx (hdc);
1592     }
1593 }
1594
1595 void write_clip (void *data, int len) {
1596     HGLOBAL clipdata;
1597     void *lock;
1598
1599     clipdata = GlobalAlloc (GMEM_DDESHARE | GMEM_MOVEABLE, len + 1);
1600     if (!clipdata)
1601         return;
1602     lock = GlobalLock (clipdata);
1603     if (!lock)
1604         return;
1605     memcpy (lock, data, len);
1606     ((unsigned char *) lock) [len] = 0;
1607     GlobalUnlock (clipdata);
1608
1609     SendMessage (hwnd, WM_IGNORE_CLIP, TRUE, 0);
1610     if (OpenClipboard (hwnd)) {
1611         EmptyClipboard();
1612         SetClipboardData (CF_TEXT, clipdata);
1613         CloseClipboard();
1614     } else
1615         GlobalFree (clipdata);
1616     SendMessage (hwnd, WM_IGNORE_CLIP, FALSE, 0);
1617 }
1618
1619 void get_clip (void **p, int *len) {
1620     static HGLOBAL clipdata = NULL;
1621
1622     if (!p) {
1623         if (clipdata)
1624             GlobalUnlock (clipdata);
1625         clipdata = NULL;
1626         return;
1627     } else {
1628         if (OpenClipboard (NULL)) {
1629             clipdata = GetClipboardData (CF_TEXT);
1630             CloseClipboard();
1631             if (clipdata) {
1632                 *p = GlobalLock (clipdata);
1633                 if (*p) {
1634                     *len = strlen(*p);
1635                     return;
1636                 }
1637             }
1638         }
1639     }
1640
1641     *p = NULL;
1642     *len = 0;
1643 }
1644
1645 /*
1646  * Move `lines' lines from position `from' to position `to' in the
1647  * window.
1648  */
1649 void optimised_move (int to, int from, int lines) {
1650     RECT r;
1651     int min, max;
1652
1653     min = (to < from ? to : from);
1654     max = to + from - min;
1655
1656     r.left = 0; r.right = cols * font_width;
1657     r.top = min * font_height; r.bottom = (max+lines) * font_height;
1658     ScrollWindow (hwnd, 0, (to - from) * font_height, &r, &r);
1659 }
1660
1661 /*
1662  * Print a message box and perform a fatal exit.
1663  */
1664 void fatalbox(char *fmt, ...) {
1665     va_list ap;
1666     char stuff[200];
1667
1668     va_start(ap, fmt);
1669     vsprintf(stuff, fmt, ap);
1670     va_end(ap);
1671     MessageBox(hwnd, stuff, "PuTTY Fatal Error", MB_ICONERROR | MB_OK);
1672     exit(1);
1673 }
1674
1675 /*
1676  * Beep.
1677  */
1678 void beep(void) {
1679     MessageBeep(MB_OK);
1680 }