shell.h

Go to the documentation of this file.

/*
 * Copyright (c) 2018 Nordic Semiconductor ASA
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef SHELL_H__
#define SHELL_H__
 
#include <zephyr.h>
#include <shell/shell_types.h>
#include <shell/shell_history.h>
#include <shell/shell_fprintf.h>
#include <shell/shell_log_backend.h>
#include <logging/log_instance.h>
#include <logging/log.h>
#include <sys/util.h>
 
#if defined CONFIG_SHELL_GETOPT
#include <shell/shell_getopt.h>
#endif
 
#ifdef __cplusplus
extern "C" {
#endif
 
#ifndef CONFIG_SHELL_CMD_BUFF_SIZE
#define CONFIG_SHELL_CMD_BUFF_SIZE 0
#endif
 
#ifndef CONFIG_SHELL_PRINTF_BUFF_SIZE
#define CONFIG_SHELL_PRINTF_BUFF_SIZE 0
#endif
 
#ifndef CONFIG_SHELL_HISTORY_BUFFER
#define CONFIG_SHELL_HISTORY_BUFFER 0
#endif
 
#define Z_SHELL_CMD_ROOT_LVL            (0u)
 
#define SHELL_HEXDUMP_BYTES_IN_LINE     16
 
#define SHELL_OPT_ARG_RAW       (0xFE)
 
#define SHELL_OPT_ARG_CHECK_SKIP (0xFF)
 
#define SHELL_OPT_ARG_MAX               (0xFD)
 
struct getopt_state;
struct shell_static_entry;
 
typedef void (*shell_dynamic_get)(size_t idx,
                                  struct shell_static_entry *entry);
 
struct shell_cmd_entry {
        bool is_dynamic;
        union union_cmd_entry {
                shell_dynamic_get dynamic_get;
 
                const struct shell_static_entry *entry;
        } u;
};
 
struct shell;
 
struct shell_static_args {
        uint8_t mandatory; 
        uint8_t optional;  
};
 
const struct device *shell_device_lookup(size_t idx,
                                   const char *prefix);
 
typedef int (*shell_cmd_handler)(const struct shell *shell,
                                 size_t argc, char **argv);
 
typedef int (*shell_dict_cmd_handler)(const struct shell *shell, size_t argc,
                                      char **argv, void *data);
 
/*
 * @brief Shell static command descriptor.
 */
struct shell_static_entry {
        const char *syntax;                     
        const char *help;                       
        const struct shell_cmd_entry *subcmd;   
        shell_cmd_handler handler;              
        struct shell_static_args args;          
};
 
#define SHELL_CMD_ARG_REGISTER(syntax, subcmd, help, handler,              \
                               mandatory, optional)                        \
        static const struct shell_static_entry UTIL_CAT(_shell_, syntax) = \
        SHELL_CMD_ARG(syntax, subcmd, help, handler, mandatory, optional); \
        static const struct shell_cmd_entry UTIL_CAT(shell_cmd_, syntax)   \
        __attribute__ ((section("."                                        \
                        STRINGIFY(UTIL_CAT(shell_root_cmd_, syntax)))))    \
        __attribute__((used)) = {                                          \
                .is_dynamic = false,                                       \
                .u = {.entry = &UTIL_CAT(_shell_, syntax)}                 \
        }
 
#define SHELL_COND_CMD_ARG_REGISTER(flag, syntax, subcmd, help, handler, \
                                        mandatory, optional) \
        COND_CODE_1(\
                flag, \
                (\
                SHELL_CMD_ARG_REGISTER(syntax, subcmd, help, handler, \
                                        mandatory, optional) \
                ), \
                (\
                static shell_cmd_handler dummy_##syntax##_handler __unused = \
                                                                handler;\
                static const struct shell_cmd_entry *dummy_subcmd_##syntax \
                        __unused = subcmd\
                )\
        )
#define SHELL_CMD_REGISTER(syntax, subcmd, help, handler) \
        SHELL_CMD_ARG_REGISTER(syntax, subcmd, help, handler, 0, 0)
 
#define SHELL_COND_CMD_REGISTER(flag, syntax, subcmd, help, handler) \
        SHELL_COND_CMD_ARG_REGISTER(flag, syntax, subcmd, help, handler, 0, 0)
 
