radare2/libr/cons/input.c

/* radare - LGPL - Copyright 2009 pancake<nopcode.org> */

#include <r_cons.h>
#include <string.h>
#if HAVE_DIETLINE
#include <r_line.h>
#endif

#if 0
#define CMDS 54
static const char *radare_argv[CMDS] ={
	NULL, // padding
	"? ", "!step ", "!stepo ", "!cont ", "!signal ", "!fd ", "!maps ", ".!maps*",
	"!bp ", "!!", "#md5", "#sha1", "#crc32", "#entropy", "Visual", "ad", "ac",
	"ag", "emenu ", "eval ", "seek ", "info ", "help ", "move ", "quit ", "flag ",
	"Po ", "Ps ", "Pi ", "H ", "H no ", "H nj ", "H fj ", "H lua ", "x ", "b ",
	"y ", "yy ", "y? ", "wx ", "ww ", "wf ", "w?", "pD ", "pG ", "pb ", "px ",
	"pX ", "po ", "pm ", "pz ", "pr > ", "p? "
};
#endif

int (*r_cons_user_fgets)(char *buf, int len) = NULL;

char *dl_readline(int argc, const char **argv);

// XXX no control for max length here?!?!
R_API int r_cons_fgets(char *buf, int len, int argc, const char **argv)
{
	if (r_cons_user_fgets)
		return r_cons_user_fgets (buf, 512);

#if HAVE_DIETLINE
	/* TODO: link against dietline if possible for autocompletion */
	char *ptr;
	buf[0]='\0';
	ptr = r_line_readline((argv)?argc:CMDS, (argv)?argv:radare_argv);
	if (ptr == NULL)
		return -1;
	strncpy(buf, ptr, len);
#else
	buf[0]='\0';
	if (fgets(buf, len, r_cons_stdin_fd) == NULL)
		return -1;
	if (feof (r_cons_stdin_fd))
		return -1;
	buf[strlen(buf)-1]='\0';
#endif
	return strlen(buf);
}


R_API void r_cons_any_key()
{
	r_cons_strcat("\n--press any key--\n");
	r_cons_flush();
	r_cons_readchar();
	r_cons_strcat("\x1b[2J\x1b[0;0H");
}

R_API int r_cons_readchar()
{
	char buf[2];
#if __WINDOWS__
	DWORD out;
	BOOL ret;
	LPDWORD mode;
	HANDLE h = GetStdHandle(STD_INPUT_HANDLE);
	GetConsoleMode(h, &mode);
	SetConsoleMode(h, 0); // RAW
	ret = ReadConsole(h, buf,1, &out, NULL);
	if (!ret)
		return -1;
	SetConsoleMode(h, mode);
#else
	r_cons_set_raw(1);
	if (read(0, buf, 1)==-1)
		return -1;
	r_cons_set_raw(0);
#endif
	return buf[0];
}