projects / u / mdw / putty / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Fix potential segfault on malloc failure
[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
73251d5d 85 putty_inst = inst;
86
374330e2 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
e277c42d 109 default_protocol = DEFAULT_PROTOCOL;
110 default_port = DEFAULT_PORT;
111
374330e2 112 do_defaults(NULL);
113
114 p = cmdline;
115 while (*p && isspace(*p)) p++;
116
117 /*
e277c42d 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 /*
374330e2 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
5e1a8e27 183 back = (cfg.protocol == PROT_SSH ? &ssh_backend :
5bc238bb 184 cfg.protocol == PROT_TELNET ? &telnet_backend :
185 &raw_backend);
186
187 ldisc = (cfg.ldisc_term ? &ldisc_term : &ldisc_simple);
374330e2 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));
1bb542b2 197 wndclass.hCursor = LoadCursor (NULL, IDC_IBEAM);
374330e2 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;
5ac36532 301 sprintf(msg, "%s - PuTTY", realhost);
374330e2 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);
0a4aa984 324 HMENU p,s;
325 int i;
374330e2 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");
374330e2 347 AppendMenu (m, MF_SEPARATOR, 0, 0);
348 }
c5e9c988 349 AppendMenu (m, MF_ENABLED, IDM_SHOWLOG, "Event Log");
350 AppendMenu (m, MF_SEPARATOR, 0, 0);
374330e2 351 AppendMenu (m, MF_ENABLED, IDM_NEWSESS, "New Session");
352 AppendMenu (m, MF_ENABLED, IDM_DUPSESS, "Duplicate Session");
0a4aa984 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");
374330e2 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 */
414static 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 */
433static 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 */
489static 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) {
14963b8f 513 f(FONT_NORMAL, cfg.fontcharset, fw_dontcare, FALSE);
514 f(FONT_UNDERLINE, cfg.fontcharset, fw_dontcare, TRUE);
374330e2 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) {
14963b8f 522 f(FONT_BOLD, cfg.fontcharset, fw_bold, FALSE);
523 f(FONT_BOLDUND, cfg.fontcharset, fw_bold, TRUE);
374330e2 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
587void 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
596static void click (Mouse_Button b, int x, int y) {
fdedf2c8 597 int thistime = GetMessageTime();
598
599 if (lastbtn == b && thistime - lasttime < dbltime) {
374330e2 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);
fdedf2c8 609 lasttime = thistime;
374330e2 610}
611
612static 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;
68130d34 622 case WM_CLOSE:
9ef49106 623 if (!cfg.warn_on_close ||
624 MessageBox(hwnd, "Are you sure you want to close this session?",
68130d34 625 "PuTTY Exit Confirmation",
626 MB_ICONWARNING | MB_OKCANCEL) == IDOK)
627 DestroyWindow(hwnd);
628 return 0;
374330e2 629 case WM_DESTROY:
630 PostQuitMessage (0);
631 return 0;
6833a413 632 case WM_SYSCOMMAND:
633 switch (wParam & ~0xF) { /* low 4 bits reserved to Windows */
374330e2 634 case IDM_SHOWLOG:
c5e9c988 635 showeventlog(hwnd);
374330e2 636 break;
637 case IDM_NEWSESS:
638 case IDM_DUPSESS:
6833a413 639 case IDM_SAVEDSESS:
374330e2 640 {
641 char b[2048];
642 char c[30], *cl;
e4e4cc7e 643 int freecl = FALSE;
374330e2 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;
0a4aa984 676 } else if (wParam == IDM_SAVEDSESS) {
e4e4cc7e 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 */
94e6450e 681 else {
682 sprintf(cl, "putty @%s", session);
683 freecl = TRUE;
684 }
374330e2 685 } else
6833a413 686 cl = NULL;
374330e2 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);
e4e4cc7e 701 if (freecl)
702 free(cl);
374330e2 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);
638787a8 719 ldisc = (cfg.ldisc_term ? &ldisc_term : &ldisc_simple);
374330e2 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;
0a4aa984 760 default:
761 if (wParam >= IDM_SAVED_MIN && wParam <= IDM_SAVED_MAX) {
762 SendMessage(hwnd, WM_SYSCOMMAND, IDM_SAVEDSESS, wParam);
763 }
374330e2 764 }
765 break;
37508af4 766
767#define X_POS(l) ((int)(short)LOWORD(l))
768#define Y_POS(l) ((int)(short)HIWORD(l))
769
fdedf2c8 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
374330e2 773 case WM_LBUTTONDOWN:
fdedf2c8 774 click (MB_SELECT, TO_CHR_X(X_POS(lParam)),
775 TO_CHR_Y(Y_POS(lParam)));
fef97f43 776 SetCapture(hwnd);
374330e2 777 return 0;
778 case WM_LBUTTONUP:
fdedf2c8 779 term_mouse (MB_SELECT, MA_RELEASE, TO_CHR_X(X_POS(lParam)),
780 TO_CHR_Y(Y_POS(lParam)));
37508af4 781 ReleaseCapture();
374330e2 782 return 0;
783 case WM_MBUTTONDOWN:
37508af4 784 SetCapture(hwnd);
374330e2 785 click (cfg.mouse_is_xterm ? MB_PASTE : MB_EXTEND,
fdedf2c8 786 TO_CHR_X(X_POS(lParam)),
787 TO_CHR_Y(Y_POS(lParam)));
374330e2 788 return 0;
789 case WM_MBUTTONUP:
790 term_mouse (cfg.mouse_is_xterm ? MB_PASTE : MB_EXTEND,
fdedf2c8 791 MA_RELEASE, TO_CHR_X(X_POS(lParam)),
792 TO_CHR_Y(Y_POS(lParam)));
37508af4 793 ReleaseCapture();
a0b1cefc 794 return 0;
374330e2 795 case WM_RBUTTONDOWN:
37508af4 796 SetCapture(hwnd);
374330e2 797 click (cfg.mouse_is_xterm ? MB_EXTEND : MB_PASTE,
fdedf2c8 798 TO_CHR_X(X_POS(lParam)),
799 TO_CHR_Y(Y_POS(lParam)));
374330e2 800 return 0;
801 case WM_RBUTTONUP:
802 term_mouse (cfg.mouse_is_xterm ? MB_EXTEND : MB_PASTE,
fdedf2c8 803 MA_RELEASE, TO_CHR_X(X_POS(lParam)),
804 TO_CHR_Y(Y_POS(lParam)));
37508af4 805 ReleaseCapture();
374330e2 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;
fdedf2c8 823 term_mouse (b, MA_DRAG, TO_CHR_X(X_POS(lParam)),
824 TO_CHR_Y(Y_POS(lParam)));
374330e2 825 }
374330e2 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;
73251d5d 890 case WM_ENTERSIZEMOVE:
891 EnableSizeTip(1);
892 break;
893 case WM_EXITSIZEMOVE:
894 EnableSizeTip(0);
895 break;
374330e2 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;
73251d5d 905 UpdateSizeTip(hwnd, w, h);
374330e2 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) {
1a6f78fe 939 int width, height, w, h;
940#if 0 /* we have fixed this using WM_SIZING now */
941 int ew, eh;
942#endif
374330e2 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);
c5e9c988 1023 if (len == -1)
1024 return DefWindowProc (hwnd, message, wParam, lParam);
5bc238bb 1025 ldisc->send (buf, len);
374330e2 1026 }
1027 return 0;
67c339f7 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;
374330e2 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 {
14963b8f 1064 char c = xlat_kbd2tty((unsigned char)wParam);
5bc238bb 1065 ldisc->send (&c, 1);
374330e2 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 */
1079void 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)) {
1a6f78fe 1189 HPEN oldpen;
1190 oldpen = SelectObject (hdc, CreatePen(PS_SOLID, 0, fg));
374330e2 1191 MoveToEx (hdc, x, y+descent, NULL);
1192 LineTo (hdc, x+len*font_width, y+descent);
1a6f78fe 1193 oldpen = SelectObject (hdc, oldpen);
1194 DeleteObject (oldpen);
374330e2 1195 }
1196 if (attr & ATTR_PASCURS) {
1197 POINT pts[5];
1a6f78fe 1198 HPEN oldpen;
374330e2 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;
1a6f78fe 1203 oldpen = SelectObject (hdc, CreatePen(PS_SOLID, 0, colours[23]));
374330e2 1204 Polyline (hdc, pts, 5);
1a6f78fe 1205 oldpen = SelectObject (hdc, oldpen);
1206 DeleteObject (oldpen);
374330e2 1207 }
1208}
1209
1210/*
1211 * Translate a WM_(SYS)?KEYDOWN message into a string of ASCII
1212 * codes. Returns number of bytes used.
1213 */
1214static int TranslateKey(WPARAM wParam, LPARAM lParam, unsigned char *output) {
1215 unsigned char *p = output;
1216 BYTE keystate[256];
1217 int ret, code;
67c339f7 1218 int cancel_alt = FALSE;
374330e2 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
67c339f7 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
374330e2 1253 /*
25d39ef6 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 /*
374330e2 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 }
c5e9c988 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;
374330e2 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
25d39ef6 1319 if (app_keypad_keys) {
c5e9c988 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 */
374330e2 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 /*
67c339f7 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.
374330e2 1367 */
67c339f7 1368 if (cancel_alt) {
1369 keystate[VK_MENU] = keystate[VK_RMENU];
1370 keystate[VK_LMENU] = 0;
1371 }
374330e2 1372
1373 /*
1374 * Attempt the Windows char-map translation.
1375 */
1376 if (ret) {
1377 WORD chr;
14963b8f 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));
374330e2 1390 if (r == 1) {
14963b8f 1391 *p++ = xlat_kbd2tty((unsigned char)(chr & 0xFF));
374330e2 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
1495void set_title (char *title) {
1496 sfree (window_name);
1497 window_name = smalloc(1+strlen(title));
1498 strcpy (window_name, title);
37508af4 1499 if (cfg.win_name_always || !IsIconic(hwnd))
374330e2 1500 SetWindowText (hwnd, title);
1501}
1502
1503void set_icon (char *title) {
1504 sfree (icon_name);
1505 icon_name = smalloc(1+strlen(title));
1506 strcpy (icon_name, title);
37508af4 1507 if (!cfg.win_name_always && IsIconic(hwnd))
374330e2 1508 SetWindowText (hwnd, title);
1509}
1510
1511void 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;
c1f5f956 1519 if (hwnd)
1520 SetScrollInfo (hwnd, SB_VERT, &si, TRUE);
374330e2 1521}
1522
f67b4e85 1523Context get_ctx(void) {
374330e2 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
1534void free_ctx (Context ctx) {
1535 SelectPalette (ctx, GetStockObject (DEFAULT_PALETTE), FALSE);
1536 ReleaseDC (hwnd, ctx);
1537}
1538
1539static 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
1551void 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
1568void 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
1595void 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
1619void 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 */
1649void optimised_move (int to, int from, int lines) {
1650 RECT r;
f67b4e85 1651 int min, max;
374330e2 1652
1653 min = (to < from ? to : from);
1654 max = to + from - min;
374330e2 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 */
1664void 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 */
1678void beep(void) {
1679 MessageBeep(MB_OK);
1680}
Local modifications for Simon Tatham's `PuTTY' SSH client and terminal emulator