fix(frontend): webconsole use语句正则校验 TencentBlueKing#7488

# Reviewed, transaction id: 22282
royalpioneer · Oct 30, 2024 · 71e277f · 71e277f
1 parent e2f8423
commit 71e277f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/dbm-ui/frontend/src/views/db-manage/common/webconsole/components/console-panel/Index.vue b/dbm-ui/frontend/src/views/db-manage/common/webconsole/components/console-panel/Index.vue
@@ -187,6 +187,7 @@
     }
 
     // 校验语句
+    cmd = cmd.substring(inputPlaceholder.length);
     if (consoleConfig.value.validate) {
       const validateResult = consoleConfig.value.validate(cmd);
       if (validateResult) {
@@ -202,7 +203,6 @@
     // 开始请求
     try {
       loading.value = true;
-      cmd = cmd.substring(inputPlaceholder.length);
       if (typeof props.raw === 'boolean') {
         baseParams = {
           ...baseParams,

diff --git a/...ws/db-manage/common/webconsole/components/console-panel/components/RenderMysqlMessage.vue b/...ws/db-manage/common/webconsole/components/console-panel/components/RenderMysqlMessage.vue
@@ -32,7 +32,7 @@
   }
 
   export const validate = (cmd: string) => {
-    if (cmd.includes('use')) {
+    if (/^\s*use\s+.*$/.test(cmd)) {
       return t('暂不支持 use 语句，请使用 db.table 指定 database');
     }
     return '';

diff --git a/...ws/db-manage/common/webconsole/components/console-panel/components/RenderRedisMessage.vue b/...ws/db-manage/common/webconsole/components/console-panel/components/RenderRedisMessage.vue
@@ -37,7 +37,7 @@
     queryResult: string;
     commandInputs: string[];
   }) => {
-    if (/select/i.test(cmd) && queryResult === 'OK') {
+    if (/^\s*select\s+.*$/.test(cmd) && queryResult === 'OK') {
       const newDbIndex = Number(cmd.substring('select '.length));
       setDbIndexByClusterId(clusterId, newDbIndex);
       const newCommandInputs = commandInputs.map((item) => item.replace(/\[(\d+)\]/, `[${newDbIndex}]`));