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