dfe3332b |
1 | #include <stdio.h> |
2 | #include <assert.h> |
3 | #include <stdlib.h> |
4 | #include "putty.h" |
5 | |
6 | /* |
7 | * Some command-line parameters need to be saved up until after |
8 | * we've loaded the saved session which will form the basis of our |
9 | * eventual running configuration. For this we use the macro |
10 | * SAVEABLE, which notices if the `need_save' parameter is set and |
11 | * saves the parameter and value on a list. |
12 | * |
13 | * We also assign priorities to saved parameters, just to slightly |
14 | * ameliorate silly ordering problems. For example, if you specify |
15 | * a saved session to load, it will be loaded _before_ all your |
16 | * local modifications such as -L are evaluated; and if you specify |
17 | * a protocol and a port, the protocol is set up first so that the |
18 | * port can override its choice of port number. |
c725e24c |
19 | * |
20 | * (In fact -load is not saved at all, since in at least Plink the |
21 | * processing of further command-line options depends on whether or |
22 | * not the loaded session contained a hostname. So it must be |
23 | * executed immediately.) |
dfe3332b |
24 | */ |
25 | |
c725e24c |
26 | #define NPRIORITIES 2 |
dfe3332b |
27 | |
28 | struct cmdline_saved_param { |
29 | char *p, *value; |
30 | }; |
31 | struct cmdline_saved_param_set { |
32 | struct cmdline_saved_param *params; |
33 | int nsaved, savesize; |
34 | }; |
35 | |
36 | /* |
37 | * C guarantees this structure will be initialised to all zero at |
38 | * program start, which is exactly what we want. |
39 | */ |
40 | static struct cmdline_saved_param_set saves[NPRIORITIES]; |
41 | |
42 | static void cmdline_save_param(char *p, char *value, int pri) |
43 | { |
44 | if (saves[pri].nsaved >= saves[pri].savesize) { |
45 | saves[pri].savesize = saves[pri].nsaved + 32; |
3d88e64d |
46 | saves[pri].params = sresize(saves[pri].params, saves[pri].savesize, |
47 | struct cmdline_saved_param); |
dfe3332b |
48 | } |
49 | saves[pri].params[saves[pri].nsaved].p = p; |
50 | saves[pri].params[saves[pri].nsaved].value = value; |
51 | saves[pri].nsaved++; |
52 | } |
53 | |
679539d7 |
54 | void cmdline_cleanup(void) |
55 | { |
56 | int pri; |
57 | |
58 | for (pri = 0; pri < NPRIORITIES; pri++) |
59 | sfree(saves[pri].params); |
60 | } |
61 | |
dfe3332b |
62 | #define SAVEABLE(pri) do { \ |
63 | if (need_save) { cmdline_save_param(p, value, pri); return ret; } \ |
64 | } while (0) |
65 | |
66 | char *cmdline_password = NULL; |
67 | |
68 | static int cmdline_get_line(const char *prompt, char *str, |
69 | int maxlen, int is_pw) |
70 | { |
71 | static int tried_once = 0; |
72 | |
73 | assert(is_pw && cmdline_password); |
74 | |
75 | if (tried_once) { |
76 | return 0; |
77 | } else { |
78 | strncpy(str, cmdline_password, maxlen); |
79 | str[maxlen - 1] = '\0'; |
80 | tried_once = 1; |
81 | return 1; |
82 | } |
83 | } |
84 | |
85 | /* |
86 | * Here we have a flags word which describes the capabilities of |
87 | * the particular tool on whose behalf we're running. We will |
88 | * refuse certain command-line options if a particular tool |
89 | * inherently can't do anything sensible. For example, the file |
90 | * transfer tools (psftp, pscp) can't do a great deal with protocol |
91 | * selections (ever tried running scp over telnet?) or with port |
92 | * forwarding (even if it wasn't a hideously bad idea, they don't |
93 | * have the select() infrastructure to make them work). |
94 | */ |
95 | int cmdline_tooltype = 0; |
96 | |
97 | static int cmdline_check_unavailable(int flag, char *p) |
98 | { |
99 | if (cmdline_tooltype & flag) { |
100 | cmdline_error("option \"%s\" not available in this tool", p); |
101 | return 1; |
102 | } |
103 | return 0; |
104 | } |
105 | |
106 | #define UNAVAILABLE_IN(flag) do { \ |
107 | if (cmdline_check_unavailable(flag, p)) return ret; \ |
108 | } while (0) |
109 | |
110 | /* |
111 | * Process a standard command-line parameter. `p' is the parameter |
112 | * in question; `value' is the subsequent element of argv, which |
113 | * may or may not be required as an operand to the parameter. |
b552f500 |
114 | * If `need_save' is 1, arguments which need to be saved as |
115 | * described at this top of this file are, for later execution; |
116 | * if 0, they are processed normally. (-1 is a special value used |
117 | * by pterm to count arguments for a preliminary pass through the |
118 | * argument list; it causes immediate return with an appropriate |
119 | * value with no action taken.) |
dfe3332b |
120 | * Return value is 2 if both arguments were used; 1 if only p was |
121 | * used; 0 if the parameter wasn't one we recognised; -2 if it |
122 | * should have been 2 but value was NULL. |
123 | */ |
124 | |
125 | #define RETURN(x) do { \ |
126 | if ((x) == 2 && !value) return -2; \ |
127 | ret = x; \ |
46a3419b |
128 | if (need_save < 0) return x; \ |
dfe3332b |
129 | } while (0) |
130 | |
5555d393 |
131 | int cmdline_process_param(char *p, char *value, int need_save, Config *cfg) |
dfe3332b |
132 | { |
133 | int ret = 0; |
134 | |
135 | if (!strcmp(p, "-load")) { |
136 | RETURN(2); |
c725e24c |
137 | /* This parameter must be processed immediately rather than being |
138 | * saved. */ |
5555d393 |
139 | do_defaults(value, cfg); |
18e62ad8 |
140 | loaded_session = TRUE; |
dfe3332b |
141 | return 2; |
142 | } |
143 | if (!strcmp(p, "-ssh")) { |
144 | RETURN(1); |
46a3419b |
145 | UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK); |
c725e24c |
146 | SAVEABLE(0); |
5555d393 |
147 | default_protocol = cfg->protocol = PROT_SSH; |
148 | default_port = cfg->port = 22; |
dfe3332b |
149 | return 1; |
150 | } |
151 | if (!strcmp(p, "-telnet")) { |
152 | RETURN(1); |
46a3419b |
153 | UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK); |
c725e24c |
154 | SAVEABLE(0); |
5555d393 |
155 | default_protocol = cfg->protocol = PROT_TELNET; |
156 | default_port = cfg->port = 23; |
dfe3332b |
157 | return 1; |
158 | } |
159 | if (!strcmp(p, "-rlogin")) { |
160 | RETURN(1); |
46a3419b |
161 | UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK); |
c725e24c |
162 | SAVEABLE(0); |
5555d393 |
163 | default_protocol = cfg->protocol = PROT_RLOGIN; |
164 | default_port = cfg->port = 513; |
dfe3332b |
165 | return 1; |
166 | } |
167 | if (!strcmp(p, "-raw")) { |
168 | RETURN(1); |
46a3419b |
169 | UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK); |
c725e24c |
170 | SAVEABLE(0); |
5555d393 |
171 | default_protocol = cfg->protocol = PROT_RAW; |
dfe3332b |
172 | } |
173 | if (!strcmp(p, "-v")) { |
174 | RETURN(1); |
175 | flags |= FLAG_VERBOSE; |
176 | } |
177 | if (!strcmp(p, "-l")) { |
178 | RETURN(2); |
46a3419b |
179 | UNAVAILABLE_IN(TOOLTYPE_NONNETWORK); |
c725e24c |
180 | SAVEABLE(0); |
5555d393 |
181 | strncpy(cfg->username, value, sizeof(cfg->username)); |
182 | cfg->username[sizeof(cfg->username) - 1] = '\0'; |
dfe3332b |
183 | } |
820ebe3b |
184 | if ((!strcmp(p, "-L") || !strcmp(p, "-R") || !strcmp(p, "-D"))) { |
6ee9b735 |
185 | char *fwd, *ptr, *q, *qq; |
820ebe3b |
186 | int dynamic, i=0; |
dfe3332b |
187 | RETURN(2); |
46a3419b |
188 | UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK); |
c725e24c |
189 | SAVEABLE(0); |
820ebe3b |
190 | dynamic = !strcmp(p, "-D"); |
dfe3332b |
191 | fwd = value; |
5555d393 |
192 | ptr = cfg->portfwd; |
116f2044 |
193 | /* if existing forwards, find end of list */ |
194 | while (*ptr) { |
195 | while (*ptr) |
196 | ptr++; |
197 | ptr++; |
dfe3332b |
198 | } |
116f2044 |
199 | i = ptr - cfg->portfwd; |
820ebe3b |
200 | ptr[0] = p[1]; /* insert a 'L', 'R' or 'D' at the start */ |
116f2044 |
201 | ptr++; |
202 | if (1 + strlen(fwd) + 2 > sizeof(cfg->portfwd) - i) { |
dfe3332b |
203 | cmdline_error("out of space for port forwardings"); |
204 | return ret; |
205 | } |
116f2044 |
206 | strncpy(ptr, fwd, sizeof(cfg->portfwd) - i - 2); |
820ebe3b |
207 | if (!dynamic) { |
208 | /* |
209 | * We expect _at least_ two colons in this string. The |
210 | * possible formats are `sourceport:desthost:destport', |
211 | * or `sourceip:sourceport:desthost:destport' if you're |
212 | * specifying a particular loopback address. We need to |
213 | * replace the one between source and dest with a \t; |
214 | * this means we must find the second-to-last colon in |
215 | * the string. |
216 | */ |
217 | q = qq = strchr(ptr, ':'); |
218 | while (qq) { |
219 | char *qqq = strchr(qq+1, ':'); |
220 | if (qqq) |
221 | q = qq; |
222 | qq = qqq; |
223 | } |
224 | if (q) *q = '\t'; /* replace second-last colon with \t */ |
6ee9b735 |
225 | } |
5555d393 |
226 | cfg->portfwd[sizeof(cfg->portfwd) - 1] = '\0'; |
227 | cfg->portfwd[sizeof(cfg->portfwd) - 2] = '\0'; |
116f2044 |
228 | ptr[strlen(ptr)+1] = '\000'; /* append 2nd '\000' */ |
dfe3332b |
229 | } |
230 | if (!strcmp(p, "-m")) { |
231 | char *filename, *command; |
232 | int cmdlen, cmdsize; |
233 | FILE *fp; |
234 | int c, d; |
235 | |
236 | RETURN(2); |
46a3419b |
237 | UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK); |
c725e24c |
238 | SAVEABLE(0); |
dfe3332b |
239 | |
240 | filename = value; |
241 | |
242 | cmdlen = cmdsize = 0; |
243 | command = NULL; |
244 | fp = fopen(filename, "r"); |
245 | if (!fp) { |
246 | cmdline_error("unable to open command " |
247 | "file \"%s\"", filename); |
248 | return ret; |
249 | } |
250 | do { |
251 | c = fgetc(fp); |
252 | d = c; |
253 | if (c == EOF) |
254 | d = 0; |
255 | if (cmdlen >= cmdsize) { |
256 | cmdsize = cmdlen + 512; |
3d88e64d |
257 | command = sresize(command, cmdsize, char); |
dfe3332b |
258 | } |
259 | command[cmdlen++] = d; |
260 | } while (c != EOF); |
5555d393 |
261 | cfg->remote_cmd_ptr = command; |
262 | cfg->remote_cmd_ptr2 = NULL; |
263 | cfg->nopty = TRUE; /* command => no terminal */ |
dfe3332b |
264 | } |
265 | if (!strcmp(p, "-P")) { |
266 | RETURN(2); |
46a3419b |
267 | UNAVAILABLE_IN(TOOLTYPE_NONNETWORK); |
c725e24c |
268 | SAVEABLE(1); /* lower priority than -ssh,-telnet */ |
5555d393 |
269 | cfg->port = atoi(value); |
dfe3332b |
270 | } |
271 | if (!strcmp(p, "-pw")) { |
272 | RETURN(2); |
46a3419b |
273 | UNAVAILABLE_IN(TOOLTYPE_NONNETWORK); |
dfe3332b |
274 | cmdline_password = value; |
275 | ssh_get_line = cmdline_get_line; |
276 | ssh_getline_pw_only = TRUE; |
277 | } |
278 | |
279 | if (!strcmp(p, "-A")) { |
280 | RETURN(1); |
46a3419b |
281 | UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK); |
c725e24c |
282 | SAVEABLE(0); |
5555d393 |
283 | cfg->agentfwd = 1; |
dfe3332b |
284 | } |
285 | if (!strcmp(p, "-a")) { |
286 | RETURN(1); |
46a3419b |
287 | UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK); |
c725e24c |
288 | SAVEABLE(0); |
5555d393 |
289 | cfg->agentfwd = 0; |
dfe3332b |
290 | } |
291 | |
292 | if (!strcmp(p, "-X")) { |
293 | RETURN(1); |
46a3419b |
294 | UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK); |
c725e24c |
295 | SAVEABLE(0); |
5555d393 |
296 | cfg->x11_forward = 1; |
dfe3332b |
297 | } |
298 | if (!strcmp(p, "-x")) { |
299 | RETURN(1); |
46a3419b |
300 | UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK); |
c725e24c |
301 | SAVEABLE(0); |
5555d393 |
302 | cfg->x11_forward = 0; |
dfe3332b |
303 | } |
304 | |
305 | if (!strcmp(p, "-t")) { |
306 | RETURN(1); |
46a3419b |
307 | UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK); |
c725e24c |
308 | SAVEABLE(0); |
5555d393 |
309 | cfg->nopty = 0; |
dfe3332b |
310 | } |
311 | if (!strcmp(p, "-T")) { |
312 | RETURN(1); |
46a3419b |
313 | UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK); |
c725e24c |
314 | SAVEABLE(0); |
5555d393 |
315 | cfg->nopty = 1; |
dfe3332b |
316 | } |
317 | |
0ed48730 |
318 | if (!strcmp(p, "-N")) { |
319 | RETURN(1); |
320 | UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK); |
321 | SAVEABLE(0); |
322 | cfg->ssh_no_shell = 1; |
323 | } |
324 | |
dfe3332b |
325 | if (!strcmp(p, "-C")) { |
326 | RETURN(1); |
46a3419b |
327 | UNAVAILABLE_IN(TOOLTYPE_NONNETWORK); |
c725e24c |
328 | SAVEABLE(0); |
5555d393 |
329 | cfg->compression = 1; |
dfe3332b |
330 | } |
331 | |
332 | if (!strcmp(p, "-1")) { |
333 | RETURN(1); |
46a3419b |
334 | UNAVAILABLE_IN(TOOLTYPE_NONNETWORK); |
c725e24c |
335 | SAVEABLE(0); |
5555d393 |
336 | cfg->sshprot = 0; /* ssh protocol 1 only */ |
dfe3332b |
337 | } |
338 | if (!strcmp(p, "-2")) { |
339 | RETURN(1); |
46a3419b |
340 | UNAVAILABLE_IN(TOOLTYPE_NONNETWORK); |
c725e24c |
341 | SAVEABLE(0); |
5555d393 |
342 | cfg->sshprot = 3; /* ssh protocol 2 only */ |
dfe3332b |
343 | } |
344 | |
345 | if (!strcmp(p, "-i")) { |
346 | RETURN(2); |
46a3419b |
347 | UNAVAILABLE_IN(TOOLTYPE_NONNETWORK); |
c725e24c |
348 | SAVEABLE(0); |
9a30e26b |
349 | cfg->keyfile = filename_from_str(value); |
dfe3332b |
350 | } |
351 | |
05581745 |
352 | if (!strcmp(p, "-4") || !strcmp(p, "-ipv4")) { |
353 | RETURN(1); |
354 | SAVEABLE(1); |
355 | cfg->addressfamily = ADDRTYPE_IPV4; |
356 | } |
357 | if (!strcmp(p, "-6") || !strcmp(p, "-ipv6")) { |
358 | RETURN(1); |
359 | SAVEABLE(1); |
360 | cfg->addressfamily = ADDRTYPE_IPV6; |
361 | } |
362 | |
dfe3332b |
363 | return ret; /* unrecognised */ |
364 | } |
365 | |
5555d393 |
366 | void cmdline_run_saved(Config *cfg) |
dfe3332b |
367 | { |
368 | int pri, i; |
369 | for (pri = 0; pri < NPRIORITIES; pri++) |
370 | for (i = 0; i < saves[pri].nsaved; i++) |
371 | cmdline_process_param(saves[pri].params[i].p, |
5555d393 |
372 | saves[pri].params[i].value, 0, cfg); |
dfe3332b |
373 | } |