mirrors/dae
1
0
Fork 0
mirror of https://github.com/daeuniverse/dae.git synced 2025-02-22 12:38:34 +07:00
Code Issues Packages Projects Releases Wiki Activity

feat: add dial_mode domain+

Browse Source
This commit is contained in:
mzz2017 2023-02-17 01:07:27 +08:00
parent cd1270bff1
commit 739682dabc
2 changed files with 15 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
common/consts/control.go
View File

@ -10,13 +10,14 @@ import "fmt"
type DialMode string
const (
DialMode_Ip DialMode = "ip"
DialMode_Domain DialMode = "domain"
DialMode_Ip DialMode = "ip"
DialMode_Domain DialMode = "domain"
DialMode_DomainPlus DialMode = "domain+"
)
func ParseDialMode(mode string) (DialMode, error) {
switch mode {
case "ip", "domain":
case "ip", "domain", "domain+":
return DialMode(mode), nil
default:
return "", fmt.Errorf("unsupported dial mode: %v", mode)

17
control/control_plane.go
View File

@ -408,15 +408,20 @@ func (c *ControlPlane) finishInitDnsUpstreamResolve(raw common.UrlOrEmpty, dnsUp
func (c *ControlPlane) ChooseDialTarget(outbound consts.OutboundIndex, dst netip.AddrPort, domain string) (dialTarget string) {
mode := consts.DialMode_Ip
if c.dialMode == consts.DialMode_Domain &&
!outbound.IsReserved() && // Direct, block, etc. should be skipped.
domain != "" {
dstIp := common.ConvergeIp(dst.Addr())
cache := c.lookupDnsRespCache(domain, common.AddrToDnsType(dstIp))
if cache != nil && cache.IncludeIp(dstIp) {
if !outbound.IsReserved() && domain != "" {
switch c.dialMode {
case consts.DialMode_Domain:
dstIp := common.ConvergeIp(dst.Addr())
cache := c.lookupDnsRespCache(domain, common.AddrToDnsType(dstIp))
if cache != nil && cache.IncludeIp(dstIp) {
mode = consts.DialMode_Domain
}
case consts.DialMode_DomainPlus:
mode = consts.DialMode_Domain
}
}
switch mode {
case consts.DialMode_Ip:
dialTarget = dst.String()