[u/mdw/putty] / cmdline.c

#include <windows.h>\r
#include <stdio.h>\r
#include <assert.h>\r
#include <stdlib.h>\r
#include "putty.h"\r
\r
/*\r
 * Some command-line parameters need to be saved up until after\r
 * we've loaded the saved session which will form the basis of our\r
 * eventual running configuration. For this we use the macro\r
 * SAVEABLE, which notices if the `need_save' parameter is set and\r
 * saves the parameter and value on a list.\r
 * \r
 * We also assign priorities to saved parameters, just to slightly\r
 * ameliorate silly ordering problems. For example, if you specify\r
 * a saved session to load, it will be loaded _before_ all your\r
 * local modifications such as -L are evaluated; and if you specify\r
 * a protocol and a port, the protocol is set up first so that the\r
 * port can override its choice of port number.\r
 */\r
\r
#define NPRIORITIES 3\r
\r
struct cmdline_saved_param {\r
    char *p, *value;\r
};\r
struct cmdline_saved_param_set {\r
    struct cmdline_saved_param *params;\r
    int nsaved, savesize;\r
};\r
\r
/*\r
 * C guarantees this structure will be initialised to all zero at\r
 * program start, which is exactly what we want.\r
 */\r
static struct cmdline_saved_param_set saves[NPRIORITIES];\r
\r
static void cmdline_save_param(char *p, char *value, int pri)\r
{\r
    if (saves[pri].nsaved >= saves[pri].savesize) {\r
	saves[pri].savesize = saves[pri].nsaved + 32;\r
	saves[pri].params =\r
	    srealloc(saves[pri].params,\r
		     saves[pri].savesize*sizeof(*saves[pri].params));\r
    }\r
    saves[pri].params[saves[pri].nsaved].p = p;\r
    saves[pri].params[saves[pri].nsaved].value = value;\r
    saves[pri].nsaved++;\r
}\r
\r
#define SAVEABLE(pri) do { \\r
    if (need_save) { cmdline_save_param(p, value, pri); return ret; } \\r
} while (0)\r
\r
char *cmdline_password = NULL;\r
\r
static int cmdline_get_line(const char *prompt, char *str,\r
                            int maxlen, int is_pw)\r
{\r
    static int tried_once = 0;\r
\r
    assert(is_pw && cmdline_password);\r
\r
    if (tried_once) {\r
	return 0;\r
    } else {\r
	strncpy(str, cmdline_password, maxlen);\r
	str[maxlen - 1] = '\0';\r
	tried_once = 1;\r
	return 1;\r
    }\r
}\r
\r
/*\r
 * Here we have a flags word which describes the capabilities of\r
 * the particular tool on whose behalf we're running. We will\r
 * refuse certain command-line options if a particular tool\r
 * inherently can't do anything sensible. For example, the file\r
 * transfer tools (psftp, pscp) can't do a great deal with protocol\r
 * selections (ever tried running scp over telnet?) or with port\r
 * forwarding (even if it wasn't a hideously bad idea, they don't\r
 * have the select() infrastructure to make them work).\r
 */\r
int cmdline_tooltype = 0;\r
\r
static int cmdline_check_unavailable(int flag, char *p)\r
{\r
    if (cmdline_tooltype & flag) {\r
	cmdline_error("option \"%s\" not available in this tool", p);\r
	return 1;\r
    }\r
    return 0;\r
}\r
\r
#define UNAVAILABLE_IN(flag) do { \\r
    if (cmdline_check_unavailable(flag, p)) return ret; \\r
} while (0)\r
\r
/*\r
 * Process a standard command-line parameter. `p' is the parameter\r
 * in question; `value' is the subsequent element of argv, which\r
 * may or may not be required as an operand to the parameter.\r
 * Return value is 2 if both arguments were used; 1 if only p was\r
 * used; 0 if the parameter wasn't one we recognised; -2 if it\r
 * should have been 2 but value was NULL.\r
 */\r
