1/* 2 * Copyright (c) 2013-2019 Huawei Technologies Co., Ltd. All rights reserved. 3 * Copyright (c) 2020-2021 Huawei Device Co., Ltd. All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without modification, 6 * are permitted provided that the following conditions are met: 7 * 8 * 1. Redistributions of source code must retain the above copyright notice, this list of 9 * conditions and the following disclaimer. 10 * 11 * 2. Redistributions in binary form must reproduce the above copyright notice, this list 12 * of conditions and the following disclaimer in the documentation and/or other materials 13 * provided with the distribution. 14 * 15 * 3. Neither the name of the copyright holder nor the names of its contributors may be used 16 * to endorse or promote products derived from this software without specific prior written 17 * permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 20 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, 21 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR 23 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 24 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 25 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; 26 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 27 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 28 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 29 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30 */ 31 32#include "shcmd.h" 33#include "los_memory.h" 34 35 36/* 37 * Filter out double quote or single-quoted strings at both ends 38 */ 39LITE_OS_SEC_TEXT_MINOR CHAR *OsCmdParseStrdup(const CHAR *str) 40{ 41 CHAR *tempStr = NULL; 42 CHAR *newStr = NULL; 43 44 newStr = (CHAR *)LOS_MemAlloc(m_aucSysMem0, strlen(str) + 1); 45 if (newStr == NULL) { 46 return NULL; 47 } 48 49 tempStr = newStr; 50 for (; *str != '\0'; str++) { 51 if ((*str == '\"') || (*str == '\'')) { 52 continue; 53 } 54 *newStr = *str; 55 newStr++; 56 } 57 *newStr = '\0'; 58 return tempStr; 59} 60 61LITE_OS_SEC_TEXT_MINOR UINT32 OsCmdParseParaGet(CHAR **value, const CHAR *paraTokenStr) 62{ 63 if ((paraTokenStr == NULL) || (value == NULL)) { 64 return (UINT32)OS_ERROR; 65 } 66 67 *value = OsCmdParseStrdup(paraTokenStr); 68 if (*value == NULL) { 69 return LOS_NOK; 70 } 71 return LOS_OK; 72} 73 74LITE_OS_SEC_TEXT_MINOR UINT32 OsCmdParseOneToken(CmdParsed *cmdParsed, UINT32 index, const CHAR *token) 75{ 76 UINT32 ret = LOS_OK; 77 UINT32 tempLen; 78 79 if (cmdParsed == NULL) { 80 return (UINT32)OS_ERROR; 81 } 82 83 if (index == 0) { 84 if (cmdParsed->cmdType != CMD_TYPE_STD) { 85 return ret; 86 } 87 } 88 89 if ((token != NULL) && (cmdParsed->paramCnt < CMD_MAX_PARAS)) { 90 tempLen = cmdParsed->paramCnt; 91 ret = OsCmdParseParaGet(&(cmdParsed->paramArray[tempLen]), token); 92 if (ret != LOS_OK) { 93 return ret; 94 } 95 cmdParsed->paramCnt++; 96 } 97 return ret; 98} 99 100LITE_OS_SEC_TEXT_MINOR UINT32 OsCmdTokenSplit(CHAR *cmdStr, CHAR split, CmdParsed *cmdParsed) 101{ 102 enum { 103 STAT_INIT, 104 STAT_TOKEN_IN, 105 STAT_TOKEN_OUT 106 } state = STAT_INIT; 107 UINT32 count = 0; 108 CHAR *p = NULL; 109 CHAR *token = cmdStr; 110 UINT32 ret = LOS_OK; 111 BOOL quotes = FALSE; 112 113 if (cmdStr == NULL) { 114 return (UINT32)OS_ERROR; 115 } 116 117 for (p = cmdStr; (*p != '\0') && (ret == LOS_OK); p++) { 118 if (*p == '\"') { 119 SWITCH_QUOTES_STATUS(quotes); 120 } 121 switch (state) { 122 case STAT_INIT: 123 case STAT_TOKEN_IN: 124 if ((*p == split) && QUOTES_STATUS_CLOSE(quotes)) { 125 *p = '\0'; 126 ret = OsCmdParseOneToken(cmdParsed, count++, token); 127 state = STAT_TOKEN_OUT; 128 } 129 break; 130 case STAT_TOKEN_OUT: 131 if (*p != split) { 132 token = p; 133 state = STAT_TOKEN_IN; 134 } 135 break; 136 default: 137 break; 138 } 139 } 140 141 if (((ret == LOS_OK) && (state == STAT_TOKEN_IN)) || (state == STAT_INIT)) { 142 ret = OsCmdParseOneToken(cmdParsed, count, token); 143 } 144 145 return ret; 146} 147 148LITE_OS_SEC_TEXT_MINOR UINT32 OsCmdParse(CHAR *cmdStr, CmdParsed *cmdParsed) 149{ 150 if ((cmdStr == NULL) || (cmdParsed == NULL) || (strlen(cmdStr) == 0)) { 151 return (UINT32)OS_ERROR; 152 } 153 return OsCmdTokenSplit(cmdStr, ' ', cmdParsed); 154} 155 156