mirror of
https://github.com/mjl-/mox.git
synced 2025-07-10 09:14:39 +03:00
mox!
This commit is contained in:
Mechiel Lukkien
237
mtasts/parse_test.go
Normal file
237
mtasts/parse_test.go
Normal file
@ -0,0 +1,237 @@
|
||||
package mtasts
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
"testing"
|
||||
|
||||
"github.com/mjl-/mox/dns"
|
||||
)
|
||||
|
||||
func TestRecord(t *testing.T) {
|
||||
good := func(txt string, want Record) {
|
||||
t.Helper()
|
||||
r, _, err := ParseRecord(txt)
|
||||
if err != nil {
|
||||
t.Fatalf("parse: %s", err)
|
||||
}
|
||||
if !reflect.DeepEqual(r, &want) {
|
||||
t.Fatalf("want %#v, got %#v", want, *r)
|
||||
}
|
||||
}
|
||||
|
||||
bad := func(txt string) {
|
||||
t.Helper()
|
||||
r, _, err := ParseRecord(txt)
|
||||
if err == nil {
|
||||
t.Fatalf("parse, expected error, got record %v", r)
|
||||
}
|
||||
}
|
||||
|
||||
good("v=STSv1; id=20160831085700Z;", Record{Version: "STSv1", ID: "20160831085700Z"})
|
||||
good("v=STSv1; \t id=20160831085700Z \t;", Record{Version: "STSv1", ID: "20160831085700Z"})
|
||||
good("v=STSv1; id=a", Record{Version: "STSv1", ID: "a"})
|
||||
good("v=STSv1; id=a; more=a; ext=2", Record{Version: "STSv1", ID: "a", Extensions: []Pair{{"more", "a"}, {"ext", "2"}}})
|
||||
|
||||
bad("v=STSv0")
|
||||
bad("v=STSv10")
|
||||
bad("v=STSv2")
|
||||
bad("v=STSv1") // missing id
|
||||
bad("v=STSv1;") // missing id
|
||||
bad("v=STSv1; ext=1") // missing id
|
||||
bad("v=STSv1; id=") // empty id
|
||||
bad("v=STSv1; id=012345678901234567890123456789012") // id too long
|
||||
bad("v=STSv1; id=test-123") // invalid id
|
||||
bad("v=STSv1; id=a; more=") // empty value in extension
|
||||
bad("v=STSv1; id=a; a12345678901234567890123456789012=1") // extension name too long
|
||||
bad("v=STSv1; id=a; 1%=a") // invalid extension name
|
||||
bad("v=STSv1; id=a; test==") // invalid extension name
|
||||
bad("v=STSv1; id=a;;") // additional semicolon
|
||||
|
||||
const want = `v=STSv1; id=a; more=a; ext=2`
|
||||
record := Record{Version: "STSv1", ID: "a", Extensions: []Pair{{"more", "a"}, {"ext", "2"}}}
|
||||
got := record.String()
|
||||
if got != want {
|
||||
t.Fatalf("record string, got %q, want %q", got, want)
|
||||
}
|
||||
}
|
||||
|
||||
func TestParsePolicy(t *testing.T) {
|
||||
good := func(s string, want Policy) {
|
||||
t.Helper()
|
||||
p, err := ParsePolicy(s)
|
||||
if err != nil {
|
||||
t.Fatalf("parse policy: %s", err)
|
||||
}
|
||||
if !reflect.DeepEqual(p, &want) {
|
||||
t.Fatalf("want %v, got %v", want, p)
|
||||
}
|
||||
}
|
||||
|
||||
good(`version: STSv1
|
||||
mode: testing
|
||||
mx: mx1.example.com
|
||||
mx: mx2.example.com
|
||||
mx: mx.backup-example.com
|
||||
max_age: 1296000
|
||||
`,
|
||||
Policy{
|
||||
Version: "STSv1",
|
||||
Mode: ModeTesting,
|
||||
MX: []STSMX{
|
||||
{Domain: dns.Domain{ASCII: "mx1.example.com"}},
|
||||
{Domain: dns.Domain{ASCII: "mx2.example.com"}},
|
||||
{Domain: dns.Domain{ASCII: "mx.backup-example.com"}},
|
||||
},
|
||||
MaxAgeSeconds: 1296000,
|
||||
},
|
||||
)
|
||||
good("version: STSv1\nmode: enforce \nmx: *.example.com \nmax_age: 0 \n",
|
||||
Policy{
|
||||
Version: "STSv1",
|
||||
Mode: ModeEnforce,
|
||||
MX: []STSMX{
|
||||
{Wildcard: true, Domain: dns.Domain{ASCII: "example.com"}},
|
||||
},
|
||||
MaxAgeSeconds: 0,
|
||||
},
|
||||
)
|
||||
good("version:STSv1\r\nmode:\tenforce\r\nmx: \t\t *.example.com\nmax_age: 1\nmore:ext e ns ion",
|
||||
Policy{
|
||||
Version: "STSv1",
|
||||
Mode: ModeEnforce,
|
||||
MX: []STSMX{
|
||||
{Wildcard: true, Domain: dns.Domain{ASCII: "example.com"}},
|
||||
},
|
||||
MaxAgeSeconds: 1,
|
||||
Extensions: []Pair{{"more", "ext e ns ion"}},
|
||||
},
|
||||
)
|
||||
|
||||
bad := func(s string) {
|
||||
t.Helper()
|
||||
p, err := ParsePolicy(s)
|
||||
if err == nil {
|
||||
t.Fatalf("parsing policy did not fail: %v", p)
|
||||
}
|
||||
}
|
||||
|
||||
bad("") // missing version
|
||||
bad("version:STSv0\nmode:none\nmax_age:0") // bad version
|
||||
bad("version:STSv10\nmode:none\nmax_age:0") // bad version
|
||||
bad("version:STSv2\nmode:none\nmax_age:0") // bad version
|
||||
bad("version:STSv1\nmax_age:0\nmx:example.com") // missing mode
|
||||
bad("version:STSv1\nmode:none") // missing max_age
|
||||
bad("version:STSv1\nmax_age:0\nmode:enforce") // missing mx for mode
|
||||
bad("version:STSv1\nmax_age:0\nmode:testing") // missing mx for mode
|
||||
bad("max_age:0\nmode:none") // missing version
|
||||
bad("version:STSv1\nmode:none\nmax_age:01234567890") // max_age too long
|
||||
bad("version:STSv1\nmode:bad\nmax_age:1") // bad mode
|
||||
bad("version:STSv1\nmode:none\nmax_age:a") // bad max_age
|
||||
bad("version:STSv1\nmode:enforce\nmax_age:0\nmx:") // missing value
|
||||
bad("version:STSv1\nmode:enforce\nmax_age:0\nmx:*.*.example") // bad mx
|
||||
bad("version:STSv1\nmode:enforce\nmax_age:0\nmx:**.example") // bad mx
|
||||
bad("version:STSv1\nmode:enforce\nmax_age:0\nmx:**.example-") // bad mx
|
||||
bad("version:STSv1\nmode:enforce\nmax_age:0\nmx:test.example-") // bad mx
|
||||
bad("version:STSv1\nmode:none\nmax_age:0\next:") // empty extension
|
||||
bad("version:STSv1\nmode:none\nmax_age:0\na12345678901234567890123456789012:123") // long extension name
|
||||
bad("version:STSv1\nmode:none\nmax_age:0\n_bad:test") // bad ext name
|
||||
bad("version:STSv1\nmode:none\nmax_age:0\nmx: møx.example") // invalid u-label in mx
|
||||
|
||||
policy := Policy{
|
||||
Version: "STSv1",
|
||||
Mode: ModeTesting,
|
||||
MX: []STSMX{
|
||||
{Domain: dns.Domain{ASCII: "mx1.example.com"}},
|
||||
{Domain: dns.Domain{ASCII: "mx2.example.com"}},
|
||||
{Domain: dns.Domain{ASCII: "mx.backup-example.com"}},
|
||||
},
|
||||
MaxAgeSeconds: 1296000,
|
||||
}
|
||||
want := `version: STSv1
|
||||
mode: testing
|
||||
max_age: 1296000
|
||||
mx: mx1.example.com
|
||||
mx: mx2.example.com
|
||||
mx: mx.backup-example.com
|
||||
`
|
||||
got := policy.String()
|
||||
if got != want {
|
||||
t.Fatalf("policy string, got %q, want %q", got, want)
|
||||
}
|
||||
}
|
||||
|
||||
func FuzzParseRecord(f *testing.F) {
|
||||
f.Add("v=STSv1; id=20160831085700Z;")
|
||||
f.Add("v=STSv1; \t id=20160831085700Z \t;")
|
||||
f.Add("v=STSv1; id=a")
|
||||
f.Add("v=STSv1; id=a; more=a; ext=2")
|
||||
|
||||
f.Add("v=STSv0")
|
||||
f.Add("v=STSv10")
|
||||
f.Add("v=STSv2")
|
||||
f.Add("v=STSv1") // missing id
|
||||
f.Add("v=STSv1;") // missing id
|
||||
f.Add("v=STSv1; ext=1") // missing id
|
||||
f.Add("v=STSv1; id=") // empty id
|
||||
f.Add("v=STSv1; id=012345678901234567890123456789012") // id too long
|
||||
f.Add("v=STSv1; id=test-123") // invalid id
|
||||
f.Add("v=STSv1; id=a; more=") // empty value in extension
|
||||
f.Add("v=STSv1; id=a; a12345678901234567890123456789012=1") // extension name too long
|
||||
f.Add("v=STSv1; id=a; 1%=a") // invalid extension name
|
||||
f.Add("v=STSv1; id=a; test==") // invalid extension name
|
||||
f.Add("v=STSv1; id=a;;") // additional semicolon
|
||||
|
||||
f.Fuzz(func(t *testing.T, s string) {
|
||||
r, _, err := ParseRecord(s)
|
||||
if err == nil {
|
||||
_ = r.String()
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func FuzzParsePolicy(f *testing.F) {
|
||||
f.Add(`version: STSv1
|
||||
mode: testing
|
||||
mx: mx1.example.com
|
||||
mx: mx2.example.com
|
||||
mx: mx.backup-example.com
|
||||
max_age: 1296000
|
||||
`)
|
||||
f.Add(`version: STSv1
|
||||
mode: enforce
|
||||
mx: *.example.com
|
||||
max_age: 0
|
||||
`)
|
||||
f.Add("version:STSv1\r\nmode:\tenforce\r\nmx: \t\t *.example.com\nmax_age: 1\nmore:ext e ns ion")
|
||||
|
||||
f.Add("") // missing version
|
||||
f.Add("version:STSv0\nmode:none\nmax_age:0") // bad version
|
||||
f.Add("version:STSv10\nmode:none\nmax_age:0") // bad version
|
||||
f.Add("version:STSv2\nmode:none\nmax_age:0") // bad version
|
||||
f.Add("version:STSv1\nmax_age:0\nmx:example.com") // missing mode
|
||||
f.Add("version:STSv1\nmode:none") // missing max_age
|
||||
f.Add("version:STSv1\nmax_age:0\nmode:enforce") // missing mx for mode
|
||||
f.Add("version:STSv1\nmax_age:0\nmode:testing") // missing mx for mode
|
||||
f.Add("max_age:0\nmode:none") // missing version
|
||||
f.Add("version:STSv1\nmode:none\nmax_age:0 ") // trailing whitespace
|
||||
f.Add("version:STSv1\nmode:none\nmax_age:01234567890") // max_age too long
|
||||
f.Add("version:STSv1\nmode:bad\nmax_age:1") // bad mode
|
||||
f.Add("version:STSv1\nmode:none\nmax_age:a") // bad max_age
|
||||
f.Add("version:STSv1\nmode:enforce\nmax_age:0\nmx:") // missing value
|
||||
f.Add("version:STSv1\nmode:enforce\nmax_age:0\nmx:*.*.example") // bad mx
|
||||
f.Add("version:STSv1\nmode:enforce\nmax_age:0\nmx:**.example") // bad mx
|
||||
f.Add("version:STSv1\nmode:enforce\nmax_age:0\nmx:**.example-") // bad mx
|
||||
f.Add("version:STSv1\nmode:enforce\nmax_age:0\nmx:test.example-") // bad mx
|
||||
f.Add("version:STSv1\nmode:none\nmax_age:0\next:") // empty extension
|
||||
f.Add("version:STSv1\nmode:none\nmax_age:0\next:abc ") // trailing space
|
||||
f.Add("version:STSv1\nmode:none\nmax_age:0\next:a\t") // invalid char
|
||||
f.Add("version:STSv1\nmode:none\nmax_age:0\na12345678901234567890123456789012:123") // long extension name
|
||||
f.Add("version:STSv1\nmode:none\nmax_age:0\n_bad:test") // bad ext name
|
||||
|
||||
f.Fuzz(func(t *testing.T, s string) {
|
||||
r, err := ParsePolicy(s)
|
||||
if err == nil {
|
||||
_ = r.String()
|
||||
}
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user