867 lines
26 KiB
C
867 lines
26 KiB
C
/*
|
|
* Copyright (c) 2013-2019 Huawei Technologies Co., Ltd. All rights reserved.
|
|
* Copyright (c) 2020-2021 Huawei Device Co., Ltd. All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without modification,
|
|
* are permitted provided that the following conditions are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright notice, this list of
|
|
* conditions and the following disclaimer.
|
|
*
|
|
* 2. Redistributions in binary form must reproduce the above copyright notice, this list
|
|
* of conditions and the following disclaimer in the documentation and/or other materials
|
|
* provided with the distribution.
|
|
*
|
|
* 3. Neither the name of the copyright holder nor the names of its contributors may be used
|
|
* to endorse or promote products derived from this software without specific prior written
|
|
* permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
|
|
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
|
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
|
|
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
|
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
|
|
* OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
|
|
* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
|
|
* OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
|
|
* ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "shcmd.h"
|
|
#include "shell_pri.h"
|
|
#include "show.h"
|
|
#include "stdlib.h"
|
|
#include "unistd.h"
|
|
#include "dirent.h"
|
|
#include "securec.h"
|
|
#include "los_mux.h"
|
|
#include "los_memory.h"
|
|
#include "los_typedef.h"
|
|
|
|
|
|
#define SHELL_INIT_MAGIC_FLAG 0xABABABAB
|
|
#define CTRL_C 0x03 /* 0x03: ctrl+c ASCII */
|
|
|
|
STATIC CmdModInfo g_cmdInfo;
|
|
|
|
LOS_HAL_TABLE_BEGIN(g_shellcmd, shellcmd);
|
|
LOS_HAL_TABLE_END(g_shellcmdEnd, shellcmd);
|
|
|
|
CmdModInfo *OsCmdInfoGet(VOID)
|
|
{
|
|
return &g_cmdInfo;
|
|
}
|
|
|
|
STATIC VOID OsFreeCmdPara(CmdParsed *cmdParsed)
|
|
{
|
|
UINT32 i;
|
|
for (i = 0; i < cmdParsed->paramCnt; i++) {
|
|
if ((cmdParsed->paramArray[i]) != NULL) {
|
|
(VOID)LOS_MemFree(m_aucSysMem0, (cmdParsed->paramArray[i]));
|
|
cmdParsed->paramArray[i] = NULL;
|
|
}
|
|
}
|
|
}
|
|
|
|
STATIC INT32 OsStrSeparateTabStrGet(CHAR **tabStr, CmdParsed *parsed, UINT32 tabStrLen)
|
|
{
|
|
CHAR *shiftStr = NULL;
|
|
CHAR *tempStr = (CHAR *)LOS_MemAlloc(m_aucSysMem0, SHOW_MAX_LEN << 1);
|
|
if (tempStr == NULL) {
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
|
|
(VOID)memset_s(tempStr, SHOW_MAX_LEN << 1, 0, SHOW_MAX_LEN << 1);
|
|
shiftStr = tempStr + SHOW_MAX_LEN;
|
|
|
|
if (strncpy_s(tempStr, SHOW_MAX_LEN - 1, *tabStr, tabStrLen)) {
|
|
(VOID)LOS_MemFree(m_aucSysMem0, tempStr);
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
|
|
parsed->cmdType = CMD_TYPE_STD;
|
|
|
|
/* cut useless or repeat space */
|
|
if (OsCmdKeyShift(tempStr, shiftStr, SHOW_MAX_LEN - 1)) {
|
|
(VOID)LOS_MemFree(m_aucSysMem0, tempStr);
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
|
|
/* get exact position of string to complete */
|
|
/* situation different if end space lost or still exist */
|
|
if ((strlen(shiftStr) == 0) || (tempStr[strlen(tempStr) - 1] != shiftStr[strlen(shiftStr) - 1])) {
|
|
*tabStr = "";
|
|
} else {
|
|
if (OsCmdTokenSplit(shiftStr, ' ', parsed)) {
|
|
(VOID)LOS_MemFree(m_aucSysMem0, tempStr);
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
*tabStr = parsed->paramArray[parsed->paramCnt - 1];
|
|
}
|
|
|
|
(VOID)LOS_MemFree(m_aucSysMem0, tempStr);
|
|
return LOS_OK;
|
|
}
|
|
|
|
STATIC INT32 OsStrSeparate(CHAR *tabStr, CHAR *strPath, CHAR *nameLooking, UINT32 tabStrLen)
|
|
{
|
|
CHAR *strEnd = NULL;
|
|
CHAR *cutPos = NULL;
|
|
CmdParsed parsed = {0};
|
|
CHAR *shellWorkingDirectory = OsShellGetWorkingDirtectory();
|
|
INT32 ret;
|
|
|
|
ret = OsStrSeparateTabStrGet(&tabStr, &parsed, tabStrLen);
|
|
if (ret != LOS_OK) {
|
|
return ret;
|
|
}
|
|
|
|
/* get fullpath str */
|
|
if (*tabStr != '/') {
|
|
if (strncpy_s(strPath, CMD_MAX_PATH, shellWorkingDirectory, CMD_MAX_PATH - 1)) {
|
|
OsFreeCmdPara(&parsed);
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
if (strcmp(shellWorkingDirectory, "/")) {
|
|
if (strncat_s(strPath, CMD_MAX_PATH, "/", CMD_MAX_PATH - strlen(strPath) - 1)) {
|
|
OsFreeCmdPara(&parsed);
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (strncat_s(strPath, CMD_MAX_PATH, tabStr, CMD_MAX_PATH - strlen(strPath) - 1)) {
|
|
OsFreeCmdPara(&parsed);
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
|
|
/* split str by last '/' */
|
|
strEnd = strrchr(strPath, '/');
|
|
cutPos = strEnd;
|
|
if (strEnd != NULL) {
|
|
if (strncpy_s(nameLooking, CMD_MAX_PATH, strEnd + 1, CMD_MAX_PATH - 1)) { /* get cmp str */
|
|
OsFreeCmdPara(&parsed);
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
*(cutPos + 1) = '\0';
|
|
}
|
|
|
|
OsFreeCmdPara(&parsed);
|
|
return LOS_OK;
|
|
}
|
|
|
|
STATIC INT32 OsShowPageInputControl(VOID)
|
|
{
|
|
CHAR readChar;
|
|
|
|
while (1) {
|
|
if (read(STDIN_FILENO, &readChar, 1) != 1) { /* get one CHAR from stdin */
|
|
PRINTK("\n");
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
if ((readChar == 'q') || (readChar == 'Q') || (readChar == CTRL_C)) {
|
|
PRINTK("\n");
|
|
return 0;
|
|
} else if (readChar == '\r') {
|
|
PRINTK("\b \b\b \b\b \b\b \b\b \b\b \b\b \b\b \b");
|
|
return 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
STATIC INT32 OsShowPageControl(UINT32 timesPrint, UINT32 lineCap, UINT32 count)
|
|
{
|
|
if (NEED_NEW_LINE(timesPrint, lineCap)) {
|
|
PRINTK("\n");
|
|
if (SCREEN_IS_FULL(timesPrint, lineCap) && (timesPrint < count)) {
|
|
PRINTK("--More--");
|
|
return OsShowPageInputControl();
|
|
}
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
STATIC INT32 OsSurePrintAll(UINT32 count)
|
|
{
|
|
CHAR readChar = 0;
|
|
PRINTK("\nDisplay all %u possibilities?(y/n)", count);
|
|
while (1) {
|
|
if (read(0, &readChar, 1) != 1) {
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
if ((readChar == 'n') || (readChar == 'N') || (readChar == CTRL_C)) {
|
|
PRINTK("\n");
|
|
return 0;
|
|
} else if ((readChar == 'y') || (readChar == 'Y') || (readChar == '\r')) {
|
|
return 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
STATIC INT32 OsPrintMatchList(UINT32 count, const CHAR *strPath, const CHAR *nameLooking, UINT32 printLen)
|
|
{
|
|
UINT32 timesPrint = 0;
|
|
UINT32 lineCap;
|
|
INT32 ret;
|
|
DIR *openDir = NULL;
|
|
struct dirent *readDir = NULL;
|
|
CHAR formatChar[10] = {0}; /* 10:for formatChar length */
|
|
|
|
printLen = (printLen > (DEFAULT_SCREEN_WIDTH - 2)) ? (DEFAULT_SCREEN_WIDTH - 2) : printLen; /* 2:revered 2 bytes */
|
|
lineCap = DEFAULT_SCREEN_WIDTH / (printLen + 2); /* 2:DEFAULT_SCREEN_WIDTH revered 2 bytes */
|
|
if (snprintf_s(formatChar, sizeof(formatChar) - 1, 7, "%%-%us ", printLen) < 0) { /* 7:format-len */
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
|
|
if (count > (lineCap * DEFAULT_SCREEN_HEIGNT)) {
|
|
ret = OsSurePrintAll(count);
|
|
if (ret != 1) {
|
|
return ret;
|
|
}
|
|
}
|
|
openDir = opendir(strPath);
|
|
if (openDir == NULL) {
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
|
|
PRINTK("\n");
|
|
for (readDir = readdir(openDir); readDir != NULL; readDir = readdir(openDir)) {
|
|
if (strncmp(nameLooking, readDir->d_name, strlen(nameLooking)) != 0) {
|
|
continue;
|
|
}
|
|
PRINTK(formatChar, readDir->d_name);
|
|
timesPrint++;
|
|
ret = OsShowPageControl(timesPrint, lineCap, count);
|
|
if (ret != 1) {
|
|
if (closedir(openDir) < 0) {
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
PRINTK("\n");
|
|
if (closedir(openDir) < 0) {
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
|
|
return LOS_OK;
|
|
}
|
|
|
|
STATIC VOID strncmp_cut(const CHAR *s1, CHAR *s2, size_t n)
|
|
{
|
|
if ((n == 0) || (s1 == NULL) || (s2 == NULL)) {
|
|
return;
|
|
}
|
|
do {
|
|
if (*s1 && *s2 && (*s1 == *s2)) {
|
|
s1++;
|
|
s2++;
|
|
} else {
|
|
break;
|
|
}
|
|
} while (--n != 0);
|
|
if (n > 0) {
|
|
/* NULL pad the remaining n-1 bytes */
|
|
while (n-- != 0)
|
|
*s2++ = 0;
|
|
}
|
|
return;
|
|
}
|
|
STATIC INT32 OsExecNameMatch(const CHAR *strPath, const CHAR *nameLooking, CHAR *strObj, UINT32 *maxLen)
|
|
{
|
|
INT32 count = 0;
|
|
DIR *openDir = NULL;
|
|
struct dirent *readDir = NULL;
|
|
|
|
openDir = opendir(strPath);
|
|
if (openDir == NULL) {
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
|
|
for (readDir = readdir(openDir); readDir != NULL; readDir = readdir(openDir)) {
|
|
if (strncmp(nameLooking, readDir->d_name, strlen(nameLooking)) != 0) {
|
|
continue;
|
|
}
|
|
if (count == 0) {
|
|
if (strncpy_s(strObj, CMD_MAX_PATH, readDir->d_name, CMD_MAX_PATH - 1)) {
|
|
(VOID)closedir(openDir);
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
*maxLen = strlen(readDir->d_name);
|
|
} else {
|
|
/* strncmp&cut the same strings of name matched */
|
|
strncmp_cut(readDir->d_name, strObj, strlen(strObj));
|
|
if (strlen(readDir->d_name) > *maxLen) {
|
|
*maxLen = strlen(readDir->d_name);
|
|
}
|
|
}
|
|
count++;
|
|
}
|
|
|
|
if (closedir(openDir) < 0) {
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
|
|
return count;
|
|
}
|
|
|
|
STATIC VOID OsCompleteStr(const CHAR *result, const CHAR *target, CHAR *cmdKey, UINT32 *len)
|
|
{
|
|
UINT32 size = strlen(result) - strlen(target);
|
|
CHAR *des = cmdKey + *len;
|
|
CHAR *src = (CHAR *)result + strlen(target);
|
|
|
|
while (size-- > 0) {
|
|
PRINTK("%c", *src);
|
|
if (*len == (SHOW_MAX_LEN - 1)) {
|
|
*des = '\0';
|
|
break;
|
|
}
|
|
*des++ = *src++;
|
|
(*len)++;
|
|
}
|
|
}
|
|
|
|
STATIC INT32 OsTabMatchCmd(CHAR *cmdKey, UINT32 *len)
|
|
{
|
|
INT32 count = 0;
|
|
INT32 ret;
|
|
CmdItemNode *cmdItemGuard = NULL;
|
|
CmdItemNode *curCmdItem = NULL;
|
|
const CHAR *cmdMajor = (const CHAR *)cmdKey;
|
|
|
|
while (*cmdMajor == 0x20) { /* cut left space */
|
|
cmdMajor++;
|
|
}
|
|
|
|
if (LOS_ListEmpty(&(g_cmdInfo.cmdList.list))) {
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
|
|
LOS_DL_LIST_FOR_EACH_ENTRY(curCmdItem, &(g_cmdInfo.cmdList.list), CmdItemNode, list) {
|
|
if ((curCmdItem == NULL) || (curCmdItem->cmd == NULL)) {
|
|
return -1;
|
|
}
|
|
|
|
if (strncmp(cmdMajor, curCmdItem->cmd->cmdKey, strlen(cmdMajor)) > 0) {
|
|
continue;
|
|
}
|
|
|
|
if (strncmp(cmdMajor, curCmdItem->cmd->cmdKey, strlen(cmdMajor)) != 0) {
|
|
break;
|
|
}
|
|
|
|
if (count == 0) {
|
|
cmdItemGuard = curCmdItem;
|
|
}
|
|
++count;
|
|
}
|
|
|
|
if (cmdItemGuard == NULL) {
|
|
return 0;
|
|
}
|
|
|
|
if (count == 1) {
|
|
OsCompleteStr(cmdItemGuard->cmd->cmdKey, cmdMajor, cmdKey, len);
|
|
}
|
|
|
|
ret = count;
|
|
if (count > 1) {
|
|
PRINTK("\n");
|
|
while (count--) {
|
|
PRINTK("%s ", cmdItemGuard->cmd->cmdKey);
|
|
cmdItemGuard = LOS_DL_LIST_ENTRY(cmdItemGuard->list.pstNext, CmdItemNode, list);
|
|
}
|
|
PRINTK("\n");
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
STATIC INT32 OsTabMatchFile(CHAR *cmdKey, UINT32 *len)
|
|
{
|
|
UINT32 maxLen = 0;
|
|
INT32 count;
|
|
CHAR *strOutput = NULL;
|
|
CHAR *strCmp = NULL;
|
|
CHAR *dirOpen = (CHAR *)LOS_MemAlloc(m_aucSysMem0, CMD_MAX_PATH * 3); /* 3:dirOpen\strOutput\strCmp */
|
|
if (dirOpen == NULL) {
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
|
|
(VOID)memset_s(dirOpen, CMD_MAX_PATH * 3, 0, CMD_MAX_PATH * 3); /* 3:dirOpen\strOutput\strCmp */
|
|
strOutput = dirOpen + CMD_MAX_PATH;
|
|
strCmp = strOutput + CMD_MAX_PATH;
|
|
|
|
if (OsStrSeparate(cmdKey, dirOpen, strCmp, *len)) {
|
|
(VOID)LOS_MemFree(m_aucSysMem0, dirOpen);
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
|
|
count = OsExecNameMatch(dirOpen, strCmp, strOutput, &maxLen);
|
|
/* one or more matched */
|
|
if (count >= 1) {
|
|
OsCompleteStr(strOutput, strCmp, cmdKey, len);
|
|
|
|
if (count == 1) {
|
|
(VOID)LOS_MemFree(m_aucSysMem0, dirOpen);
|
|
return 1;
|
|
}
|
|
if (OsPrintMatchList((UINT32)count, dirOpen, strCmp, maxLen) == -1) {
|
|
(VOID)LOS_MemFree(m_aucSysMem0, dirOpen);
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
}
|
|
|
|
(VOID)LOS_MemFree(m_aucSysMem0, dirOpen);
|
|
return count;
|
|
}
|
|
|
|
/*
|
|
* Description: Pass in the string and clear useless space ,which inlcude:
|
|
* 1) The overmatch space which is not be marked by Quote's area
|
|
* Squeeze the overmatch space into one space
|
|
* 2) Clear all space before first vaild charatctor
|
|
* Input: cmdKey : Pass in the buff string, which is ready to be operated
|
|
* cmdOut : Pass out the buffer string ,which has already been operated
|
|
* size : cmdKey length
|
|
*/
|
|
LITE_OS_SEC_TEXT_MINOR UINT32 OsCmdKeyShift(const CHAR *cmdKey, CHAR *cmdOut, UINT32 size)
|
|
{
|
|
CHAR *output = NULL;
|
|
CHAR *outputBak = NULL;
|
|
UINT32 len;
|
|
INT32 ret;
|
|
BOOL quotes = FALSE;
|
|
|
|
if ((cmdKey == NULL) || (cmdOut == NULL)) {
|
|
return (UINT32)OS_ERROR;
|
|
}
|
|
|
|
len = strlen(cmdKey);
|
|
if (len >= size) {
|
|
return (UINT32)OS_ERROR;
|
|
}
|
|
output = (CHAR*)LOS_MemAlloc(m_aucSysMem0, len + 1);
|
|
if (output == NULL) {
|
|
PRINTK("malloc failure in %s[%d]", __FUNCTION__, __LINE__);
|
|
return (UINT32)OS_ERROR;
|
|
}
|
|
/* Backup the 'output' start address */
|
|
outputBak = output;
|
|
/* Scan each charactor in 'cmdKey',and squeeze the overmuch space and ignore invaild charactor */
|
|
for (; *cmdKey != '\0'; cmdKey++) {
|
|
/* Detected a Double Quotes, switch the matching status */
|
|
if (*(cmdKey) == '\"') {
|
|
SWITCH_QUOTES_STATUS(quotes);
|
|
}
|
|
/* Ignore the current charactor in following situation */
|
|
/* 1) Quotes matching status is FALSE (which said that the space is not been marked by double quotes) */
|
|
/* 2) Current charactor is a space */
|
|
/* 3) Next charactor is a space too, or the string is been seeked to the end already(\0) */
|
|
/* 4) Invaild charactor, such as single quotes */
|
|
if ((*cmdKey == ' ') && ((*(cmdKey + 1) == ' ') || (*(cmdKey + 1) == '\0')) && QUOTES_STATUS_CLOSE(quotes)) {
|
|
continue;
|
|
}
|
|
if (*cmdKey == '\'') {
|
|
continue;
|
|
}
|
|
*output = *cmdKey;
|
|
output++;
|
|
}
|
|
*output = '\0';
|
|
/* Restore the 'output' start address */
|
|
output = outputBak;
|
|
len = strlen(output);
|
|
/* Clear the space which is located at the first charactor in buffer */
|
|
if (*outputBak == ' ') {
|
|
output++;
|
|
len--;
|
|
}
|
|
/* Copy out the buffer which is been operated already */
|
|
ret = strncpy_s(cmdOut, size, output, len);
|
|
if (ret != EOK) {
|
|
PRINT_ERR("%s,%d strncpy_s failed, err:%d!\n", __FUNCTION__, __LINE__, ret);
|
|
(VOID)LOS_MemFree(m_aucSysMem0, output);
|
|
return OS_ERROR;
|
|
}
|
|
cmdOut[len] = '\0';
|
|
|
|
(VOID)LOS_MemFree(m_aucSysMem0, output);
|
|
|
|
return LOS_OK;
|
|
}
|
|
|
|
LITE_OS_SEC_TEXT_MINOR BOOL OsCmdKeyCheck(const CHAR *cmdKey)
|
|
{
|
|
const CHAR *temp = cmdKey;
|
|
enum Stat {
|
|
STAT_NONE,
|
|
STAT_DIGIT,
|
|
STAT_OTHER
|
|
} state = STAT_NONE;
|
|
|
|
if (strlen(cmdKey) >= CMD_KEY_LEN) {
|
|
return FALSE;
|
|
}
|
|
|
|
while (*temp != '\0') {
|
|
if (!((*temp <= '9') && (*temp >= '0')) &&
|
|
!((*temp <= 'z') && (*temp >= 'a')) &&
|
|
!((*temp <= 'Z') && (*temp >= 'A')) &&
|
|
(*temp != '_') && (*temp != '-')) {
|
|
return FALSE;
|
|
}
|
|
|
|
if ((*temp >= '0') && (*temp <= '9')) {
|
|
if (state == STAT_NONE) {
|
|
state = STAT_DIGIT;
|
|
}
|
|
} else {
|
|
state = STAT_OTHER;
|
|
}
|
|
|
|
temp++;
|
|
}
|
|
|
|
if (state == STAT_DIGIT) {
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
LITE_OS_SEC_TEXT_MINOR INT32 OsTabCompletion(CHAR *cmdKey, UINT32 *len)
|
|
{
|
|
INT32 count = 0;
|
|
CHAR *space = NULL;
|
|
CHAR *cmdMainStr = cmdKey;
|
|
|
|
if ((cmdKey == NULL) || (len == NULL)) {
|
|
return (INT32)OS_ERROR;
|
|
}
|
|
|
|
/* cut left space */
|
|
while (*cmdMainStr == 0x20) {
|
|
cmdMainStr++;
|
|
}
|
|
|
|
/* try to find space in remain */
|
|
space = strrchr(cmdMainStr, 0x20);
|
|
if ((space == NULL) && (*cmdMainStr != '\0')) {
|
|
count = OsTabMatchCmd(cmdKey, len);
|
|
}
|
|
|
|
if (count == 0) {
|
|
count = OsTabMatchFile(cmdKey, len);
|
|
}
|
|
|
|
return count;
|
|
}
|
|
|
|
LITE_OS_SEC_TEXT_MINOR VOID OsCmdAscendingInsert(CmdItemNode *cmd)
|
|
{
|
|
CmdItemNode *cmdItem = NULL;
|
|
CmdItemNode *cmdNext = NULL;
|
|
|
|
if (cmd == NULL) {
|
|
return;
|
|
}
|
|
|
|
for (cmdItem = LOS_DL_LIST_ENTRY((&g_cmdInfo.cmdList.list)->pstPrev, CmdItemNode, list);
|
|
&cmdItem->list != &(g_cmdInfo.cmdList.list);) {
|
|
cmdNext = LOS_DL_LIST_ENTRY(cmdItem->list.pstPrev, CmdItemNode, list);
|
|
if (&cmdNext->list != &(g_cmdInfo.cmdList.list)) {
|
|
if ((strncmp(cmdItem->cmd->cmdKey, cmd->cmd->cmdKey, strlen(cmd->cmd->cmdKey)) >= 0) &&
|
|
(strncmp(cmdNext->cmd->cmdKey, cmd->cmd->cmdKey, strlen(cmd->cmd->cmdKey)) < 0)) {
|
|
LOS_ListTailInsert(&(cmdItem->list), &(cmd->list));
|
|
return;
|
|
}
|
|
cmdItem = cmdNext;
|
|
} else {
|
|
if (strncmp(cmd->cmd->cmdKey, cmdItem->cmd->cmdKey, strlen(cmd->cmd->cmdKey)) > 0) {
|
|
cmdItem = cmdNext;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
LOS_ListTailInsert(&(cmdItem->list), &(cmd->list));
|
|
}
|
|
|
|
LITE_OS_SEC_TEXT_MINOR UINT32 OsShellKeyInit(ShellCB *shellCB)
|
|
{
|
|
CmdKeyLink *cmdKeyLink = NULL;
|
|
CmdKeyLink *cmdHistoryLink = NULL;
|
|
|
|
if (shellCB == NULL) {
|
|
return OS_ERROR;
|
|
}
|
|
cmdKeyLink = (CmdKeyLink *)LOS_MemAlloc(m_aucSysMem0, sizeof(CmdKeyLink));
|
|
if (cmdKeyLink == NULL) {
|
|
PRINT_ERR("Shell CmdKeyLink memory alloc error!\n");
|
|
return OS_ERROR;
|
|
}
|
|
cmdHistoryLink = (CmdKeyLink *)LOS_MemAlloc(m_aucSysMem0, sizeof(CmdKeyLink));
|
|
if (cmdHistoryLink == NULL) {
|
|
(VOID)LOS_MemFree(m_aucSysMem0, cmdKeyLink);
|
|
PRINT_ERR("Shell CmdHistoryLink memory alloc error!\n");
|
|
return OS_ERROR;
|
|
}
|
|
|
|
cmdKeyLink->count = 0;
|
|
LOS_ListInit(&(cmdKeyLink->list));
|
|
shellCB->cmdKeyLink = (VOID *)cmdKeyLink;
|
|
|
|
cmdHistoryLink->count = 0;
|
|
LOS_ListInit(&(cmdHistoryLink->list));
|
|
shellCB->cmdHistoryKeyLink = (VOID *)cmdHistoryLink;
|
|
shellCB->cmdMaskKeyLink = (VOID *)cmdHistoryLink;
|
|
return LOS_OK;
|
|
}
|
|
|
|
LITE_OS_SEC_TEXT_MINOR VOID OsShellKeyDeInit(CmdKeyLink *cmdKeyLink)
|
|
{
|
|
CmdKeyLink *cmdtmp = NULL;
|
|
if (cmdKeyLink == NULL) {
|
|
return;
|
|
}
|
|
|
|
while (!LOS_ListEmpty(&(cmdKeyLink->list))) {
|
|
cmdtmp = LOS_DL_LIST_ENTRY(cmdKeyLink->list.pstNext, CmdKeyLink, list);
|
|
LOS_ListDelete(&cmdtmp->list);
|
|
(VOID)LOS_MemFree(m_aucSysMem0, cmdtmp);
|
|
}
|
|
|
|
cmdKeyLink->count = 0;
|
|
(VOID)LOS_MemFree(m_aucSysMem0, cmdKeyLink);
|
|
}
|
|
|
|
LITE_OS_SEC_TEXT_MINOR UINT32 OsShellSysCmdRegister(VOID)
|
|
{
|
|
UINT32 i;
|
|
UINT8 *cmdItemGroup = NULL;
|
|
UINT32 index = ((UINTPTR)(&g_shellcmdEnd) - (UINTPTR)(&g_shellcmd[0])) / sizeof(CmdItem);
|
|
CmdItemNode *cmdItem = NULL;
|
|
|
|
cmdItemGroup = (UINT8 *)LOS_MemAlloc(m_aucSysMem0, index * sizeof(CmdItemNode));
|
|
if (cmdItemGroup == NULL) {
|
|
PRINT_ERR("[%s]System memory allocation failure!\n", __FUNCTION__);
|
|
return (UINT32)OS_ERROR;
|
|
}
|
|
|
|
for (i = 0; i < index; ++i) {
|
|
cmdItem = (CmdItemNode *)(cmdItemGroup + i * sizeof(CmdItemNode));
|
|
cmdItem->cmd = &g_shellcmd[i];
|
|
OsCmdAscendingInsert(cmdItem);
|
|
}
|
|
g_cmdInfo.listNum += index;
|
|
return LOS_OK;
|
|
}
|
|
|
|
LITE_OS_SEC_TEXT_MINOR VOID OsShellCmdPush(const CHAR *string, CmdKeyLink *cmdKeyLink)
|
|
{
|
|
CmdKeyLink *cmdNewNode = NULL;
|
|
UINT32 len;
|
|
|
|
if ((string == NULL) || (strlen(string) == 0)) {
|
|
return;
|
|
}
|
|
|
|
len = strlen(string);
|
|
cmdNewNode = (CmdKeyLink *)LOS_MemAlloc(m_aucSysMem0, sizeof(CmdKeyLink) + len + 1);
|
|
if (cmdNewNode == NULL) {
|
|
return;
|
|
}
|
|
|
|
(VOID)memset_s(cmdNewNode, sizeof(CmdKeyLink) + len + 1, 0, sizeof(CmdKeyLink) + len + 1);
|
|
if (strncpy_s(cmdNewNode->cmdString, len + 1, string, len)) {
|
|
(VOID)LOS_MemFree(m_aucSysMem0, cmdNewNode);
|
|
return;
|
|
}
|
|
|
|
LOS_ListTailInsert(&(cmdKeyLink->list), &(cmdNewNode->list));
|
|
|
|
return;
|
|
}
|
|
|
|
LITE_OS_SEC_TEXT_MINOR VOID OsShellHistoryShow(UINT32 value, ShellCB *shellCB)
|
|
{
|
|
CmdKeyLink *cmdtmp = NULL;
|
|
CmdKeyLink *cmdNode = shellCB->cmdHistoryKeyLink;
|
|
CmdKeyLink *cmdMask = shellCB->cmdMaskKeyLink;
|
|
errno_t ret;
|
|
|
|
(VOID)pthread_mutex_lock(&shellCB->historyMutex);
|
|
if (value == CMD_KEY_DOWN) {
|
|
if (cmdMask == cmdNode) {
|
|
goto END;
|
|
}
|
|
|
|
cmdtmp = LOS_DL_LIST_ENTRY(cmdMask->list.pstNext, CmdKeyLink, list);
|
|
if (cmdtmp != cmdNode) {
|
|
cmdMask = cmdtmp;
|
|
} else {
|
|
goto END;
|
|
}
|
|
} else if (value == CMD_KEY_UP) {
|
|
cmdtmp = LOS_DL_LIST_ENTRY(cmdMask->list.pstPrev, CmdKeyLink, list);
|
|
if (cmdtmp != cmdNode) {
|
|
cmdMask = cmdtmp;
|
|
} else {
|
|
goto END;
|
|
}
|
|
}
|
|
|
|
while (shellCB->shellBufOffset--) {
|
|
PRINTK("\b \b");
|
|
}
|
|
PRINTK("%s", cmdMask->cmdString);
|
|
shellCB->shellBufOffset = strlen(cmdMask->cmdString);
|
|
(VOID)memset_s(shellCB->shellBuf, SHOW_MAX_LEN, 0, SHOW_MAX_LEN);
|
|
ret = memcpy_s(shellCB->shellBuf, SHOW_MAX_LEN, cmdMask->cmdString, shellCB->shellBufOffset);
|
|
if (ret != EOK) {
|
|
PRINT_ERR("%s, %d memcpy failed!\n", __FUNCTION__, __LINE__);
|
|
goto END;
|
|
}
|
|
shellCB->cmdMaskKeyLink = (VOID *)cmdMask;
|
|
|
|
END:
|
|
(VOID)pthread_mutex_unlock(&shellCB->historyMutex);
|
|
return;
|
|
}
|
|
|
|
LITE_OS_SEC_TEXT_MINOR UINT32 OsCmdExec(CmdParsed *cmdParsed, CHAR *cmdStr)
|
|
{
|
|
UINT32 ret;
|
|
CmdCallBackFunc cmdHook = NULL;
|
|
CmdItemNode *curCmdItem = NULL;
|
|
UINT32 i;
|
|
const CHAR *cmdKey = NULL;
|
|
|
|
if ((cmdParsed == NULL) || (cmdStr == NULL) || (strlen(cmdStr) == 0)) {
|
|
return (UINT32)OS_ERROR;
|
|
}
|
|
|
|
ret = OsCmdParse(cmdStr, cmdParsed);
|
|
if (ret != LOS_OK) {
|
|
goto OUT;
|
|
}
|
|
|
|
LOS_DL_LIST_FOR_EACH_ENTRY(curCmdItem, &(g_cmdInfo.cmdList.list), CmdItemNode, list) {
|
|
cmdKey = curCmdItem->cmd->cmdKey;
|
|
if ((cmdParsed->cmdType == curCmdItem->cmd->cmdType) &&
|
|
(strlen(cmdKey) == strlen(cmdParsed->cmdKeyword)) &&
|
|
(strncmp(cmdKey, (CHAR *)(cmdParsed->cmdKeyword), strlen(cmdKey)) == 0)) {
|
|
cmdHook = curCmdItem->cmd->cmdHook;
|
|
break;
|
|
}
|
|
}
|
|
|
|
ret = OS_ERROR;
|
|
if (cmdHook != NULL) {
|
|
ret = (cmdHook)(cmdParsed->paramCnt, (const CHAR **)cmdParsed->paramArray);
|
|
}
|
|
|
|
OUT:
|
|
for (i = 0; i < cmdParsed->paramCnt; i++) {
|
|
if (cmdParsed->paramArray[i] != NULL) {
|
|
(VOID)LOS_MemFree(m_aucSysMem0, cmdParsed->paramArray[i]);
|
|
cmdParsed->paramArray[i] = NULL;
|
|
}
|
|
}
|
|
|
|
return (UINT32)ret;
|
|
}
|
|
|
|
LITE_OS_SEC_TEXT_MINOR UINT32 OsCmdInit(VOID)
|
|
{
|
|
UINT32 ret;
|
|
LOS_ListInit(&(g_cmdInfo.cmdList.list));
|
|
g_cmdInfo.listNum = 0;
|
|
g_cmdInfo.initMagicFlag = SHELL_INIT_MAGIC_FLAG;
|
|
ret = LOS_MuxInit(&g_cmdInfo.muxLock, NULL);
|
|
if (ret != LOS_OK) {
|
|
PRINT_ERR("Create mutex for shell cmd info failed\n");
|
|
return OS_ERROR;
|
|
}
|
|
return LOS_OK;
|
|
}
|
|
|
|
STATIC UINT32 OsCmdItemCreate(CmdType cmdType, const CHAR *cmdKey, UINT32 paraNum, CmdCallBackFunc cmdProc)
|
|
{
|
|
CmdItem *cmdItem = NULL;
|
|
CmdItemNode *cmdItemNode = NULL;
|
|
|
|
cmdItem = (CmdItem *)LOS_MemAlloc(m_aucSysMem0, sizeof(CmdItem));
|
|
if (cmdItem == NULL) {
|
|
return OS_ERRNO_SHELL_CMDREG_MEMALLOC_ERROR;
|
|
}
|
|
(VOID)memset_s(cmdItem, sizeof(CmdItem), '\0', sizeof(CmdItem));
|
|
|
|
cmdItemNode = (CmdItemNode *)LOS_MemAlloc(m_aucSysMem0, sizeof(CmdItemNode));
|
|
if (cmdItemNode == NULL) {
|
|
(VOID)LOS_MemFree(m_aucSysMem0, cmdItem);
|
|
return OS_ERRNO_SHELL_CMDREG_MEMALLOC_ERROR;
|
|
}
|
|
(VOID)memset_s(cmdItemNode, sizeof(CmdItemNode), '\0', sizeof(CmdItemNode));
|
|
cmdItemNode->cmd = cmdItem;
|
|
cmdItemNode->cmd->cmdHook = cmdProc;
|
|
cmdItemNode->cmd->paraNum = paraNum;
|
|
cmdItemNode->cmd->cmdType = cmdType;
|
|
cmdItemNode->cmd->cmdKey = cmdKey;
|
|
|
|
(VOID)LOS_MuxLock(&g_cmdInfo.muxLock, LOS_WAIT_FOREVER);
|
|
OsCmdAscendingInsert(cmdItemNode);
|
|
g_cmdInfo.listNum++;
|
|
(VOID)LOS_MuxUnlock(&g_cmdInfo.muxLock);
|
|
|
|
return LOS_OK;
|
|
}
|
|
|
|
/* open API */
|
|
LITE_OS_SEC_TEXT_MINOR UINT32 osCmdReg(CmdType cmdType, const CHAR *cmdKey, UINT32 paraNum, CmdCallBackFunc cmdProc)
|
|
{
|
|
CmdItemNode *cmdItemNode = NULL;
|
|
|
|
(VOID)LOS_MuxLock(&g_cmdInfo.muxLock, LOS_WAIT_FOREVER);
|
|
if (g_cmdInfo.initMagicFlag != SHELL_INIT_MAGIC_FLAG) {
|
|
(VOID)LOS_MuxUnlock(&g_cmdInfo.muxLock);
|
|
PRINT_ERR("[%s] shell is not yet initialized!\n", __FUNCTION__);
|
|
return OS_ERRNO_SHELL_NOT_INIT;
|
|
}
|
|
(VOID)LOS_MuxUnlock(&g_cmdInfo.muxLock);
|
|
|
|
if ((cmdProc == NULL) || (cmdKey == NULL) ||
|
|
(cmdType >= CMD_TYPE_BUTT) || (strlen(cmdKey) >= CMD_KEY_LEN) || !strlen(cmdKey)) {
|
|
return OS_ERRNO_SHELL_CMDREG_PARA_ERROR;
|
|
}
|
|
|
|
if (paraNum > CMD_MAX_PARAS) {
|
|
if (paraNum != XARGS) {
|
|
return OS_ERRNO_SHELL_CMDREG_PARA_ERROR;
|
|
}
|
|
}
|
|
|
|
if (OsCmdKeyCheck(cmdKey) != TRUE) {
|
|
return OS_ERRNO_SHELL_CMDREG_CMD_ERROR;
|
|
}
|
|
|
|
(VOID)LOS_MuxLock(&g_cmdInfo.muxLock, LOS_WAIT_FOREVER);
|
|
LOS_DL_LIST_FOR_EACH_ENTRY(cmdItemNode, &(g_cmdInfo.cmdList.list), CmdItemNode, list) {
|
|
if ((cmdType == cmdItemNode->cmd->cmdType) &&
|
|
((strlen(cmdKey) == strlen(cmdItemNode->cmd->cmdKey)) &&
|
|
(strncmp((CHAR *)(cmdItemNode->cmd->cmdKey), cmdKey, strlen(cmdKey)) == 0))) {
|
|
(VOID)LOS_MuxUnlock(&g_cmdInfo.muxLock);
|
|
return OS_ERRNO_SHELL_CMDREG_CMD_EXIST;
|
|
}
|
|
}
|
|
(VOID)LOS_MuxUnlock(&g_cmdInfo.muxLock);
|
|
|
|
return OsCmdItemCreate(cmdType, cmdKey, paraNum, cmdProc);
|
|
}
|
|
|