#define SHELL_STATIC_SUBCMD_SET_CREATE(name, ...)                       \
        static const struct shell_static_entry shell_##name[] = {       \
                __VA_ARGS__                                             \
        };                                                              \
        static const struct shell_cmd_entry name = {                    \
                .is_dynamic = false,                                    \
                .u = { .entry = shell_##name }                          \
        }
 
#define SHELL_SUBCMD_SET_END {NULL}
 
#define SHELL_DYNAMIC_CMD_CREATE(name, get)             \
        static const struct shell_cmd_entry name = {    \
                .is_dynamic = true,                     \
                .u = { .dynamic_get = get }             \
        }
 
#define SHELL_CMD_ARG(syntax, subcmd, help, handler, mand, opt) \
        SHELL_EXPR_CMD_ARG(1, syntax, subcmd, help, handler, mand, opt)
 
#define SHELL_COND_CMD_ARG(flag, syntax, subcmd, help, handler, mand, opt) \
        SHELL_EXPR_CMD_ARG(IS_ENABLED(flag), syntax, subcmd, help, \
                          handler, mand, opt)
 
#define SHELL_EXPR_CMD_ARG(_expr, _syntax, _subcmd, _help, _handler, \
                           _mand, _opt) \
        { \
                .syntax = (_expr) ? (const char *)STRINGIFY(_syntax) : "", \
                .help  = (_expr) ? (const char *)_help : NULL, \
                .subcmd = (const struct shell_cmd_entry *)((_expr) ? \
                                _subcmd : NULL), \
                .handler = (shell_cmd_handler)((_expr) ? _handler : NULL), \
                .args = { .mandatory = _mand, .optional = _opt} \
        }
 
#define SHELL_CMD(_syntax, _subcmd, _help, _handler) \
        SHELL_CMD_ARG(_syntax, _subcmd, _help, _handler, 0, 0)
 
#define SHELL_COND_CMD(_flag, _syntax, _subcmd, _help, _handler) \
        SHELL_COND_CMD_ARG(_flag, _syntax, _subcmd, _help, _handler, 0, 0)
 
#define SHELL_EXPR_CMD(_expr, _syntax, _subcmd, _help, _handler) \
        SHELL_EXPR_CMD_ARG(_expr, _syntax, _subcmd, _help, _handler, 0, 0)
 
/* Internal macro used for creating handlers for dictionary commands. */
#define Z_SHELL_CMD_DICT_HANDLER_CREATE(_data, _handler)                \
static int UTIL_CAT(cmd_dict_, GET_ARG_N(1, __DEBRACKET _data))(        \
                const struct shell *shell, size_t argc, char **argv)    \
{                                                                       \
        return _handler(shell, argc, argv,                              \
                        (void *)GET_ARG_N(2, __DEBRACKET _data));       \
}
 
/* Internal macro used for creating dictionary commands. */
#define SHELL_CMD_DICT_CREATE(_data)                                    \
        SHELL_CMD_ARG(GET_ARG_N(1, __DEBRACKET _data), NULL, NULL,      \
                UTIL_CAT(cmd_dict_, GET_ARG_N(1, __DEBRACKET _data)), 1, 0)
 
#define SHELL_SUBCMD_DICT_SET_CREATE(_name, _handler, ...)              \
        FOR_EACH_FIXED_ARG(Z_SHELL_CMD_DICT_HANDLER_CREATE, (),         \
                           _handler, __VA_ARGS__)                       \
        SHELL_STATIC_SUBCMD_SET_CREATE(_name,                           \
                FOR_EACH(SHELL_CMD_DICT_CREATE, (,), __VA_ARGS__),      \
                SHELL_SUBCMD_SET_END                                    \
        )
 
enum shell_receive_state {
        SHELL_RECEIVE_DEFAULT,
        SHELL_RECEIVE_ESC,
        SHELL_RECEIVE_ESC_SEQ,
        SHELL_RECEIVE_TILDE_EXP
};
 
enum shell_state {
        SHELL_STATE_UNINITIALIZED,
        SHELL_STATE_INITIALIZED,
        SHELL_STATE_ACTIVE,
        SHELL_STATE_PANIC_MODE_ACTIVE,  
        SHELL_STATE_PANIC_MODE_INACTIVE 
};
 
enum shell_transport_evt {
        SHELL_TRANSPORT_EVT_RX_RDY,
        SHELL_TRANSPORT_EVT_TX_RDY
};
 
typedef void (*shell_transport_handler_t)(enum shell_transport_evt evt,
                                          void *context);
 
 
typedef void (*shell_uninit_cb_t)(const struct shell *shell, int res);
 
typedef void (*shell_bypass_cb_t)(const struct shell *shell,
                                  uint8_t *data,
                                  size_t len);
 
struct shell_transport;
 
struct shell_transport_api {
        int (*init)(const struct shell_transport *transport,
                    const void *config,
                    shell_transport_handler_t evt_handler,
                    void *context);
 
        int (*uninit)(const struct shell_transport *transport);
 
        int (*enable)(const struct shell_transport *transport,
                      bool blocking_tx);
 
