kemko/reproxy
1
0
Fork 0
mirror of https://github.com/kemko/reproxy.git synced 2026-01-01 15:55:49 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

drop leading and trailing dbl-quotes from the headers #100

Browse Source
This commit is contained in:
Umputun
2021-07-08 16:27:42 -05:00
parent 207a467960
commit cbe7f6f84d
2 changed files with 18 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
app/main.go
View File

@@ -486,15 +486,29 @@ func sizeParse(inp string) (uint64, error) {
}
// splitAtCommas split s at commas, ignoring commas in strings.
// Eliminate leading and trailing dbl quotes in each element only if both presented
// based on https://stackoverflow.com/a/59318708
func splitAtCommas(s string) []string {
cleanup := func(s string) string {
if len(s) == 0 {
return s
}
res := strings.TrimSpace(s)
if s[0] == '"' && s[len(s)-1] == '"' {
res = strings.TrimPrefix(res, `"`)
res = strings.TrimSuffix(res, `"`)
}
return res
}
var res []string
var beg int
var inString bool
for i := 0; i < len(s); i++ {
if s[i] == ',' && !inString {
res = append(res, strings.TrimSpace(s[beg:i]))
res = append(res, cleanup(s[beg:i]))
beg = i + 1
continue
}
@@ -502,12 +516,12 @@ func splitAtCommas(s string) []string {
if s[i] == '"' {
if !inString {
inString = true
} else if i > 0 && s[i-1] != '\\' {
} else if i > 0 && s[i-1] != '\\' { // also allow \"
inString = false
}
}
}
res = append(res, strings.TrimSpace(s[beg:]))
res = append(res, cleanup(s[beg:]))
if len(res) == 1 && res[0] == "" {
return []string{}
}

2
app/main_test.go
View File

@@ -361,7 +361,7 @@ func Test_splitAtCommas(t *testing.T) {
}{
{"a string", []string{"a string"}},
{"vv1, vv2, vv3", []string{"vv1", "vv2", "vv3"}},
{`"vv1, blah", vv2, vv3`, []string{"\"vv1, blah\"", "vv2", "vv3"}},
{`"vv1, blah", vv2, vv3`, []string{"vv1, blah", "vv2", "vv3"}},
{
`Access-Control-Allow-Headers:"DNT,X-CustomHeader,Keep-Alive,User-Agent,X-Requested-With,If-Modified-Since,Cache-Control,Content-Type",header123:val, foo:"bar1,bar2"`,
[]string{"Access-Control-Allow-Headers:\"DNT,X-CustomHeader,Keep-Alive,User-Agent,X-Requested-With,If-Modified-Since,Cache-Control,Content-Type\"", "header123:val", "foo:\"bar1,bar2\""},