[u/mdw/putty] / cmdline.c

/*
 * cmdline.c - command-line parsing shared between many of the
 * PuTTY applications
 */

#include <stdio.h>
#include <assert.h>
#include <stdlib.h>
#include "putty.h"

/*
 * Some command-line parameters need to be saved up until after
 * we've loaded the saved session which will form the basis of our
 * eventual running configuration. For this we use the macro
 * SAVEABLE, which notices if the `need_save' parameter is set and
 * saves the parameter and value on a list.
 * 
 * We also assign priorities to saved parameters, just to slightly
 * ameliorate silly ordering problems. For example, if you specify
 * a saved session to load, it will be loaded _before_ all your
 * local modifications such as -L are evaluated; and if you specify
 * a protocol and a port, the protocol is set up first so that the
 * port can override its choice of port number.
 * 
 * (In fact -load is not saved at all, since in at least Plink the
 * processing of further command-line options depends on whether or
 * not the loaded session contained a hostname. So it must be
 * executed immediately.)
 */

#define NPRIORITIES 2

struct cmdline_saved_param {
    char *p, *value;
};
struct cmdline_saved_param_set {
    struct cmdline_saved_param *params;
    int nsaved, savesize;
};

/*
 * C guarantees this structure will be initialised to all zero at
 * program start, which is exactly what we want.
 */
static struct cmdline_saved_param_set saves[NPRIORITIES];

static void cmdline_save_param(char *p, char *value, int pri)
{
    if (saves[pri].nsaved >= saves[pri].savesize) {
	saves[pri].savesize = saves[pri].nsaved + 32;
	saves[pri].params = sresize(saves[pri].params, saves[pri].savesize,
				    struct cmdline_saved_param);
    }
    saves[pri].params[saves[pri].nsaved].p = p;
    saves[pri].params[saves[pri].nsaved].value = value;
    saves[pri].nsaved++;
}

static char *cmdline_password = NULL;

void cmdline_cleanup(void)
{
    int pri;

    if (cmdline_password) {
	memset(cmdline_password, 0, strlen(cmdline_password));
	sfree(cmdline_password);
	cmdline_password = NULL;
    }
    
    for (pri = 0; pri < NPRIORITIES; pri++) {
	sfree(saves[pri].params);
	saves[pri].params = NULL;
	saves[pri].savesize = 0;
	saves[pri].nsaved = 0;
    }
}

#define SAVEABLE(pri) do { \
    if (need_save) { cmdline_save_param(p, value, pri); return ret; } \
} while (0)

/*
 * Similar interface to get_userpass_input(), except that here a -1
 * return means that we aren't capable of processing the prompt and
 * someone else should do it.
 */
int cmdline_get_passwd_input(prompts_t *p, unsigned char *in, int inlen) {

    static int tried_once = 0;

    /*
     * We only handle prompts which don't echo (which we assume to be
     * passwords), and (currently) we only cope with a password prompt
     * that comes in a prompt-set on its own.
     */
    if (!cmdline_password || in || p->n_prompts != 1 || p->prompts[0]->echo) {
	return -1;
    }

    /*
     * If we've tried once, return utter failure (no more passwords left
     * to try).
     */
    if (tried_once)
	return 0;

    strncpy(p->prompts[0]->result, cmdline_password,
	    p->prompts[0]->result_len);
    p->prompts[0]->result[p->prompts[0]->result_len-1] = '\0';
    memset(cmdline_password, 0, strlen(cmdline_password));
    sfree(cmdline_password);
    cmdline_password = NULL;
    tried_once = 1;
    return 1;

}

/*
 * Here we have a flags word which describes the capabilities of
 * the particular tool on whose behalf we're running. We will
 * refuse certain command-line options if a particular tool
 * inherently can't do anything sensible. For example, the file
 * transfer tools (psftp, pscp) can't do a great deal with protocol
 * selections (ever tried running scp over telnet?) or with port
 * forwarding (even if it wasn't a hideously bad idea, they don't
 * have the select() infrastructure to make them work).
 */
int cmdline_tooltype = 0;

static int cmdline_check_unavailable(int flag, char *p)
{
    if (cmdline_tooltype & flag) {
	cmdline_error("option \"%s\" not available in this tool", p);
	return 1;
    }
    return 0;
}

#define UNAVAILABLE_IN(flag) do { \
    if (cmdline_check_unavailable(flag, p)) return ret; \
} while (0)

/*
 * Process a standard command-line parameter. `p' is the parameter
 * in question; `value' is the subsequent element of argv, which
 * may or may not be required as an operand to the parameter.
 * If `need_save' is 1, arguments which need to be saved as
 * described at this top of this file are, for later execution;
 * if 0, they are processed normally. (-1 is a special value used
 * by pterm to count arguments for a preliminary pass through the
 * argument list; it causes immediate return with an appropriate
 * value with no action taken.)
 * Return value is 2 if both arguments were used; 1 if only p was
 * used; 0 if the parameter wasn't one we recognised; -2 if it
 * should have been 2 but value was NULL.
 */

#define RETURN(x) do { \
    if ((x) == 2 && !value) return -2; \
    ret = x; \
    if (need_save < 0) return x; \
} while (0)

