2 * PLink - a command-line (stdin/stdout) variant of PuTTY.
12 #define PUTTY_DO_GLOBALS /* actually _define_ globals */
15 void fatalbox (char *p, ...) {
17 fprintf(stderr, "FATAL ERROR: ", p);
19 vfprintf(stderr, p, ap);
25 void connection_fatal (char *p, ...) {
27 fprintf(stderr, "FATAL ERROR: ", p);
29 vfprintf(stderr, p, ap);
36 static char *password = NULL;
39 * Stubs for linking with other modules.
41 void write_clip (void *data, int len) { }
42 void term_deselect(void) { }
45 DWORD orig_console_mode;
47 void begin_session(void) {
49 SetConsoleMode(GetStdHandle(STD_INPUT_HANDLE), ENABLE_PROCESSED_INPUT);
51 SetConsoleMode(GetStdHandle(STD_INPUT_HANDLE), orig_console_mode);
59 while (reap < inbuf_head) {
60 if (!WriteFile(outhandle, inbuf+reap, inbuf_head-reap, &ret, NULL))
61 return; /* give up in panic */
73 static int get_password(const char *prompt, char *str, int maxlen)
79 static int tried_once = 0;
84 strncpy(str, password, maxlen);
91 hin = GetStdHandle(STD_INPUT_HANDLE);
92 hout = GetStdHandle(STD_OUTPUT_HANDLE);
93 if (hin == INVALID_HANDLE_VALUE || hout == INVALID_HANDLE_VALUE) {
94 fprintf(stderr, "Cannot get standard input/output handles");
98 GetConsoleMode(hin, &savemode);
99 SetConsoleMode(hin, (savemode & (~ENABLE_ECHO_INPUT)) |
100 ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT);
102 WriteFile(hout, prompt, strlen(prompt), &i, NULL);
103 ReadFile(hin, str, maxlen-1, &i, NULL);
105 SetConsoleMode(hin, savemode);
107 if ((int)i > maxlen) i = maxlen-1; else i = i - 2;
110 WriteFile(hout, "\r\n", 2, &i, NULL);
115 int WINAPI stdin_read_thread(void *param) {
116 struct input_data *idata = (struct input_data *)param;
119 inhandle = GetStdHandle(STD_INPUT_HANDLE);
121 while (ReadFile(inhandle, idata->buffer, sizeof(idata->buffer),
122 &idata->len, NULL)) {
123 SetEvent(idata->event);
127 SetEvent(idata->event);
133 * Short description of parameters.
135 static void usage(void)
137 printf("PuTTY Link: command-line connection utility\n");
139 printf("Usage: plink [options] [user@]host [command]\n");
140 printf("Options:\n");
141 printf(" -v show verbose messages\n");
142 printf(" -ssh force use of ssh protocol\n");
143 printf(" -P port connect to specified port\n");
144 printf(" -pw passw login with specified password\n");
148 int main(int argc, char **argv) {
151 WSAEVENT netevent, stdinevent;
155 struct input_data idata;
159 ssh_get_password = get_password;
163 * Process the command line.
166 default_protocol = cfg.protocol;
167 default_port = cfg.port;
170 * Override the default protocol if PLINK_PROTOCOL is set.
172 char *p = getenv("PLINK_PROTOCOL");
175 for (i = 0; backends[i].backend != NULL; i++) {
176 if (!strcmp(backends[i].name, p)) {
177 default_protocol = cfg.protocol = backends[i].protocol;
178 default_port = cfg.port = backends[i].backend->default_port;
187 if (!strcmp(p, "-ssh")) {
188 default_protocol = cfg.protocol = PROT_SSH;
189 default_port = cfg.port = 22;
190 } else if (!strcmp(p, "-telnet")) {
191 default_protocol = cfg.protocol = PROT_TELNET;
192 default_port = cfg.port = 23;
193 } else if (!strcmp(p, "-raw")) {
194 default_protocol = cfg.protocol = PROT_RAW;
195 } else if (!strcmp(p, "-v")) {
196 flags |= FLAG_VERBOSE;
197 } else if (!strcmp(p, "-log")) {
198 logfile = "putty.log";
199 } else if (!strcmp(p, "-pw") && argc > 1) {
200 --argc, password = *++argv;
201 } else if (!strcmp(p, "-l") && argc > 1) {
203 --argc, username = *++argv;
204 strncpy(cfg.username, username, sizeof(cfg.username));
205 cfg.username[sizeof(cfg.username)-1] = '\0';
206 } else if (!strcmp(p, "-P") && argc > 1) {
207 --argc, portnumber = atoi(*++argv);
213 * If the hostname starts with "telnet:", set the
214 * protocol to Telnet and process the string as a
217 if (!strncmp(q, "telnet:", 7)) {
221 if (q[0] == '/' && q[1] == '/')
223 cfg.protocol = PROT_TELNET;
225 while (*p && *p != ':' && *p != '/') p++;
233 strncpy (cfg.host, q, sizeof(cfg.host)-1);
234 cfg.host[sizeof(cfg.host)-1] = '\0';
238 * Before we process the [user@]host string, we
239 * first check for the presence of a protocol
240 * prefix (a protocol name followed by ",").
245 for (i = 0; backends[i].backend != NULL; i++) {
246 j = strlen(backends[i].name);
248 !memcmp(backends[i].name, p, j)) {
249 default_protocol = cfg.protocol = backends[i].protocol;
250 portnumber = backends[i].backend->default_port;
258 * Three cases. Either (a) there's a nonzero
259 * length string followed by an @, in which
260 * case that's user and the remainder is host.
261 * Or (b) there's only one string, not counting
262 * a potential initial @, and it exists in the
263 * saved-sessions database. Or (c) only one
264 * string and it _doesn't_ exist in the
268 if (r == p) p++, r = NULL; /* discount initial @ */
274 if (cfg.host[0] == '\0') {
275 /* No settings for this host; use defaults */
276 strncpy(cfg.host, p, sizeof(cfg.host)-1);
277 cfg.host[sizeof(cfg.host)-1] = '\0';
282 strncpy(cfg.username, p, sizeof(cfg.username)-1);
283 cfg.username[sizeof(cfg.username)-1] = '\0';
284 strncpy(cfg.host, r, sizeof(cfg.host)-1);
285 cfg.host[sizeof(cfg.host)-1] = '\0';
290 int len = sizeof(cfg.remote_cmd) - 1;
291 char *cp = cfg.remote_cmd;
294 strncpy(cp, p, len); cp[len] = '\0';
295 len2 = strlen(cp); len -= len2; cp += len2;
299 strncpy(cp, *++argv, len); cp[len] = '\0';
300 len2 = strlen(cp); len -= len2; cp += len2;
302 cfg.nopty = TRUE; /* command => no terminal */
303 cfg.ldisc_term = TRUE; /* use stdin like a line buffer */
304 break; /* done with cmdline */
313 if (!*cfg.remote_cmd)
314 flags |= FLAG_INTERACTIVE;
317 * Select protocol. This is farmed out into a table in a
318 * separate file to enable an ssh-free variant.
323 for (i = 0; backends[i].backend != NULL; i++)
324 if (backends[i].protocol == cfg.protocol) {
325 back = backends[i].backend;
329 fprintf(stderr, "Internal fault: Unsupported protocol found\n");
337 if (portnumber != -1)
338 cfg.port = portnumber;
341 * Initialise WinSock.
343 winsock_ver = MAKEWORD(2, 0);
344 if (WSAStartup(winsock_ver, &wsadata)) {
345 MessageBox(NULL, "Unable to initialise WinSock", "WinSock Error",
346 MB_OK | MB_ICONEXCLAMATION);
349 if (LOBYTE(wsadata.wVersion) != 2 || HIBYTE(wsadata.wVersion) != 0) {
350 MessageBox(NULL, "WinSock version is incompatible with 2.0",
351 "WinSock Error", MB_OK | MB_ICONEXCLAMATION);
357 * Start up the connection.
363 error = back->init (NULL, cfg.host, cfg.port, &realhost);
365 fprintf(stderr, "Unable to open connection:\n%s", error);
370 netevent = CreateEvent(NULL, FALSE, FALSE, NULL);
371 stdinevent = CreateEvent(NULL, FALSE, FALSE, NULL);
373 GetConsoleMode(GetStdHandle(STD_INPUT_HANDLE), &orig_console_mode);
374 SetConsoleMode(GetStdHandle(STD_INPUT_HANDLE), ENABLE_PROCESSED_INPUT);
375 outhandle = GetStdHandle(STD_OUTPUT_HANDLE);
378 * Now we must send the back end oodles of stuff.
380 socket = back->socket();
382 * Turn off ECHO and LINE input modes. We don't care if this
383 * call fails, because we know we aren't necessarily running in
386 WSAEventSelect(socket, netevent, FD_READ | FD_CLOSE);
387 handles[0] = netevent;
388 handles[1] = stdinevent;
393 if (!sending && back->sendok()) {
395 * Create a separate thread to read from stdin. This is
396 * a total pain, but I can't find another way to do it:
398 * - an overlapped ReadFile or ReadFileEx just doesn't
399 * happen; we get failure from ReadFileEx, and
400 * ReadFile blocks despite being given an OVERLAPPED
401 * structure. Perhaps we can't do overlapped reads
402 * on consoles. WHY THE HELL NOT?
404 * - WaitForMultipleObjects(netevent, console) doesn't
405 * work, because it signals the console when
406 * _anything_ happens, including mouse motions and
407 * other things that don't cause data to be readable
408 * - so we're back to ReadFile blocking.
410 idata.event = stdinevent;
411 if (!CreateThread(NULL, 0, stdin_read_thread,
412 &idata, 0, &threadid)) {
413 fprintf(stderr, "Unable to create second thread\n");
419 n = WaitForMultipleObjects(2, handles, FALSE, INFINITE);
421 WSANETWORKEVENTS things;
422 if (!WSAEnumNetworkEvents(socket, netevent, &things)) {
423 if (things.lNetworkEvents & FD_READ)
424 back->msg(0, FD_READ);
425 if (things.lNetworkEvents & FD_CLOSE) {
426 back->msg(0, FD_CLOSE);
433 back->send(idata.buffer, idata.len);
435 back->special(TS_EOF);