Fix potential buffer overrun in new saved-session menu
[u/mdw/putty] / window.c
CommitLineData
374330e2 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
6833a413 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
374330e2 35
36#define WM_IGNORE_SIZE (WM_USER + 2)
37#define WM_IGNORE_CLIP (WM_USER + 3)
38
39static int WINAPI WndProc (HWND, UINT, WPARAM, LPARAM);
40static int TranslateKey(WPARAM wParam, LPARAM lParam, unsigned char *output);
41static void cfgtopalette(void);
42static void init_palette(void);
43static void init_fonts(void);
44
45static 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
55static HFONT fonts[8];
56static enum {
57 BOLD_COLOURS, BOLD_SHADOW, BOLD_FONT
58} bold_mode;
59static enum {
60 UND_LINE, UND_FONT
61} und_mode;
62static int descent;
63
64#define NCOLOURS 24
65static COLORREF colours[NCOLOURS];
66static HPALETTE pal;
67static LPLOGPALETTE logpal;
68static RGBTRIPLE defpal[NCOLOURS];
69
70static HWND hwnd;
71
72static int dbltime, lasttime, lastact;
73static Mouse_Button lastbtn;
74
75static char *window_name, *icon_name;
76
77int 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
e277c42d 107 default_protocol = DEFAULT_PROTOCOL;
108 default_port = DEFAULT_PORT;
109
374330e2 110 do_defaults(NULL);
111
112 p = cmdline;
113 while (*p && isspace(*p)) p++;
114
115 /*
e277c42d 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 /*
374330e2 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
5e1a8e27 181 back = (cfg.protocol == PROT_SSH ? &ssh_backend :
182 cfg.protocol == PROT_TELNET ? &telnet_backend : &raw_backend );
374330e2 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));
1bb542b2 192 wndclass.hCursor = LoadCursor (NULL, IDC_IBEAM);
374330e2 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;
5ac36532 296 sprintf(msg, "%s - PuTTY", realhost);
374330e2 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);
0a4aa984 319 HMENU p,s;
320 int i;
374330e2 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");
0a4aa984 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");
374330e2 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 */
408static 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 */
427static 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 */
483static 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
581void 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
590static void click (Mouse_Button b, int x, int y) {
fdedf2c8 591 int thistime = GetMessageTime();
592
593 if (lastbtn == b && thistime - lasttime < dbltime) {
374330e2 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);
fdedf2c8 603 lasttime = thistime;
374330e2 604}
605
606static 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;
6833a413 619 case WM_SYSCOMMAND:
620 switch (wParam & ~0xF) { /* low 4 bits reserved to Windows */
374330e2 621 case IDM_SHOWLOG:
622 shownegot(hwnd);
623 break;
624 case IDM_NEWSESS:
625 case IDM_DUPSESS:
6833a413 626 case IDM_SAVEDSESS:
374330e2 627 {
628 char b[2048];
629 char c[30], *cl;
e4e4cc7e 630 int freecl = FALSE;
374330e2 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;
0a4aa984 663 } else if (wParam == IDM_SAVEDSESS) {
e4e4cc7e 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;
374330e2 670 } else
6833a413 671 cl = NULL;
374330e2 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);
e4e4cc7e 686 if (freecl)
687 free(cl);
374330e2 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;
0a4aa984 744 default:
745 if (wParam >= IDM_SAVED_MIN && wParam <= IDM_SAVED_MAX) {
746 SendMessage(hwnd, WM_SYSCOMMAND, IDM_SAVEDSESS, wParam);
747 }
374330e2 748 }
749 break;
37508af4 750
751#define X_POS(l) ((int)(short)LOWORD(l))
752#define Y_POS(l) ((int)(short)HIWORD(l))
753
fdedf2c8 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
374330e2 757 case WM_LBUTTONDOWN:
fdedf2c8 758 click (MB_SELECT, TO_CHR_X(X_POS(lParam)),
759 TO_CHR_Y(Y_POS(lParam)));
fef97f43 760 SetCapture(hwnd);
374330e2 761 return 0;
762 case WM_LBUTTONUP:
fdedf2c8 763 term_mouse (MB_SELECT, MA_RELEASE, TO_CHR_X(X_POS(lParam)),
764 TO_CHR_Y(Y_POS(lParam)));
37508af4 765 ReleaseCapture();
374330e2 766 return 0;
767 case WM_MBUTTONDOWN:
37508af4 768 SetCapture(hwnd);
374330e2 769 click (cfg.mouse_is_xterm ? MB_PASTE : MB_EXTEND,
fdedf2c8 770 TO_CHR_X(X_POS(lParam)),
771 TO_CHR_Y(Y_POS(lParam)));
374330e2 772 return 0;
773 case WM_MBUTTONUP:
774 term_mouse (cfg.mouse_is_xterm ? MB_PASTE : MB_EXTEND,
fdedf2c8 775 MA_RELEASE, TO_CHR_X(X_POS(lParam)),
776 TO_CHR_Y(Y_POS(lParam)));
37508af4 777 ReleaseCapture();
a0b1cefc 778 return 0;
374330e2 779 case WM_RBUTTONDOWN:
37508af4 780 SetCapture(hwnd);
374330e2 781 click (cfg.mouse_is_xterm ? MB_EXTEND : MB_PASTE,
fdedf2c8 782 TO_CHR_X(X_POS(lParam)),
783 TO_CHR_Y(Y_POS(lParam)));
374330e2 784 return 0;
785 case WM_RBUTTONUP:
786 term_mouse (cfg.mouse_is_xterm ? MB_EXTEND : MB_PASTE,
fdedf2c8 787 MA_RELEASE, TO_CHR_X(X_POS(lParam)),
788 TO_CHR_Y(Y_POS(lParam)));
37508af4 789 ReleaseCapture();
374330e2 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;
fdedf2c8 807 term_mouse (b, MA_DRAG, TO_CHR_X(X_POS(lParam)),
808 TO_CHR_Y(Y_POS(lParam)));
374330e2 809 }
374330e2 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) {
1a6f78fe 916 int width, height, w, h;
917#if 0 /* we have fixed this using WM_SIZING now */
918 int ew, eh;
919#endif
374330e2 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;
67c339f7 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;
374330e2 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 */
1054void 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)) {
1a6f78fe 1164 HPEN oldpen;
1165 oldpen = SelectObject (hdc, CreatePen(PS_SOLID, 0, fg));
374330e2 1166 MoveToEx (hdc, x, y+descent, NULL);
1167 LineTo (hdc, x+len*font_width, y+descent);
1a6f78fe 1168 oldpen = SelectObject (hdc, oldpen);
1169 DeleteObject (oldpen);
374330e2 1170 }
1171 if (attr & ATTR_PASCURS) {
1172 POINT pts[5];
1a6f78fe 1173 HPEN oldpen;
374330e2 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;
1a6f78fe 1178 oldpen = SelectObject (hdc, CreatePen(PS_SOLID, 0, colours[23]));
374330e2 1179 Polyline (hdc, pts, 5);
1a6f78fe 1180 oldpen = SelectObject (hdc, oldpen);
1181 DeleteObject (oldpen);
374330e2 1182 }
1183}
1184
1185/*
1186 * Translate a WM_(SYS)?KEYDOWN message into a string of ASCII
1187 * codes. Returns number of bytes used.
1188 */
1189static int TranslateKey(WPARAM wParam, LPARAM lParam, unsigned char *output) {
1190 unsigned char *p = output;
1191 BYTE keystate[256];
1192 int ret, code;
67c339f7 1193 int cancel_alt = FALSE;
374330e2 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
67c339f7 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
374330e2 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 /*
67c339f7 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.
374330e2 1316 */
67c339f7 1317 if (cancel_alt) {
1318 keystate[VK_MENU] = keystate[VK_RMENU];
1319 keystate[VK_LMENU] = 0;
1320 }
374330e2 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
1434void set_title (char *title) {
1435 sfree (window_name);
1436 window_name = smalloc(1+strlen(title));
1437 strcpy (window_name, title);
37508af4 1438 if (cfg.win_name_always || !IsIconic(hwnd))
374330e2 1439 SetWindowText (hwnd, title);
1440}
1441
1442void set_icon (char *title) {
1443 sfree (icon_name);
1444 icon_name = smalloc(1+strlen(title));
1445 strcpy (icon_name, title);
37508af4 1446 if (!cfg.win_name_always && IsIconic(hwnd))
374330e2 1447 SetWindowText (hwnd, title);
1448}
1449
1450void 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;
c1f5f956 1458 if (hwnd)
1459 SetScrollInfo (hwnd, SB_VERT, &si, TRUE);
374330e2 1460}
1461
f67b4e85 1462Context get_ctx(void) {
374330e2 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
1473void free_ctx (Context ctx) {
1474 SelectPalette (ctx, GetStockObject (DEFAULT_PALETTE), FALSE);
1475 ReleaseDC (hwnd, ctx);
1476}
1477
1478static 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
1490void 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
1507void 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
1534void 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
1558void 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 */
1588void optimised_move (int to, int from, int lines) {
1589 RECT r;
f67b4e85 1590 int min, max;
374330e2 1591
1592 min = (to < from ? to : from);
1593 max = to + from - min;
374330e2 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 */
1603void 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 */
1617void beep(void) {
1618 MessageBeep(MB_OK);
1619}