mirror of
https://github.com/daeuniverse/dae.git
synced 2025-01-07 05:51:20 +07:00
48 lines
1.4 KiB
Go
48 lines
1.4 KiB
Go
/*
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
* Copyright (c) 2023, daeuniverse Organization <dae@v2raya.org>
|
|
*/
|
|
|
|
package dns
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/daeuniverse/dae/component/routing"
|
|
"github.com/daeuniverse/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)
|
|
}
|
|
}
|