console.c

   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
  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
  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);
 110             cleanup_exit(1);
 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);
 118             cleanup_exit(1);
 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);
 136         cleanup_exit(0);
 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);
 167         cleanup_exit(1);
 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);
 187         cleanup_exit(0);
 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 (cfg.logxfovr != LGXF_ASK) {
 216         return ((cfg.logxfovr == LGXF_OVR) ? 2 : 1);
 217     }
 218     if (console_batch_mode) {
 219         fprintf(stderr, msgtemplate_batch, FILENAME_MAX, filename);
 220         fflush(stderr);
 221         return 0;
 222     }
 223     fprintf(stderr, msgtemplate, FILENAME_MAX, filename);
 224     fflush(stderr);
 225
 226     hin = GetStdHandle(STD_INPUT_HANDLE);
 227     GetConsoleMode(hin, &savemode);
 228     SetConsoleMode(hin, (savemode | ENABLE_ECHO_INPUT |
 229                          ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT));
 230     ReadFile(hin, line, sizeof(line) - 1, &i, NULL);
 231     SetConsoleMode(hin, savemode);
 232
 233     if (line[0] == 'y' || line[0] == 'Y')
 234         return 2;
 235     else if (line[0] == 'n' || line[0] == 'N')
 236         return 1;
 237     else
 238         return 0;
 239 }
 240
 241 /*
 242  * Warn about the obsolescent key file format.
 243  */
 244 void old_keyfile_warning(void)
 245 {
 246     static const char message[] =
 247         "You are loading an SSH 2 private key which has an\n"
 248         "old version of the file format. This means your key\n"
 249         "file is not fully tamperproof. Future versions of\n"
 250         "PuTTY may stop supporting this private key format,\n"
 251         "so we recommend you convert your key to the new\n"
 252         "format.\n"
 253         "\n"
 254         "Once the key is loaded into PuTTYgen, you can perform\n"
 255         "this conversion simply by saving it again.\n";
 256
 257     fputs(message, stderr);
 258 }
 259
 260 void logevent(char *string)
 261 {
 262 }
 263
 264 int console_get_line(const char *prompt, char *str,
 265                             int maxlen, int is_pw)
 266 {
 267     HANDLE hin, hout;
 268     DWORD savemode, newmode, i;
 269
 270     if (console_batch_mode) {
 271         if (maxlen > 0)
 272             str[0] = '\0';
 273     } else {
 274         hin = GetStdHandle(STD_INPUT_HANDLE);
 275         hout = GetStdHandle(STD_OUTPUT_HANDLE);
 276         if (hin == INVALID_HANDLE_VALUE || hout == INVALID_HANDLE_VALUE) {
 277             fprintf(stderr, "Cannot get standard input/output handles\n");
 278             cleanup_exit(1);
 279         }
 280
 281         GetConsoleMode(hin, &savemode);
 282         newmode = savemode | ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT;
 283         if (is_pw)
 284             newmode &= ~ENABLE_ECHO_INPUT;
 285         else
 286             newmode |= ENABLE_ECHO_INPUT;
 287         SetConsoleMode(hin, newmode);
 288
 289         WriteFile(hout, prompt, strlen(prompt), &i, NULL);
 290         ReadFile(hin, str, maxlen - 1, &i, NULL);
 291
 292         SetConsoleMode(hin, savemode);
 293
 294         if ((int) i > maxlen)
 295             i = maxlen - 1;
 296         else
 297             i = i - 2;
 298         str[i] = '\0';
 299
 300         if (is_pw)
 301             WriteFile(hout, "\r\n", 2, &i, NULL);
 302
 303     }
 304     return 1;
 305 }