\r
#define RETURN(x) do { \\r
    if ((x) == 2 && !value) return -2; \\r
    ret = x; \\r
} while (0)\r
\r
int cmdline_process_param(char *p, char *value, int need_save)\r
{\r
    int ret = 0;\r
\r
    if (!strcmp(p, "-load")) {\r
	RETURN(2);\r
	SAVEABLE(0);		       /* very high priority */\r
	do_defaults(value, &cfg);\r
	return 2;\r
    }\r
    if (!strcmp(p, "-ssh")) {\r
	RETURN(1);\r
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
	SAVEABLE(1);\r
	default_protocol = cfg.protocol = PROT_SSH;\r
	default_port = cfg.port = 22;\r
	return 1;\r
    }\r
    if (!strcmp(p, "-telnet")) {\r
	RETURN(1);\r
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
	SAVEABLE(1);\r
	default_protocol = cfg.protocol = PROT_TELNET;\r
	default_port = cfg.port = 23;\r
	return 1;\r
    }\r
    if (!strcmp(p, "-rlogin")) {\r
	RETURN(1);\r
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
	SAVEABLE(1);\r
	default_protocol = cfg.protocol = PROT_RLOGIN;\r
	default_port = cfg.port = 513;\r
	return 1;\r
    }\r
    if (!strcmp(p, "-raw")) {\r
	RETURN(1);\r
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
	SAVEABLE(1);\r
	default_protocol = cfg.protocol = PROT_RAW;\r
    }\r
    if (!strcmp(p, "-v")) {\r
	RETURN(1);\r
	flags |= FLAG_VERBOSE;\r
    }\r
    if (!strcmp(p, "-l")) {\r
	RETURN(2);\r
	SAVEABLE(1);\r
	strncpy(cfg.username, value, sizeof(cfg.username));\r
	cfg.username[sizeof(cfg.username) - 1] = '\0';\r
    }\r
    if ((!strcmp(p, "-L") || !strcmp(p, "-R"))) {\r
	char *fwd, *ptr, *q;\r
	int i=0;\r
	RETURN(2);\r
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
	SAVEABLE(1);\r
	fwd = value;\r
	ptr = cfg.portfwd;\r
	/* if multiple forwards, find end of list */\r
	if (ptr[0]=='R' || ptr[0]=='L') {\r
	    for (i = 0; i < sizeof(cfg.portfwd) - 2; i++)\r
		if (ptr[i]=='\000' && ptr[i+1]=='\000')\r
		    break;\r
	    ptr = ptr + i + 1;  /* point to next forward slot */\r
	}\r
	ptr[0] = p[1];  /* insert a 'L' or 'R' at the start */\r
	if (strlen(fwd) > sizeof(cfg.portfwd) - i - 2) {\r
	    cmdline_error("out of space for port forwardings");\r
	    return ret;\r
	}\r
	strncpy(ptr+1, fwd, sizeof(cfg.portfwd) - i);\r
	q = strchr(ptr, ':');\r
	if (q) *q = '\t';      /* replace first : with \t */\r
	cfg.portfwd[sizeof(cfg.portfwd) - 1] = '\0';\r
	cfg.portfwd[sizeof(cfg.portfwd) - 2] = '\0';\r
	ptr[strlen(ptr)+1] = '\000';    /* append two '\000' */\r
    }\r
    if (!strcmp(p, "-m")) {\r
	char *filename, *command;\r
	int cmdlen, cmdsize;\r
	FILE *fp;\r
	int c, d;\r
\r
	RETURN(2);\r
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
	SAVEABLE(1);\r
\r
	filename = value;\r
\r
	cmdlen = cmdsize = 0;\r
	command = NULL;\r
	fp = fopen(filename, "r");\r
	if (!fp) {\r
	    cmdline_error("unable to open command "\r
			  "file \"%s\"", filename);\r
	    return ret;\r
	}\r
	do {\r
	    c = fgetc(fp);\r
	    d = c;\r
	    if (c == EOF)\r
		d = 0;\r
	    if (cmdlen >= cmdsize) {\r
		cmdsize = cmdlen + 512;\r
		command = srealloc(command, cmdsize);\r
	    }\r
	    command[cmdlen++] = d;\r
	} while (c != EOF);\r
	cfg.remote_cmd_ptr = command;\r
	cfg.remote_cmd_ptr2 = NULL;\r
	cfg.nopty = TRUE;      /* command => no terminal */\r
    }\r
    if (!strcmp(p, "-P")) {\r
	RETURN(2);\r
	SAVEABLE(2);		       /* lower priority than -ssh,-telnet */\r
	cfg.port = atoi(value);\r
    }\r
    if (!strcmp(p, "-pw")) {\r
	RETURN(2);\r
	cmdline_password = value;\r
	ssh_get_line = cmdline_get_line;\r
	ssh_getline_pw_only = TRUE;\r
    }\r
\r
    if (!strcmp(p, "-A")) {\r
	RETURN(1);\r
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
	SAVEABLE(1);\r
	cfg.agentfwd = 1;\r
    }\r
    if (!strcmp(p, "-a")) {\r
	RETURN(1);\r
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
	SAVEABLE(1);\r
	cfg.agentfwd = 0;\r
    }\r
\r
    if (!strcmp(p, "-X")) {\r
	RETURN(1);\r
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
	SAVEABLE(1);\r
	cfg.x11_forward = 1;\r
    }\r
    if (!strcmp(p, "-x")) {\r
	RETURN(1);\r
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
	SAVEABLE(1);\r
	cfg.x11_forward = 0;\r
    }\r
\r
    if (!strcmp(p, "-t")) {\r
	RETURN(1);\r
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
	SAVEABLE(1);\r
	cfg.nopty = 0;\r
    }\r
    if (!strcmp(p, "-T")) {\r
	RETURN(1);\r
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
	SAVEABLE(1);\r
	cfg.nopty = 1;\r
    }\r
\r
    if (!strcmp(p, "-C")) {\r
	RETURN(1);\r
	SAVEABLE(1);\r
	cfg.compression = 1;\r
    }\r
\r
    if (!strcmp(p, "-1")) {\r
	RETURN(1);\r
	SAVEABLE(1);\r
	cfg.sshprot = 0;	       /* ssh protocol 1 only */\r
    }\r
    if (!strcmp(p, "-2")) {\r
	RETURN(1);\r
	SAVEABLE(1);\r
	cfg.sshprot = 3;	       /* ssh protocol 2 only */\r
    }\r
\r
    if (!strcmp(p, "-i")) {\r
	RETURN(2);\r
	SAVEABLE(1);\r
	strncpy(cfg.keyfile, value, sizeof(cfg.keyfile));\r
	cfg.keyfile[sizeof(cfg.keyfile)-1] = '\0';\r
    }\r
\r
    return ret;			       /* unrecognised */\r
}\r
\r
void cmdline_run_saved(void)\r
{\r
    int pri, i;\r
    for (pri = 0; pri < NPRIORITIES; pri++)\r
	for (i = 0; i < saves[pri].nsaved; i++)\r
	    cmdline_process_param(saves[pri].params[i].p,\r
				  saves[pri].params[i].value, 0);\r
}\r
Commit	Line	Data
c0a81592	1	#include <windows.h>\r
	2	#include <stdio.h>\r
	3	#include <assert.h>\r
	4	#include <stdlib.h>\r
	5	#include "putty.h"\r
	6	\r
	7	/*\r
	8	* Some command-line parameters need to be saved up until after\r
	9	* we've loaded the saved session which will form the basis of our\r
	10	* eventual running configuration. For this we use the macro\r
	11	* SAVEABLE, which notices if the `need_save' parameter is set and\r
	12	* saves the parameter and value on a list.\r
	13	* \r
	14	* We also assign priorities to saved parameters, just to slightly\r
	15	* ameliorate silly ordering problems. For example, if you specify\r
	16	* a saved session to load, it will be loaded _before_ all your\r
	17	* local modifications such as -L are evaluated; and if you specify\r
	18	* a protocol and a port, the protocol is set up first so that the\r
	19	* port can override its choice of port number.\r
	20	*/\r
	21	\r
	22	#define NPRIORITIES 3\r
	23	\r
	24	struct cmdline_saved_param {\r
	25	char p, value;\r
	26	};\r
	27	struct cmdline_saved_param_set {\r
	28	struct cmdline_saved_param *params;\r
	29	int nsaved, savesize;\r
	30	};\r
	31	\r
	32	/*\r
	33	* C guarantees this structure will be initialised to all zero at\r
	34	* program start, which is exactly what we want.\r
	35	*/\r
	36	static struct cmdline_saved_param_set saves[NPRIORITIES];\r
	37	\r
	38	static void cmdline_save_param(char p, char value, int pri)\r
	39	{\r
	40	if (saves[pri].nsaved >= saves[pri].savesize) {\r
	41	saves[pri].savesize = saves[pri].nsaved + 32;\r
	42	saves[pri].params =\r
	43	srealloc(saves[pri].params,\r
	44	saves[pri].savesizesizeof(saves[pri].params));\r
	45	}\r
	46	saves[pri].params[saves[pri].nsaved].p = p;\r
	47	saves[pri].params[saves[pri].nsaved].value = value;\r
	48	saves[pri].nsaved++;\r
	49	}\r
	50	\r
	51	#define SAVEABLE(pri) do { \\r
	52	if (need_save) { cmdline_save_param(p, value, pri); return ret; } \\r
	53	} while (0)\r
	54	\r
	55	char *cmdline_password = NULL;\r
	56	\r
	57	static int cmdline_get_line(const char prompt, char str,\r
	58	int maxlen, int is_pw)\r
	59	{\r
	60	static int tried_once = 0;\r
	61	\r
	62	assert(is_pw && cmdline_password);\r
	63	\r
	64	if (tried_once) {\r
65	return 0;\r
66	} else {\r
67	strncpy(str, cmdline_password, maxlen);\r
68	str[maxlen - 1] = '\0';\r
69	tried_once = 1;\r
70	return 1;\r
71	}\r
72	}\r
73	\r
74	/*\r
75	* Here we have a flags word which describes the capabilities of\r
76	* the particular tool on whose behalf we're running. We will\r
77	* refuse certain command-line options if a particular tool\r
78	* inherently can't do anything sensible. For example, the file\r
79	* transfer tools (psftp, pscp) can't do a great deal with protocol\r
80	* selections (ever tried running scp over telnet?) or with port\r
81	* forwarding (even if it wasn't a hideously bad idea, they don't\r
82	* have the select() infrastructure to make them work).\r
83	*/\r
84	int cmdline_tooltype = 0;\r
85	\r
86	static int cmdline_check_unavailable(int flag, char *p)\r
87	{\r
88	if (cmdline_tooltype & flag) {\r
89	cmdline_error("option \"%s\" not available in this tool", p);\r
90	return 1;\r
91	}\r
92	return 0;\r
93	}\r
94	\r
95	#define UNAVAILABLE_IN(flag) do { \\r
96	if (cmdline_check_unavailable(flag, p)) return ret; \\r
97	} while (0)\r
98	\r
99	/*\r
100	* Process a standard command-line parameter. `p' is the parameter\r
101	* in question; `value' is the subsequent element of argv, which\r
102	* may or may not be required as an operand to the parameter.\r
103	* Return value is 2 if both arguments were used; 1 if only p was\r
104	* used; 0 if the parameter wasn't one we recognised; -2 if it\r
105	* should have been 2 but value was NULL.\r
106	*/\r
107	\r
108	#define RETURN(x) do { \\r
109	if ((x) == 2 && !value) return -2; \\r
110	ret = x; \\r
111	} while (0)\r
112	\r
113	int cmdline_process_param(char p, char value, int need_save)\r
114	{\r
115	int ret = 0;\r
116	\r
117	if (!strcmp(p, "-load")) {\r
118	RETURN(2);\r
119	SAVEABLE(0); /* very high priority */\r
120	do_defaults(value, &cfg);\r
121	return 2;\r
122	}\r
123	if (!strcmp(p, "-ssh")) {\r
124	RETURN(1);\r
125	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
126	SAVEABLE(1);\r
127	default_protocol = cfg.protocol = PROT_SSH;\r
128	default_port = cfg.port = 22;\r
129	return 1;\r
130	}\r
131	if (!strcmp(p, "-telnet")) {\r
132	RETURN(1);\r
133	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
134	SAVEABLE(1);\r
135	default_protocol = cfg.protocol = PROT_TELNET;\r
136	default_port = cfg.port = 23;\r
137	return 1;\r
138	}\r
139	if (!strcmp(p, "-rlogin")) {\r
140	RETURN(1);\r
141	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
142	SAVEABLE(1);\r
143	default_protocol = cfg.protocol = PROT_RLOGIN;\r
144	default_port = cfg.port = 513;\r
145	return 1;\r
146	}\r
147	if (!strcmp(p, "-raw")) {\r
148	RETURN(1);\r
149	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
150	SAVEABLE(1);\r
151	default_protocol = cfg.protocol = PROT_RAW;\r
152	}\r
153	if (!strcmp(p, "-v")) {\r
154	RETURN(1);\r
155	flags \|= FLAG_VERBOSE;\r
156	}\r
157	if (!strcmp(p, "-l")) {\r
158	RETURN(2);\r
159	SAVEABLE(1);\r
160	strncpy(cfg.username, value, sizeof(cfg.username));\r
161	cfg.username[sizeof(cfg.username) - 1] = '\0';\r
162	}\r
163	if ((!strcmp(p, "-L") \|\| !strcmp(p, "-R"))) {\r
164	char fwd, ptr, *q;\r
165	int i=0;\r
166	RETURN(2);\r
167	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
168	SAVEABLE(1);\r
169	fwd = value;\r
170	ptr = cfg.portfwd;\r
171	/* if multiple forwards, find end of list */\r
172	if (ptr[0]=='R' \|\| ptr[0]=='L') {\r
173	for (i = 0; i < sizeof(cfg.portfwd) - 2; i++)\r
174	if (ptr[i]=='\000' && ptr[i+1]=='\000')\r
175	break;\r
176	ptr = ptr + i + 1; /* point to next forward slot */\r
177	}\r
178	ptr[0] = p[1]; /* insert a 'L' or 'R' at the start */\r
179	if (strlen(fwd) > sizeof(cfg.portfwd) - i - 2) {\r
180	cmdline_error("out of space for port forwardings");\r
181	return ret;\r
182	}\r
183	strncpy(ptr+1, fwd, sizeof(cfg.portfwd) - i);\r
184	q = strchr(ptr, ':');\r
185	if (q) q = '\t'; / replace first : with \t */\r
186	cfg.portfwd[sizeof(cfg.portfwd) - 1] = '\0';\r
187	cfg.portfwd[sizeof(cfg.portfwd) - 2] = '\0';\r
188	ptr[strlen(ptr)+1] = '\000'; /* append two '\000' */\r
189	}\r
190	if (!strcmp(p, "-m")) {\r
191	char filename, command;\r
192	int cmdlen, cmdsize;\r
193	FILE *fp;\r
194	int c, d;\r
195	\r
196	RETURN(2);\r
197	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
198	SAVEABLE(1);\r
199	\r
200	filename = value;\r
201	\r
202	cmdlen = cmdsize = 0;\r
203	command = NULL;\r
204	fp = fopen(filename, "r");\r
205	if (!fp) {\r
206	cmdline_error("unable to open command "\r
207	"file \"%s\"", filename);\r
208	return ret;\r
209	}\r
210	do {\r
211	c = fgetc(fp);\r
212	d = c;\r
213	if (c == EOF)\r
214	d = 0;\r
215	if (cmdlen >= cmdsize) {\r
216	cmdsize = cmdlen + 512;\r
217	command = srealloc(command, cmdsize);\r
218	}\r
219	command[cmdlen++] = d;\r
220	} while (c != EOF);\r
221	cfg.remote_cmd_ptr = command;\r
222	cfg.remote_cmd_ptr2 = NULL;\r
223	cfg.nopty = TRUE; /* command => no terminal */\r
224	}\r
b42572b6	225	if (!strcmp(p, "-P")) {\r
c0a81592	226	RETURN(2);\r
	227	SAVEABLE(2); /* lower priority than -ssh,-telnet */\r
	228	cfg.port = atoi(value);\r
	229	}\r
	230	if (!strcmp(p, "-pw")) {\r
	231	RETURN(2);\r
	232	cmdline_password = value;\r
	233	ssh_get_line = cmdline_get_line;\r
	234	ssh_getline_pw_only = TRUE;\r
	235	}\r
34fb176e	236	\r
	237	if (!strcmp(p, "-A")) {\r
	238	RETURN(1);\r
3233f3af	239	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
34fb176e	240	SAVEABLE(1);\r
	241	cfg.agentfwd = 1;\r
	242	}\r
	243	if (!strcmp(p, "-a")) {\r
	244	RETURN(1);\r
3233f3af	245	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
34fb176e	246	SAVEABLE(1);\r
	247	cfg.agentfwd = 0;\r
	248	}\r
	249	\r
	250	if (!strcmp(p, "-X")) {\r
	251	RETURN(1);\r
3233f3af	252	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
34fb176e	253	SAVEABLE(1);\r
	254	cfg.x11_forward = 1;\r
	255	}\r
	256	if (!strcmp(p, "-x")) {\r
	257	RETURN(1);\r
3233f3af	258	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
34fb176e	259	SAVEABLE(1);\r
	260	cfg.x11_forward = 0;\r
	261	}\r
	262	\r
	263	if (!strcmp(p, "-t")) {\r
	264	RETURN(1);\r
3233f3af	265	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
34fb176e	266	SAVEABLE(1);\r
	267	cfg.nopty = 0;\r
	268	}\r
	269	if (!strcmp(p, "-T")) {\r
	270	RETURN(1);\r
3233f3af	271	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER);\r
34fb176e	272	SAVEABLE(1);\r
	273	cfg.nopty = 1;\r
	274	}\r
	275	\r
	276	if (!strcmp(p, "-C")) {\r
	277	RETURN(1);\r
	278	SAVEABLE(1);\r
	279	cfg.compression = 1;\r
	280	}\r
	281	\r
	282	if (!strcmp(p, "-1")) {\r
	283	RETURN(1);\r
	284	SAVEABLE(1);\r
	285	cfg.sshprot = 0; /* ssh protocol 1 only */\r
	286	}\r
	287	if (!strcmp(p, "-2")) {\r
	288	RETURN(1);\r
	289	SAVEABLE(1);\r
	290	cfg.sshprot = 3; /* ssh protocol 2 only */\r
	291	}\r
	292	\r
	293	if (!strcmp(p, "-i")) {\r
	294	RETURN(2);\r
	295	SAVEABLE(1);\r
	296	strncpy(cfg.keyfile, value, sizeof(cfg.keyfile));\r
	297	cfg.keyfile[sizeof(cfg.keyfile)-1] = '\0';\r
	298	}\r
	299	\r
c0a81592	300	return ret; /* unrecognised */\r
	301	}\r
	302	\r
	303	void cmdline_run_saved(void)\r
	304	{\r
	305	int pri, i;\r
	306	for (pri = 0; pri < NPRIORITIES; pri++)\r
	307	for (i = 0; i < saves[pri].nsaved; i++)\r
	308	cmdline_process_param(saves[pri].params[i].p,\r
	309	saves[pri].params[i].value, 0);\r
	310	}\r