diff --git a/utils/utils.go b/utils/utils.go index 674be01a..7e172fe0 100644 --- a/utils/utils.go +++ b/utils/utils.go @@ -64,21 +64,20 @@ func GetWafBlockStatusCode(target, mHost string) (isWaf bool, statusCode int, er return } -func GetAllFiles(path string) ([]string, error) { +func GetAllFiles(glob string) ([]string, error) { var files []string - - err := filepath.Walk(path, func(filePath string, info os.FileInfo, err error) error { + matches, err := filepath.Glob(glob) + if err != nil { + return nil, err + } + for _, filePath := range matches { + info, err := os.Stat(filePath) if err != nil { - return err + return nil, err } if !info.IsDir() { files = append(files, filePath) } - return nil - }) - if err != nil { - return nil, err } - return files, nil }