File:  [ELWIX - Embedded LightWeight unIX -] / libaitcli / src / aitcli.c
Revision 1.1.1.1.2.4: download - view: text, annotated - select for diffs - revision graph
Tue Apr 20 11:48:49 2010 UTC (14 years, 3 months ago) by misho
Branches: cli1_0
Diff to: branchpoint 1.1.1.1: preferred, unified
added new func

    1: /*************************************************************************
    2: * (C) 2010 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
    3: *  by Michael Pounov <misho@openbsd-bg.org>
    4: *
    5: * $Author: misho $
    6: * $Id: aitcli.c,v 1.1.1.1.2.4 2010/04/20 11:48:49 misho Exp $
    7: *
    8: *************************************************************************/
    9: #include "global.h"
   10: 
   11: 
   12: #pragma GCC visibility push(hidden)
   13: 
   14: cliCommands_t cli_stdCmds[] = {
   15: 	{ "test", cli_Cmd_Unsupported, "Test - Don`t use default command structure!", "test <cr>", cli_Comp_Filename }, 
   16: 	{ "-------", NULL, "---------------------", NULL, NULL }, 
   17: 	{ "help", cli_Cmd_Help, "Help screen", "help [command] <cr>", NULL }, 
   18: 	{ "exit", cli_Cmd_Exit, "Exit from console", "exit <cr>", NULL }, 
   19: 	{ NULL, NULL, NULL, NULL }
   20: };
   21: 
   22: // ------------------------------------------------
   23: 
   24: int cli_Errno;
   25: char cli_Error[STRSIZ];
   26: 
   27: #pragma GCC visibility pop
   28: 
   29: 
   30: static void cli_Null_Prep_Term(int meta)
   31: {
   32: }
   33: 
   34: static void cli_Null_Deprep_Term()
   35: {
   36: }
   37: 
   38: static int cli_Pre_Input_Change_Mode()
   39: {
   40: 	return 0;
   41: }
   42: 
   43: static int cli_GetC(FILE *dummy)
   44: {
   45: 	int ch = rl_getc(stdin);
   46: 
   47: 	/*
   48: 	if (is_special_char(ch)) {
   49: 		pending_special_char = ch;
   50: 		return '\r';
   51: 	}
   52: 	*/
   53: 
   54: 	return ch;
   55: }
   56: 
   57: 
   58: // cli_GetErrno() Get error code of last operation
   59: inline int cli_GetErrno()
   60: {
   61: 	return cli_Errno;
   62: }
   63: 
   64: // io_GetError() Get error text of last operation
   65: inline const char *cli_GetError()
   66: {
   67: 	return cli_Error;
   68: }
   69: 
   70: // cli_SetErr() Set error to variables for internal use!!!
   71: inline void cli_SetErr(int eno, char *estr, ...)
   72: {
   73: 	va_list lst;
   74: 
   75: 	cli_Errno = eno;
   76: 	memset(cli_Error, 0, STRSIZ);
   77: 	va_start(lst, estr);
   78: 	vsnprintf(cli_Error, STRSIZ, estr, lst);
   79: 	va_end(lst);
   80: }
   81: 
   82: // ------------------------------------------------------------
   83: 
   84: /*
   85:  * cli_Printf() Printf CLI features
   86:  * @out = Output stream
   87:  * @csFormat = Printf format string
   88:  * return: -1 error, != -1 printed chars
   89: */
   90: inline int cli_Printf(FILE *out, const char *csFormat, ...)
   91: {
   92: 	va_list lst;
   93: 	int ret;
   94: 
   95: 	va_start(lst, csFormat);
   96: 
   97: 	ret = vfprintf(out, csFormat, lst);
   98: 	if (-1 == ret)
   99: 		LOGERR;
  100: 
  101: 	va_end(lst);
  102: 	return ret;
  103: }
  104: 
  105: 
  106: /*
  107:  * cliComp() Initialize completion CLI features
  108:  * @cmdComplete = Completion function
  109:  * @cmdEntry = Compentry function
  110:  * return: none
  111: */
  112: inline void cliComp(cli_Completion_t *cmdComplete, cli_CompEntry_t *cmdEntry)
  113: {
  114: 	// command completon
  115: 	rl_attempted_completion_function = cmdComplete;
  116: 	rl_completion_entry_function = cmdEntry;
  117: }
  118: 
  119: /*
  120:  * cliTTY() Initialize I/O TTY CLI features
  121:  * @term = terminal name
  122:  * @inp = input handle
  123:  * @out = output handle
  124:  * @win = window size
  125:  * return: -1 error, != -1 ok
  126: */
  127: inline int cliTTY(const char *term, FILE *inp, FILE *out, struct winsize *win)
  128: {
  129: 	if (term)
  130: 		rl_terminal_name = term;
  131: 
  132: 	if (inp)
  133: 		rl_instream = inp;
  134: 	if (out)
  135: 		rl_outstream = out;
  136: 
  137: 	if (win)
  138: 	       if (ioctl(!rl_outstream ? STDOUT_FILENO : fileno(rl_outstream), TIOCSWINSZ, win) == -1) {
  139: 		       LOGERR;
  140: 		       return -1;
  141: 	       }
  142: 
  143: 	return 0;
  144: }
  145: 
  146: /*
  147:  * cliInit() Initialize Readline
  148:  * @csProg = program name
  149:  * return: none
  150: */
  151: inline void cliInit(const char *csProg)
  152: {
  153: 	rl_readline_name = csProg;
  154: 
  155: 	rl_variable_bind("editing-mode", "emacs");
  156: }
  157: 
  158: /*
  159:  * cliNetInit() Initialize Readline if CLI bind to socket
  160:  * @csProg = program name
  161:  * @pty = Master pty
  162:  * @term = stdin termios
  163:  * return: none
  164: */
  165: void cliNetInit(const char *csProg, int pty, struct termios *term)
  166: {
  167: 	struct termios t;
  168: 
  169: 	if (term) {
  170: 		t = *term;
  171: 		t.c_lflag &= ~(ICANON | ISIG | ECHO | ECHOCTL | ECHOE | ECHOK | ECHOKE | ECHONL | ECHOPRT);
  172: 		t.c_iflag &= ~ICRNL;
  173: 		t.c_iflag |= IGNBRK;
  174: 		t.c_cc[VMIN] = 1;
  175: 		t.c_cc[VTIME] = 0;
  176: 		tcsetattr(pty, TCSANOW, &t);
  177: 	}
  178: 
  179: 	cliInit(csProg);
  180: 
  181: 	rl_instream = fdopen(pty, "r");
  182: 
  183: 	rl_prep_term_function = cli_Null_Prep_Term;
  184: 	rl_deprep_term_function = cli_Null_Deprep_Term;
  185: 	rl_pre_input_hook = cli_Pre_Input_Change_Mode;
  186: 
  187: 	rl_getc_function = cli_GetC;
  188: 
  189: }
  190: 
  191: /*
  192:  * cliExec() Execute CLI main loop
  193:  * @cmdList = Commands list
  194:  * @csPrompt = Prompt text
  195:  * return: -1 error, 0 = exit w/^+D, 1 done.
  196: */
  197: int cliExec(cliCommands_t *cmdList, const char *csPrompt)
  198: {
  199: 	char *line, *s, *t, **app, *items[MAX_PROMPT_ITEMS];
  200: 	int ret = 0;
  201: 	register int i;
  202: 	cliCommands_t *cmd = NULL;
  203: 	FILE *out;
  204: 
  205: 	inline int inline_help()
  206: 	{
  207: 		cli_Cmd_Help(cmdList ? cmdList : cli_stdCmds, -1, out, NULL);
  208: 		rl_on_new_line();
  209: 		return 0;
  210: 	}
  211: 
  212: 	char **cli_stdCompletion(const char *text, int start, int end)
  213: 	{
  214: 		register int i;
  215: 		char **matches = NULL;
  216: 
  217: 		char *cmdCompGet(const char *text, int state)
  218: 		{
  219: 			int len = strlen(text);
  220: 
  221: 			for (i = state; cmdList[i].cmd_name; i++) {
  222: 				if (strncmp(cmdList[i].cmd_name, "---", 3) && 
  223: 						!strncmp(cmdList[i].cmd_name, text, len))
  224: 					return strdup(cmdList[i].cmd_name);
  225: 			}
  226: 
  227: 			return NULL;
  228: 		}
  229: 
  230: 		if (!start)
  231: 			matches = rl_completion_matches(text, cmdCompGet);
  232: 		else
  233: 			for (i = 0; cmdList[i].cmd_name; i++) {
  234: 				if (!cmdList[i].cmd_comp)
  235: 					continue;
  236: 				if (!strncmp(rl_line_buffer, cmdList[i].cmd_name, strlen(cmdList[i].cmd_name)))
  237: 					matches = rl_completion_matches(text, cmdList[i].cmd_comp);
  238: 			}
  239: 
  240: 		return matches;
  241: 	}
  242: 	char *cli_stdCompEntry(const char *ignore, int invoking_key)
  243: 	{
  244: 		return NULL;
  245: 	}
  246: 
  247: 	/* --- main body of CLI --- */
  248: 
  249: 	out = rl_outstream;
  250: 	if (!out)
  251: 		out = stdout;
  252: 
  253: 	rl_bind_key('?', inline_help);
  254: 	if (!rl_attempted_completion_function) 
  255: 		cliComp(cli_stdCompletion, cli_stdCompEntry);
  256: 
  257: 	do {
  258: 		line = readline(csPrompt);
  259: 		if (!line) {	// ^+d
  260: 			cli_Printf(out, "\n");
  261: 			break;
  262: 		}
  263: 		// clear whitespaces
  264: 		for (s = line; isspace(*s); s++);
  265: 		if (*s) {
  266: 			for (t = s + strlen(s) - 1; t > s && isspace(*t); t--);
  267: 			*++t = 0;
  268: 		}
  269: 
  270: 		if (*s) {
  271: 			add_history(s);
  272: 
  273: 			memset(items, 0, sizeof(char*) * MAX_PROMPT_ITEMS);
  274: 			for (app = items; app < items + MAX_PROMPT_ITEMS - 1 && (*app = strsep(&s, " \t")); 
  275: 					*app ? app++ : app);
  276: 
  277: 			/*
  278: 			for (i = 0; i < MAX_PROMPT_ITEMS; i++)
  279: 				cli_Printf(out, "i=%d %s\n", i, items[i]);
  280: 				*/
  281: 
  282: 			// exec_cmd ...
  283: 			for (cmd = NULL, i = 0; cmdList[i].cmd_name; i++)
  284: 				if (*items[0] && !strncmp(cmdList[i].cmd_name, items[0], strlen(items[0]))) {
  285: 					cmd = &cmdList[i];
  286: 					break;
  287: 				}
  288: 			if (!cmd) {
  289: 				cli_Printf(out, "Command '%s' not found!\n", items[0]);
  290: 				ret = -1;
  291: 			} else
  292: 				ret = cmd->cmd_func(cmdList, i, out, items);
  293: 		}
  294: 
  295: 		free(line);
  296: 	} while (ret < 1);
  297: 
  298: 	return ret;
  299: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>