2 * PLink - a command-line (stdin/stdout) variant of PuTTY.
13 #define PUTTY_DO_GLOBALS /* actually _define_ globals */
18 void fatalbox (char *p, ...) {
20 fprintf(stderr, "FATAL ERROR: ");
22 vfprintf(stderr, p, ap);
28 void connection_fatal (char *p, ...) {
30 fprintf(stderr, "FATAL ERROR: ");
32 vfprintf(stderr, p, ap);
39 static char *password = NULL;
41 void logevent(char *string) { }
43 void verify_ssh_host_key(char *host, int port, char *keytype,
44 char *keystr, char *fingerprint) {
49 static const char absentmsg[] =
50 "The server's host key is not cached in the registry. You\n"
51 "have no guarantee that the server is the computer you\n"
53 "The server's key fingerprint is:\n"
55 "If you trust this host, enter \"y\" to add the key to\n"
56 "PuTTY's cache and carry on connecting.\n"
57 "If you do not trust this host, enter \"n\" to abandon the\n"
59 "Continue connecting? (y/n) ";
61 static const char wrongmsg[] =
62 "WARNING - POTENTIAL SECURITY BREACH!\n"
63 "The server's host key does not match the one PuTTY has\n"
64 "cached in the registry. This means that either the\n"
65 "server administrator has changed the host key, or you\n"
66 "have actually connected to another computer pretending\n"
68 "The new key fingerprint is:\n"
70 "If you were expecting this change and trust the new key,\n"
71 "enter \"y\" to update PuTTY's cache and continue connecting.\n"
72 "If you want to carry on connecting but without updating\n"
73 "the cache, enter \"n\".\n"
74 "If you want to abandon the connection completely, press\n"
75 "Return to cancel. Pressing Return is the ONLY guaranteed\n"
77 "Update cached key? (y/n, Return cancels connection) ";
79 static const char abandoned[] = "Connection abandoned.\n";
84 * Verify the key against the registry.
86 ret = verify_host_key(host, port, keytype, keystr);
88 if (ret == 0) /* success - key matched OK */
91 if (ret == 2) /* key was different */
92 fprintf(stderr, wrongmsg, fingerprint);
93 if (ret == 1) /* key was absent */
94 fprintf(stderr, absentmsg, fingerprint);
96 hin = GetStdHandle(STD_INPUT_HANDLE);
97 GetConsoleMode(hin, &savemode);
98 SetConsoleMode(hin, (savemode | ENABLE_ECHO_INPUT |
99 ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT));
100 ReadFile(hin, line, sizeof(line)-1, &i, NULL);
101 SetConsoleMode(hin, savemode);
103 if (ret == 2) { /* key was different */
104 if (line[0] != '\0' && line[0] != '\r' && line[0] != '\n') {
105 if (line[0] == 'y' || line[0] == 'Y')
106 store_host_key(host, port, keytype, keystr);
108 fprintf(stderr, abandoned);
112 if (ret == 1) { /* key was absent */
113 if (line[0] == 'y' || line[0] == 'Y')
114 store_host_key(host, port, keytype, keystr);
116 fprintf(stderr, abandoned);
122 HANDLE inhandle, outhandle, errhandle;
123 DWORD orig_console_mode;
127 void from_backend(int is_stderr, char *data, int len) {
130 HANDLE h = (is_stderr ? errhandle : outhandle);
134 if (!WriteFile(h, data+pos, len-pos, &ret, NULL))
135 return; /* give up in panic */
140 int term_ldisc(int mode) { return FALSE; }
141 void ldisc_update(int echo, int edit) {
142 /* Update stdin read mode to reflect changes in line discipline. */
145 mode = ENABLE_PROCESSED_INPUT;
147 mode = mode | ENABLE_ECHO_INPUT;
149 mode = mode &~ ENABLE_ECHO_INPUT;
151 mode = mode | ENABLE_LINE_INPUT;
153 mode = mode &~ ENABLE_LINE_INPUT;
154 SetConsoleMode(inhandle, mode);
160 HANDLE event, eventback;
163 static int get_line(const char *prompt, char *str, int maxlen, int is_pw)
166 DWORD savemode, newmode, i;
168 if (is_pw && password) {
169 static int tried_once = 0;
174 strncpy(str, password, maxlen);
175 str[maxlen-1] = '\0';
181 hin = GetStdHandle(STD_INPUT_HANDLE);
182 hout = GetStdHandle(STD_OUTPUT_HANDLE);
183 if (hin == INVALID_HANDLE_VALUE || hout == INVALID_HANDLE_VALUE) {
184 fprintf(stderr, "Cannot get standard input/output handles");
188 GetConsoleMode(hin, &savemode);
189 newmode = savemode | ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT;
191 newmode &= ~ENABLE_ECHO_INPUT;
193 newmode |= ENABLE_ECHO_INPUT;
194 SetConsoleMode(hin, newmode);
196 WriteFile(hout, prompt, strlen(prompt), &i, NULL);
197 ReadFile(hin, str, maxlen-1, &i, NULL);
199 SetConsoleMode(hin, savemode);
201 if ((int)i > maxlen) i = maxlen-1; else i = i - 2;
205 WriteFile(hout, "\r\n", 2, &i, NULL);
210 static DWORD WINAPI stdin_read_thread(void *param) {
211 struct input_data *idata = (struct input_data *)param;
214 inhandle = GetStdHandle(STD_INPUT_HANDLE);
216 while (ReadFile(inhandle, idata->buffer, sizeof(idata->buffer),
217 &idata->len, NULL) && idata->len > 0) {
218 SetEvent(idata->event);
219 WaitForSingleObject(idata->eventback, INFINITE);
223 SetEvent(idata->event);
229 * Short description of parameters.
231 static void usage(void)
233 printf("PuTTY Link: command-line connection utility\n");
235 printf("Usage: plink [options] [user@]host [command]\n");
236 printf("Options:\n");
237 printf(" -v show verbose messages\n");
238 printf(" -ssh force use of ssh protocol\n");
239 printf(" -P port connect to specified port\n");
240 printf(" -pw passw login with specified password\n");
241 printf(" -m file read remote command(s) from file\n");
245 char *do_select(SOCKET skt, int startup) {
248 events = FD_READ | FD_WRITE | FD_OOB | FD_CLOSE;
252 if (WSAEventSelect (skt, netevent, events) == SOCKET_ERROR) {
253 switch (WSAGetLastError()) {
254 case WSAENETDOWN: return "Network is down";
255 default: return "WSAAsyncSelect(): unknown error";
261 int main(int argc, char **argv) {
267 struct input_data idata;
274 ssh_get_line = get_line;
276 sklist = NULL; skcount = sksize = 0;
280 * Process the command line.
282 do_defaults(NULL, &cfg);
283 default_protocol = cfg.protocol;
284 default_port = cfg.port;
287 * Override the default protocol if PLINK_PROTOCOL is set.
289 char *p = getenv("PLINK_PROTOCOL");
292 for (i = 0; backends[i].backend != NULL; i++) {
293 if (!strcmp(backends[i].name, p)) {
294 default_protocol = cfg.protocol = backends[i].protocol;
295 default_port = cfg.port = backends[i].backend->default_port;
304 if (!strcmp(p, "-ssh")) {
305 default_protocol = cfg.protocol = PROT_SSH;
306 default_port = cfg.port = 22;
307 } else if (!strcmp(p, "-telnet")) {
308 default_protocol = cfg.protocol = PROT_TELNET;
309 default_port = cfg.port = 23;
310 } else if (!strcmp(p, "-raw")) {
311 default_protocol = cfg.protocol = PROT_RAW;
312 } else if (!strcmp(p, "-v")) {
313 flags |= FLAG_VERBOSE;
314 } else if (!strcmp(p, "-log")) {
315 logfile = "putty.log";
316 } else if (!strcmp(p, "-pw") && argc > 1) {
317 --argc, password = *++argv;
318 } else if (!strcmp(p, "-l") && argc > 1) {
320 --argc, username = *++argv;
321 strncpy(cfg.username, username, sizeof(cfg.username));
322 cfg.username[sizeof(cfg.username)-1] = '\0';
323 } else if (!strcmp(p, "-m") && argc > 1) {
324 char *filename, *command;
329 --argc, filename = *++argv;
331 cmdlen = cmdsize = 0;
333 fp = fopen(filename, "r");
335 fprintf(stderr, "plink: unable to open command "
336 "file \"%s\"\n", filename);
344 if (cmdlen >= cmdsize) {
345 cmdsize = cmdlen + 512;
346 command = srealloc(command, cmdsize);
348 command[cmdlen++] = d;
350 cfg.remote_cmd_ptr = command;
351 cfg.nopty = TRUE; /* command => no terminal */
352 } else if (!strcmp(p, "-P") && argc > 1) {
353 --argc, portnumber = atoi(*++argv);
359 * If the hostname starts with "telnet:", set the
360 * protocol to Telnet and process the string as a
363 if (!strncmp(q, "telnet:", 7)) {
367 if (q[0] == '/' && q[1] == '/')
369 cfg.protocol = PROT_TELNET;
371 while (*p && *p != ':' && *p != '/') p++;
379 strncpy (cfg.host, q, sizeof(cfg.host)-1);
380 cfg.host[sizeof(cfg.host)-1] = '\0';
384 * Before we process the [user@]host string, we
385 * first check for the presence of a protocol
386 * prefix (a protocol name followed by ",").
391 for (i = 0; backends[i].backend != NULL; i++) {
392 j = strlen(backends[i].name);
394 !memcmp(backends[i].name, p, j)) {
395 default_protocol = cfg.protocol = backends[i].protocol;
396 portnumber = backends[i].backend->default_port;
404 * Three cases. Either (a) there's a nonzero
405 * length string followed by an @, in which
406 * case that's user and the remainder is host.
407 * Or (b) there's only one string, not counting
408 * a potential initial @, and it exists in the
409 * saved-sessions database. Or (c) only one
410 * string and it _doesn't_ exist in the
414 if (r == p) p++, r = NULL; /* discount initial @ */
420 do_defaults (p, &cfg2);
421 if (cfg2.host[0] == '\0') {
422 /* No settings for this host; use defaults */
423 strncpy(cfg.host, p, sizeof(cfg.host)-1);
424 cfg.host[sizeof(cfg.host)-1] = '\0';
425 cfg.port = default_port;
428 cfg.remote_cmd_ptr = cfg.remote_cmd;
432 strncpy(cfg.username, p, sizeof(cfg.username)-1);
433 cfg.username[sizeof(cfg.username)-1] = '\0';
434 strncpy(cfg.host, r, sizeof(cfg.host)-1);
435 cfg.host[sizeof(cfg.host)-1] = '\0';
436 cfg.port = default_port;
440 int len = sizeof(cfg.remote_cmd) - 1;
441 char *cp = cfg.remote_cmd;
444 strncpy(cp, p, len); cp[len] = '\0';
445 len2 = strlen(cp); len -= len2; cp += len2;
449 strncpy(cp, *++argv, len); cp[len] = '\0';
450 len2 = strlen(cp); len -= len2; cp += len2;
452 cfg.nopty = TRUE; /* command => no terminal */
453 break; /* done with cmdline */
462 if (!*cfg.remote_cmd_ptr)
463 flags |= FLAG_INTERACTIVE;
466 * Select protocol. This is farmed out into a table in a
467 * separate file to enable an ssh-free variant.
472 for (i = 0; backends[i].backend != NULL; i++)
473 if (backends[i].protocol == cfg.protocol) {
474 back = backends[i].backend;
478 fprintf(stderr, "Internal fault: Unsupported protocol found\n");
486 if (portnumber != -1)
487 cfg.port = portnumber;
490 * Initialise WinSock.
492 winsock_ver = MAKEWORD(2, 0);
493 if (WSAStartup(winsock_ver, &wsadata)) {
494 MessageBox(NULL, "Unable to initialise WinSock", "WinSock Error",
495 MB_OK | MB_ICONEXCLAMATION);
498 if (LOBYTE(wsadata.wVersion) != 2 || HIBYTE(wsadata.wVersion) != 0) {
499 MessageBox(NULL, "WinSock version is incompatible with 2.0",
500 "WinSock Error", MB_OK | MB_ICONEXCLAMATION);
507 * Start up the connection.
509 netevent = CreateEvent(NULL, FALSE, FALSE, NULL);
514 error = back->init (cfg.host, cfg.port, &realhost);
516 fprintf(stderr, "Unable to open connection:\n%s", error);
522 stdinevent = CreateEvent(NULL, FALSE, FALSE, NULL);
524 inhandle = GetStdHandle(STD_INPUT_HANDLE);
525 outhandle = GetStdHandle(STD_OUTPUT_HANDLE);
526 errhandle = GetStdHandle(STD_ERROR_HANDLE);
527 GetConsoleMode(inhandle, &orig_console_mode);
528 SetConsoleMode(inhandle, ENABLE_PROCESSED_INPUT);
531 * Turn off ECHO and LINE input modes. We don't care if this
532 * call fails, because we know we aren't necessarily running in
535 handles[0] = netevent;
536 handles[1] = stdinevent;
541 if (!sending && back->sendok()) {
543 * Create a separate thread to read from stdin. This is
544 * a total pain, but I can't find another way to do it:
546 * - an overlapped ReadFile or ReadFileEx just doesn't
547 * happen; we get failure from ReadFileEx, and
548 * ReadFile blocks despite being given an OVERLAPPED
549 * structure. Perhaps we can't do overlapped reads
550 * on consoles. WHY THE HELL NOT?
552 * - WaitForMultipleObjects(netevent, console) doesn't
553 * work, because it signals the console when
554 * _anything_ happens, including mouse motions and
555 * other things that don't cause data to be readable
556 * - so we're back to ReadFile blocking.
558 idata.event = stdinevent;
559 idata.eventback = CreateEvent(NULL, FALSE, FALSE, NULL);
560 if (!CreateThread(NULL, 0, stdin_read_thread,
561 &idata, 0, &threadid)) {
562 fprintf(stderr, "Unable to create second thread\n");
568 n = WaitForMultipleObjects(2, handles, FALSE, INFINITE);
570 WSANETWORKEVENTS things;
573 extern SOCKET first_socket(enum234 *), next_socket(enum234 *);
574 extern int select_result(WPARAM, LPARAM);
578 * We must not call select_result() for any socket
579 * until we have finished enumerating within the tree.
580 * This is because select_result() may close the socket
581 * and modify the tree.
583 /* Count the active sockets. */
585 for (socket = first_socket(&e); socket != INVALID_SOCKET;
586 socket = next_socket(&e))
589 /* Expand the buffer if necessary. */
592 sklist = srealloc(sklist, sksize * sizeof(*sklist));
595 /* Retrieve the sockets into sklist. */
597 for (socket = first_socket(&e); socket != INVALID_SOCKET;
598 socket = next_socket(&e)) {
599 sklist[skcount++] = socket;
602 /* Now we're done enumerating; go through the list. */
603 for (i = 0; i < skcount; i++) {
607 if (!WSAEnumNetworkEvents(socket, NULL, &things)) {
608 noise_ultralight(socket);
609 noise_ultralight(things.lNetworkEvents);
610 if (things.lNetworkEvents & FD_READ)
611 connopen &= select_result(wp, (LPARAM)FD_READ);
612 if (things.lNetworkEvents & FD_CLOSE)
613 connopen &= select_result(wp, (LPARAM)FD_CLOSE);
614 if (things.lNetworkEvents & FD_OOB)
615 connopen &= select_result(wp, (LPARAM)FD_OOB);
616 if (things.lNetworkEvents & FD_WRITE)
617 connopen &= select_result(wp, (LPARAM)FD_WRITE);
621 noise_ultralight(idata.len);
623 back->send(idata.buffer, idata.len);
625 back->special(TS_EOF);
627 SetEvent(idata.eventback);
629 if (!connopen || back->socket() == NULL)
630 break; /* we closed the connection */