鸿蒙内核源码分析(Shell解析篇) | 应用窥视内核的窗口

系列篇从内核视角用一句话概括shell的底层实现为：两个任务，三个阶段。其本质是独立进程，因而划到进程管理模块。每次创建shell进程都会再创建两个任务。

客户端任务(ShellEntry)： 负责接受来自终端(控制台)敲入的一个个字符，字符按VT规范组装成一句句的命令。
服务端任务(ShellTask)： 对命令进行解析并执行,将结果输出到控制台。
而按命令生命周期可分三个阶段.
编辑： 鸿蒙在这个部分实现了一个简单的编辑器功能,处理控制台输入的每个字符,主要包括了对控制字符例如 <ESC>,\t,\b,\n,\r,四个方向键0x41 ~ 0x44 的处理。
解析： 对编辑后的字符串进行解析，解析出命令项和参数项，找到对应的命令项执行函数。
执行： 命令可通过静态和动态两种方式注册到内核，解析出具体命令后在注册表中找到对应函数回调。将结果输出到控制台。

编辑部分由客户端任务完成，后两个部分由服务端任务完成，命令全局注册由内核完成。

本篇主要说 服务端任务 和 解析/执行过程.
客户端任务 和 编辑过程 已在 (Shell编辑篇) 中说明,请自行翻看.

总体过程

第一步：将支持的shell命令注册进全局链表,支持静态和动态两种方式,内容包括命令项,参数信息和回调函数.
第二步：由独立任务解析出用户输入的命令行,拆分出命令项和参数内容
第三步：通过命令项在全局链表中遍历找到已注册的回调函数,并执行.

结构体

鸿蒙对命令的注册用了三个结构体,个人感觉前两个可以合成一个,降低代码阅读难度.

STATIC CmdModInfo g_cmdInfo;//shell 命令模块信息,上面挂了所有的命令项(ls,cd ,cp ==)typedef struct {//命令项CmdType cmdType;	//命令类型//CMD_TYPE_EX：不支持标准命令参数输入，会把用户填写的命令关键字屏蔽掉，例如：输入ls /ramfs，传入给注册函数的参数只有/ramfs，而ls命令关键字并不会被传入。//CMD_TYPE_STD：支持的标准命令参数输入，所有输入的字符都会通过命令解析后被传入。const CHAR *cmdKey;	//命令关键字，例如:ls 函数在Shell中访问的名称。UINT32 paraNum;		//调用的执行函数的入参最大个数，暂不支持。CmdCallBackFunc cmdHook;//命令执行函数地址，即命令实际执行函数。
} CmdItem;
typedef struct {	//命令节点LOS_DL_LIST list;	//双向链表CmdItem *cmd;	//命令项
} CmdItemNode;/* global info for shell module */
typedef struct {//shell 模块的全局信息CmdItemNode cmdList;	//命令项节点UINT32 listNum;//节点数量UINT32 initMagicFlag;//初始魔法标签 0xABABABABLosMux muxLock;	//操作链表互斥锁CmdVerifyTransID transIdHook;//暂不知何意
} CmdModInfo;

解读

CmdItem为注册的内容载体结构体,cmdHook为回调函数,是命令的真正执行体.
通过双向链表CmdItemNode.list将所有命令穿起来
CmdModInfo记录命令数量和操作的互斥锁,shell的魔法数字为 0xABABABAB

第一步 | Shell 注册

静态宏方式注册,链接时处理
静态注册命令方式一般用在系统常用命令注册,鸿蒙已支持以下命令.

        arp           cat           cd            chgrp         chmod         chown         cp            cpup          date          dhclient      dmesg         dns           format        free          help          hwi           ifconfig      ipdebug       kill          log           ls            lsfd          memcheck      mkdir         mount         netstat       oom           partinfo      partition     ping          ping6         pwd           reset         rm            rmdir         sem           statfs        su            swtmr         sync          systeminfo    task          telnet        test          tftp          touch         umount        uname         watch         writeproc

例如注册 `ls`命令

    SHELLCMD_ENTRY(ls_shellcmd,  CMD_TYPE_EX, "ls", XARGS,  (CMD_CBK_FUNC)osShellCmdLs)

需在链接选项中添加链接该新增命令项参数,具体在liteos_tables_ldflags.mk文件的LITEOS_TABLES_LDFLAGS项下添加-uls_shellcmd。至于SHELLCMD_ENTRY是如何实现的在链接阶段的注册,请自行翻看 (内联汇编篇) ,有详细说明实现细节.

