feat: add CheckBadCmd()

fufuok · May 16, 2024 · bde92df · bde92df
1 parent 3ffb9f8
commit bde92df
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/cmder/helper.go b/cmder/helper.go
@@ -3,6 +3,7 @@ package cmder
 import (
 	"context"
 	"errors"
+	"strings"
 	"time"
 
 	"github.com/fufuok/utils"
@@ -68,3 +69,13 @@ func RunCmdWithContext(ctx context.Context, cmdArgs []string) cmd.Status {
 		}
 	}
 }
+
+// CheckBadCmd 检查命令是否包含潜在非法字符
+func CheckBadCmd(s string) bool {
+	if strings.Contains(s, "&") || strings.Contains(s, "|") || strings.Contains(s, ";") ||
+		strings.Contains(s, "$") || strings.Contains(s, "'") || strings.Contains(s, "`") ||
+		strings.Contains(s, "(") || strings.Contains(s, ")") || strings.Contains(s, "\"") {
+		return true
+	}
+	return false
+}