ff2ae367 |
1 | /* |
2 | * console.c: various interactive-prompt routines shared between |
3 | * the console PuTTY tools |
4 | */ |
5 | |
6 | #include <windows.h> |
7 | |
8 | #include <stdio.h> |
9 | #include <stdlib.h> |
10 | #include <stdarg.h> |
11 | |
12 | #include "putty.h" |
13 | #include "storage.h" |
14 | #include "ssh.h" |
15 | |
16 | int console_batch_mode = FALSE; |
17 | |
93b581bd |
18 | /* |
19 | * Clean up and exit. |
20 | */ |
21 | void cleanup_exit(int code) |
22 | { |
23 | /* |
24 | * Clean up. |
25 | */ |
26 | sk_cleanup(); |
27 | WSACleanup(); |
28 | |
29 | if (cfg.protocol == PROT_SSH) { |
30 | random_save_seed(); |
31 | #ifdef MSCRYPTOAPI |
32 | crypto_wrapup(); |
33 | #endif |
34 | } |
35 | |
36 | exit(code); |
37 | } |
38 | |
ff2ae367 |
39 | void verify_ssh_host_key(char *host, int port, char *keytype, |
40 | char *keystr, char *fingerprint) |
41 | { |
42 | int ret; |
43 | HANDLE hin; |
44 | DWORD savemode, i; |
45 | |
46 | static const char absentmsg_batch[] = |
47 | "The server's host key is not cached in the registry. You\n" |
48 | "have no guarantee that the server is the computer you\n" |
49 | "think it is.\n" |
50 | "The server's key fingerprint is:\n" |
51 | "%s\n" |
52 | "Connection abandoned.\n"; |
53 | static const char absentmsg[] = |
54 | "The server's host key is not cached in the registry. You\n" |
55 | "have no guarantee that the server is the computer you\n" |
56 | "think it is.\n" |
57 | "The server's key fingerprint is:\n" |
58 | "%s\n" |
59 | "If you trust this host, enter \"y\" to add the key to\n" |
60 | "PuTTY's cache and carry on connecting.\n" |
61 | "If you want to carry on connecting just once, without\n" |
62 | "adding the key to the cache, enter \"n\".\n" |
63 | "If you do not trust this host, press Return to abandon the\n" |
64 | "connection.\n" |
65 | "Store key in cache? (y/n) "; |
66 | |
67 | static const char wrongmsg_batch[] = |
68 | "WARNING - POTENTIAL SECURITY BREACH!\n" |
69 | "The server's host key does not match the one PuTTY has\n" |
70 | "cached in the registry. This means that either the\n" |
71 | "server administrator has changed the host key, or you\n" |
72 | "have actually connected to another computer pretending\n" |
73 | "to be the server.\n" |
74 | "The new key fingerprint is:\n" |
75 | "%s\n" |
76 | "Connection abandoned.\n"; |
77 | static const char wrongmsg[] = |
78 | "WARNING - POTENTIAL SECURITY BREACH!\n" |
79 | "The server's host key does not match the one PuTTY has\n" |
80 | "cached in the registry. This means that either the\n" |
81 | "server administrator has changed the host key, or you\n" |
82 | "have actually connected to another computer pretending\n" |
83 | "to be the server.\n" |
84 | "The new key fingerprint is:\n" |
85 | "%s\n" |
86 | "If you were expecting this change and trust the new key,\n" |
87 | "enter \"y\" to update PuTTY's cache and continue connecting.\n" |
88 | "If you want to carry on connecting but without updating\n" |
89 | "the cache, enter \"n\".\n" |
90 | "If you want to abandon the connection completely, press\n" |
91 | "Return to cancel. Pressing Return is the ONLY guaranteed\n" |
92 | "safe choice.\n" |
93 | "Update cached key? (y/n, Return cancels connection) "; |
94 | |
95 | static const char abandoned[] = "Connection abandoned.\n"; |
96 | |
97 | char line[32]; |
98 | |
99 | /* |
100 | * Verify the key against the registry. |
101 | */ |
102 | ret = verify_host_key(host, port, keytype, keystr); |
103 | |
104 | if (ret == 0) /* success - key matched OK */ |
105 | return; |
106 | |
107 | if (ret == 2) { /* key was different */ |
108 | if (console_batch_mode) { |
109 | fprintf(stderr, wrongmsg_batch, fingerprint); |
93b581bd |
110 | cleanup_exit(1); |
ff2ae367 |
111 | } |
112 | fprintf(stderr, wrongmsg, fingerprint); |
113 | fflush(stderr); |
114 | } |
115 | if (ret == 1) { /* key was absent */ |
116 | if (console_batch_mode) { |
117 | fprintf(stderr, absentmsg_batch, fingerprint); |
93b581bd |
118 | cleanup_exit(1); |
ff2ae367 |
119 | } |
120 | fprintf(stderr, absentmsg, fingerprint); |
121 | fflush(stderr); |
122 | } |
123 | |
124 | hin = GetStdHandle(STD_INPUT_HANDLE); |
125 | GetConsoleMode(hin, &savemode); |
126 | SetConsoleMode(hin, (savemode | ENABLE_ECHO_INPUT | |
127 | ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT)); |
128 | ReadFile(hin, line, sizeof(line) - 1, &i, NULL); |
129 | SetConsoleMode(hin, savemode); |
130 | |
131 | if (line[0] != '\0' && line[0] != '\r' && line[0] != '\n') { |
132 | if (line[0] == 'y' || line[0] == 'Y') |
133 | store_host_key(host, port, keytype, keystr); |
134 | } else { |
135 | fprintf(stderr, abandoned); |
93b581bd |
136 | cleanup_exit(0); |
ff2ae367 |
137 | } |
138 | } |
139 | |
140 | /* |
141 | * Ask whether the selected cipher is acceptable (since it was |
142 | * below the configured 'warn' threshold). |
143 | * cs: 0 = both ways, 1 = client->server, 2 = server->client |
144 | */ |
145 | void askcipher(char *ciphername, int cs) |
146 | { |
147 | HANDLE hin; |
148 | DWORD savemode, i; |
149 | |
150 | static const char msg[] = |
151 | "The first %scipher supported by the server is\n" |
152 | "%s, which is below the configured warning threshold.\n" |
153 | "Continue with connection? (y/n) "; |
154 | static const char msg_batch[] = |
155 | "The first %scipher supported by the server is\n" |
156 | "%s, which is below the configured warning threshold.\n" |
157 | "Connection abandoned.\n"; |
158 | static const char abandoned[] = "Connection abandoned.\n"; |
159 | |
160 | char line[32]; |
161 | |
162 | if (console_batch_mode) { |
163 | fprintf(stderr, msg_batch, |
164 | (cs == 0) ? "" : |
165 | (cs == 1) ? "client-to-server " : "server-to-client ", |
166 | ciphername); |
93b581bd |
167 | cleanup_exit(1); |
ff2ae367 |
168 | } |
169 | |
170 | fprintf(stderr, msg, |
171 | (cs == 0) ? "" : |
172 | (cs == 1) ? "client-to-server " : "server-to-client ", |
173 | ciphername); |
174 | fflush(stderr); |
175 | |
176 | hin = GetStdHandle(STD_INPUT_HANDLE); |
177 | GetConsoleMode(hin, &savemode); |
178 | SetConsoleMode(hin, (savemode | ENABLE_ECHO_INPUT | |
179 | ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT)); |
180 | ReadFile(hin, line, sizeof(line) - 1, &i, NULL); |
181 | SetConsoleMode(hin, savemode); |
182 | |
183 | if (line[0] == 'y' || line[0] == 'Y') { |
184 | return; |
185 | } else { |
186 | fprintf(stderr, abandoned); |
93b581bd |
187 | cleanup_exit(0); |
ff2ae367 |
188 | } |
189 | } |
190 | |
191 | /* |
192 | * Ask whether to wipe a session log file before writing to it. |
193 | * Returns 2 for wipe, 1 for append, 0 for cancel (don't log). |
194 | */ |
195 | int askappend(char *filename) |
196 | { |
197 | HANDLE hin; |
198 | DWORD savemode, i; |
199 | |
200 | static const char msgtemplate[] = |
201 | "The session log file \"%.*s\" already exists.\n" |
202 | "You can overwrite it with a new session log,\n" |
203 | "append your session log to the end of it,\n" |
204 | "or disable session logging for this session.\n" |
205 | "Enter \"y\" to wipe the file, \"n\" to append to it,\n" |
206 | "or just press Return to disable logging.\n" |
207 | "Wipe the log file? (y/n, Return cancels logging) "; |
208 | |
209 | static const char msgtemplate_batch[] = |
210 | "The session log file \"%.*s\" already exists.\n" |
211 | "Logging will not be enabled.\n"; |
212 | |
213 | char line[32]; |
214 | |
215 | if (console_batch_mode) { |
216 | fprintf(stderr, msgtemplate_batch, FILENAME_MAX, filename); |
217 | fflush(stderr); |
218 | return 0; |
219 | } |
220 | fprintf(stderr, msgtemplate, FILENAME_MAX, filename); |
221 | fflush(stderr); |
222 | |
223 | hin = GetStdHandle(STD_INPUT_HANDLE); |
224 | GetConsoleMode(hin, &savemode); |
225 | SetConsoleMode(hin, (savemode | ENABLE_ECHO_INPUT | |
226 | ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT)); |
227 | ReadFile(hin, line, sizeof(line) - 1, &i, NULL); |
228 | SetConsoleMode(hin, savemode); |
229 | |
230 | if (line[0] == 'y' || line[0] == 'Y') |
231 | return 2; |
232 | else if (line[0] == 'n' || line[0] == 'N') |
233 | return 1; |
234 | else |
235 | return 0; |
236 | } |
237 | |
238 | /* |
239 | * Warn about the obsolescent key file format. |
240 | */ |
241 | void old_keyfile_warning(void) |
242 | { |
243 | static const char message[] = |
244 | "You are loading an SSH 2 private key which has an\n" |
245 | "old version of the file format. This means your key\n" |
246 | "file is not fully tamperproof. Future versions of\n" |
247 | "PuTTY may stop supporting this private key format,\n" |
248 | "so we recommend you convert your key to the new\n" |
249 | "format.\n" |
250 | "\n" |
251 | "Once the key is loaded into PuTTYgen, you can perform\n" |
252 | "this conversion simply by saving it again.\n"; |
253 | |
254 | fputs(message, stderr); |
255 | } |
256 | |
257 | void logevent(char *string) |
258 | { |
259 | } |
260 | |
261 | char *console_password = NULL; |
262 | |
263 | int console_get_line(const char *prompt, char *str, |
264 | int maxlen, int is_pw) |
265 | { |
266 | HANDLE hin, hout; |
267 | DWORD savemode, newmode, i; |
268 | |
269 | if (is_pw && console_password) { |
270 | static int tried_once = 0; |
271 | |
272 | if (tried_once) { |
273 | return 0; |
274 | } else { |
275 | strncpy(str, console_password, maxlen); |
276 | str[maxlen - 1] = '\0'; |
277 | tried_once = 1; |
278 | return 1; |
279 | } |
280 | } |
281 | |
282 | if (console_batch_mode) { |
283 | if (maxlen > 0) |
284 | str[0] = '\0'; |
285 | } else { |
286 | hin = GetStdHandle(STD_INPUT_HANDLE); |
287 | hout = GetStdHandle(STD_OUTPUT_HANDLE); |
288 | if (hin == INVALID_HANDLE_VALUE || hout == INVALID_HANDLE_VALUE) { |
289 | fprintf(stderr, "Cannot get standard input/output handles\n"); |
93b581bd |
290 | cleanup_exit(1); |
ff2ae367 |
291 | } |
292 | |
293 | GetConsoleMode(hin, &savemode); |
294 | newmode = savemode | ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT; |
295 | if (is_pw) |
296 | newmode &= ~ENABLE_ECHO_INPUT; |
297 | else |
298 | newmode |= ENABLE_ECHO_INPUT; |
299 | SetConsoleMode(hin, newmode); |
300 | |
301 | WriteFile(hout, prompt, strlen(prompt), &i, NULL); |
302 | ReadFile(hin, str, maxlen - 1, &i, NULL); |
303 | |
304 | SetConsoleMode(hin, savemode); |
305 | |
306 | if ((int) i > maxlen) |
307 | i = maxlen - 1; |
308 | else |
309 | i = i - 2; |
310 | str[i] = '\0'; |
311 | |
312 | if (is_pw) |
313 | WriteFile(hout, "\r\n", 2, &i, NULL); |
314 | |
315 | } |
316 | return 1; |
317 | } |