mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-09-19 15:12:33 +00:00
39 lines
758 B
Go
39 lines
758 B
Go
package commons
|
|
|
|
import (
|
|
"net/netip"
|
|
"time"
|
|
|
|
"github.com/Dreamacro/clash/component/resolver"
|
|
|
|
D "github.com/miekg/dns"
|
|
)
|
|
|
|
const DefaultDnsReadTimeout = time.Second * 10
|
|
|
|
func ShouldHijackDns(dnsAdds []netip.AddrPort, targetAddr netip.AddrPort) bool {
|
|
for _, addrPort := range dnsAdds {
|
|
if addrPort == targetAddr || (addrPort.Addr().IsUnspecified() && targetAddr.Port() == 53) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func RelayDnsPacket(payload []byte) ([]byte, error) {
|
|
msg := &D.Msg{}
|
|
if err := msg.Unpack(payload); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
r, err := resolver.ServeMsg(msg)
|
|
if err != nil {
|
|
m := new(D.Msg)
|
|
m.SetRcode(msg, D.RcodeServerFailure)
|
|
return m.Pack()
|
|
}
|
|
|
|
r.SetRcode(msg, r.Rcode)
|
|
r.Compress = true
|
|
return r.Pack()
|
|
}
|