feat: dns routing (#26)

component/dns/function_parser.go

@@ -0,0 +1,47 @@
+/*
+ * SPDX-License-Identifier: AGPL-3.0-only
+ * Copyright (c) 2023, v2rayA Organization <team@v2raya.org>
+ */
+
+package dns
+
+import (
+	"fmt"
+	"github.com/sirupsen/logrus"
+	"github.com/v2rayA/dae/component/routing"
+	"github.com/v2rayA/dae/pkg/config_parser"
+	"golang.org/x/net/dns/dnsmessage"
+	"strings"
+)
+
+var typeNames = map[string]dnsmessage.Type{
+	"A":     dnsmessage.TypeA,
+	"NS":    dnsmessage.TypeNS,
+	"CNAME": dnsmessage.TypeCNAME,
+	"SOA":   dnsmessage.TypeSOA,
+	"PTR":   dnsmessage.TypePTR,
+	"MX":    dnsmessage.TypeMX,
+	"TXT":   dnsmessage.TypeTXT,
+	"AAAA":  dnsmessage.TypeAAAA,
+	"SRV":   dnsmessage.TypeSRV,
+	"OPT":   dnsmessage.TypeOPT,
+	"WKS":   dnsmessage.TypeWKS,
+	"HINFO": dnsmessage.TypeHINFO,
+	"MINFO": dnsmessage.TypeMINFO,
+	"AXFR":  dnsmessage.TypeAXFR,
+	"ALL":   dnsmessage.TypeALL,
+}
+
+func TypeParserFactory(callback func(f *config_parser.Function, types []dnsmessage.Type, overrideOutbound *routing.Outbound) (err error)) routing.FunctionParser {
+	return func(log *logrus.Logger, f *config_parser.Function, key string, paramValueGroup []string, overrideOutbound *routing.Outbound) (err error) {
+		var types []dnsmessage.Type
+		for _, v := range paramValueGroup {
+			t, ok := typeNames[strings.ToUpper(v)]
+			if !ok {
+				return fmt.Errorf("unknown DNS request type: %v", v)
+			}
+			types = append(types, t)
+		}
+		return callback(f, types, overrideOutbound)
+	}
+}