        int (*write)(const struct shell_transport *transport,
                     const void *data, size_t length, size_t *cnt);
 
        int (*read)(const struct shell_transport *transport,
                    void *data, size_t length, size_t *cnt);
 
        void (*update)(const struct shell_transport *transport);
 
};
 
struct shell_transport {
        const struct shell_transport_api *api;
        void *ctx;
};
 
struct shell_stats {
        atomic_t log_lost_cnt; 
};
 
#ifdef CONFIG_SHELL_STATS
#define Z_SHELL_STATS_DEFINE(_name) static struct shell_stats _name##_stats
#define Z_SHELL_STATS_PTR(_name) (&(_name##_stats))
#else
#define Z_SHELL_STATS_DEFINE(_name)
#define Z_SHELL_STATS_PTR(_name) NULL
#endif /* CONFIG_SHELL_STATS */
 
struct shell_flags {
        uint32_t insert_mode :1; 
        uint32_t use_colors  :1; 
        uint32_t echo        :1; 
        uint32_t obscure     :1; 
        uint32_t processing  :1; 
        uint32_t tx_rdy      :1;
        uint32_t mode_delete :1; 
        uint32_t history_exit:1; 
        uint32_t last_nl     :8; 
        uint32_t cmd_ctx     :1; 
        uint32_t print_noinit:1; 
};
 
BUILD_ASSERT((sizeof(struct shell_flags) == sizeof(uint32_t)),
             "Structure must fit in 4 bytes");
 
union shell_internal {
        uint32_t value;
        struct shell_flags flags;
};
 
enum shell_signal {
        SHELL_SIGNAL_RXRDY,
        SHELL_SIGNAL_LOG_MSG,
        SHELL_SIGNAL_KILL,
        SHELL_SIGNAL_TXDONE, /* TXDONE must be last one before SHELL_SIGNALS */
        SHELL_SIGNALS
};
 
struct shell_ctx {
        const char *prompt; 
        enum shell_state state; 
        enum shell_receive_state receive_state;
        struct shell_static_entry active_cmd;
 
        /* New root command. If NULL shell uses default root commands. */
        const struct shell_static_entry *selected_cmd;
 
        struct shell_vt100_ctx vt100_ctx;
 
        shell_uninit_cb_t uninit_cb;
 
        shell_bypass_cb_t bypass;
 
#if defined CONFIG_SHELL_GETOPT
        struct getopt_state getopt_state;
#endif
 
        uint16_t cmd_buff_len; 
        uint16_t cmd_buff_pos; 
        uint16_t cmd_tmp_buff_len; 
        char cmd_buff[CONFIG_SHELL_CMD_BUFF_SIZE];
 
        char temp_buff[CONFIG_SHELL_CMD_BUFF_SIZE];
 
        char printf_buff[CONFIG_SHELL_PRINTF_BUFF_SIZE];
 
        volatile union shell_internal internal; 
        struct k_poll_signal signals[SHELL_SIGNALS];
        struct k_poll_event events[SHELL_SIGNALS];
 
        struct k_mutex wr_mtx;
        k_tid_t tid;
};
 
extern const struct log_backend_api log_backend_shell_api;
 
enum shell_flag {
        SHELL_FLAG_CRLF_DEFAULT = (1<<0),       /* Do not map CR or LF */
        SHELL_FLAG_OLF_CRLF     = (1<<1)        /* Map LF to CRLF on output */
};
 
struct shell {
        const char *default_prompt; 
        const struct shell_transport *iface; 
        struct shell_ctx *ctx; 
        struct shell_history *history;
 
        const enum shell_flag shell_flag;
 
        const struct shell_fprintf *fprintf_ctx;
 
        struct shell_stats *stats;
 
        const struct shell_log_backend *log_backend;
 
        LOG_INSTANCE_PTR_DECLARE(log);
 
        const char *thread_name;
        struct k_thread *thread;
        k_thread_stack_t *stack;
};
 
extern void z_shell_print_stream(const void *user_ctx, const char *data,
                                 size_t data_len);
