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
);