feat: complete basic functions

Support http,dns and mysql connection. Support custom http, dns response. Support dns rebinding. Support mysql load local files and jdbc deserialize exploit.
2026-09-27 00:51:55 +08:00 · 2021-04-21 18:55:16 +08:00
parent e9db8ddb24
commit 3559894acc
114 changed files with 41617 additions and 0 deletions
@@ -0,0 +1,94 @@
+package newdns
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestRecordValidate(t *testing.T) {
+	table := []struct {
+		typ Type
+		rec Record
+		err string
+	}{
+		{
+			typ: A,
+			rec: Record{Address: "foo"},
+			err: "invalid IPv4 address: foo",
+		},
+		{
+			typ: AAAA,
+			rec: Record{Address: "foo"},
+			err: "invalid IPv6 address: foo",
+		},
+		{
+			typ: A,
+			rec: Record{Address: "1:2:3:4::"},
+			err: "invalid IPv4 address: 1:2:3:4::",
+		},
+		{
+			typ: A,
+			rec: Record{Address: "1.2.3.4"},
+		},
+		{
+			typ: AAAA,
+			rec: Record{Address: "1:2:3:4::"},
+		},
+		{
+			typ: CNAME,
+			rec: Record{Address: "---"},
+			err: "invalid domain name: ---",
+		},
+		{
+			typ: CNAME,
+			rec: Record{Address: "foo.com"},
+			err: "invalid domain name: foo.com",
+		},
+		{
+			typ: CNAME,
+			rec: Record{Address: "foo.com."},
+		},
+		{
+			typ: MX,
+			rec: Record{Address: "foo.com"},
+			err: "invalid domain name: foo.com",
+		},
+		{
+			typ: MX,
+			rec: Record{Address: "foo.com."},
+		},
+		{
+			typ: TXT,
+			rec: Record{Data: nil},
+			err: "missing data",
+		},
+		{
+			typ: TXT,
+			rec: Record{Data: []string{"z4e6ycRMp6MP3WvWQMxIAOXglxANbj3oB0xD8BffktO4eo3VCR0s6TyGHKixvarOFJU0fqNkXeFOeI7sTXH5X0iXZukfLgnGTxLXNC7KkVFwtVFsh1P0IUNXtNBlOVWrVbxkS62ezbLpENNkiBwbkCvcTjwF2kyI0curAt9JhhJFb3AAq0q1iHWlJLn1KSrev9PIsY3alndDKjYTPxAojxzGKdK3A7rWLJ8Uzb3Z5OhLwP7jTKqbWVUocJRFLYpL"}},
+			err: "data too long",
+		},
+		{
+			typ: TXT,
+			rec: Record{Data: []string{"foo"}},
+		},
+		{
+			typ: NS,
+			rec: Record{Address: "foo.com"},
+			err: "invalid ns name: foo.com",
+		},
+		{
+			typ: NS,
+			rec: Record{Address: "foo.com."},
+		},
+	}
+
+	for i, item := range table {
+		err := item.rec.Validate(item.typ)
+		if err != nil {
+			assert.Equal(t, item.err, err.Error(), i)
+		} else {
+			assert.Equal(t, item.err, "", item)
+		}
+	}
+}