esp-idf/components/console/commands.c

334 lines
9.6 KiB
C
Raw Normal View History

2021-05-16 21:41:32 -04:00
/*
* SPDX-FileCopyrightText: 2016-2021 Espressif Systems (Shanghai) CO LTD
*
* SPDX-License-Identifier: Apache-2.0
*/
2017-08-15 04:11:19 -04:00
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/param.h>
2023-08-02 02:24:27 -04:00
#include "esp_heap_caps.h"
2017-08-15 04:11:19 -04:00
#include "esp_log.h"
#include "esp_console.h"
#include "esp_system.h"
2017-08-15 04:11:19 -04:00
#include "linenoise/linenoise.h"
#include "argtable3/argtable3.h"
#include "sys/queue.h"
2017-08-15 04:11:19 -04:00
#define ANSI_COLOR_DEFAULT 39 /** Default foreground color */
typedef struct cmd_item_ {
/**
* Command name (statically allocated by application)
*/
2018-04-30 04:52:45 -04:00
const char *command;
2017-08-15 04:11:19 -04:00
/**
* Help text (statically allocated by application), may be NULL.
*/
2018-04-30 04:52:45 -04:00
const char *help;
2017-08-15 04:11:19 -04:00
/**
* Hint text, usually lists possible arguments, dynamically allocated.
* May be NULL.
*/
2018-04-30 04:52:45 -04:00
char *hint;
esp_console_cmd_func_t func; //!< pointer to the command handler (without user context)
esp_console_cmd_func_with_context_t func_w_context; //!< pointer to the command handler (with user context)
void *argtable; //!< optional pointer to arg table
void *context; //!< optional pointer to user context
SLIST_ENTRY(cmd_item_) next; //!< next command in the list
2017-08-15 04:11:19 -04:00
} cmd_item_t;
/** linked list of command structures */
static SLIST_HEAD(cmd_list_, cmd_item_) s_cmd_list;
/** run-time configuration options */
static esp_console_config_t s_config = {
.heap_alloc_caps = MALLOC_CAP_DEFAULT
};
2017-08-15 04:11:19 -04:00
/** temporary buffer used for command line parsing */
2018-04-30 04:52:45 -04:00
static char *s_tmp_line_buf;
2017-08-15 04:11:19 -04:00
2018-04-30 04:52:45 -04:00
static const cmd_item_t *find_command_by_name(const char *name);
2017-08-15 04:11:19 -04:00
2018-04-30 04:52:45 -04:00
esp_err_t esp_console_init(const esp_console_config_t *config)
2017-08-15 04:11:19 -04:00
{
if (!config) {
return ESP_ERR_INVALID_ARG;
}
2017-08-15 04:11:19 -04:00
if (s_tmp_line_buf) {
return ESP_ERR_INVALID_STATE;
}
memcpy(&s_config, config, sizeof(s_config));
if (s_config.hint_color == 0) {
s_config.hint_color = ANSI_COLOR_DEFAULT;
}
if (s_config.heap_alloc_caps == 0) {
s_config.heap_alloc_caps = MALLOC_CAP_DEFAULT;
2023-08-02 02:24:27 -04:00
}
s_tmp_line_buf = heap_caps_calloc(1, config->max_cmdline_length, s_config.heap_alloc_caps);
2017-08-15 04:11:19 -04:00
if (s_tmp_line_buf == NULL) {
return ESP_ERR_NO_MEM;
}
return ESP_OK;
}
esp_err_t esp_console_deinit(void)
2017-08-15 04:11:19 -04:00
{
if (!s_tmp_line_buf) {
return ESP_ERR_INVALID_STATE;
}
free(s_tmp_line_buf);
s_tmp_line_buf = NULL;
2017-08-15 04:11:19 -04:00
cmd_item_t *it, *tmp;
SLIST_FOREACH_SAFE(it, &s_cmd_list, next, tmp) {
SLIST_REMOVE(&s_cmd_list, it, cmd_item_, next);
2017-08-15 04:11:19 -04:00
free(it->hint);
free(it);
}
return ESP_OK;
}
esp_err_t esp_console_cmd_register(const esp_console_cmd_t *cmd)
{
cmd_item_t *item = NULL;
if (!cmd || cmd->command == NULL) {
2017-08-15 04:11:19 -04:00
return ESP_ERR_INVALID_ARG;
}
if (strchr(cmd->command, ' ') != NULL) {
return ESP_ERR_INVALID_ARG;
}
if ((cmd->func == NULL && cmd->func_w_context == NULL)
|| (cmd->func != NULL && cmd->func_w_context != NULL)) {
return ESP_ERR_INVALID_ARG;
}
item = (cmd_item_t *)find_command_by_name(cmd->command);
if (!item) {
// not registered before
item = heap_caps_calloc(1, sizeof(*item), s_config.heap_alloc_caps);
if (item == NULL) {
return ESP_ERR_NO_MEM;
}
} else {
// remove from list and free the old hint, because we will alloc new hint for the command
SLIST_REMOVE(&s_cmd_list, item, cmd_item_, next);
free(item->hint);
}
2017-08-15 04:11:19 -04:00
item->command = cmd->command;
item->help = cmd->help;
if (cmd->hint) {
/* Prepend a space before the hint. It separates command name and
* the hint. arg_print_syntax below adds this space as well.
*/
int unused __attribute__((unused));
unused = asprintf(&item->hint, " %s", cmd->hint);
2017-08-15 04:11:19 -04:00
} else if (cmd->argtable) {
/* Generate hint based on cmd->argtable */
2018-04-30 04:52:45 -04:00
char *buf = NULL;
size_t buf_size = 0;
2018-04-30 04:52:45 -04:00
FILE *f = open_memstream(&buf, &buf_size);
if (f != NULL) {
arg_print_syntax(f, cmd->argtable, NULL);
fclose(f);
}
2017-08-15 04:11:19 -04:00
item->hint = buf;
}
item->argtable = cmd->argtable;
item->func = cmd->func;
item->func_w_context = cmd->func_w_context;
cmd_item_t *last = NULL;
cmd_item_t *it;
SLIST_FOREACH(it, &s_cmd_list, next) {
if (strcmp(it->command, item->command) > 0) {
break;
}
last = it;
}
2017-08-15 04:11:19 -04:00
if (last == NULL) {
SLIST_INSERT_HEAD(&s_cmd_list, item, next);
} else {
SLIST_INSERT_AFTER(last, item, next);
}
return ESP_OK;
}
esp_err_t esp_console_cmd_set_context(const char *cmd, void *context)
{
if (cmd == NULL ) {
return ESP_ERR_INVALID_ARG;
}
cmd_item_t *it;
SLIST_FOREACH(it, &s_cmd_list, next) {
if (strcmp(cmd, it->command) == 0) {
it->context = context;
return ESP_OK;
}
}
return ESP_ERR_NOT_FOUND;
}
2017-08-15 04:11:19 -04:00
void esp_console_get_completion(const char *buf, linenoiseCompletions *lc)
{
size_t len = strlen(buf);
if (len == 0) {
return;
}
2018-04-30 04:52:45 -04:00
cmd_item_t *it;
2017-08-15 04:11:19 -04:00
SLIST_FOREACH(it, &s_cmd_list, next) {
/* Check if command starts with buf */
if (strncmp(buf, it->command, len) == 0) {
linenoiseAddCompletion(lc, it->command);
}
}
}
2018-04-30 04:52:45 -04:00
const char *esp_console_get_hint(const char *buf, int *color, int *bold)
2017-08-15 04:11:19 -04:00
{
2020-11-16 23:48:35 -05:00
size_t len = strlen(buf);
2018-04-30 04:52:45 -04:00
cmd_item_t *it;
2017-08-15 04:11:19 -04:00
SLIST_FOREACH(it, &s_cmd_list, next) {
if (strlen(it->command) == len &&
strncmp(buf, it->command, len) == 0) {
*color = s_config.hint_color;
*bold = s_config.hint_bold;
return it->hint;
}
}
return NULL;
}
2018-04-30 04:52:45 -04:00
static const cmd_item_t *find_command_by_name(const char *name)
2017-08-15 04:11:19 -04:00
{
2018-04-30 04:52:45 -04:00
const cmd_item_t *cmd = NULL;
cmd_item_t *it;
2020-11-16 23:48:35 -05:00
size_t len = strlen(name);
2017-08-15 04:11:19 -04:00
SLIST_FOREACH(it, &s_cmd_list, next) {
if (strlen(it->command) == len &&
strcmp(name, it->command) == 0) {
2017-08-15 04:11:19 -04:00
cmd = it;
break;
}
}
return cmd;
}
2018-04-30 04:52:45 -04:00
esp_err_t esp_console_run(const char *cmdline, int *cmd_ret)
2017-08-15 04:11:19 -04:00
{
if (s_tmp_line_buf == NULL) {
return ESP_ERR_INVALID_STATE;
}
char **argv = (char **) heap_caps_calloc(s_config.max_cmdline_args, sizeof(char *), s_config.heap_alloc_caps);
2017-08-15 04:11:19 -04:00
if (argv == NULL) {
return ESP_ERR_NO_MEM;
}
strlcpy(s_tmp_line_buf, cmdline, s_config.max_cmdline_length);
size_t argc = esp_console_split_argv(s_tmp_line_buf, argv,
2018-04-30 04:52:45 -04:00
s_config.max_cmdline_args);
if (argc == 0) {
2018-04-30 04:52:45 -04:00
free(argv);
return ESP_ERR_INVALID_ARG;
}
2018-04-30 04:52:45 -04:00
const cmd_item_t *cmd = find_command_by_name(argv[0]);
2017-08-15 04:11:19 -04:00
if (cmd == NULL) {
2018-04-30 04:52:45 -04:00
free(argv);
2017-08-15 04:11:19 -04:00
return ESP_ERR_NOT_FOUND;
}
if (cmd->func) {
*cmd_ret = (*cmd->func)(argc, argv);
}
if (cmd->func_w_context) {
*cmd_ret = (*cmd->func_w_context)(cmd->context, argc, argv);
}
2017-08-15 04:11:19 -04:00
free(argv);
return ESP_OK;
}
static struct {
struct arg_str *help_cmd;
struct arg_end *end;
} help_args;
static void print_arg_help(cmd_item_t *it)
{
/* First line: command name and hint
* Pad all the hints to the same column
*/
const char *hint = (it->hint) ? it->hint : "";
printf("%-s %s\n", it->command, hint);
/* Second line: print help.
* Argtable has a nice helper function for this which does line
* wrapping.
*/
printf(" "); // arg_print_formatted does not indent the first line
arg_print_formatted(stdout, 2, 78, it->help);
/* Finally, print the list of arguments */
if (it->argtable) {
arg_print_glossary(stdout, (void **) it->argtable, " %12s %s\n");
}
printf("\n");
}
2018-04-30 04:52:45 -04:00
static int help_command(int argc, char **argv)
2017-08-15 04:11:19 -04:00
{
int nerrors = arg_parse(argc, argv, (void **) &help_args);
if (nerrors != 0) {
arg_print_errors(stderr, help_args.end, argv[0]);
return 1;
}
2018-04-30 04:52:45 -04:00
cmd_item_t *it;
int ret_value = 1;
2017-08-15 04:11:19 -04:00
if (help_args.help_cmd->count == 0) {
/* Print summary of each command */
SLIST_FOREACH(it, &s_cmd_list, next) {
if (it->help == NULL) {
continue;
}
print_arg_help(it);
2017-08-15 04:11:19 -04:00
}
ret_value = 0;
} else {
/* Print summary of given command */
bool found_command = false;
SLIST_FOREACH(it, &s_cmd_list, next) {
if (it->help == NULL) {
continue;
}
if (strcmp(help_args.help_cmd->sval[0], it->command) == 0) {
print_arg_help(it);
found_command = true;
ret_value = 0;
break;
}
}
/* If given command has not been found, print error message*/
if (!found_command) {
printf("help: Unrecognized option '%s'. Please use correct command as argument "
"or type 'help' only to print help for all commands\n", help_args.help_cmd->sval[0]);
2017-08-15 04:11:19 -04:00
}
}
return ret_value;
2017-08-15 04:11:19 -04:00
}
esp_err_t esp_console_register_help_command(void)
2017-08-15 04:11:19 -04:00
{
help_args.help_cmd = arg_str0(NULL, NULL, "<string>", "Name of command");
help_args.end = arg_end(1);
2017-08-15 04:11:19 -04:00
esp_console_cmd_t command = {
2018-04-30 04:52:45 -04:00
.command = "help",
.help = "Print the summary of all registered commands if no arguments "
"are given, otherwise print summary of given command.",
.func = &help_command,
.argtable = &help_args
2017-08-15 04:11:19 -04:00
};
return esp_console_cmd_register(&command);
}