From 524581393c335bcbcb8f4fb9c2deafe8b1018351 Mon Sep 17 00:00:00 2001 From: Paul Selkirk Date: Thu, 21 Jul 2016 14:52:46 -0400 Subject: Import of libcli from https://github.com/dparrish/libcli.git Upstream commit 958e44e7a69d3c71e89908fa8ee15232c55a821a --- libcli.c | 2353 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 2353 insertions(+) create mode 100644 libcli.c (limited to 'libcli.c') diff --git a/libcli.c b/libcli.c new file mode 100644 index 0000000..3893b2a --- /dev/null +++ b/libcli.c @@ -0,0 +1,2353 @@ +#ifdef WIN32 +#include +#include +#endif + +#define _GNU_SOURCE +#include +#include +#include +#include +#include +#if !defined(__APPLE__) && !defined(__FreeBSD__) +#include +#endif +#include +#include +#include +#include +#ifndef WIN32 +#include +#endif +#include "libcli.h" + +// vim:sw=4 tw=120 et + +#ifdef __GNUC__ +# define UNUSED(d) d __attribute__ ((unused)) +#else +# define UNUSED(d) d +#endif + +#define MATCH_REGEX 1 +#define MATCH_INVERT 2 + +#ifdef WIN32 +/* + * Stupid windows has multiple namespaces for filedescriptors, with different + * read/write functions required for each .. + */ +int read(int fd, void *buf, unsigned int count) { + return recv(fd, buf, count, 0); +} + +int write(int fd, const void *buf, unsigned int count) { + return send(fd, buf, count, 0); +} + +int vasprintf(char **strp, const char *fmt, va_list args) { + int size; + + size = vsnprintf(NULL, 0, fmt, args); + if ((*strp = malloc(size + 1)) == NULL) { + return -1; + } + + size = vsnprintf(*strp, size + 1, fmt, args); + return size; +} + +int asprintf(char **strp, const char *fmt, ...) { + va_list args; + int size; + + va_start(args, fmt); + size = vasprintf(strp, fmt, args); + + va_end(args); + return size; +} + +int fprintf(FILE *stream, const char *fmt, ...) { + va_list args; + int size; + char *buf; + + va_start(args, fmt); + size = vasprintf(&buf, fmt, args); + if (size < 0) { + goto out; + } + size = write(stream->_file, buf, size); + free(buf); + +out: + va_end(args); + return size; +} + +/* + * Dummy definitions to allow compilation on Windows + */ +int regex_dummy() {return 0;}; +#define regfree(...) regex_dummy() +#define regexec(...) regex_dummy() +#define regcomp(...) regex_dummy() +#define regex_t int +#define REG_NOSUB 0 +#define REG_EXTENDED 0 +#define REG_ICASE 0 +#endif + +enum cli_states { + STATE_LOGIN, + STATE_PASSWORD, + STATE_NORMAL, + STATE_ENABLE_PASSWORD, + STATE_ENABLE +}; + +struct unp { + char *username; + char *password; + struct unp *next; +}; + +struct cli_filter_cmds +{ + const char *cmd; + const char *help; +}; + +/* free and zero (to avoid double-free) */ +#define free_z(p) do { if (p) { free(p); (p) = 0; } } while (0) + +int cli_match_filter_init(struct cli_def *cli, int argc, char **argv, struct cli_filter *filt); +int cli_range_filter_init(struct cli_def *cli, int argc, char **argv, struct cli_filter *filt); +int cli_count_filter_init(struct cli_def *cli, int argc, char **argv, struct cli_filter *filt); +int cli_match_filter(struct cli_def *cli, const char *string, void *data); +int cli_range_filter(struct cli_def *cli, const char *string, void *data); +int cli_count_filter(struct cli_def *cli, const char *string, void *data); + +static struct cli_filter_cmds filter_cmds[] = +{ + { "begin", "Begin with lines that match" }, + { "between", "Between lines that match" }, + { "count", "Count of lines" }, + { "exclude", "Exclude lines that match" }, + { "include", "Include lines that match" }, + { "grep", "Include lines that match regex (options: -v, -i, -e)" }, + { "egrep", "Include lines that match extended regex" }, + { NULL, NULL} +}; + +static ssize_t _write(int fd, const void *buf, size_t count) +{ + size_t written = 0; + ssize_t thisTime =0; + while (count != written) + { + thisTime = write(fd, (char*)buf + written, count - written); + if (thisTime == -1) + { + if (errno == EINTR) + continue; + else + return -1; + } + written += thisTime; + } + return written; +} +char *cli_command_name(struct cli_def *cli, struct cli_command *command) +{ + char *name = cli->commandname; + char *o; + + if (name) free(name); + if (!(name = calloc(1, 1))) + return NULL; + + while (command) + { + o = name; + if (asprintf(&name, "%s%s%s", command->command, *o ? " " : "", o) == -1) + { + fprintf(stderr, "Couldn't allocate memory for command_name: %s", strerror(errno)); + free(o); + return NULL; + } + command = command->parent; + free(o); + } + cli->commandname = name; + return name; +} + +void cli_set_auth_callback(struct cli_def *cli, int (*auth_callback)(const char *, const char *)) +{ + cli->auth_callback = auth_callback; +} + +void cli_set_enable_callback(struct cli_def *cli, int (*enable_callback)(const char *)) +{ + cli->enable_callback = enable_callback; +} + +void cli_allow_user(struct cli_def *cli, const char *username, const char *password) +{ + struct unp *u, *n; + if (!(n = malloc(sizeof(struct unp)))) + { + fprintf(stderr, "Couldn't allocate memory for user: %s", strerror(errno)); + return; + } + if (!(n->username = strdup(username))) + { + fprintf(stderr, "Couldn't allocate memory for username: %s", strerror(errno)); + free(n); + return; + } + if (!(n->password = strdup(password))) + { + fprintf(stderr, "Couldn't allocate memory for password: %s", strerror(errno)); + free(n->username); + free(n); + return; + } + n->next = NULL; + + if (!cli->users) + { + cli->users = n; + } + else + { + for (u = cli->users; u && u->next; u = u->next); + if (u) u->next = n; + } +} + +void cli_allow_enable(struct cli_def *cli, const char *password) +{ + free_z(cli->enable_password); + if (!(cli->enable_password = strdup(password))) + { + fprintf(stderr, "Couldn't allocate memory for enable password: %s", strerror(errno)); + } +} + +void cli_deny_user(struct cli_def *cli, const char *username) +{ + struct unp *u, *p = NULL; + if (!cli->users) return; + for (u = cli->users; u; u = u->next) + { + if (strcmp(username, u->username) == 0) + { + if (p) + p->next = u->next; + else + cli->users = u->next; + free(u->username); + free(u->password); + free(u); + break; + } + p = u; + } +} + +void cli_set_banner(struct cli_def *cli, const char *banner) +{ + free_z(cli->banner); + if (banner && *banner) + cli->banner = strdup(banner); +} + +void cli_set_hostname(struct cli_def *cli, const char *hostname) +{ + free_z(cli->hostname); + if (hostname && *hostname) + cli->hostname = strdup(hostname); +} + +void cli_set_promptchar(struct cli_def *cli, const char *promptchar) +{ + free_z(cli->promptchar); + cli->promptchar = strdup(promptchar); +} + +static int cli_build_shortest(struct cli_def *cli, struct cli_command *commands) +{ + struct cli_command *c, *p; + char *cp, *pp; + unsigned len; + + for (c = commands; c; c = c->next) + { + c->unique_len = strlen(c->command); + if ((c->mode != MODE_ANY && c->mode != cli->mode) || c->privilege > cli->privilege) + continue; + + c->unique_len = 1; + for (p = commands; p; p = p->next) + { + if (c == p) + continue; + + if ((p->mode != MODE_ANY && p->mode != cli->mode) || p->privilege > cli->privilege) + continue; + + cp = c->command; + pp = p->command; + len = 1; + + while (*cp && *pp && *cp++ == *pp++) + len++; + + if (len > c->unique_len) + c->unique_len = len; + } + + if (c->children) + cli_build_shortest(cli, c->children); + } + + return CLI_OK; +} + +int cli_set_privilege(struct cli_def *cli, int priv) +{ + int old = cli->privilege; + cli->privilege = priv; + + if (priv != old) + { + cli_set_promptchar(cli, priv == PRIVILEGE_PRIVILEGED ? "# " : "> "); + cli_build_shortest(cli, cli->commands); + } + + return old; +} + +void cli_set_modestring(struct cli_def *cli, const char *modestring) +{ + free_z(cli->modestring); + if (modestring) + cli->modestring = strdup(modestring); +} + +int cli_set_configmode(struct cli_def *cli, int mode, const char *config_desc) +{ + int old = cli->mode; + cli->mode = mode; + + if (mode != old) + { + if (!cli->mode) + { + // Not config mode + cli_set_modestring(cli, NULL); + } + else if (config_desc && *config_desc) + { + char string[64]; + snprintf(string, sizeof(string), "(config-%s)", config_desc); + cli_set_modestring(cli, string); + } + else + { + cli_set_modestring(cli, "(config)"); + } + + cli_build_shortest(cli, cli->commands); + } + + return old; +} + +struct cli_command *cli_register_command(struct cli_def *cli, struct cli_command *parent, const char *command, int + (*callback)(struct cli_def *cli, const char *, char **, int), int privilege, + int mode, const char *help) +{ + struct cli_command *c, *p; + + if (!command) return NULL; + if (!(c = calloc(sizeof(struct cli_command), 1))) return NULL; + + c->callback = callback; + c->next = NULL; + if (!(c->command = strdup(command))) + return NULL; + c->parent = parent; + c->privilege = privilege; + c->mode = mode; + if (help && !(c->help = strdup(help))) + return NULL; + + if (parent) + { + if (!parent->children) + { + parent->children = c; + } + else + { + for (p = parent->children; p && p->next; p = p->next); + if (p) p->next = c; + } + } + else + { + if (!cli->commands) + { + cli->commands = c; + } + else + { + for (p = cli->commands; p && p->next; p = p->next); + if (p) p->next = c; + } + } + return c; +} + +static void cli_free_command(struct cli_command *cmd) +{ + struct cli_command *c, *p; + + for (c = cmd->children; c;) + { + p = c->next; + cli_free_command(c); + c = p; + } + + free(cmd->command); + if (cmd->help) free(cmd->help); + free(cmd); +} + +int cli_unregister_command(struct cli_def *cli, const char *command) +{ + struct cli_command *c, *p = NULL; + + if (!command) return -1; + if (!cli->commands) return CLI_OK; + + for (c = cli->commands; c; c = c->next) + { + if (strcmp(c->command, command) == 0) + { + if (p) + p->next = c->next; + else + cli->commands = c->next; + + cli_free_command(c); + return CLI_OK; + } + p = c; + } + + return CLI_OK; +} + +int cli_show_help(struct cli_def *cli, struct cli_command *c) +{ + struct cli_command *p; + + for (p = c; p; p = p->next) + { + if (p->command && p->callback && cli->privilege >= p->privilege && + (p->mode == cli->mode || p->mode == MODE_ANY)) + { + cli_error(cli, " %-20s %s", cli_command_name(cli, p), (p->help != NULL ? p->help : "")); + } + + if (p->children) + cli_show_help(cli, p->children); + } + + return CLI_OK; +} + +int cli_int_enable(struct cli_def *cli, UNUSED(const char *command), UNUSED(char *argv[]), UNUSED(int argc)) +{ + if (cli->privilege == PRIVILEGE_PRIVILEGED) + return CLI_OK; + + if (!cli->enable_password && !cli->enable_callback) + { + /* no password required, set privilege immediately */ + cli_set_privilege(cli, PRIVILEGE_PRIVILEGED); + cli_set_configmode(cli, MODE_EXEC, NULL); + } + else + { + /* require password entry */ + cli->state = STATE_ENABLE_PASSWORD; + } + + return CLI_OK; +} + +int cli_int_disable(struct cli_def *cli, UNUSED(const char *command), UNUSED(char *argv[]), UNUSED(int argc)) +{ + cli_set_privilege(cli, PRIVILEGE_UNPRIVILEGED); + cli_set_configmode(cli, MODE_EXEC, NULL); + return CLI_OK; +} + +int cli_int_help(struct cli_def *cli, UNUSED(const char *command), UNUSED(char *argv[]), UNUSED(int argc)) +{ + cli_error(cli, "\nCommands available:"); + cli_show_help(cli, cli->commands); + return CLI_OK; +} + +int cli_int_history(struct cli_def *cli, UNUSED(const char *command), UNUSED(char *argv[]), UNUSED(int argc)) +{ + int i; + + cli_error(cli, "\nCommand history:"); + for (i = 0; i < MAX_HISTORY; i++) + { + if (cli->history[i]) + cli_error(cli, "%3d. %s", i, cli->history[i]); + } + + return CLI_OK; +} + +int cli_int_quit(struct cli_def *cli, UNUSED(const char *command), UNUSED(char *argv[]), UNUSED(int argc)) +{ + cli_set_privilege(cli, PRIVILEGE_UNPRIVILEGED); + cli_set_configmode(cli, MODE_EXEC, NULL); + return CLI_QUIT; +} + +int cli_int_exit(struct cli_def *cli, const char *command, char *argv[], int argc) +{ + if (cli->mode == MODE_EXEC) + return cli_int_quit(cli, command, argv, argc); + + if (cli->mode > MODE_CONFIG) + cli_set_configmode(cli, MODE_CONFIG, NULL); + else + cli_set_configmode(cli, MODE_EXEC, NULL); + + cli->service = NULL; + return CLI_OK; +} + +int cli_int_idle_timeout(struct cli_def *cli) +{ + cli_print(cli, "Idle timeout"); + return CLI_QUIT; +} + +int cli_int_configure_terminal(struct cli_def *cli, UNUSED(const char *command), UNUSED(char *argv[]), UNUSED(int argc)) +{ + cli_set_configmode(cli, MODE_CONFIG, NULL); + return CLI_OK; +} + +struct cli_def *cli_init() +{ + struct cli_def *cli; + struct cli_command *c; + + if (!(cli = calloc(sizeof(struct cli_def), 1))) + return 0; + + cli->buf_size = 1024; + if (!(cli->buffer = calloc(cli->buf_size, 1))) + { + free_z(cli); + return 0; + } + cli->telnet_protocol = 1; + + cli_register_command(cli, 0, "help", cli_int_help, PRIVILEGE_UNPRIVILEGED, MODE_ANY, "Show available commands"); + cli_register_command(cli, 0, "quit", cli_int_quit, PRIVILEGE_UNPRIVILEGED, MODE_ANY, "Disconnect"); + cli_register_command(cli, 0, "logout", cli_int_quit, PRIVILEGE_UNPRIVILEGED, MODE_ANY, "Disconnect"); + cli_register_command(cli, 0, "exit", cli_int_exit, PRIVILEGE_UNPRIVILEGED, MODE_ANY, "Exit from current mode"); + cli_register_command(cli, 0, "history", cli_int_history, PRIVILEGE_UNPRIVILEGED, MODE_ANY, + "Show a list of previously run commands"); + cli_register_command(cli, 0, "enable", cli_int_enable, PRIVILEGE_UNPRIVILEGED, MODE_EXEC, + "Turn on privileged commands"); + cli_register_command(cli, 0, "disable", cli_int_disable, PRIVILEGE_PRIVILEGED, MODE_EXEC, + "Turn off privileged commands"); + + c = cli_register_command(cli, 0, "configure", 0, PRIVILEGE_PRIVILEGED, MODE_EXEC, "Enter configuration mode"); + cli_register_command(cli, c, "terminal", cli_int_configure_terminal, PRIVILEGE_PRIVILEGED, MODE_EXEC, + "Configure from the terminal"); + + cli->privilege = cli->mode = -1; + cli_set_privilege(cli, PRIVILEGE_UNPRIVILEGED); + cli_set_configmode(cli, MODE_EXEC, 0); + + // Default to 1 second timeout intervals + cli->timeout_tm.tv_sec = 1; + cli->timeout_tm.tv_usec = 0; + + // Set default idle timeout callback, but no timeout + cli_set_idle_timeout_callback(cli, 0, cli_int_idle_timeout); + return cli; +} + +void cli_unregister_all(struct cli_def *cli, struct cli_command *command) +{ + struct cli_command *c, *p = NULL; + + if (!command) command = cli->commands; + if (!command) return; + + for (c = command; c; ) + { + p = c->next; + + // Unregister all child commands + if (c->children) + cli_unregister_all(cli, c->children); + + if (c->command) free(c->command); + if (c->help) free(c->help); + free(c); + + c = p; + } +} + +int cli_done(struct cli_def *cli) +{ + struct unp *u = cli->users, *n; + + if (!cli) return CLI_OK; + cli_free_history(cli); + + // Free all users + while (u) + { + if (u->username) free(u->username); + if (u->password) free(u->password); + n = u->next; + free(u); + u = n; + } + + /* free all commands */ + cli_unregister_all(cli, 0); + + free_z(cli->commandname); + free_z(cli->modestring); + free_z(cli->banner); + free_z(cli->promptchar); + free_z(cli->hostname); + free_z(cli->buffer); + free_z(cli); + + return CLI_OK; +} + +static int cli_add_history(struct cli_def *cli, const char *cmd) +{ + int i; + for (i = 0; i < MAX_HISTORY; i++) + { + if (!cli->history[i]) + { + if (i == 0 || strcasecmp(cli->history[i-1], cmd)) + if (!(cli->history[i] = strdup(cmd))) + return CLI_ERROR; + return CLI_OK; + } + } + // No space found, drop one off the beginning of the list + free(cli->history[0]); + for (i = 0; i < MAX_HISTORY-1; i++) + cli->history[i] = cli->history[i+1]; + if (!(cli->history[MAX_HISTORY - 1] = strdup(cmd))) + return CLI_ERROR; + return CLI_OK; +} + +void cli_free_history(struct cli_def *cli) +{ + int i; + for (i = 0; i < MAX_HISTORY; i++) + { + if (cli->history[i]) + free_z(cli->history[i]); + } +} + +static int cli_parse_line(const char *line, char *words[], int max_words) +{ + int nwords = 0; + const char *p = line; + const char *word_start = 0; + int inquote = 0; + + while (*p) + { + if (!isspace(*p)) + { + word_start = p; + break; + } + p++; + } + + while (nwords < max_words - 1) + { + if (!*p || *p == inquote || (word_start && !inquote && (isspace(*p) || *p == '|'))) + { + if (word_start) + { + int len = p - word_start; + + memcpy(words[nwords] = malloc(len + 1), word_start, len); + words[nwords++][len] = 0; + } + + if (!*p) + break; + + if (inquote) + p++; /* skip over trailing quote */ + + inquote = 0; + word_start = 0; + } + else if (*p == '"' || *p == '\'') + { + inquote = *p++; + word_start = p; + } + else + { + if (!word_start) + { + if (*p == '|') + { + if (!(words[nwords++] = strdup("|"))) + return 0; + } + else if (!isspace(*p)) + word_start = p; + } + + p++; + } + } + + return nwords; +} + +static char *join_words(int argc, char **argv) +{ + char *p; + int len = 0; + int i; + + for (i = 0; i < argc; i++) + { + if (i) + len += 1; + + len += strlen(argv[i]); + } + + p = malloc(len + 1); + p[0] = 0; + + for (i = 0; i < argc; i++) + { + if (i) + strcat(p, " "); + + strcat(p, argv[i]); + } + + return p; +} + +static int cli_find_command(struct cli_def *cli, struct cli_command *commands, int num_words, char *words[], + int start_word, int filters[]) +{ + struct cli_command *c, *again_config = NULL, *again_any = NULL; + int c_words = num_words; + + if (filters[0]) + c_words = filters[0]; + + // Deal with ? for help + if (!words[start_word]) + return CLI_ERROR; + + if (words[start_word][strlen(words[start_word]) - 1] == '?') + { + int l = strlen(words[start_word])-1; + + if (commands->parent && commands->parent->callback) + cli_error(cli, "%-20s %s", cli_command_name(cli, commands->parent), + (commands->parent->help != NULL ? commands->parent->help : "")); + + for (c = commands; c; c = c->next) + { + if (strncasecmp(c->command, words[start_word], l) == 0 + && (c->callback || c->children) + && cli->privilege >= c->privilege + && (c->mode == cli->mode || c->mode == MODE_ANY)) + cli_error(cli, " %-20s %s", c->command, (c->help != NULL ? c->help : "")); + } + + return CLI_OK; + } + + for (c = commands; c; c = c->next) + { + if (cli->privilege < c->privilege) + continue; + + if (strncasecmp(c->command, words[start_word], c->unique_len)) + continue; + + if (strncasecmp(c->command, words[start_word], strlen(words[start_word]))) + continue; + + AGAIN: + if (c->mode == cli->mode || (c->mode == MODE_ANY && again_any != NULL)) + { + int rc = CLI_OK; + int f; + struct cli_filter **filt = &cli->filters; + + // Found a word! + if (!c->children) + { + // Last word + if (!c->callback) + { + cli_error(cli, "No callback for \"%s\"", cli_command_name(cli, c)); + return CLI_ERROR; + } + } + else + { + if (start_word == c_words - 1) + { + if (c->callback) + goto CORRECT_CHECKS; + + cli_error(cli, "Incomplete command"); + return CLI_ERROR; + } + rc = cli_find_command(cli, c->children, num_words, words, start_word + 1, filters); + if (rc == CLI_ERROR_ARG) + { + if (c->callback) + { + rc = CLI_OK; + goto CORRECT_CHECKS; + } + else + { + cli_error(cli, "Invalid %s \"%s\"", commands->parent ? "argument" : "command", + words[start_word]); + } + } + return rc; + } + + if (!c->callback) + { + cli_error(cli, "Internal server error processing \"%s\"", cli_command_name(cli, c)); + return CLI_ERROR; + } + + CORRECT_CHECKS: + for (f = 0; rc == CLI_OK && filters[f]; f++) + { + int n = num_words; + char **argv; + int argc; + int len; + + if (filters[f+1]) + n = filters[f+1]; + + if (filters[f] == n - 1) + { + cli_error(cli, "Missing filter"); + return CLI_ERROR; + } + + argv = words + filters[f] + 1; + argc = n - (filters[f] + 1); + len = strlen(argv[0]); + if (argv[argc - 1][strlen(argv[argc - 1]) - 1] == '?') + { + if (argc == 1) + { + int i; + for (i = 0; filter_cmds[i].cmd; i++) + cli_error(cli, " %-20s %s", filter_cmds[i].cmd, filter_cmds[i].help ); + } + else + { + if (argv[0][0] != 'c') // count + cli_error(cli, " WORD"); + + if (argc > 2 || argv[0][0] == 'c') // count + cli_error(cli, " "); + } + + return CLI_OK; + } + + if (argv[0][0] == 'b' && len < 3) // [beg]in, [bet]ween + { + cli_error(cli, "Ambiguous filter \"%s\" (begin, between)", argv[0]); + return CLI_ERROR; + } + *filt = calloc(sizeof(struct cli_filter), 1); + + if (!strncmp("include", argv[0], len) || !strncmp("exclude", argv[0], len) || + !strncmp("grep", argv[0], len) || !strncmp("egrep", argv[0], len)) + rc = cli_match_filter_init(cli, argc, argv, *filt); + else if (!strncmp("begin", argv[0], len) || !strncmp("between", argv[0], len)) + rc = cli_range_filter_init(cli, argc, argv, *filt); + else if (!strncmp("count", argv[0], len)) + rc = cli_count_filter_init(cli, argc, argv, *filt); + else + { + cli_error(cli, "Invalid filter \"%s\"", argv[0]); + rc = CLI_ERROR; + } + + if (rc == CLI_OK) + { + filt = &(*filt)->next; + } + else + { + free(*filt); + *filt = 0; + } + } + + if (rc == CLI_OK) + rc = c->callback(cli, cli_command_name(cli, c), words + start_word + 1, c_words - start_word - 1); + + while (cli->filters) + { + struct cli_filter *filt = cli->filters; + + // call one last time to clean up + filt->filter(cli, NULL, filt->data); + cli->filters = filt->next; + free(filt); + } + + return rc; + } + else if (cli->mode > MODE_CONFIG && c->mode == MODE_CONFIG) + { + // command matched but from another mode, + // remember it if we fail to find correct command + again_config = c; + } + else if (c->mode == MODE_ANY) + { + // command matched but for any mode, + // remember it if we fail to find correct command + again_any = c; + } + } + + // drop out of config submode if we have matched command on MODE_CONFIG + if (again_config) + { + c = again_config; + cli_set_configmode(cli, MODE_CONFIG, NULL); + goto AGAIN; + } + if (again_any) + { + c = again_any; + goto AGAIN; + } + + if (start_word == 0) + cli_error(cli, "Invalid %s \"%s\"", commands->parent ? "argument" : "command", words[start_word]); + + return CLI_ERROR_ARG; +} + +int cli_run_command(struct cli_def *cli, const char *command) +{ + int r; + unsigned int num_words, i, f; + char *words[CLI_MAX_LINE_WORDS] = {0}; + int filters[CLI_MAX_LINE_WORDS] = {0}; + + if (!command) return CLI_ERROR; + while (isspace(*command)) + command++; + + if (!*command) return CLI_OK; + + num_words = cli_parse_line(command, words, CLI_MAX_LINE_WORDS); + for (i = f = 0; i < num_words && f < CLI_MAX_LINE_WORDS - 1; i++) + { + if (words[i][0] == '|') + filters[f++] = i; + } + + filters[f] = 0; + + if (num_words) + r = cli_find_command(cli, cli->commands, num_words, words, 0, filters); + else + r = CLI_ERROR; + + for (i = 0; i < num_words; i++) + free(words[i]); + + if (r == CLI_QUIT) + return r; + + return CLI_OK; +} + +static int cli_get_completions(struct cli_def *cli, const char *command, char **completions, int max_completions) +{ + struct cli_command *c; + struct cli_command *n; + int num_words, save_words, i, k=0; + char *words[CLI_MAX_LINE_WORDS] = {0}; + int filter = 0; + + if (!command) return 0; + while (isspace(*command)) + command++; + + save_words = num_words = cli_parse_line(command, words, sizeof(words)/sizeof(words[0])); + if (!command[0] || command[strlen(command)-1] == ' ') + num_words++; + + if (!num_words) + goto out; + + for (i = 0; i < num_words; i++) + { + if (words[i] && words[i][0] == '|') + filter = i; + } + + if (filter) // complete filters + { + unsigned len = 0; + + if (filter < num_words - 1) // filter already completed + goto out; + + if (filter == num_words - 1) + len = strlen(words[num_words-1]); + + for (i = 0; filter_cmds[i].cmd && k < max_completions; i++) + { + if (!len || (len < strlen(filter_cmds[i].cmd) && !strncmp(filter_cmds[i].cmd, words[num_words - 1], len))) + completions[k++] = (char *)filter_cmds[i].cmd; + } + + completions[k] = NULL; + goto out; + } + + for (c = cli->commands, i = 0; c && i < num_words && k < max_completions; c = n) + { + n = c->next; + + if (cli->privilege < c->privilege) + continue; + + if (c->mode != cli->mode && c->mode != MODE_ANY) + continue; + + if (words[i] && strncasecmp(c->command, words[i], strlen(words[i]))) + continue; + + if (i < num_words - 1) + { + if (strlen(words[i]) < c->unique_len) + continue; + + n = c->children; + i++; + continue; + } + + completions[k++] = c->command; + } + +out: + for (i = 0; i < save_words; i++) + free(words[i]); + + return k; +} + +static void cli_clear_line(int sockfd, char *cmd, int l, int cursor) +{ + int i; + if (cursor < l) + { + for (i = 0; i < (l - cursor); i++) + _write(sockfd, " ", 1); + } + for (i = 0; i < l; i++) + cmd[i] = '\b'; + for (; i < l * 2; i++) + cmd[i] = ' '; + for (; i < l * 3; i++) + cmd[i] = '\b'; + _write(sockfd, cmd, i); + memset((char *)cmd, 0, i); + l = cursor = 0; +} + +void cli_reprompt(struct cli_def *cli) +{ + if (!cli) return; + cli->showprompt = 1; +} + +void cli_regular(struct cli_def *cli, int (*callback)(struct cli_def *cli)) +{ + if (!cli) return; + cli->regular_callback = callback; +} + +void cli_regular_interval(struct cli_def *cli, int seconds) +{ + if (seconds < 1) seconds = 1; + cli->timeout_tm.tv_sec = seconds; + cli->timeout_tm.tv_usec = 0; +} + +#define DES_PREFIX "{crypt}" /* to distinguish clear text from DES crypted */ +#define MD5_PREFIX "$1$" + +static int pass_matches(const char *pass, const char *try) +{ + int des; + if ((des = !strncasecmp(pass, DES_PREFIX, sizeof(DES_PREFIX)-1))) + pass += sizeof(DES_PREFIX)-1; + +#ifndef WIN32 + /* + * TODO - find a small crypt(3) function for use on windows + */ + if (des || !strncmp(pass, MD5_PREFIX, sizeof(MD5_PREFIX)-1)) + try = crypt(try, pass); +#endif + + return !strcmp(pass, try); +} + +#define CTRL(c) (c - '@') + +static int show_prompt(struct cli_def *cli, int sockfd) +{ + int len = 0; + + if (cli->hostname) + len += write(sockfd, cli->hostname, strlen(cli->hostname)); + + if (cli->modestring) + len += write(sockfd, cli->modestring, strlen(cli->modestring)); + + return len + write(sockfd, cli->promptchar, strlen(cli->promptchar)); +} + +int cli_loop(struct cli_def *cli, int sockfd) +{ + unsigned char c; + int n, l, oldl = 0, is_telnet_option = 0, skip = 0, esc = 0; + int cursor = 0, insertmode = 1; + char *cmd = NULL, *oldcmd = 0; + char *username = NULL, *password = NULL; + + cli_build_shortest(cli, cli->commands); + cli->state = STATE_LOGIN; + + cli_free_history(cli); + if (cli->telnet_protocol) + { + static const char *negotiate = + "\xFF\xFB\x03" + "\xFF\xFB\x01" + "\xFF\xFD\x03" + "\xFF\xFD\x01"; + _write(sockfd, negotiate, strlen(negotiate)); + } + + if ((cmd = malloc(CLI_MAX_LINE_LENGTH)) == NULL) + return CLI_ERROR; + +#ifdef WIN32 + /* + * OMG, HACK + */ + if (!(cli->client = fdopen(_open_osfhandle(sockfd, 0), "w+"))) + return CLI_ERROR; + cli->client->_file = sockfd; +#else + if (!(cli->client = fdopen(sockfd, "w+"))) + return CLI_ERROR; +#endif + + setbuf(cli->client, NULL); + if (cli->banner) + cli_error(cli, "%s", cli->banner); + + // Set the last action now so we don't time immediately + if (cli->idle_timeout) + time(&cli->last_action); + + /* start off in unprivileged mode */ + cli_set_privilege(cli, PRIVILEGE_UNPRIVILEGED); + cli_set_configmode(cli, MODE_EXEC, NULL); + + /* no auth required? */ + if (!cli->users && !cli->auth_callback) + cli->state = STATE_NORMAL; + + while (1) + { + signed int in_history = 0; + int lastchar = 0; + struct timeval tm; + + cli->showprompt = 1; + + if (oldcmd) + { + l = cursor = oldl; + oldcmd[l] = 0; + cli->showprompt = 1; + oldcmd = NULL; + oldl = 0; + } + else + { + memset(cmd, 0, CLI_MAX_LINE_LENGTH); + l = 0; + cursor = 0; + } + + memcpy(&tm, &cli->timeout_tm, sizeof(tm)); + + while (1) + { + int sr; + fd_set r; + if (cli->showprompt) + { + if (cli->state != STATE_PASSWORD && cli->state != STATE_ENABLE_PASSWORD) + _write(sockfd, "\r\n", 2); + + switch (cli->state) + { + case STATE_LOGIN: + _write(sockfd, "Username: ", strlen("Username: ")); + break; + + case STATE_PASSWORD: + _write(sockfd, "Password: ", strlen("Password: ")); + break; + + case STATE_NORMAL: + case STATE_ENABLE: + show_prompt(cli, sockfd); + _write(sockfd, cmd, l); + if (cursor < l) + { + int n = l - cursor; + while (n--) + _write(sockfd, "\b", 1); + } + break; + + case STATE_ENABLE_PASSWORD: + _write(sockfd, "Password: ", strlen("Password: ")); + break; + + } + + cli->showprompt = 0; + } + + FD_ZERO(&r); + FD_SET(sockfd, &r); + + if ((sr = select(sockfd + 1, &r, NULL, NULL, &tm)) < 0) + { + /* select error */ + if (errno == EINTR) + continue; + + perror("select"); + l = -1; + break; + } + + if (sr == 0) + { + /* timeout every second */ + if (cli->regular_callback && cli->regular_callback(cli) != CLI_OK) + { + l = -1; + break; + } + + if (cli->idle_timeout) + { + if (time(NULL) - cli->last_action >= cli->idle_timeout) + { + if (cli->idle_timeout_callback) + { + // Call the callback and continue on if successful + if (cli->idle_timeout_callback(cli) == CLI_OK) + { + // Reset the idle timeout counter + time(&cli->last_action); + continue; + } + } + // Otherwise, break out of the main loop + l = -1; + break; + } + } + + memcpy(&tm, &cli->timeout_tm, sizeof(tm)); + continue; + } + + if ((n = read(sockfd, &c, 1)) < 0) + { + if (errno == EINTR) + continue; + + perror("read"); + l = -1; + break; + } + + if (cli->idle_timeout) + time(&cli->last_action); + + if (n == 0) + { + l = -1; + break; + } + + if (skip) + { + skip--; + continue; + } + + if (c == 255 && !is_telnet_option) + { + is_telnet_option++; + continue; + } + + if (is_telnet_option) + { + if (c >= 251 && c <= 254) + { + is_telnet_option = c; + continue; + } + + if (c != 255) + { + is_telnet_option = 0; + continue; + } + + is_telnet_option = 0; + } + + /* handle ANSI arrows */ + if (esc) + { + if (esc == '[') + { + /* remap to readline control codes */ + switch (c) + { + case 'A': /* Up */ + c = CTRL('P'); + break; + + case 'B': /* Down */ + c = CTRL('N'); + break; + + case 'C': /* Right */ + c = CTRL('F'); + break; + + case 'D': /* Left */ + c = CTRL('B'); + break; + + default: + c = 0; + } + + esc = 0; + } + else + { + esc = (c == '[') ? c : 0; + continue; + } + } + + if (c == 0) continue; + if (c == '\n') continue; + + if (c == '\r') + { + if (cli->state != STATE_PASSWORD && cli->state != STATE_ENABLE_PASSWORD) + _write(sockfd, "\r\n", 2); + break; + } + + if (c == 27) + { + esc = 1; + continue; + } + + if (c == CTRL('C')) + { + _write(sockfd, "\a", 1); + continue; + } + + /* back word, backspace/delete */ + if (c == CTRL('W') || c == CTRL('H') || c == 0x7f) + { + int back = 0; + + if (c == CTRL('W')) /* word */ + { + int nc = cursor; + + if (l == 0 || cursor == 0) + continue; + + while (nc && cmd[nc - 1] == ' ') + { + nc--; + back++; + } + + while (nc && cmd[nc - 1] != ' ') + { + nc--; + back++; + } + } + else /* char */ + { + if (l == 0 || cursor == 0) + { + _write(sockfd, "\a", 1); + continue; + } + + back = 1; + } + + if (back) + { + while (back--) + { + if (l == cursor) + { + cmd[--cursor] = 0; + if (cli->state != STATE_PASSWORD && cli->state != STATE_ENABLE_PASSWORD) + _write(sockfd, "\b \b", 3); + } + else + { + int i; + cursor--; + if (cli->state != STATE_PASSWORD && cli->state != STATE_ENABLE_PASSWORD) + { + for (i = cursor; i <= l; i++) cmd[i] = cmd[i+1]; + _write(sockfd, "\b", 1); + _write(sockfd, cmd + cursor, strlen(cmd + cursor)); + _write(sockfd, " ", 1); + for (i = 0; i <= (int)strlen(cmd + cursor); i++) + _write(sockfd, "\b", 1); + } + } + l--; + } + + continue; + } + } + + /* redraw */ + if (c == CTRL('L')) + { + int i; + int cursorback = l - cursor; + + if (cli->state == STATE_PASSWORD || cli->state == STATE_ENABLE_PASSWORD) + continue; + + _write(sockfd, "\r\n", 2); + show_prompt(cli, sockfd); + _write(sockfd, cmd, l); + + for (i = 0; i < cursorback; i++) + _write(sockfd, "\b", 1); + + continue; + } + + /* clear line */ + if (c == CTRL('U')) + { + if (cli->state == STATE_PASSWORD || cli->state == STATE_ENABLE_PASSWORD) + memset(cmd, 0, l); + else + cli_clear_line(sockfd, cmd, l, cursor); + + l = cursor = 0; + continue; + } + + /* kill to EOL */ + if (c == CTRL('K')) + { + if (cursor == l) + continue; + + if (cli->state != STATE_PASSWORD && cli->state != STATE_ENABLE_PASSWORD) + { + int c; + for (c = cursor; c < l; c++) + _write(sockfd, " ", 1); + + for (c = cursor; c < l; c++) + _write(sockfd, "\b", 1); + } + + memset(cmd + cursor, 0, l - cursor); + l = cursor; + continue; + } + + /* EOT */ + if (c == CTRL('D')) + { + if (cli->state == STATE_PASSWORD || cli->state == STATE_ENABLE_PASSWORD) + break; + + if (l) + continue; + + l = -1; + break; + } + + /* disable */ + if (c == CTRL('Z')) + { + if (cli->mode != MODE_EXEC) + { + cli_clear_line(sockfd, cmd, l, cursor); + cli_set_configmode(cli, MODE_EXEC, NULL); + cli->showprompt = 1; + } + + continue; + } + + /* TAB completion */ + if (c == CTRL('I')) + { + char *completions[CLI_MAX_LINE_WORDS]; + int num_completions = 0; + + if (cli->state == STATE_LOGIN || cli->state == STATE_PASSWORD || cli->state == STATE_ENABLE_PASSWORD) + continue; + + if (cursor != l) continue; + + num_completions = cli_get_completions(cli, cmd, completions, CLI_MAX_LINE_WORDS); + if (num_completions == 0) + { + _write(sockfd, "\a", 1); + } + else if (num_completions == 1) + { + // Single completion + for (; l > 0; l--, cursor--) + { + if (cmd[l-1] == ' ' || cmd[l-1] == '|') + break; + _write(sockfd, "\b", 1); + } + strcpy((cmd + l), completions[0]); + l += strlen(completions[0]); + cmd[l++] = ' '; + cursor = l; + _write(sockfd, completions[0], strlen(completions[0])); + _write(sockfd, " ", 1); + } + else if (lastchar == CTRL('I')) + { + // double tab + int i; + _write(sockfd, "\r\n", 2); + for (i = 0; i < num_completions; i++) + { + _write(sockfd, completions[i], strlen(completions[i])); + if (i % 4 == 3) + _write(sockfd, "\r\n", 2); + else + _write(sockfd, " ", 1); + } + if (i % 4 != 3) _write(sockfd, "\r\n", 2); + cli->showprompt = 1; + } + else + { + // More than one completion + lastchar = c; + _write(sockfd, "\a", 1); + } + continue; + } + + /* history */ + if (c == CTRL('P') || c == CTRL('N')) + { + int history_found = 0; + + if (cli->state == STATE_LOGIN || cli->state == STATE_PASSWORD || cli->state == STATE_ENABLE_PASSWORD) + continue; + + if (c == CTRL('P')) // Up + { + in_history--; + if (in_history < 0) + { + for (in_history = MAX_HISTORY-1; in_history >= 0; in_history--) + { + if (cli->history[in_history]) + { + history_found = 1; + break; + } + } + } + else + { + if (cli->history[in_history]) history_found = 1; + } + } + else // Down + { + in_history++; + if (in_history >= MAX_HISTORY || !cli->history[in_history]) + { + int i = 0; + for (i = 0; i < MAX_HISTORY; i++) + { + if (cli->history[i]) + { + in_history = i; + history_found = 1; + break; + } + } + } + else + { + if (cli->history[in_history]) history_found = 1; + } + } + if (history_found && cli->history[in_history]) + { + // Show history item + cli_clear_line(sockfd, cmd, l, cursor); + memset(cmd, 0, CLI_MAX_LINE_LENGTH); + strncpy(cmd, cli->history[in_history], CLI_MAX_LINE_LENGTH - 1); + l = cursor = strlen(cmd); + _write(sockfd, cmd, l); + } + + continue; + } + + /* left/right cursor motion */ + if (c == CTRL('B') || c == CTRL('F')) + { + if (c == CTRL('B')) /* Left */ + { + if (cursor) + { + if (cli->state != STATE_PASSWORD && cli->state != STATE_ENABLE_PASSWORD) + _write(sockfd, "\b", 1); + + cursor--; + } + } + else /* Right */ + { + if (cursor < l) + { + if (cli->state != STATE_PASSWORD && cli->state != STATE_ENABLE_PASSWORD) + _write(sockfd, &cmd[cursor], 1); + + cursor++; + } + } + + continue; + } + + /* start of line */ + if (c == CTRL('A')) + { + if (cursor) + { + if (cli->state != STATE_PASSWORD && cli->state != STATE_ENABLE_PASSWORD) + { + _write(sockfd, "\r", 1); + show_prompt(cli, sockfd); + } + + cursor = 0; + } + + continue; + } + + /* end of line */ + if (c == CTRL('E')) + { + if (cursor < l) + { + if (cli->state != STATE_PASSWORD && cli->state != STATE_ENABLE_PASSWORD) + _write(sockfd, &cmd[cursor], l - cursor); + + cursor = l; + } + + continue; + } + + /* normal character typed */ + if (cursor == l) + { + /* append to end of line */ + cmd[cursor] = c; + if (l < CLI_MAX_LINE_LENGTH - 1) + { + l++; + cursor++; + } + else + { + _write(sockfd, "\a", 1); + continue; + } + } + else + { + // Middle of text + if (insertmode) + { + int i; + // Move everything one character to the right + if (l >= CLI_MAX_LINE_LENGTH - 2) l--; + for (i = l; i >= cursor; i--) + cmd[i + 1] = cmd[i]; + // Write what we've just added + cmd[cursor] = c; + + _write(sockfd, &cmd[cursor], l - cursor + 1); + for (i = 0; i < (l - cursor + 1); i++) + _write(sockfd, "\b", 1); + l++; + } + else + { + cmd[cursor] = c; + } + cursor++; + } + + if (cli->state != STATE_PASSWORD && cli->state != STATE_ENABLE_PASSWORD) + { + if (c == '?' && cursor == l) + { + _write(sockfd, "\r\n", 2); + oldcmd = cmd; + oldl = cursor = l - 1; + break; + } + _write(sockfd, &c, 1); + } + + oldcmd = 0; + oldl = 0; + lastchar = c; + } + + if (l < 0) break; + + if (cli->state == STATE_LOGIN) + { + if (l == 0) continue; + + /* require login */ + free_z(username); + if (!(username = strdup(cmd))) + return 0; + cli->state = STATE_PASSWORD; + cli->showprompt = 1; + } + else if (cli->state == STATE_PASSWORD) + { + /* require password */ + int allowed = 0; + + free_z(password); + if (!(password = strdup(cmd))) + return 0; + if (cli->auth_callback) + { + if (cli->auth_callback(username, password) == CLI_OK) + allowed++; + } + + if (!allowed) + { + struct unp *u; + for (u = cli->users; u; u = u->next) + { + if (!strcmp(u->username, username) && pass_matches(u->password, password)) + { + allowed++; + break; + } + } + } + + if (allowed) + { + cli_error(cli, " "); + cli->state = STATE_NORMAL; + } + else + { + cli_error(cli, "\n\nAccess denied"); + free_z(username); + free_z(password); + cli->state = STATE_LOGIN; + } + + cli->showprompt = 1; + } + else if (cli->state == STATE_ENABLE_PASSWORD) + { + int allowed = 0; + if (cli->enable_password) + { + /* check stored static enable password */ + if (pass_matches(cli->enable_password, cmd)) + allowed++; + } + + if (!allowed && cli->enable_callback) + { + /* check callback */ + if (cli->enable_callback(cmd)) + allowed++; + } + + if (allowed) + { + cli->state = STATE_ENABLE; + cli_set_privilege(cli, PRIVILEGE_PRIVILEGED); + } + else + { + cli_error(cli, "\n\nAccess denied"); + cli->state = STATE_NORMAL; + } + } + else + { + if (l == 0) continue; + if (cmd[l - 1] != '?' && strcasecmp(cmd, "history") != 0) + cli_add_history(cli, cmd); + + if (cli_run_command(cli, cmd) == CLI_QUIT) + break; + } + + // Update the last_action time now as the last command run could take a + // long time to return + if (cli->idle_timeout) + time(&cli->last_action); + } + + cli_free_history(cli); + free_z(username); + free_z(password); + free_z(cmd); + + fclose(cli->client); + cli->client = 0; + return CLI_OK; +} + +int cli_file(struct cli_def *cli, FILE *fh, int privilege, int mode) +{ + int oldpriv = cli_set_privilege(cli, privilege); + int oldmode = cli_set_configmode(cli, mode, NULL); + char buf[CLI_MAX_LINE_LENGTH]; + + while (1) + { + char *p; + char *cmd; + char *end; + + if (fgets(buf, CLI_MAX_LINE_LENGTH - 1, fh) == NULL) + break; /* end of file */ + + if ((p = strpbrk(buf, "#\r\n"))) + *p = 0; + + cmd = buf; + while (isspace(*cmd)) + cmd++; + + if (!*cmd) + continue; + + for (p = end = cmd; *p; p++) + if (!isspace(*p)) + end = p; + + *++end = 0; + if (strcasecmp(cmd, "quit") == 0) + break; + + if (cli_run_command(cli, cmd) == CLI_QUIT) + break; + } + + cli_set_privilege(cli, oldpriv); + cli_set_configmode(cli, oldmode, NULL /* didn't save desc */); + + return CLI_OK; +} + +static void _print(struct cli_def *cli, int print_mode, const char *format, va_list ap) +{ + va_list aq; + int n; + char *p; + + if (!cli) return; // sanity check + + while (1) + { + va_copy(aq, ap); + if ((n = vsnprintf(cli->buffer, cli->buf_size, format, ap)) == -1) + return; + + if ((unsigned)n >= cli->buf_size) + { + cli->buf_size = n + 1; + cli->buffer = realloc(cli->buffer, cli->buf_size); + if (!cli->buffer) + return; + va_end(ap); + va_copy(ap, aq); + continue; + } + break; + } + + + p = cli->buffer; + do + { + char *next = strchr(p, '\n'); + struct cli_filter *f = (print_mode & PRINT_FILTERED) ? cli->filters : 0; + int print = 1; + + if (next) + *next++ = 0; + else if (print_mode & PRINT_BUFFERED) + break; + + while (print && f) + { + print = (f->filter(cli, p, f->data) == CLI_OK); + f = f->next; + } + if (print) + { + if (cli->print_callback) + cli->print_callback(cli, p); + else if (cli->client) + fprintf(cli->client, "%s\r\n", p); + } + + p = next; + } while (p); + + if (p && *p) + { + if (p != cli->buffer) + memmove(cli->buffer, p, strlen(p)); + } + else *cli->buffer = 0; +} + +void cli_bufprint(struct cli_def *cli, const char *format, ...) +{ + va_list ap; + + va_start(ap, format); + _print(cli, PRINT_BUFFERED|PRINT_FILTERED, format, ap); + va_end(ap); +} + +void cli_vabufprint(struct cli_def *cli, const char *format, va_list ap) +{ + _print(cli, PRINT_BUFFERED, format, ap); +} + +void cli_print(struct cli_def *cli, const char *format, ...) +{ + va_list ap; + + va_start(ap, format); + _print(cli, PRINT_FILTERED, format, ap); + va_end(ap); +} + +void cli_error(struct cli_def *cli, const char *format, ...) +{ + va_list ap; + + va_start(ap, format); + _print(cli, PRINT_PLAIN, format, ap); + va_end(ap); +} + +struct cli_match_filter_state +{ + int flags; + union { + char *string; + regex_t re; + } match; +}; + +int cli_match_filter_init(struct cli_def *cli, int argc, char **argv, struct cli_filter *filt) +{ + struct cli_match_filter_state *state; + int rflags; + int i; + char *p; + + if (argc < 2) + { + if (cli->client) + fprintf(cli->client, "Match filter requires an argument\r\n"); + + return CLI_ERROR; + } + + filt->filter = cli_match_filter; + filt->data = state = calloc(sizeof(struct cli_match_filter_state), 1); + + if (argv[0][0] == 'i' || (argv[0][0] == 'e' && argv[0][1] == 'x')) // include/exclude + { + if (argv[0][0] == 'e') + state->flags = MATCH_INVERT; + + state->match.string = join_words(argc-1, argv+1); + return CLI_OK; + } + +#ifdef WIN32 + /* + * No regex functions in windows, so return an error + */ + return CLI_ERROR; +#endif + + state->flags = MATCH_REGEX; + + // grep/egrep + rflags = REG_NOSUB; + if (argv[0][0] == 'e') // egrep + rflags |= REG_EXTENDED; + + i = 1; + while (i < argc - 1 && argv[i][0] == '-' && argv[i][1]) + { + int last = 0; + p = &argv[i][1]; + + if (strspn(p, "vie") != strlen(p)) + break; + + while (*p) + { + switch (*p++) + { + case 'v': + state->flags |= MATCH_INVERT; + break; + + case 'i': + rflags |= REG_ICASE; + break; + + case 'e': + last++; + break; + } + } + + i++; + if (last) + break; + } + + p = join_words(argc-i, argv+i); + if ((i = regcomp(&state->match.re, p, rflags))) + { + if (cli->client) + fprintf(cli->client, "Invalid pattern \"%s\"\r\n", p); + + free_z(p); + return CLI_ERROR; + } + + free_z(p); + return CLI_OK; +} + +int cli_match_filter(UNUSED(struct cli_def *cli), const char *string, void *data) +{ + struct cli_match_filter_state *state = data; + int r = CLI_ERROR; + + if (!string) // clean up + { + if (state->flags & MATCH_REGEX) + regfree(&state->match.re); + else + free(state->match.string); + + free(state); + return CLI_OK; + } + + if (state->flags & MATCH_REGEX) + { + if (!regexec(&state->match.re, string, 0, NULL, 0)) + r = CLI_OK; + } + else + { + if (strstr(string, state->match.string)) + r = CLI_OK; + } + + if (state->flags & MATCH_INVERT) + { + if (r == CLI_OK) + r = CLI_ERROR; + else + r = CLI_OK; + } + + return r; +} + +struct cli_range_filter_state { + int matched; + char *from; + char *to; +}; + +int cli_range_filter_init(struct cli_def *cli, int argc, char **argv, struct cli_filter *filt) +{ + struct cli_range_filter_state *state; + char *from = 0; + char *to = 0; + + if (!strncmp(argv[0], "bet", 3)) // between + { + if (argc < 3) + { + if (cli->client) + fprintf(cli->client, "Between filter requires 2 arguments\r\n"); + + return CLI_ERROR; + } + + if (!(from = strdup(argv[1]))) + return CLI_ERROR; + to = join_words(argc-2, argv+2); + } + else // begin + { + if (argc < 2) + { + if (cli->client) + fprintf(cli->client, "Begin filter requires an argument\r\n"); + + return CLI_ERROR; + } + + from = join_words(argc-1, argv+1); + } + + filt->filter = cli_range_filter; + filt->data = state = calloc(sizeof(struct cli_range_filter_state), 1); + + state->from = from; + state->to = to; + + return CLI_OK; +} + +int cli_range_filter(UNUSED(struct cli_def *cli), const char *string, void *data) +{ + struct cli_range_filter_state *state = data; + int r = CLI_ERROR; + + if (!string) // clean up + { + free_z(state->from); + free_z(state->to); + free_z(state); + return CLI_OK; + } + + if (!state->matched) + state->matched = !!strstr(string, state->from); + + if (state->matched) + { + r = CLI_OK; + if (state->to && strstr(string, state->to)) + state->matched = 0; + } + + return r; +} + +int cli_count_filter_init(struct cli_def *cli, int argc, UNUSED(char **argv), struct cli_filter *filt) +{ + if (argc > 1) + { + if (cli->client) + fprintf(cli->client, "Count filter does not take arguments\r\n"); + + return CLI_ERROR; + } + + filt->filter = cli_count_filter; + if (!(filt->data = calloc(sizeof(int), 1))) + return CLI_ERROR; + + return CLI_OK; +} + +int cli_count_filter(struct cli_def *cli, const char *string, void *data) +{ + int *count = data; + + if (!string) // clean up + { + // print count + if (cli->client) + fprintf(cli->client, "%d\r\n", *count); + + free(count); + return CLI_OK; + } + + while (isspace(*string)) + string++; + + if (*string) + (*count)++; // only count non-blank lines + + return CLI_ERROR; // no output +} + +void cli_print_callback(struct cli_def *cli, void (*callback)(struct cli_def *, const char *)) +{ + cli->print_callback = callback; +} + +void cli_set_idle_timeout(struct cli_def *cli, unsigned int seconds) +{ + if (seconds < 1) + seconds = 0; + cli->idle_timeout = seconds; + time(&cli->last_action); +} + +void cli_set_idle_timeout_callback(struct cli_def *cli, unsigned int seconds, int (*callback)(struct cli_def *)) +{ + cli_set_idle_timeout(cli, seconds); + cli->idle_timeout_callback = callback; +} + +void cli_telnet_protocol(struct cli_def *cli, int telnet_protocol) { + cli->telnet_protocol = !!telnet_protocol; +} + +void cli_set_context(struct cli_def *cli, void *context) { + cli->user_context = context; +} + +void *cli_get_context(struct cli_def *cli) { + return cli->user_context; +} -- cgit v1.2.3