int cmdline_process_param(char *p, char *value, int need_save, Conf *conf)
{
    int ret = 0;

    if (!strcmp(p, "-load")) {
	RETURN(2);
	/* This parameter must be processed immediately rather than being
	 * saved. */
	do_defaults(value, conf);
	loaded_session = TRUE;
	cmdline_session_name = dupstr(value);
	return 2;
    }
    if (!strcmp(p, "-ssh")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	default_protocol = PROT_SSH;
	default_port = 22;
	conf_set_int(conf, CONF_protocol, default_protocol);
	conf_set_int(conf, CONF_port, default_port);
	return 1;
    }
    if (!strcmp(p, "-telnet")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	default_protocol = PROT_TELNET;
	default_port = 23;
	conf_set_int(conf, CONF_protocol, default_protocol);
	conf_set_int(conf, CONF_port, default_port);
	return 1;
    }
    if (!strcmp(p, "-rlogin")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	default_protocol = PROT_RLOGIN;
	default_port = 513;
	conf_set_int(conf, CONF_protocol, default_protocol);
	conf_set_int(conf, CONF_port, default_port);
	return 1;
    }
    if (!strcmp(p, "-raw")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	default_protocol = PROT_RAW;
	conf_set_int(conf, CONF_protocol, default_protocol);
    }
    if (!strcmp(p, "-serial")) {
	RETURN(1);
	/* Serial is not NONNETWORK in an odd sense of the word */
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	default_protocol = PROT_SERIAL;
	conf_set_int(conf, CONF_protocol, default_protocol);
	/* The host parameter will already be loaded into CONF_host,
	 * so copy it across */
	conf_set_str(conf, CONF_serline, conf_get_str(conf, CONF_host));
    }
    if (!strcmp(p, "-v")) {
	RETURN(1);
	flags |= FLAG_VERBOSE;
    }
    if (!strcmp(p, "-l")) {
	RETURN(2);
	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	conf_set_str(conf, CONF_username, value);
    }
    if (!strcmp(p, "-loghost")) {
	RETURN(2);
	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	conf_set_str(conf, CONF_loghost, value);
    }
    if ((!strcmp(p, "-L") || !strcmp(p, "-R") || !strcmp(p, "-D"))) {
	char type, *q, *qq, *key, *val;
	RETURN(2);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	if (strcmp(p, "-D")) {
	    /*
             * For -L or -R forwarding types:
             *
	     * We expect _at least_ two colons in this string. The
	     * possible formats are `sourceport:desthost:destport',
	     * or `sourceip:sourceport:desthost:destport' if you're
	     * specifying a particular loopback address. We need to
	     * replace the one between source and dest with a \t;
	     * this means we must find the second-to-last colon in
	     * the string.
	     *
	     * (This looks like a foolish way of doing it given the
	     * existence of strrchr, but it's more efficient than
	     * two strrchrs - not to mention that the second strrchr
	     * would require us to modify the input string!)
	     */

            type = p[1];               /* 'L' or 'R' */

	    q = qq = strchr(value, ':');
	    while (qq) {
		char *qqq = strchr(qq+1, ':');
		if (qqq)
		    q = qq;
		qq = qqq;
	    }

	    if (!q) {
		cmdline_error("-%c expects at least two colons in its"
			      " argument", type);
		return ret;
	    }

	    key = dupprintf("%c%.*s", type, q - value, value);
	    val = dupstr(q+1);
	} else {
            /*
             * Dynamic port forwardings are entered under the same key
             * as if they were local (because they occupy the same
             * port space - a local and a dynamic forwarding on the
             * same local port are mutually exclusive), with the
             * special value "D" (which can be distinguished from
             * anything in the ordinary -L case by containing no
             * colon).
             */
	    key = dupprintf("L%s", value);
	    val = dupstr("D");
	}
	conf_set_str_str(conf, CONF_portfwd, key, val);
	sfree(key);
	sfree(val);
    }
    if ((!strcmp(p, "-nc"))) {
	char *host, *portp;

	RETURN(2);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(0);

	portp = strchr(value, ':');
	if (!portp) {
	    cmdline_error("-nc expects argument of form 'host:port'");
	    return ret;
	}

	host = dupprintf("%.*s", portp - value, value);
	conf_set_str(conf, CONF_ssh_nc_host, host);
	conf_set_int(conf, CONF_ssh_nc_port, atoi(portp + 1));
    }
    if (!strcmp(p, "-m")) {
	char *filename, *command;
	int cmdlen, cmdsize;
	FILE *fp;
	int c, d;

	RETURN(2);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(0);

	filename = value;

	cmdlen = cmdsize = 0;
	command = NULL;
	fp = fopen(filename, "r");
	if (!fp) {
	    cmdline_error("unable to open command file \"%s\"", filename);
	    return ret;
	}
	do {
	    c = fgetc(fp);
	    d = c;
	    if (c == EOF)
		d = 0;
	    if (cmdlen >= cmdsize) {
		cmdsize = cmdlen + 512;
		command = sresize(command, cmdsize, char);
	    }
	    command[cmdlen++] = d;
	} while (c != EOF);
	fclose(fp);
	conf_set_str(conf, CONF_remote_cmd, command);
	conf_set_str(conf, CONF_remote_cmd2, "");
	conf_set_int(conf, CONF_nopty, TRUE);   /* command => no terminal */
	sfree(command);
    }
    if (!strcmp(p, "-P")) {
	RETURN(2);
	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
	SAVEABLE(1);		       /* lower priority than -ssh,-telnet */
	conf_set_int(conf, CONF_port, atoi(value));
    }
    if (!strcmp(p, "-pw")) {
	RETURN(2);
	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
	SAVEABLE(1);
	/* We delay evaluating this until after the protocol is decided,
	 * so that we can warn if it's of no use with the selected protocol */
	if (conf_get_int(conf, CONF_protocol) != PROT_SSH)
	    cmdline_error("the -pw option can only be used with the "
			  "SSH protocol");
	else {
	    cmdline_password = dupstr(value);
	    /* Assuming that `value' is directly from argv, make a good faith
	     * attempt to trample it, to stop it showing up in `ps' output
	     * on Unix-like systems. Not guaranteed, of course. */
	    memset(value, 0, strlen(value));
	}
    }

    if (!strcmp(p, "-agent") || !strcmp(p, "-pagent") ||
	!strcmp(p, "-pageant")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	conf_set_int(conf, CONF_tryagent, TRUE);
    }
    if (!strcmp(p, "-noagent") || !strcmp(p, "-nopagent") ||
	!strcmp(p, "-nopageant")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	conf_set_int(conf, CONF_tryagent, FALSE);
    }

    if (!strcmp(p, "-A")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	conf_set_int(conf, CONF_agentfwd, 1);
    }
    if (!strcmp(p, "-a")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	conf_set_int(conf, CONF_agentfwd, 0);
    }

    if (!strcmp(p, "-X")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	conf_set_int(conf, CONF_x11_forward, 1);
    }
    if (!strcmp(p, "-x")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	conf_set_int(conf, CONF_x11_forward, 0);
    }

    if (!strcmp(p, "-t")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(1);	/* lower priority than -m */
	conf_set_int(conf, CONF_nopty, 0);
    }
    if (!strcmp(p, "-T")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(1);
	conf_set_int(conf, CONF_nopty, 1);
    }

    if (!strcmp(p, "-N")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	conf_set_int(conf, CONF_ssh_no_shell, 1);
    }

    if (!strcmp(p, "-C")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	conf_set_int(conf, CONF_compression, 1);
    }

    if (!strcmp(p, "-1")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	conf_set_int(conf, CONF_sshprot, 0);   /* ssh protocol 1 only */
    }
    if (!strcmp(p, "-2")) {
	RETURN(1);
	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	conf_set_int(conf, CONF_sshprot, 3);   /* ssh protocol 2 only */
    }

    if (!strcmp(p, "-i")) {
	Filename fn;
	RETURN(2);
	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
	SAVEABLE(0);
	fn = filename_from_str(value);
	conf_set_filename(conf, CONF_keyfile, &fn);
    }

    if (!strcmp(p, "-4") || !strcmp(p, "-ipv4")) {
	RETURN(1);
	SAVEABLE(1);
	conf_set_int(conf, CONF_addressfamily, ADDRTYPE_IPV4);
    }
    if (!strcmp(p, "-6") || !strcmp(p, "-ipv6")) {
	RETURN(1);
	SAVEABLE(1);
	conf_set_int(conf, CONF_addressfamily, ADDRTYPE_IPV6);
    }
    if (!strcmp(p, "-sercfg")) {
	char* nextitem;
	RETURN(2);
	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER | TOOLTYPE_NONNETWORK);
	SAVEABLE(1);
	if (conf_get_int(conf, CONF_protocol) != PROT_SERIAL)
	    cmdline_error("the -sercfg option can only be used with the "
			  "serial protocol");
	/* Value[0] contains one or more , separated values, like 19200,8,n,1,X */
	nextitem = value;
	while (nextitem[0] != '\0') {
	    int length, skip;
	    char *end = strchr(nextitem, ',');
	    if (!end) {
		length = strlen(nextitem);
		skip = 0;
	    } else {
		length = end - nextitem;
		nextitem[length] = '\0';
		skip = 1;
	    }
	    if (length == 1) {
		switch (*nextitem) {
		  case '1':
		  case '2':
		    conf_set_int(conf, CONF_serstopbits, 2 * (*nextitem-'0'));
		    break;

		  case '5':
		    conf_set_int(conf, CONF_serdatabits, *nextitem-'0');
		    break;

		  case 'n':
		    conf_set_int(conf, CONF_serparity, SER_PAR_NONE);
		    break;
		  case 'o':
		    conf_set_int(conf, CONF_serparity, SER_PAR_ODD);
		    break;
		  case 'e':
		    conf_set_int(conf, CONF_serparity, SER_PAR_EVEN);
		    break;
		  case 'm':
		    conf_set_int(conf, CONF_serparity, SER_PAR_MARK);
		    break;
		  case 's':
		    conf_set_int(conf, CONF_serparity, SER_PAR_SPACE);
		    break;

		  case 'N':
		    conf_set_int(conf, CONF_serflow, SER_FLOW_NONE);
		    break;
		  case 'X':
		    conf_set_int(conf, CONF_serflow, SER_FLOW_XONXOFF);
		    break;
		  case 'R':
		    conf_set_int(conf, CONF_serflow, SER_FLOW_RTSCTS);
		    break;
		  case 'D':
		    conf_set_int(conf, CONF_serflow, SER_FLOW_DSRDTR);
		    break;

		  default:
		    cmdline_error("Unrecognised suboption \"-sercfg %c\"",
				  *nextitem);
		}
	    } else if (length == 3 && !strncmp(nextitem,"1.5",3)) {
		/* Messy special case */
		conf_set_int(conf, CONF_serstopbits, 3);
	    } else {
		int serspeed = atoi(nextitem);
		if (serspeed != 0) {
		    conf_set_int(conf, CONF_serspeed, serspeed);
		} else {
		    cmdline_error("Unrecognised suboption \"-sercfg %s\"",
				  nextitem);
		}
	    }
	    nextitem += length + skip;
	}
    }
    return ret;			       /* unrecognised */
}

