소스 검색

限制python代码执行

xyxie 1 년 전
부모
커밋
acc94b7ffe
1개의 변경된 파일2개의 추가작업 그리고 1개의 파일을 삭제
  1. 2 1
      services/base_from_python.go

+ 2 - 1
services/base_from_python.go

@@ -69,8 +69,9 @@ func ExecPythonCode(edbCode, reqCode string) (dataMap models.EdbDataFromPython,
 		}
 	}()
 
+	codeStr := strings.ToLower(reqCode)
 	// 限制python代码的执行,如果包含特殊的关键词,则报错
-	if strings.Contains(edbCode, "drop") || strings.Contains(edbCode, "delete") || strings.Contains(edbCode, "update") || strings.Contains(edbCode, "alter") || strings.Contains(edbCode, "truncate") {
+	if strings.Contains(codeStr, "drop") || strings.Contains(codeStr, "delete") || strings.Contains(codeStr, "update") || strings.Contains(codeStr, "alter") || strings.Contains(codeStr, "truncate") {
 		err = fmt.Errorf("不支持该命令")
 		return
 	}