动态命令方式,运行时处理
动态注册命令方式一般用在用户命令注册,具体实现代码如下:

    osCmdReg(CMD_TYPE_EX, "ls", XARGS,  (CMD_CBK_FUNC)osShellCmdLs){// ....//5.正式创建命令,挂入链表return OsCmdItemCreate(cmdType, cmdKey, paraNum, cmdProc);//不存在就注册命令}//创建一个命令项,例如 chmodSTATIC UINT32 OsCmdItemCreate(CmdType cmdType, const CHAR *cmdKey, UINT32 paraNum, CmdCallBackFunc cmdProc){CmdItem *cmdItem = NULL;CmdItemNode *cmdItemNode = NULL;//1.构造命令节点过程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;//回调函数 osShellCmdLscmdItemNode->cmd->paraNum = paraNum;//`777`,'/home'cmdItemNode->cmd->cmdType = cmdType;//关键字类型cmdItemNode->cmd->cmdKey = cmdKey;	//`chmod`//2.完成构造后挂入全局链表(VOID)LOS_MuxLock(&g_cmdInfo.muxLock, LOS_WAIT_FOREVER);OsCmdAscendingInsert(cmdItemNode);//按升序方式插入g_cmdInfo.listNum++;//命令总数增加(VOID)LOS_MuxUnlock(&g_cmdInfo.muxLock);return LOS_OK;}

第二步解析 | ShellTask

//shell 服务端任务初始化,这个任务负责解析和执行命令
LITE_OS_SEC_TEXT_MINOR UINT32 ShellTaskInit(ShellCB *shellCB)
{CHAR *name = NULL;TSK_INIT_PARAM_S initParam = {0};//输入Shell命令的两种方式if (shellCB->consoleID == CONSOLE_SERIAL) {	//通过串口工具name = SERIAL_SHELL_TASK_NAME;} else if (shellCB->consoleID == CONSOLE_TELNET) {//通过远程工具name = TELNET_SHELL_TASK_NAME;} else {return LOS_NOK;}initParam.pfnTaskEntry = (TSK_ENTRY_FUNC)ShellTask;//任务入口函数,主要是解析shell命令initParam.usTaskPrio   = 9; /* 9:shell task priority */initParam.auwArgs[0]   = (UINTPTR)shellCB;initParam.uwStackSize  = 0x3000;initParam.pcName       = name;initParam.uwResved     = LOS_TASK_STATUS_DETACHED;(VOID)LOS_EventInit(&shellCB->shellEvent);//初始化事件,以事件方式通知任务解析命令return LOS_TaskCreate(&shellCB->shellTaskHandle, &initParam);//创建任务
}
LITE_OS_SEC_TEXT_MINOR UINT32 ShellTask(UINTPTR param1,UINTPTR param2,UINTPTR param3,UINTPTR param4)
{UINT32 ret;ShellCB *shellCB = (ShellCB *)param1;(VOID)param2;(VOID)param3;(VOID)param4;while (1) {PRINTK("\nOHOS # ");//读取shell 输入事件 例如: cat weharmony.net 命令ret = LOS_EventRead(&shellCB->shellEvent,0xFFF, LOS_WAITMODE_OR | LOS_WAITMODE_CLR, LOS_WAIT_FOREVER);if (ret == SHELL_CMD_PARSE_EVENT) {//获得解析命令事件ShellCmdProcess(shellCB);//处理命令 } else if (ret == CONSOLE_SHELL_KEY_EVENT) {//退出shell事件break;}}OsShellKeyDeInit((CmdKeyLink *)shellCB->cmdKeyLink);//OsShellKeyDeInit((CmdKeyLink *)shellCB->cmdHistoryKeyLink);(VOID)LOS_EventDestroy(&shellCB->shellEvent);//注销事件(VOID)LOS_MemFree((VOID *)m_aucSysMem0, shellCB);//释放shell控制块return 0;
}

解读

任务优先级和客户端任务一样同为 9
指定内核栈大小为0x3000 = 12K ,因任务负责命令的解析和执行,所以需要更大的内核空间.
任务的入口函数ShellTask,一个死循环在以LOS_WAIT_FOREVER方式死等事件发生.
- SHELL_CMD_PARSE_EVENT 通知开始解析事件,该事件由客户端任务ShellEntry检测到回车键时发出.

      STATIC VOID ShellNotify(ShellCB *shellCB){(VOID)LOS_EventWrite(&shellCB->shellEvent, SHELL_CMD_PARSE_EVENT);}

*   `CONSOLE_SHELL_KEY_EVENT` 收到 `exit`命令时将发出该事件,退出`shell`回收资源

层层跟进ShellCmdProcess,解析出命令项和参数内容,最终跑到OsCmdExec中遍历已注册的命令表,找出命令对应的函数完成回调.

   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)) {//找到命令的回调函数 例如: ls <-> osShellCmdLscmdHook = 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;}

第三步 | 执行

想知道有哪些系统shell命令,可以搜索关键词SHELLCMD_ENTRY拿到所有通过静态方式注册的命令.

其中有网络的,进程的,任务的,内存的等等,此处列出几个常用的shell命令的实现.

ls 命令

SHELLCMD_ENTRY(ls_shellcmd, CMD_TYPE_EX, "ls", XARGS, (CmdCallBackFunc)osShellCmdLs);/*******************************************************
命令功能
ls命令用来显示当前目录的内容。命令格式
ls [path]path为空时，显示当前目录的内容。
path为无效文件名时，显示失败，提示：
ls error: No such directory。
path为有效目录路径时，会显示对应目录下的内容。使用指南
ls命令显示当前目录的内容。
ls可以显示文件的大小。
proc下ls无法统计文件大小，显示为0。*******************************************************/
int osShellCmdLs(int argc, const char **argv)
{char *fullpath = NULL;const char *filename = NULL;int ret;char *shell_working_directory = OsShellGetWorkingDirtectory();//获取当前工作目录if (shell_working_directory == NULL){return -1;}ERROR_OUT_IF(argc > 1, PRINTK("ls or ls [DIRECTORY]\n"), return -1);if (argc == 0)//木有参数时 -> #ls {ls(shell_working_directory);//执行ls 当前工作目录return 0;}filename = argv[0];//有参数时 -> #ls ../harmony  or #ls /no such file or directoryret = vfs_normalize_path(shell_working_directory, filename, &fullpath);//获取全路径,注意这里带出来fullpath，而fullpath已经在内核空间ERROR_OUT_IF(ret < 0, set_err(-ret, "ls error"), return -1);ls(fullpath);//执行 ls 全路径free(fullpath);//释放全路径，为啥要释放，因为fullpath已经由内核空间分配return 0;
}

task 命令

SHELLCMD_ENTRY(task_shellcmd, CMD_TYPE_EX, "task", 1, (CmdCallBackFunc)OsShellCmdDumpTask);LITE_OS_SEC_TEXT_MINOR UINT32 OsShellCmdDumpTask(INT32 argc, const CHAR **argv)
{UINT32 flag = 0;
#ifdef LOSCFG_KERNEL_VMflag |= OS_PROCESS_MEM_INFO;
#endifif (argc >= 2) { /* 2: The task shell name restricts the parameters */goto TASK_HELP;}if (argc == 1) {if (strcmp("-a", argv[0]) == 0) {flag |= OS_PROCESS_INFO_ALL;} else if (strcmp("-i", argv[0]) == 0) {if (!OsShellShowTickRespo()) {return LOS_OK;}goto TASK_HELP;} else if (strcmp("-t", argv[0]) == 0) {if (!OsShellShowSchedParam()) {return LOS_OK;}goto TASK_HELP;} else {goto TASK_HELP;}}return OsShellCmdTskInfoGet(OS_ALL_TASK_MASK, NULL, flag);TASK_HELP:PRINTK("Unknown option: %s\n", argv[0]);PRINTK("usage: task or task -a\n");return LOS_NOK;
}

cat 命令

SHELLCMD_ENTRY(cat_shellcmd, CMD_TYPE_EX, "cat", XARGS, (CmdCallBackFunc)osShellCmdCat);/*****************************************************************
cat用于显示文本文件的内容。cat [pathname]
cat weharmony.txt
*****************************************************************/
int osShellCmdCat(int argc, const char **argv)
{char *fullpath = NULL;int ret;unsigned int ca_task;struct Vnode *vnode = NULL;TSK_INIT_PARAM_S init_param;char *shell_working_directory = OsShellGetWorkingDirtectory();//显示当前目录 pwdif (shell_working_directory == NULL){return -1;}ERROR_OUT_IF(argc != 1, PRINTK("cat [FILE]\n"), return -1);ret = vfs_normalize_path(shell_working_directory, argv[0], &fullpath);//由相对路径获取绝对路径ERROR_OUT_IF(ret < 0, set_err(-ret, "cat error"), return -1);VnodeHold();ret = VnodeLookup(fullpath, &vnode, O_RDONLY);if (ret != LOS_OK){set_errno(-ret);perror("cat error");VnodeDrop();free(fullpath);return -1;}if (vnode->type != VNODE_TYPE_REG){set_errno(EINVAL);perror("cat error");VnodeDrop();free(fullpath);return -1;}VnodeDrop();(void)memset_s(&init_param, sizeof(init_param), 0, sizeof(TSK_INIT_PARAM_S));init_param.pfnTaskEntry = (TSK_ENTRY_FUNC)osShellCmdDoCatShow;init_param.usTaskPrio   = CAT_TASK_PRIORITY;	//优先级10init_param.auwArgs[0]   = (UINTPTR)fullpath;	//入口参数init_param.uwStackSize  = CAT_TASK_STACK_SIZE;//内核栈大小init_param.pcName       = "shellcmd_cat";	//任务名称init_param.uwResved     = LOS_TASK_STATUS_DETACHED | OS_TASK_FLAG_SPECIFIES_PROCESS;init_param.processID    = 2; /* 2: kProcess */ //内核任务ret = (int)LOS_TaskCreate(&ca_task, &init_param);//创建任务显示cat内容if (ret != LOS_OK){free(fullpath);}return ret;
}

你能看明白这些命令的底层实现吗? 如果看明白了,可能会不由得发出 原来如此 的感叹!