#define SHELL_DEFINE(_name, _prompt, _transport_iface,                        \
                     _log_queue_size, _log_timeout, _shell_flag)              \
        static const struct shell _name;                                      \
        static struct shell_ctx UTIL_CAT(_name, _ctx);                        \
        static uint8_t _name##_out_buffer[CONFIG_SHELL_PRINTF_BUFF_SIZE];     \
        Z_SHELL_LOG_BACKEND_DEFINE(_name, _name##_out_buffer,                 \
                                 CONFIG_SHELL_PRINTF_BUFF_SIZE,               \
                                 _log_queue_size, _log_timeout);              \
        Z_SHELL_HISTORY_DEFINE(_name##_history, CONFIG_SHELL_HISTORY_BUFFER); \
        Z_SHELL_FPRINTF_DEFINE(_name##_fprintf, &_name, _name##_out_buffer,   \
                             CONFIG_SHELL_PRINTF_BUFF_SIZE,                   \
                             true, z_shell_print_stream);                     \
        LOG_INSTANCE_REGISTER(shell, _name, CONFIG_SHELL_LOG_LEVEL);          \
        Z_SHELL_STATS_DEFINE(_name);                                          \
        static K_KERNEL_STACK_DEFINE(_name##_stack, CONFIG_SHELL_STACK_SIZE); \
        static struct k_thread _name##_thread;                                \
        static const STRUCT_SECTION_ITERABLE(shell, _name) = {                \
                .default_prompt = _prompt,                                    \
                .iface = _transport_iface,                                    \
                .ctx = &UTIL_CAT(_name, _ctx),                                \
                .history = IS_ENABLED(CONFIG_SHELL_HISTORY) ?                 \
                                &_name##_history : NULL,                      \
                .shell_flag = _shell_flag,                                    \
                .fprintf_ctx = &_name##_fprintf,                              \
                .stats = Z_SHELL_STATS_PTR(_name),                            \
                .log_backend = Z_SHELL_LOG_BACKEND_PTR(_name),                \
                LOG_INSTANCE_PTR_INIT(log, shell, _name)                      \
                .thread_name = STRINGIFY(_name),                              \
                .thread = &_name##_thread,                                    \
                .stack = _name##_stack                                        \
        }
 
int shell_init(const struct shell *shell, const void *transport_config,
               bool use_colors, bool log_backend, uint32_t init_log_level);
 
void shell_uninit(const struct shell *shell, shell_uninit_cb_t cb);
 
int shell_start(const struct shell *shell);
 
int shell_stop(const struct shell *shell);
 
#define SHELL_NORMAL    SHELL_VT100_COLOR_DEFAULT
 
#define SHELL_INFO      SHELL_VT100_COLOR_GREEN
 
#define SHELL_OPTION    SHELL_VT100_COLOR_CYAN
 
#define SHELL_WARNING   SHELL_VT100_COLOR_YELLOW
 
#define SHELL_ERROR     SHELL_VT100_COLOR_RED
 
void shell_fprintf(const struct shell *shell, enum shell_vt100_color color,
                   const char *fmt, ...);
 
void shell_vfprintf(const struct shell *shell, enum shell_vt100_color color,
                   const char *fmt, va_list args);
 
void shell_hexdump_line(const struct shell *shell, unsigned int offset,
                        const uint8_t *data, size_t len);
 
void shell_hexdump(const struct shell *shell, const uint8_t *data, size_t len);
 
#define shell_info(_sh, _ft, ...) \
        shell_fprintf(_sh, SHELL_INFO, _ft "\n", ##__VA_ARGS__)
 
#define shell_print(_sh, _ft, ...) \
        shell_fprintf(_sh, SHELL_NORMAL, _ft "\n", ##__VA_ARGS__)
 
#define shell_warn(_sh, _ft, ...) \
        shell_fprintf(_sh, SHELL_WARNING, _ft "\n", ##__VA_ARGS__)
 
#define shell_error(_sh, _ft, ...) \
        shell_fprintf(_sh, SHELL_ERROR, _ft "\n", ##__VA_ARGS__)
 
void shell_process(const struct shell *shell);
 
int shell_prompt_change(const struct shell *shell, const char *prompt);
 
void shell_help(const struct shell *shell);
 
/* @brief Command's help has been printed */
#define SHELL_CMD_HELP_PRINTED  (1)
 
#if defined CONFIG_SHELL_GETOPT
int shell_getopt(const struct shell *shell, int argc, char *const argv[],
                 const char *ostr);
 
struct getopt_state *shell_getopt_state_get(const struct shell *shell);
#endif /* CONFIG_SHELL_GETOPT */
 
int shell_execute_cmd(const struct shell *shell, const char *cmd);
 
int shell_set_root_cmd(const char *cmd);
 
void shell_set_bypass(const struct shell *shell, shell_bypass_cb_t bypass);
 
int shell_insert_mode_set(const struct shell *shell, bool val);
 
int shell_use_colors_set(const struct shell *shell, bool val);
 
int shell_echo_set(const struct shell *shell, bool val);
 
int shell_obscure_set(const struct shell *shell, bool obscure);
 
int shell_mode_delete_set(const struct shell *shell, bool val);
 
#ifdef __cplusplus
}
#endif
 
#endif /* SHELL_H__ */