void cmdline_run_saved(Conf *conf)
{
    int pri, i;
    for (pri = 0; pri < NPRIORITIES; pri++)
	for (i = 0; i < saves[pri].nsaved; i++)
	    cmdline_process_param(saves[pri].params[i].p,
				  saves[pri].params[i].value, 0, conf);
}
Commit	Line	Data
eaf1e20a	1	/*
	2	* cmdline.c - command-line parsing shared between many of the
	3	* PuTTY applications
	4	*/
	5
dfe3332b	6	#include <stdio.h>
	7	#include <assert.h>
	8	#include <stdlib.h>
	9	#include "putty.h"
	10
	11	/*
	12	* Some command-line parameters need to be saved up until after
	13	* we've loaded the saved session which will form the basis of our
	14	* eventual running configuration. For this we use the macro
	15	* SAVEABLE, which notices if the `need_save' parameter is set and
	16	* saves the parameter and value on a list.
	17	*
	18	* We also assign priorities to saved parameters, just to slightly
	19	* ameliorate silly ordering problems. For example, if you specify
	20	* a saved session to load, it will be loaded _before_ all your
	21	* local modifications such as -L are evaluated; and if you specify
	22	* a protocol and a port, the protocol is set up first so that the
	23	* port can override its choice of port number.
c725e24c	24	*
	25	* (In fact -load is not saved at all, since in at least Plink the
	26	* processing of further command-line options depends on whether or
	27	* not the loaded session contained a hostname. So it must be
	28	* executed immediately.)
dfe3332b	29	*/
dfe3332b	30
c725e24c	31	#define NPRIORITIES 2
dfe3332b	32
	33	struct cmdline_saved_param {
	34	char p, value;
	35	};
	36	struct cmdline_saved_param_set {
	37	struct cmdline_saved_param *params;
	38	int nsaved, savesize;
	39	};
	40
	41	/*
	42	* C guarantees this structure will be initialised to all zero at
	43	* program start, which is exactly what we want.
	44	*/
	45	static struct cmdline_saved_param_set saves[NPRIORITIES];
	46
	47	static void cmdline_save_param(char p, char value, int pri)
	48	{
	49	if (saves[pri].nsaved >= saves[pri].savesize) {
	50	saves[pri].savesize = saves[pri].nsaved + 32;
3d88e64d	51	saves[pri].params = sresize(saves[pri].params, saves[pri].savesize,
3d88e64d	52	struct cmdline_saved_param);
dfe3332b	53	}
	54	saves[pri].params[saves[pri].nsaved].p = p;
	55	saves[pri].params[saves[pri].nsaved].value = value;
	56	saves[pri].nsaved++;
	57	}
	58
