From 81c398eeb049b61b1d1bbe6a7d23d668366f1c2b Mon Sep 17 00:00:00 2001
From: Joona Hoikkala <joona@kuori.org>
Date: Sat, 30 Mar 2019 01:38:37 +0200
Subject: [PATCH] Make regex filter match headers too

---
 pkg/filter/regex.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/pkg/filter/regex.go b/pkg/filter/regex.go
index 6a745a0..9e72ca2 100644
--- a/pkg/filter/regex.go
+++ b/pkg/filter/regex.go
@@ -21,7 +21,15 @@ func NewRegexpFilter(value string) (ffuf.FilterProvider, error) {
 }
 
 func (f *RegexpFilter) Filter(response *ffuf.Response) (bool, error) {
-	return f.Value.Match(response.Data), nil
+	matchheaders := ""
+	for k, v := range response.Headers {
+		for _, iv := range v {
+			matchheaders += k + ": " + iv + "\r\n"
+		}
+	}
+	matchdata := []byte(matchheaders)
+	matchdata = append(matchdata, response.Data...)
+	return f.Value.Match(matchdata), nil
 }
 
 func (f *RegexpFilter) Repr() string {