55 lines
1.1 KiB
Go
55 lines
1.1 KiB
Go
package filter
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/ffuf/ffuf/pkg/ffuf"
|
|
)
|
|
|
|
type StatusFilter struct {
|
|
Value []int64
|
|
}
|
|
|
|
func NewStatusFilter(value string) (ffuf.FilterProvider, error) {
|
|
var intvals []int64
|
|
for _, sv := range strings.Split(value, ",") {
|
|
if sv == "all" {
|
|
intvals = append(intvals, 0)
|
|
} else {
|
|
intval, err := strconv.ParseInt(sv, 10, 0)
|
|
if err != nil {
|
|
return &StatusFilter{}, fmt.Errorf("Status filter or matcher (-fc / -mc): invalid value %s", value)
|
|
}
|
|
intvals = append(intvals, intval)
|
|
}
|
|
}
|
|
return &StatusFilter{Value: intvals}, nil
|
|
}
|
|
|
|
func (f *StatusFilter) Filter(response *ffuf.Response) (bool, error) {
|
|
for _, iv := range f.Value {
|
|
if iv == 0 {
|
|
// Handle the "all" case
|
|
return true, nil
|
|
}
|
|
if iv == response.StatusCode {
|
|
return true, nil
|
|
}
|
|
}
|
|
return false, nil
|
|
}
|
|
|
|
func (f *StatusFilter) Repr() string {
|
|
var strval []string
|
|
for _, iv := range f.Value {
|
|
if iv == 0 {
|
|
strval = append(strval, "all")
|
|
} else {
|
|
strval = append(strval, strconv.Itoa(int(iv)))
|
|
}
|
|
}
|
|
return fmt.Sprintf("Response status: %s", strings.Join(strval, ","))
|
|
}
|