Fix: fakeip udp should not replace with another ip

This commit is contained in:
Dreamacro
2022-07-05 21:09:29 +08:00
parent aaf08dadff
commit 5497adaba1
2 changed files with 26 additions and 17 deletions

View File

@ -4,6 +4,7 @@ import (
"context"
"fmt"
"net"
"net/netip"
"runtime"
"strconv"
"sync"
@ -166,9 +167,10 @@ func handleUDPConn(packet *inbound.PacketAdapter) {
}
// make a fAddr if request ip is fakeip
var fAddr net.Addr
var fAddr netip.Addr
if resolver.IsExistFakeIP(metadata.DstIP) {
fAddr = metadata.UDPAddr()
fAddr, _ = netip.AddrFromSlice(metadata.DstIP)
fAddr = fAddr.Unmap()
}
if err := preHandleMetadata(metadata); err != nil {
@ -176,6 +178,15 @@ func handleUDPConn(packet *inbound.PacketAdapter) {
return
}
// local resolve UDP dns
if !metadata.Resolved() {
ip, err := resolver.ResolveIP(metadata.Host)
if err != nil {
return
}
metadata.DstIP = ip
}
key := packet.LocalAddr().String()
handle := func() bool {
@ -240,7 +251,9 @@ func handleUDPConn(packet *inbound.PacketAdapter) {
log.Infoln("[UDP] %s --> %s doesn't match any rule using DIRECT", metadata.SourceAddress(), metadata.RemoteAddress())
}
go handleUDPToLocal(packet.UDPPacket, pc, key, fAddr)
oAddr, _ := netip.AddrFromSlice(metadata.DstIP)
oAddr = oAddr.Unmap()
go handleUDPToLocal(packet.UDPPacket, pc, key, oAddr, fAddr)
natTable.Set(key, pc)
handle()