projectdiscovery/httpx
1
1
Fork 0
mirror of https://github.com/projectdiscovery/httpx.git synced 2024-09-11 17:46:08 +03:00
Code Issues Packages Projects Releases Wiki Activity

adding string/regex match/filter

Browse Source
This commit is contained in:
Mzack9999 2020-09-14 19:40:19 +02:00
parent 04d01f8268
commit edb9e1baaa
1 changed files with 33 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
cmd/httpx/httpx.go
View File

@ -8,6 +8,7 @@ import (
"io/ioutil"
"os"
"path"
"regexp"
"strconv"
"strings"
"time"
@ -162,12 +163,24 @@ func main() {
if len(options.filterContentLength) > 0 && slice.IntSliceContains(options.filterContentLength, r.ContentLength) {
continue
}
if options.filterRegex != nil && options.filterRegex.MatchString(r.Response) {
continue
}
if options.OutputFilterString != "" && strings.Contains(r.Response, options.OutputFilterString) {
continue
}
if len(options.matchStatusCode) > 0 && !slice.IntSliceContains(options.matchStatusCode, r.StatusCode) {
continue
}
if len(options.matchContentLength) > 0 && !slice.IntSliceContains(options.matchContentLength, r.ContentLength) {
continue
}
if options.matchRegex != nil && !options.matchRegex.MatchString(r.Response) {
continue
}
if options.OutputMatchString != "" && !strings.Contains(r.Response, options.OutputMatchString) {
continue
}
row := r.str
if options.JSONOutput {
@ -608,6 +621,12 @@ type Options struct {
Debug bool
Pipeline bool
HTTP2Probe bool
OutputFilterString string
OutputMatchString string
OutputFilterRegex string
filterRegex *regexp.Regexp
OutputMatchRegex string
matchRegex *regexp.Regexp
}
// ParseOptions parses the command line options for application
@ -656,6 +675,10 @@ func ParseOptions() *Options {
flag.BoolVar(&options.Pipeline, "pipeline", false, "HTTP1.1 Pipeline")
flag.BoolVar(&options.HTTP2Probe, "http2", false, "HTTP2 probe")
flag.BoolVar(&options.OutputIP, "ip", false, "Output target ip")
flag.StringVar(&options.OutputFilterString, "filter-string", "", "Filter String")
flag.StringVar(&options.OutputMatchString, "match-string", "", "Match string")
flag.StringVar(&options.OutputFilterRegex, "filter-regex", "", "Filter Regex")
flag.StringVar(&options.OutputMatchRegex, "match-regex", "", "Match Regex")
flag.Parse()
// Read the inputs and configure the logging
@ -695,6 +718,16 @@ func (options *Options) validateOptions() {
if options.filterContentLength, err = stringz.StringToSliceInt(options.OutputFilterContentLength); err != nil {
gologger.Fatalf("Invalid value for filter content length option: %s\n", err)
}
if options.OutputFilterRegex != "" {
if options.filterRegex, err = regexp.Compile(options.OutputFilterRegex); err != nil {
gologger.Fatalf("Invalid value for regex filter option: %s\n", err)
}
}
if options.OutputMatchRegex != "" {
if options.matchRegex, err = regexp.Compile(options.OutputMatchRegex); err != nil {
gologger.Fatalf("Invalid value for match regex option: %s\n", err)
}
}
}
// configureOutput configures the output on the screen