b72bdf11	59	static char *cmdline_password = NULL;
b72bdf11	60
679539d7	61	void cmdline_cleanup(void)
	62	{
	63	int pri;
	64
b72bdf11	65	if (cmdline_password) {
	66	memset(cmdline_password, 0, strlen(cmdline_password));
	67	sfree(cmdline_password);
	68	cmdline_password = NULL;
	69	}
	70
	71	for (pri = 0; pri < NPRIORITIES; pri++) {
679539d7	72	sfree(saves[pri].params);
b72bdf11	73	saves[pri].params = NULL;
	74	saves[pri].savesize = 0;
	75	saves[pri].nsaved = 0;
	76	}
679539d7	77	}
679539d7	78
dfe3332b	79	#define SAVEABLE(pri) do { \
	80	if (need_save) { cmdline_save_param(p, value, pri); return ret; } \
	81	} while (0)
	82
edd0cb8a	83	/*
	84	* Similar interface to get_userpass_input(), except that here a -1
	85	* return means that we aren't capable of processing the prompt and
	86	* someone else should do it.
	87	*/
	88	int cmdline_get_passwd_input(prompts_t p, unsigned char in, int inlen) {
dfe3332b	89
dfe3332b	90	static int tried_once = 0;
dfe3332b	91
edd0cb8a	92	/*
	93	* We only handle prompts which don't echo (which we assume to be
	94	* passwords), and (currently) we only cope with a password prompt
	95	* that comes in a prompt-set on its own.
	96	*/
	97	if (!cmdline_password \|\| in \|\| p->n_prompts != 1 \|\| p->prompts[0]->echo) {
	98	return -1;
	99	}
dfe3332b	100
edd0cb8a	101	/*
	102	* If we've tried once, return utter failure (no more passwords left
	103	* to try).
	104	*/
	105	if (tried_once)
dfe3332b	106	return 0;
edd0cb8a	107
	108	strncpy(p->prompts[0]->result, cmdline_password,
	109	p->prompts[0]->result_len);
	110	p->prompts[0]->result[p->prompts[0]->result_len-1] = '\0';
	111	memset(cmdline_password, 0, strlen(cmdline_password));
b72bdf11	112	sfree(cmdline_password);
b72bdf11	113	cmdline_password = NULL;
edd0cb8a	114	tried_once = 1;
	115	return 1;
	116
dfe3332b	117	}
	118
	119	/*
	120	* Here we have a flags word which describes the capabilities of
	121	* the particular tool on whose behalf we're running. We will
	122	* refuse certain command-line options if a particular tool
	123	* inherently can't do anything sensible. For example, the file
	124	* transfer tools (psftp, pscp) can't do a great deal with protocol
	125	* selections (ever tried running scp over telnet?) or with port
	126	* forwarding (even if it wasn't a hideously bad idea, they don't
	127	* have the select() infrastructure to make them work).
	128	*/
	129	int cmdline_tooltype = 0;
	130
	131	static int cmdline_check_unavailable(int flag, char *p)
	132	{
	133	if (cmdline_tooltype & flag) {
	134	cmdline_error("option \"%s\" not available in this tool", p);
	135	return 1;
	136	}
	137	return 0;
	138	}
	139
	140	#define UNAVAILABLE_IN(flag) do { \
	141	if (cmdline_check_unavailable(flag, p)) return ret; \
	142	} while (0)
	143
	144	/*
	145	* Process a standard command-line parameter. `p' is the parameter
	146	* in question; `value' is the subsequent element of argv, which
	147	* may or may not be required as an operand to the parameter.
b552f500	148	* If `need_save' is 1, arguments which need to be saved as
	149	* described at this top of this file are, for later execution;
	150	* if 0, they are processed normally. (-1 is a special value used
	151	* by pterm to count arguments for a preliminary pass through the
	152	* argument list; it causes immediate return with an appropriate
	153	* value with no action taken.)
dfe3332b	154	* Return value is 2 if both arguments were used; 1 if only p was
	155	* used; 0 if the parameter wasn't one we recognised; -2 if it
	156	* should have been 2 but value was NULL.
	157	*/
	158
	159	#define RETURN(x) do { \
	160	if ((x) == 2 && !value) return -2; \
	161	ret = x; \
