diff --git a/internal/pool.go b/internal/pool.go index a1f4c45..78f0a09 100644 --- a/internal/pool.go +++ b/internal/pool.go @@ -299,7 +299,7 @@ func (p *Pool) BaseCompare(bl *pkg.Baseline) { if ok { if status = base.Compare(bl); status == 1 { - p.PutToInvalid(bl, "compare failed") + p.PutToInvalid(bl, ErrCompareFailed.Error()) return } } @@ -307,13 +307,13 @@ func (p *Pool) BaseCompare(bl *pkg.Baseline) { bl.Collect() for _, f := range bl.Frameworks { if f.Tag == "waf/cdn" { - p.PutToInvalid(bl, "waf") + p.PutToInvalid(bl, ErrWaf.Error()) return } } if ok && status == 0 && base.FuzzyCompare(bl) { - p.PutToInvalid(bl, "fuzzy compare failed") + p.PutToInvalid(bl, ErrFuzzyCompareFailed.Error()) p.PutToFuzzy(bl) return } diff --git a/internal/types.go b/internal/types.go index 123429b..3b04e8d 100644 --- a/internal/types.go +++ b/internal/types.go @@ -8,6 +8,8 @@ const ( ErrRequestFailed ErrWaf ErrRedirect + ErrCompareFailed + ErrFuzzyCompareFailed ) func (e ErrorType) Error() string { @@ -22,6 +24,10 @@ func (e ErrorType) Error() string { return "maybe ban of waf" case ErrRedirect: return "duplicate redirect url" + case ErrCompareFailed: + return "compare failed" + case ErrFuzzyCompareFailed: + return "fuzzy compare failed" default: return "unknown error" }