596 lines
14 KiB
C
596 lines
14 KiB
C
/*
|
|
* shell implementation for finsh shell.
|
|
*
|
|
* COPYRIGHT (C) 2006 - 2013, RT-Thread Development Team
|
|
*
|
|
* This file is part of RT-Thread (http://www.rt-thread.org)
|
|
* Maintainer: bernard.xiong <bernard.xiong at gmail.com>
|
|
*
|
|
* All rights reserved.
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*
|
|
* Change Logs:
|
|
* Date Author Notes
|
|
* 2006-04-30 Bernard the first version for FinSH
|
|
* 2006-05-08 Bernard change finsh thread stack to 2048
|
|
* 2006-06-03 Bernard add support for skyeye
|
|
* 2006-09-24 Bernard remove the code related with hardware
|
|
* 2010-01-18 Bernard fix down then up key bug.
|
|
* 2010-03-19 Bernard fix backspace issue and fix device read in shell.
|
|
* 2010-04-01 Bernard add prompt output when start and remove the empty history
|
|
* 2011-02-23 Bernard fix variable section end issue of finsh shell
|
|
* initialization when use GNU GCC compiler.
|
|
*/
|
|
|
|
#include <rtthread.h>
|
|
#include <rthw.h>
|
|
|
|
#include "finsh.h"
|
|
#include "shell.h"
|
|
|
|
#ifdef FINSH_USING_MSH
|
|
#include "msh.h"
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
#include <stdio.h> /* for putchar */
|
|
#endif
|
|
|
|
/* finsh thread */
|
|
static struct rt_thread finsh_thread;
|
|
ALIGN(RT_ALIGN_SIZE)
|
|
static char finsh_thread_stack[FINSH_THREAD_STACK_SIZE];
|
|
struct finsh_shell* shell;
|
|
|
|
#if !defined (RT_USING_NEWLIB) && !defined (RT_USING_MINILIBC)
|
|
int strcmp (const char *s1, const char *s2)
|
|
{
|
|
while (*s1 && *s1 == *s2) s1++, s2++;
|
|
|
|
return (*s1 - *s2);
|
|
}
|
|
|
|
#ifdef RT_USING_HEAP
|
|
char *strdup(const char *s)
|
|
{
|
|
size_t len = strlen(s) + 1;
|
|
char *tmp = (char *)rt_malloc(len);
|
|
|
|
if(!tmp) return NULL;
|
|
|
|
rt_memcpy(tmp, s, len);
|
|
return tmp;
|
|
}
|
|
#endif
|
|
|
|
#endif
|
|
|
|
#if defined(RT_USING_DFS) && defined(DFS_USING_WORKDIR)
|
|
#include <dfs_posix.h>
|
|
const char* finsh_get_prompt()
|
|
{
|
|
#define _PROMPT "finsh "
|
|
static char finsh_prompt[RT_CONSOLEBUF_SIZE + 1] = {_PROMPT};
|
|
|
|
/* get current working directory */
|
|
getcwd(&finsh_prompt[6], RT_CONSOLEBUF_SIZE - 8);
|
|
strcat(finsh_prompt, ">");
|
|
|
|
return finsh_prompt;
|
|
}
|
|
#endif
|
|
|
|
static rt_err_t finsh_rx_ind(rt_device_t dev, rt_size_t size)
|
|
{
|
|
RT_ASSERT(shell != RT_NULL);
|
|
|
|
/* release semaphore to let finsh thread rx data */
|
|
rt_sem_release(&shell->rx_sem);
|
|
|
|
return RT_EOK;
|
|
}
|
|
|
|
/**
|
|
* @ingroup finsh
|
|
*
|
|
* This function sets the input device of finsh shell.
|
|
*
|
|
* @param device_name the name of new input device.
|
|
*/
|
|
void finsh_set_device(const char* device_name)
|
|
{
|
|
rt_device_t dev = RT_NULL;
|
|
|
|
RT_ASSERT(shell != RT_NULL);
|
|
dev = rt_device_find(device_name);
|
|
if (dev == RT_NULL)
|
|
{
|
|
rt_kprintf("finsh: can not find device: %s\n", device_name);
|
|
return;
|
|
}
|
|
|
|
/* check whether it's a same device */
|
|
if (dev == shell->device) return;
|
|
/* open this device and set the new device in finsh shell */
|
|
if (rt_device_open(dev, RT_DEVICE_OFLAG_RDWR) == RT_EOK)
|
|
{
|
|
if (shell->device != RT_NULL)
|
|
{
|
|
/* close old finsh device */
|
|
rt_device_close(shell->device);
|
|
rt_device_set_rx_indicate(dev, RT_NULL);
|
|
}
|
|
|
|
shell->device = dev;
|
|
rt_device_set_rx_indicate(dev, finsh_rx_ind);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @ingroup finsh
|
|
*
|
|
* This function returns current finsh shell input device.
|
|
*
|
|
* @return the finsh shell input device name is returned.
|
|
*/
|
|
const char* finsh_get_device()
|
|
{
|
|
RT_ASSERT(shell != RT_NULL);
|
|
return shell->device->parent.name;
|
|
}
|
|
|
|
/**
|
|
* @ingroup finsh
|
|
*
|
|
* This function set the echo mode of finsh shell.
|
|
*
|
|
* FINSH_OPTION_ECHO=0x01 is echo mode, other values are none-echo mode.
|
|
*
|
|
* @param echo the echo mode
|
|
*/
|
|
void finsh_set_echo(rt_uint32_t echo)
|
|
{
|
|
RT_ASSERT(shell != RT_NULL);
|
|
shell->echo_mode = (rt_uint8_t)echo;
|
|
}
|
|
|
|
/**
|
|
* @ingroup finsh
|
|
*
|
|
* This function gets the echo mode of finsh shell.
|
|
*
|
|
* @return the echo mode
|
|
*/
|
|
rt_uint32_t finsh_get_echo()
|
|
{
|
|
RT_ASSERT(shell != RT_NULL);
|
|
|
|
return shell->echo_mode;
|
|
}
|
|
|
|
void finsh_auto_complete(char* prefix)
|
|
{
|
|
extern void list_prefix(char* prefix);
|
|
|
|
rt_kprintf("\n");
|
|
#ifdef FINSH_USING_MSH
|
|
if (msh_is_used() == RT_TRUE)
|
|
{
|
|
msh_auto_complete(prefix);
|
|
}
|
|
else
|
|
#endif
|
|
{
|
|
list_prefix(prefix);
|
|
}
|
|
|
|
rt_kprintf("%s%s", FINSH_PROMPT, prefix);
|
|
}
|
|
|
|
void finsh_run_line(struct finsh_parser* parser, const char *line)
|
|
{
|
|
const char* err_str;
|
|
|
|
rt_kprintf("\n");
|
|
finsh_parser_run(parser, (unsigned char*)line);
|
|
|
|
/* compile node root */
|
|
if (finsh_errno() == 0)
|
|
{
|
|
finsh_compiler_run(parser->root);
|
|
}
|
|
else
|
|
{
|
|
err_str = finsh_error_string(finsh_errno());
|
|
rt_kprintf("%s\n", err_str);
|
|
}
|
|
|
|
/* run virtual machine */
|
|
if (finsh_errno() == 0)
|
|
{
|
|
char ch;
|
|
finsh_vm_run();
|
|
|
|
ch = (unsigned char)finsh_stack_bottom();
|
|
if (ch > 0x20 && ch < 0x7e)
|
|
{
|
|
rt_kprintf("\t'%c', %d, 0x%08x\n",
|
|
(unsigned char)finsh_stack_bottom(),
|
|
(unsigned int)finsh_stack_bottom(),
|
|
(unsigned int)finsh_stack_bottom());
|
|
}
|
|
else
|
|
{
|
|
rt_kprintf("\t%d, 0x%08x\n",
|
|
(unsigned int)finsh_stack_bottom(),
|
|
(unsigned int)finsh_stack_bottom());
|
|
}
|
|
}
|
|
|
|
finsh_flush(parser);
|
|
}
|
|
|
|
#ifdef FINSH_USING_HISTORY
|
|
rt_bool_t finsh_handle_history(struct finsh_shell* shell, char ch)
|
|
{
|
|
/*
|
|
* handle up and down key
|
|
* up key : 0x1b 0x5b 0x41
|
|
* down key: 0x1b 0x5b 0x42
|
|
*/
|
|
if (ch == 0x1b)
|
|
{
|
|
shell->stat = WAIT_SPEC_KEY;
|
|
return RT_TRUE;
|
|
}
|
|
|
|
if ((shell->stat == WAIT_SPEC_KEY))
|
|
{
|
|
if (ch == 0x5b)
|
|
{
|
|
shell->stat = WAIT_FUNC_KEY;
|
|
return RT_TRUE;
|
|
}
|
|
|
|
shell->stat = WAIT_NORMAL;
|
|
return RT_FALSE;
|
|
}
|
|
|
|
if (shell->stat == WAIT_FUNC_KEY)
|
|
{
|
|
shell->stat = WAIT_NORMAL;
|
|
|
|
if (ch == 0x41) /* up key */
|
|
{
|
|
/* prev history */
|
|
if (shell->current_history > 0)shell->current_history --;
|
|
else
|
|
{
|
|
shell->current_history = 0;
|
|
return RT_TRUE;
|
|
}
|
|
|
|
/* copy the history command */
|
|
memcpy(shell->line, &shell->cmd_history[shell->current_history][0],
|
|
FINSH_CMD_SIZE);
|
|
shell->line_position = strlen(shell->line);
|
|
shell->use_history = 1;
|
|
}
|
|
else if (ch == 0x42) /* down key */
|
|
{
|
|
/* next history */
|
|
if (shell->current_history < shell->history_count - 1)
|
|
shell->current_history ++;
|
|
else
|
|
{
|
|
/* set to the end of history */
|
|
if (shell->history_count != 0)
|
|
{
|
|
shell->current_history = shell->history_count - 1;
|
|
}
|
|
else return RT_TRUE;
|
|
}
|
|
|
|
memcpy(shell->line, &shell->cmd_history[shell->current_history][0],
|
|
FINSH_CMD_SIZE);
|
|
shell->line_position = strlen(shell->line);
|
|
shell->use_history = 1;
|
|
}
|
|
|
|
if (shell->use_history)
|
|
{
|
|
#if defined(_WIN32)
|
|
int i;
|
|
rt_kprintf("\r");
|
|
|
|
for(i=0; i<= 60; i++)
|
|
putchar(' ');
|
|
rt_kprintf("\r");
|
|
|
|
#else
|
|
rt_kprintf("\033[2K\r");
|
|
#endif
|
|
rt_kprintf("%s%s", FINSH_PROMPT, shell->line);
|
|
return RT_TRUE;;
|
|
}
|
|
}
|
|
|
|
return RT_FALSE;
|
|
}
|
|
|
|
void finsh_push_history(struct finsh_shell* shell)
|
|
{
|
|
if ((shell->use_history == 0) && (shell->line_position != 0))
|
|
{
|
|
/* push history */
|
|
if (shell->history_count >= FINSH_HISTORY_LINES)
|
|
{
|
|
/* move history */
|
|
int index;
|
|
for (index = 0; index < FINSH_HISTORY_LINES - 1; index ++)
|
|
{
|
|
memcpy(&shell->cmd_history[index][0],
|
|
&shell->cmd_history[index + 1][0], FINSH_CMD_SIZE);
|
|
}
|
|
memset(&shell->cmd_history[index][0], 0, FINSH_CMD_SIZE);
|
|
memcpy(&shell->cmd_history[index][0], shell->line, shell->line_position);
|
|
|
|
/* it's the maximum history */
|
|
shell->history_count = FINSH_HISTORY_LINES;
|
|
}
|
|
else
|
|
{
|
|
memset(&shell->cmd_history[shell->history_count][0], 0, FINSH_CMD_SIZE);
|
|
memcpy(&shell->cmd_history[shell->history_count][0], shell->line, shell->line_position);
|
|
|
|
/* increase count and set current history position */
|
|
shell->history_count ++;
|
|
}
|
|
}
|
|
shell->current_history = shell->history_count;
|
|
}
|
|
#endif
|
|
|
|
#ifndef RT_USING_HEAP
|
|
struct finsh_shell _shell;
|
|
#endif
|
|
void finsh_thread_entry(void* parameter)
|
|
{
|
|
char ch;
|
|
|
|
/* normal is echo mode */
|
|
shell->echo_mode = 1;
|
|
|
|
finsh_init(&shell->parser);
|
|
rt_kprintf(FINSH_PROMPT);
|
|
|
|
/* set console device as shell device */
|
|
shell->device = rt_console_get_device();
|
|
if (shell->device != RT_NULL)
|
|
{
|
|
rt_device_open(shell->device, RT_DEVICE_OFLAG_RDWR);
|
|
rt_device_set_rx_indicate(shell->device, finsh_rx_ind);
|
|
}
|
|
|
|
while (1)
|
|
{
|
|
/* wait receive */
|
|
if (rt_sem_take(&shell->rx_sem, RT_WAITING_FOREVER) != RT_EOK) continue;
|
|
|
|
/* read one character from device */
|
|
while (rt_device_read(shell->device, 0, &ch, 1) == 1)
|
|
{
|
|
/* handle history key */
|
|
#ifdef FINSH_USING_HISTORY
|
|
if (finsh_handle_history(shell, ch) == RT_TRUE) continue;
|
|
#endif
|
|
|
|
/* handle CR key */
|
|
if (ch == '\r')
|
|
{
|
|
char next;
|
|
|
|
if (rt_device_read(shell->device, 0, &next, 1) == 1)
|
|
ch = next;
|
|
else ch = '\r';
|
|
}
|
|
/* handle tab key */
|
|
else if (ch == '\t')
|
|
{
|
|
/* auto complete */
|
|
finsh_auto_complete(&shell->line[0]);
|
|
/* re-calculate position */
|
|
shell->line_position = strlen(shell->line);
|
|
continue;
|
|
}
|
|
/* handle backspace key */
|
|
else if (ch == 0x7f || ch == 0x08)
|
|
{
|
|
if (shell->line_position != 0)
|
|
{
|
|
rt_kprintf("%c %c", ch, ch);
|
|
}
|
|
if (shell->line_position <= 0) shell->line_position = 0;
|
|
else shell->line_position --;
|
|
shell->line[shell->line_position] = 0;
|
|
continue;
|
|
}
|
|
|
|
/* handle end of line, break */
|
|
if (ch == '\r' || ch == '\n')
|
|
{
|
|
#ifdef FINSH_USING_MSH
|
|
if (msh_is_used() == RT_TRUE && shell->line_position != 0)
|
|
{
|
|
rt_kprintf("\n");
|
|
msh_exec(shell->line, shell->line_position);
|
|
#ifdef FINSH_USING_HISTORY
|
|
finsh_push_history(shell);
|
|
#endif
|
|
}
|
|
else
|
|
#endif
|
|
{
|
|
/* add ';' and run the command line */
|
|
shell->line[shell->line_position] = ';';
|
|
|
|
#ifdef FINSH_USING_HISTORY
|
|
finsh_push_history(shell);
|
|
#endif
|
|
|
|
if (shell->line_position != 0) finsh_run_line(&shell->parser, shell->line);
|
|
else rt_kprintf("\n");
|
|
}
|
|
|
|
rt_kprintf(FINSH_PROMPT);
|
|
memset(shell->line, 0, sizeof(shell->line));
|
|
shell->line_position = 0;
|
|
|
|
break;
|
|
}
|
|
|
|
/* it's a large line, discard it */
|
|
if (shell->line_position >= FINSH_CMD_SIZE) shell->line_position = 0;
|
|
|
|
/* normal character */
|
|
shell->line[shell->line_position] = ch; ch = 0;
|
|
if (shell->echo_mode) rt_kprintf("%c", shell->line[shell->line_position]);
|
|
shell->line_position ++;
|
|
shell->use_history = 0; /* it's a new command */
|
|
} /* end of device read */
|
|
}
|
|
}
|
|
|
|
void finsh_system_function_init(const void* begin, const void* end)
|
|
{
|
|
_syscall_table_begin = (struct finsh_syscall*) begin;
|
|
_syscall_table_end = (struct finsh_syscall*) end;
|
|
}
|
|
|
|
void finsh_system_var_init(const void* begin, const void* end)
|
|
{
|
|
_sysvar_table_begin = (struct finsh_sysvar*) begin;
|
|
_sysvar_table_end = (struct finsh_sysvar*) end;
|
|
}
|
|
|
|
#if defined(__ICCARM__) /* for IAR compiler */
|
|
#ifdef FINSH_USING_SYMTAB
|
|
#pragma section="FSymTab"
|
|
#pragma section="VSymTab"
|
|
#endif
|
|
#elif defined(__ADSPBLACKFIN__) /* for VisaulDSP++ Compiler*/
|
|
#ifdef FINSH_USING_SYMTAB
|
|
extern "asm" int __fsymtab_start;
|
|
extern "asm" int __fsymtab_end;
|
|
extern "asm" int __vsymtab_start;
|
|
extern "asm" int __vsymtab_end;
|
|
#endif
|
|
#elif defined(_MSC_VER)
|
|
#pragma section("FSymTab$a", read)
|
|
const char __fsym_begin_name[] = "__start";
|
|
const char __fsym_begin_desc[] = "begin of finsh";
|
|
__declspec(allocate("FSymTab$a")) const struct finsh_syscall __fsym_begin =
|
|
{
|
|
__fsym_begin_name,
|
|
__fsym_begin_desc,
|
|
NULL
|
|
};
|
|
|
|
#pragma section("FSymTab$z", read)
|
|
const char __fsym_end_name[] = "__end";
|
|
const char __fsym_end_desc[] = "end of finsh";
|
|
__declspec(allocate("FSymTab$z")) const struct finsh_syscall __fsym_end =
|
|
{
|
|
__fsym_end_name,
|
|
__fsym_end_desc,
|
|
NULL
|
|
};
|
|
#endif
|
|
|
|
/*
|
|
* @ingroup finsh
|
|
*
|
|
* This function will initialize finsh shell
|
|
*/
|
|
int finsh_system_init(void)
|
|
{
|
|
rt_err_t result;
|
|
|
|
#ifdef FINSH_USING_SYMTAB
|
|
#ifdef __CC_ARM /* ARM C Compiler */
|
|
extern const int FSymTab$$Base;
|
|
extern const int FSymTab$$Limit;
|
|
extern const int VSymTab$$Base;
|
|
extern const int VSymTab$$Limit;
|
|
finsh_system_function_init(&FSymTab$$Base, &FSymTab$$Limit);
|
|
finsh_system_var_init(&VSymTab$$Base, &VSymTab$$Limit);
|
|
#elif defined (__ICCARM__) /* for IAR Compiler */
|
|
finsh_system_function_init(__section_begin("FSymTab"),
|
|
__section_end("FSymTab"));
|
|
finsh_system_var_init(__section_begin("VSymTab"),
|
|
__section_end("VSymTab"));
|
|
#elif defined (__GNUC__) || defined(__TI_COMPILER_VERSION__)
|
|
/* GNU GCC Compiler and TI CCS */
|
|
extern const int __fsymtab_start;
|
|
extern const int __fsymtab_end;
|
|
extern const int __vsymtab_start;
|
|
extern const int __vsymtab_end;
|
|
finsh_system_function_init(&__fsymtab_start, &__fsymtab_end);
|
|
finsh_system_var_init(&__vsymtab_start, &__vsymtab_end);
|
|
#elif defined(__ADSPBLACKFIN__) /* for VisualDSP++ Compiler */
|
|
finsh_system_function_init(&__fsymtab_start, &__fsymtab_end);
|
|
finsh_system_var_init(&__vsymtab_start, &__vsymtab_end);
|
|
#elif defined(_MSC_VER)
|
|
unsigned int *ptr_begin, *ptr_end;
|
|
|
|
ptr_begin = (unsigned int*)&__fsym_begin; ptr_begin += (sizeof(struct finsh_syscall)/sizeof(unsigned int));
|
|
while (*ptr_begin == 0) ptr_begin ++;
|
|
|
|
ptr_end = (unsigned int*) &__fsym_end; ptr_end --;
|
|
while (*ptr_end == 0) ptr_end --;
|
|
|
|
finsh_system_function_init(ptr_begin, ptr_end);
|
|
#endif
|
|
#endif
|
|
|
|
/* create or set shell structure */
|
|
#ifdef RT_USING_HEAP
|
|
shell = (struct finsh_shell*)rt_malloc(sizeof(struct finsh_shell));
|
|
#else
|
|
shell = &_shell;
|
|
#endif
|
|
if (shell == RT_NULL)
|
|
{
|
|
rt_kprintf("no memory for shell\n");
|
|
return -1;
|
|
}
|
|
|
|
memset(shell, 0, sizeof(struct finsh_shell));
|
|
|
|
rt_sem_init(&(shell->rx_sem), "shrx", 0, 0);
|
|
result = rt_thread_init(&finsh_thread,
|
|
"tshell",
|
|
finsh_thread_entry, RT_NULL,
|
|
&finsh_thread_stack[0], sizeof(finsh_thread_stack),
|
|
FINSH_THREAD_PRIORITY, 10);
|
|
|
|
if (result == RT_EOK)
|
|
rt_thread_startup(&finsh_thread);
|
|
return 0;
|
|
}
|
|
INIT_COMPONENT_EXPORT(finsh_system_init);
|
|
|