46a3419b	162	if (need_save < 0) return x; \
dfe3332b	163	} while (0)
dfe3332b	164
4a693cfc	165	int cmdline_process_param(char p, char value, int need_save, Conf *conf)
dfe3332b	166	{
	167	int ret = 0;
	168
	169	if (!strcmp(p, "-load")) {
	170	RETURN(2);
c725e24c	171	/* This parameter must be processed immediately rather than being
c725e24c	172	* saved. */
4a693cfc	173	do_defaults(value, conf);
18e62ad8	174	loaded_session = TRUE;
073e9f42	175	cmdline_session_name = dupstr(value);
dfe3332b	176	return 2;
	177	}
	178	if (!strcmp(p, "-ssh")) {
	179	RETURN(1);
46a3419b	180	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
c725e24c	181	SAVEABLE(0);
4a693cfc	182	default_protocol = PROT_SSH;
	183	default_port = 22;
	184	conf_set_int(conf, CONF_protocol, default_protocol);
	185	conf_set_int(conf, CONF_port, default_port);
dfe3332b	186	return 1;
	187	}
	188	if (!strcmp(p, "-telnet")) {
	189	RETURN(1);
46a3419b	190	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
c725e24c	191	SAVEABLE(0);
4a693cfc	192	default_protocol = PROT_TELNET;
	193	default_port = 23;
	194	conf_set_int(conf, CONF_protocol, default_protocol);
	195	conf_set_int(conf, CONF_port, default_port);
dfe3332b	196	return 1;
	197	}
	198	if (!strcmp(p, "-rlogin")) {
	199	RETURN(1);
46a3419b	200	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
c725e24c	201	SAVEABLE(0);
4a693cfc	202	default_protocol = PROT_RLOGIN;
	203	default_port = 513;
	204	conf_set_int(conf, CONF_protocol, default_protocol);
	205	conf_set_int(conf, CONF_port, default_port);
dfe3332b	206	return 1;
	207	}
	208	if (!strcmp(p, "-raw")) {
	209	RETURN(1);
46a3419b	210	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
c725e24c	211	SAVEABLE(0);
4a693cfc	212	default_protocol = PROT_RAW;
4a693cfc	213	conf_set_int(conf, CONF_protocol, default_protocol);
dfe3332b	214	}
9621bbab	215	if (!strcmp(p, "-serial")) {
	216	RETURN(1);
	217	/* Serial is not NONNETWORK in an odd sense of the word */
	218	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
	219	SAVEABLE(0);
4a693cfc	220	default_protocol = PROT_SERIAL;
	221	conf_set_int(conf, CONF_protocol, default_protocol);
	222	/* The host parameter will already be loaded into CONF_host,
	223	* so copy it across */
	224	conf_set_str(conf, CONF_serline, conf_get_str(conf, CONF_host));
9621bbab	225	}
dfe3332b	226	if (!strcmp(p, "-v")) {
	227	RETURN(1);
	228	flags \|= FLAG_VERBOSE;
	229	}
	230	if (!strcmp(p, "-l")) {
	231	RETURN(2);
46a3419b	232	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
c725e24c	233	SAVEABLE(0);
4a693cfc	234	conf_set_str(conf, CONF_username, value);
dfe3332b	235	}
881da168	236	if (!strcmp(p, "-loghost")) {
	237	RETURN(2);
	238	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
	239	SAVEABLE(0);
4a693cfc	240	conf_set_str(conf, CONF_loghost, value);
881da168	241	}
820ebe3b	242	if ((!strcmp(p, "-L") \|\| !strcmp(p, "-R") \|\| !strcmp(p, "-D"))) {
4a693cfc	243	char type, q, qq, key, val;
dfe3332b	244	RETURN(2);
46a3419b	245	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
c725e24c	246	SAVEABLE(0);
4a693cfc	247	if (strcmp(p, "-D")) {
820ebe3b	248	/*
4a693cfc	249	* For -L or -R forwarding types:
4a693cfc	250	*
820ebe3b	251	* We expect _at least_ two colons in this string. The
	252	* possible formats are `sourceport:desthost:destport',
	253	* or `sourceip:sourceport:desthost:destport' if you're
	254	* specifying a particular loopback address. We need to
	255	* replace the one between source and dest with a \t;
	256	* this means we must find the second-to-last colon in
	257	* the string.
4a693cfc	258	*
	259	* (This looks like a foolish way of doing it given the
	260	* existence of strrchr, but it's more efficient than
	261	* two strrchrs - not to mention that the second strrchr
	262	* would require us to modify the input string!)
820ebe3b	263	*/
4a693cfc	264
	265	type = p[1]; /* 'L' or 'R' */
	266
	267	q = qq = strchr(value, ':');
820ebe3b	268	while (qq) {
	269	char *qqq = strchr(qq+1, ':');
	270	if (qqq)
	271	q = qq;
	272	qq = qqq;
	273	}
4a693cfc	274
	275	if (!q) {
	276	cmdline_error("-%c expects at least two colons in its"
	277	" argument", type);
	278	return ret;
	279	}
	280
	281	key = dupprintf("%c%.*s", type, q - value, value);
	282	val = dupstr(q+1);
	283	} else {
	284	/*
	285	* Dynamic port forwardings are entered under the same key
	286	* as if they were local (because they occupy the same
	287	* port space - a local and a dynamic forwarding on the
	288	* same local port are mutually exclusive), with the
	289	* special value "D" (which can be distinguished from
	290	* anything in the ordinary -L case by containing no
	291	* colon).
	292	*/
	293	key = dupprintf("L%s", value);
	294	val = dupstr("D");
6ee9b735	295	}
4a693cfc	296	conf_set_str_str(conf, CONF_portfwd, key, val);
	297	sfree(key);
	298	sfree(val);
dfe3332b	299	}
feb02b4e	300	if ((!strcmp(p, "-nc"))) {
	301	char host, portp;
	302
	303	RETURN(2);
	304	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
	305	SAVEABLE(0);
	306
4a693cfc	307	portp = strchr(value, ':');
4a693cfc	308	if (!portp) {
feb02b4e	309	cmdline_error("-nc expects argument of form 'host:port'");
	310	return ret;
	311	}
4a693cfc	312
	313	host = dupprintf("%.*s", portp - value, value);
	314	conf_set_str(conf, CONF_ssh_nc_host, host);
	315	conf_set_int(conf, CONF_ssh_nc_port, atoi(portp + 1));
feb02b4e	316	}
dfe3332b	317	if (!strcmp(p, "-m")) {
	318	char filename, command;
	319	int cmdlen, cmdsize;
	320	FILE *fp;
	321	int c, d;
	322
	323	RETURN(2);
46a3419b	324	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
c725e24c	325	SAVEABLE(0);
dfe3332b	326
	327	filename = value;
	328
	329	cmdlen = cmdsize = 0;
	330	command = NULL;
	331	fp = fopen(filename, "r");
	332	if (!fp) {
4a693cfc	333	cmdline_error("unable to open command file \"%s\"", filename);
dfe3332b	334	return ret;
	335	}
	336	do {
	337	c = fgetc(fp);
	338	d = c;
	339	if (c == EOF)
	340	d = 0;
	341	if (cmdlen >= cmdsize) {
	342	cmdsize = cmdlen + 512;
3d88e64d	343	command = sresize(command, cmdsize, char);
dfe3332b	344	}
	345	command[cmdlen++] = d;
	346	} while (c != EOF);
2fa6ce2c	347	fclose(fp);
4a693cfc	348	conf_set_str(conf, CONF_remote_cmd, command);
	349	conf_set_str(conf, CONF_remote_cmd2, "");
	350	conf_set_int(conf, CONF_nopty, TRUE); /* command => no terminal */
	351	sfree(command);
dfe3332b	352	}
	353	if (!strcmp(p, "-P")) {
	354	RETURN(2);
46a3419b	355	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
c725e24c	356	SAVEABLE(1); /* lower priority than -ssh,-telnet */
4a693cfc	357	conf_set_int(conf, CONF_port, atoi(value));
dfe3332b	358	}
	359	if (!strcmp(p, "-pw")) {
	360	RETURN(2);
46a3419b	361	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
01ccff1b	362	SAVEABLE(1);
	363	/* We delay evaluating this until after the protocol is decided,
	364	* so that we can warn if it's of no use with the selected protocol */
4a693cfc	365	if (conf_get_int(conf, CONF_protocol) != PROT_SSH)
4858529b	366	cmdline_error("the -pw option can only be used with the "
01ccff1b	367	"SSH protocol");
4858529b	368	else {
	369	cmdline_password = dupstr(value);
	370	/* Assuming that `value' is directly from argv, make a good faith
	371	* attempt to trample it, to stop it showing up in `ps' output
	372	* on Unix-like systems. Not guaranteed, of course. */
	373	memset(value, 0, strlen(value));
	374	}
dfe3332b	375	}
dfe3332b	376
973612f5	377	if (!strcmp(p, "-agent") \|\| !strcmp(p, "-pagent") \|\|
	378	!strcmp(p, "-pageant")) {
	379	RETURN(1);
	380	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
a3a48ced	381	SAVEABLE(0);
4a693cfc	382	conf_set_int(conf, CONF_tryagent, TRUE);
973612f5	383	}
	384	if (!strcmp(p, "-noagent") \|\| !strcmp(p, "-nopagent") \|\|
	385	!strcmp(p, "-nopageant")) {
	386	RETURN(1);
	387	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
a3a48ced	388	SAVEABLE(0);
4a693cfc	389	conf_set_int(conf, CONF_tryagent, FALSE);
973612f5	390	}
973612f5	391
dfe3332b	392	if (!strcmp(p, "-A")) {
dfe3332b	393	RETURN(1);
46a3419b	394	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
c725e24c	395	SAVEABLE(0);
4a693cfc	396	conf_set_int(conf, CONF_agentfwd, 1);
dfe3332b	397	}
	398	if (!strcmp(p, "-a")) {
	399	RETURN(1);
46a3419b	400	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
c725e24c	401	SAVEABLE(0);
4a693cfc	402	conf_set_int(conf, CONF_agentfwd, 0);
dfe3332b	403	}
	404
	405	if (!strcmp(p, "-X")) {
	406	RETURN(1);
46a3419b	407	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
c725e24c	408	SAVEABLE(0);
4a693cfc	409	conf_set_int(conf, CONF_x11_forward, 1);
dfe3332b	410	}
	411	if (!strcmp(p, "-x")) {
	412	RETURN(1);
46a3419b	413	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
c725e24c	414	SAVEABLE(0);
4a693cfc	415	conf_set_int(conf, CONF_x11_forward, 0);
dfe3332b	416	}
	417
	418	if (!strcmp(p, "-t")) {
	419	RETURN(1);
46a3419b	420	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
e7cf9486	421	SAVEABLE(1); /* lower priority than -m */
4a693cfc	422	conf_set_int(conf, CONF_nopty, 0);
dfe3332b	423	}
	424	if (!strcmp(p, "-T")) {
	425	RETURN(1);
46a3419b	426	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
e7cf9486	427	SAVEABLE(1);
4a693cfc	428	conf_set_int(conf, CONF_nopty, 1);
dfe3332b	429	}
dfe3332b	430
0ed48730	431	if (!strcmp(p, "-N")) {
	432	RETURN(1);
	433	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
	434	SAVEABLE(0);
4a693cfc	435	conf_set_int(conf, CONF_ssh_no_shell, 1);
0ed48730	436	}
0ed48730	437
dfe3332b	438	if (!strcmp(p, "-C")) {
dfe3332b	439	RETURN(1);
46a3419b	440	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
c725e24c	441	SAVEABLE(0);
4a693cfc	442	conf_set_int(conf, CONF_compression, 1);
dfe3332b	443	}
	444
	445	if (!strcmp(p, "-1")) {
	446	RETURN(1);
46a3419b	447	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
c725e24c	448	SAVEABLE(0);
4a693cfc	449	conf_set_int(conf, CONF_sshprot, 0); /* ssh protocol 1 only */
dfe3332b	450	}
	451	if (!strcmp(p, "-2")) {
	452	RETURN(1);
46a3419b	453	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
c725e24c	454	SAVEABLE(0);
4a693cfc	455	conf_set_int(conf, CONF_sshprot, 3); /* ssh protocol 2 only */
dfe3332b	456	}
	457
	458	if (!strcmp(p, "-i")) {
4a693cfc	459	Filename fn;
dfe3332b	460	RETURN(2);
46a3419b	461	UNAVAILABLE_IN(TOOLTYPE_NONNETWORK);
c725e24c	462	SAVEABLE(0);
4a693cfc	463	fn = filename_from_str(value);
4a693cfc	464	conf_set_filename(conf, CONF_keyfile, &fn);
dfe3332b	465	}
dfe3332b	466
05581745	467	if (!strcmp(p, "-4") \|\| !strcmp(p, "-ipv4")) {
	468	RETURN(1);
	469	SAVEABLE(1);
4a693cfc	470	conf_set_int(conf, CONF_addressfamily, ADDRTYPE_IPV4);
05581745	471	}
	472	if (!strcmp(p, "-6") \|\| !strcmp(p, "-ipv6")) {
	473	RETURN(1);
	474	SAVEABLE(1);
4a693cfc	475	conf_set_int(conf, CONF_addressfamily, ADDRTYPE_IPV6);
05581745	476	}
9621bbab	477	if (!strcmp(p, "-sercfg")) {
	478	char* nextitem;
	479	RETURN(2);
	480	UNAVAILABLE_IN(TOOLTYPE_FILETRANSFER \| TOOLTYPE_NONNETWORK);
	481	SAVEABLE(1);
4a693cfc	482	if (conf_get_int(conf, CONF_protocol) != PROT_SERIAL)
9621bbab	483	cmdline_error("the -sercfg option can only be used with the "
	484	"serial protocol");
	485	/* Value[0] contains one or more , separated values, like 19200,8,n,1,X */
	486	nextitem = value;
	487	while (nextitem[0] != '\0') {
	488	int length, skip;
	489	char *end = strchr(nextitem, ',');
	490	if (!end) {
	491	length = strlen(nextitem);
	492	skip = 0;
	493	} else {
	494	length = end - nextitem;
	495	nextitem[length] = '\0';
	496	skip = 1;
	497	}
	498	if (length == 1) {
	499	switch (*nextitem) {
	500	case '1':
9621bbab	501	case '2':
4a693cfc	502	conf_set_int(conf, CONF_serstopbits, 2 * (*nextitem-'0'));
9621bbab	503	break;
	504
	505	case '5':
4a693cfc	506	conf_set_int(conf, CONF_serdatabits, *nextitem-'0');
9621bbab	507	break;
	508
	509	case 'n':
4a693cfc	510	conf_set_int(conf, CONF_serparity, SER_PAR_NONE);
9621bbab	511	break;
9621bbab	512	case 'o':
4a693cfc	513	conf_set_int(conf, CONF_serparity, SER_PAR_ODD);
9621bbab	514	break;
9621bbab	515	case 'e':
4a693cfc	516	conf_set_int(conf, CONF_serparity, SER_PAR_EVEN);
9621bbab	517	break;
9621bbab	518	case 'm':
4a693cfc	519	conf_set_int(conf, CONF_serparity, SER_PAR_MARK);
9621bbab	520	break;
9621bbab	521	case 's':
4a693cfc	522	conf_set_int(conf, CONF_serparity, SER_PAR_SPACE);
9621bbab	523	break;
	524
	525	case 'N':
4a693cfc	526	conf_set_int(conf, CONF_serflow, SER_FLOW_NONE);
9621bbab	527	break;
9621bbab	528	case 'X':
4a693cfc	529	conf_set_int(conf, CONF_serflow, SER_FLOW_XONXOFF);
9621bbab	530	break;
9621bbab	531	case 'R':
4a693cfc	532	conf_set_int(conf, CONF_serflow, SER_FLOW_RTSCTS);
9621bbab	533	break;
9621bbab	534	case 'D':
4a693cfc	535	conf_set_int(conf, CONF_serflow, SER_FLOW_DSRDTR);
9621bbab	536	break;
	537
	538	default:
	539	cmdline_error("Unrecognised suboption \"-sercfg %c\"",
	540	*nextitem);
	541	}
	542	} else if (length == 3 && !strncmp(nextitem,"1.5",3)) {
	543	/* Messy special case */
4a693cfc	544	conf_set_int(conf, CONF_serstopbits, 3);
9621bbab	545	} else {
	546	int serspeed = atoi(nextitem);
	547	if (serspeed != 0) {
4a693cfc	548	conf_set_int(conf, CONF_serspeed, serspeed);
9621bbab	549	} else {
	550	cmdline_error("Unrecognised suboption \"-sercfg %s\"",
	551	nextitem);
	552	}
	553	}
	554	nextitem += length + skip;
	555	}
	556	}
dfe3332b	557	return ret; /* unrecognised */
	558	}
	559
4a693cfc	560	void cmdline_run_saved(Conf *conf)
dfe3332b	561	{
	562	int pri, i;
	563	for (pri = 0; pri < NPRIORITIES; pri++)
	564	for (i = 0; i < saves[pri].nsaved; i++)
	565	cmdline_process_param(saves[pri].params[i].p,
4a693cfc	566	saves[pri].params[i].value, 0, conf);
dfe3332b	567	}