parent
b1c09f7512
commit
2a7e8383c8
3 changed files with 286 additions and 1 deletions
78
pkg/appsec/query_utils.go
Normal file
78
pkg/appsec/query_utils.go
Normal file
|
@ -0,0 +1,78 @@
|
|||
package appsec
|
||||
|
||||
// This file is mostly stolen from net/url package, but with some modifications to allow less strict parsing of query strings
|
||||
|
||||
import (
|
||||
"net/url"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// parseQuery and parseQuery are copied net/url package, but allow semicolon in values
|
||||
func ParseQuery(query string) url.Values {
|
||||
m := make(url.Values)
|
||||
parseQuery(m, query)
|
||||
return m
|
||||
}
|
||||
|
||||
func parseQuery(m url.Values, query string) {
|
||||
for query != "" {
|
||||
var key string
|
||||
key, query, _ = strings.Cut(query, "&")
|
||||
|
||||
if key == "" {
|
||||
continue
|
||||
}
|
||||
key, value, _ := strings.Cut(key, "=")
|
||||
//for now we'll just ignore the errors, but ideally we want to fire some "internal" rules when we see invalid query strings
|
||||
key = unescape(key)
|
||||
value = unescape(value)
|
||||
m[key] = append(m[key], value)
|
||||
}
|
||||
}
|
||||
|
||||
func hexDigitToByte(digit byte) (byte, bool) {
|
||||
switch {
|
||||
case digit >= '0' && digit <= '9':
|
||||
return digit - '0', true
|
||||
case digit >= 'a' && digit <= 'f':
|
||||
return digit - 'a' + 10, true
|
||||
case digit >= 'A' && digit <= 'F':
|
||||
return digit - 'A' + 10, true
|
||||
default:
|
||||
return 0, false
|
||||
}
|
||||
}
|
||||
|
||||
func unescape(input string) string {
|
||||
ilen := len(input)
|
||||
res := strings.Builder{}
|
||||
res.Grow(ilen)
|
||||
for i := 0; i < ilen; i++ {
|
||||
ci := input[i]
|
||||
if ci == '+' {
|
||||
res.WriteByte(' ')
|
||||
continue
|
||||
}
|
||||
if ci == '%' {
|
||||
if i+2 >= ilen {
|
||||
res.WriteByte(ci)
|
||||
continue
|
||||
}
|
||||
hi, ok := hexDigitToByte(input[i+1])
|
||||
if !ok {
|
||||
res.WriteByte(ci)
|
||||
continue
|
||||
}
|
||||
lo, ok := hexDigitToByte(input[i+2])
|
||||
if !ok {
|
||||
res.WriteByte(ci)
|
||||
continue
|
||||
}
|
||||
res.WriteByte(hi<<4 | lo)
|
||||
i += 2
|
||||
continue
|
||||
}
|
||||
res.WriteByte(ci)
|
||||
}
|
||||
return res.String()
|
||||
}
|
207
pkg/appsec/query_utils_test.go
Normal file
207
pkg/appsec/query_utils_test.go
Normal file
|
@ -0,0 +1,207 @@
|
|||
package appsec
|
||||
|
||||
import (
|
||||
"net/url"
|
||||
"reflect"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestParseQuery(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
query string
|
||||
expected url.Values
|
||||
}{
|
||||
{
|
||||
name: "Simple query",
|
||||
query: "foo=bar",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Multiple values",
|
||||
query: "foo=bar&foo=baz",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar", "baz"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Empty value",
|
||||
query: "foo=",
|
||||
expected: url.Values{
|
||||
"foo": []string{""},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Empty key",
|
||||
query: "=bar",
|
||||
expected: url.Values{
|
||||
"": []string{"bar"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Empty query",
|
||||
query: "",
|
||||
expected: url.Values{},
|
||||
},
|
||||
{
|
||||
name: "Multiple keys",
|
||||
query: "foo=bar&baz=qux",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar"},
|
||||
"baz": []string{"qux"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Multiple keys with empty value",
|
||||
query: "foo=bar&baz=qux&quux=",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar"},
|
||||
"baz": []string{"qux"},
|
||||
"quux": []string{""},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Multiple keys with empty value and empty key",
|
||||
query: "foo=bar&baz=qux&quux=&=quuz",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar"},
|
||||
"baz": []string{"qux"},
|
||||
"quux": []string{""},
|
||||
"": []string{"quuz"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Multiple keys with empty value and empty key and multiple values",
|
||||
query: "foo=bar&baz=qux&quux=&=quuz&foo=baz",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar", "baz"},
|
||||
"baz": []string{"qux"},
|
||||
"quux": []string{""},
|
||||
"": []string{"quuz"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Multiple keys with empty value and empty key and multiple values and escaped characters",
|
||||
query: "foo=bar&baz=qux&quux=&=quuz&foo=baz&foo=bar%20baz",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar", "baz", "bar baz"},
|
||||
"baz": []string{"qux"},
|
||||
"quux": []string{""},
|
||||
"": []string{"quuz"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Multiple keys with empty value and empty key and multiple values and escaped characters and semicolon",
|
||||
query: "foo=bar&baz=qux&quux=&=quuz&foo=baz&foo=bar%20baz&foo=bar%3Bbaz",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar", "baz", "bar baz", "bar;baz"},
|
||||
"baz": []string{"qux"},
|
||||
"quux": []string{""},
|
||||
"": []string{"quuz"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Multiple keys with empty value and empty key and multiple values and escaped characters and semicolon and ampersand",
|
||||
query: "foo=bar&baz=qux&quux=&=quuz&foo=baz&foo=bar%20baz&foo=bar%3Bbaz&foo=bar%26baz",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar", "baz", "bar baz", "bar;baz", "bar&baz"},
|
||||
"baz": []string{"qux"},
|
||||
"quux": []string{""},
|
||||
"": []string{"quuz"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Multiple keys with empty value and empty key and multiple values and escaped characters and semicolon and ampersand and equals",
|
||||
query: "foo=bar&baz=qux&quux=&=quuz&foo=baz&foo=bar%20baz&foo=bar%3Bbaz&foo=bar%26baz&foo=bar%3Dbaz",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar", "baz", "bar baz", "bar;baz", "bar&baz", "bar=baz"},
|
||||
"baz": []string{"qux"},
|
||||
"quux": []string{""},
|
||||
"": []string{"quuz"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Multiple keys with empty value and empty key and multiple values and escaped characters and semicolon and ampersand and equals and question mark",
|
||||
query: "foo=bar&baz=qux&quux=&=quuz&foo=baz&foo=bar%20baz&foo=bar%3Bbaz&foo=bar%26baz&foo=bar%3Dbaz&foo=bar%3Fbaz",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar", "baz", "bar baz", "bar;baz", "bar&baz", "bar=baz", "bar?baz"},
|
||||
"baz": []string{"qux"},
|
||||
"quux": []string{""},
|
||||
"": []string{"quuz"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "keys with escaped characters",
|
||||
query: "foo=ba;r&baz=qu;;x&quux=x\\&ww&xx=qu?uz&",
|
||||
expected: url.Values{
|
||||
"foo": []string{"ba;r"},
|
||||
"baz": []string{"qu;;x"},
|
||||
"quux": []string{"x\\"},
|
||||
"ww": []string{""},
|
||||
"xx": []string{"qu?uz"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "hexadecimal characters",
|
||||
query: "foo=bar%20baz",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar baz"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "hexadecimal characters upper and lower case",
|
||||
query: "foo=Ba%42%42&bar=w%2f%2F",
|
||||
expected: url.Values{
|
||||
"foo": []string{"BaBB"},
|
||||
"bar": []string{"w//"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "hexadecimal characters with invalid characters",
|
||||
query: "foo=bar%20baz%2",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar baz%2"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "hexadecimal characters with invalid hex characters",
|
||||
query: "foo=bar%xx",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar%xx"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "hexadecimal characters with invalid 2nd hex character",
|
||||
query: "foo=bar%2x",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar%2x"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "url +",
|
||||
query: "foo=bar+x",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar x"},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "url &&",
|
||||
query: "foo=bar&&lol=bur",
|
||||
expected: url.Values{
|
||||
"foo": []string{"bar"},
|
||||
"lol": []string{"bur"},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
t.Run(test.name, func(t *testing.T) {
|
||||
res := ParseQuery(test.query)
|
||||
if !reflect.DeepEqual(res, test.expected) {
|
||||
t.Fatalf("unexpected result: %v", res)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
|
@ -367,7 +367,7 @@ func NewParsedRequestFromRequest(r *http.Request, logger *logrus.Entry) (ParsedR
|
|||
URL: r.URL,
|
||||
Proto: r.Proto,
|
||||
Body: body,
|
||||
Args: parsedURL.Query(), //TODO: Check if there's not potential bypass as it excludes malformed args
|
||||
Args: ParseQuery(parsedURL.RawQuery),
|
||||
TransferEncoding: r.TransferEncoding,
|
||||
ResponseChannel: make(chan AppsecTempResponse),
|
||||
RemoteAddrNormalized: remoteAddrNormalized,
|
||||
|
|
Loading…
Reference in a new issue