/*
 * This module implements the command dispatch for fc-uicc-tool
 */

#include <ctype.h>
#include <stdio.h>
#include <string.h>
#include <strings.h>
#include <stdlib.h>

extern int cmd_apdu();
extern int cmd_cd();
extern int cmd_change_pin();
extern int cmd_dir();
extern int cmd_disable_pin();
extern int cmd_enable_pin();
extern int cmd_exec();
extern int cmd_iccid();
extern int cmd_pin_attempt_cnt();
extern int cmd_puk_attempt_cnt();
extern int cmd_readbin();
extern int cmd_readef();
extern int cmd_readrec();
extern int cmd_select();
extern int cmd_select_aid();
extern int cmd_select_isim();
extern int cmd_select_usim();
extern int cmd_sim_resp();
extern int cmd_unblock_pin();
extern int cmd_update_bin();
extern int cmd_update_bin_imm();
extern int cmd_update_rec();
extern int cmd_verify_hex();
extern int cmd_verify_pin();

extern int good_exit();
extern int retrieve_atr();

static struct cmdtab {
	char *cmd;
	int minargs;
	int maxargs;
	int allow_redir;
	int (*func)();
} cmdtab[] = {
	{"apdu", 1, 1, 0, cmd_apdu},
	{"atr", 0, 0, 0, retrieve_atr},
	{"cd", 1, 1, 0, cmd_cd},
	{"change-pin", 3, 3, 0, cmd_change_pin},
	{"dir", 0, 0, 1, cmd_dir},
	{"disable-pin", 2, 2, 0, cmd_disable_pin},
	{"enable-pin", 2, 2, 0, cmd_enable_pin},
	{"exec", 1, 1, 0, cmd_exec},
	{"exit", 0, 0, 0, good_exit},
	{"iccid", 0, 0, 1, cmd_iccid},
	{"pin-attempt-cnt", 1, 1, 0, cmd_pin_attempt_cnt},
	{"puk-attempt-cnt", 1, 1, 0, cmd_puk_attempt_cnt},
	{"quit", 0, 0, 0, good_exit},
	{"readbin", 2, 2, 1, cmd_readbin},
	{"readef", 1, 1, 1, cmd_readef},
	{"readrec", 1, 2, 1, cmd_readrec},
	{"select", 1, 1, 0, cmd_select},
	{"select-aid", 1, 1, 0, cmd_select_aid},
	{"select-isim", 0, 0, 0, cmd_select_isim},
	{"select-usim", 0, 0, 0, cmd_select_usim},
	{"sim-resp", 0, 0, 1, cmd_sim_resp},
	{"unblock-pin", 3, 3, 0, cmd_unblock_pin},
	{"update-bin", 2, 2, 0, cmd_update_bin},
	{"update-bin-imm", 2, 2, 0, cmd_update_bin_imm},
	{"update-rec", 2, 2, 0, cmd_update_rec},
	{"verify-ext", 2, 2, 0, cmd_verify_pin},
	{"verify-hex", 2, 2, 0, cmd_verify_hex},
	{"verify-pin", 2, 2, 0, cmd_verify_pin},
	{0, 0, 0, 0, 0}
};

static FILE *
handle_output_redir(str)
	char *str;
{
	char *cp, *fn;
	FILE *outf;

	for (cp = str; isspace(*cp); cp++)
		;
	if (!*cp || *cp == '#') {
		fprintf(stderr, "error: no filename after '>'\n");
		return(0);
	}
	for (fn = cp; *cp && !isspace(*cp); cp++)
		;
	if (*cp)
		*cp++ = '\0';
	while (isspace(*cp))
		cp++;
	if (*cp && *cp != '#') {
		fprintf(stderr, "error: invalid syntax after '>'\n");
		return(0);
	}
	outf = fopen(fn, "w");
	if (!outf)
		perror(fn);
	return outf;
}

simtool_dispatch_cmd(cmd, is_script)
	char *cmd;
{
	char *argv[10];
	char *cp, **ap;
	struct cmdtab *tp;
	FILE *outf;
	int rc;

	for (cp = cmd; isspace(*cp); cp++)
		;
	if (!*cp || *cp == '#')
		return(0);
	if (is_script)
		printf("Script command: %s\n", cp);
	if (*cp == '!')
		return system(cp + 1);
	argv[0] = cp;
	while (*cp && !isspace(*cp))
		cp++;
	if (*cp)
		*cp++ = '\0';
	for (tp = cmdtab; tp->cmd; tp++)
		if (!strcmp(tp->cmd, argv[0]))
			break;
	if (!tp->func) {
		fprintf(stderr, "error: no such command\n");
		return(-1);
	}
	for (ap = argv + 1; ; ) {
		while (isspace(*cp))
			cp++;
		if (!*cp || *cp == '#' || *cp == '>')
			break;
		if (ap - argv - 1 >= tp->maxargs) {
			fprintf(stderr, "error: too many arguments\n");
			return(-1);
		}
		if (*cp == '"') {
			*ap++ = ++cp;
			for (;;) {
				if (!*cp) {
unterm_qstring:				fprintf(stderr,
					"error: unterminated quoted string\n");
					return(-1);
				}
				if (*cp == '"')
					break;
				if (*cp++ == '\\') {
					if (!*cp)
						goto unterm_qstring;
					cp++;
				}
			}
			*cp++ = '\0';
		} else {
			*ap++ = cp;
			while (*cp && !isspace(*cp))
				cp++;
			if (*cp)
				*cp++ = '\0';
		}
	}
	if (ap - argv - 1 < tp->minargs) {
		fprintf(stderr, "error: too few arguments\n");
		return(-1);
	}
	*ap = 0;
	if (*cp == '>') {
		if (!tp->allow_redir) {
			fprintf(stderr,
			"error: command does not support output redirection\n");
			return(-1);
		}
		outf = handle_output_redir(cp + 1);
		if (!outf)
			return(-1);
	} else
		outf = stdout;
	rc = tp->func(ap - argv, argv, outf);
	if (outf != stdout)
		fclose(outf);
	return rc;
}

dispatch_ready_argv(argc, argv)
	char **argv;
{
	struct cmdtab *tp;

	for (tp = cmdtab; tp->cmd; tp++)
		if (!strcmp(tp->cmd, argv[0]))
			break;
	if (!tp->func) {
		fprintf(stderr, "error: no such command\n");
		return(-1);
	}
	if (argc - 1 > tp->maxargs) {
		fprintf(stderr, "error: too many arguments\n");
		return(-1);
	}
	if (argc - 1 < tp->minargs) {
		fprintf(stderr, "error: too few arguments\n");
		return(-1);
	}
	return tp->func(argc, argv, stdout);
}